rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use Express router for secrets endpoint

Browse Source
This commit is contained in:
valadaptive
2023-12-04 12:55:13 -05:00
parent 414c9bd5fb
commit 091255d451
2 changed files with 55 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server.js
View File

@ -3585,7 +3585,7 @@ require('./src/endpoints/tokenizers').registerEndpoints(app, jsonParser);
require('./src/endpoints/presets').registerEndpoints(app, jsonParser);
// Secrets managemenet
require('./src/endpoints/secrets').registerEndpoints(app, jsonParser);
app.use('/api/secrets', require('./src/endpoints/secrets').router);
// Thumbnail generation
require('./src/endpoints/thumbnails').registerEndpoints(app, jsonParser);

29
src/endpoints/secrets.js
View File

@ -1,7 +1,9 @@
const fs = require('fs');
const path = require('path');
const express = require('express');
const { getConfigValue } = require('../util');
const writeFileAtomicSync = require('write-file-atomic').sync;
const { jsonParser } = require('../express-common');
const SECRETS_FILE = path.join(process.cwd(), './secrets.json');
const SECRET_KEYS = {
@ -143,23 +145,17 @@ function getAllSecrets() {
return secrets;
}
/**
* Registers endpoints for the secret management API
* @param {import('express').Express} app Express app
* @param {any} jsonParser JSON parser middleware
*/
function registerEndpoints(app, jsonParser) {
const router = express.Router();
app.post('/api/secrets/write', jsonParser, (request, response) => {
router.post('/write', jsonParser, (request, response) => {
const key = request.body.key;
const value = request.body.value;
writeSecret(key, value);
return response.send('ok');
});
app.post('/api/secrets/read', jsonParser, (_, response) => {
});
router.post('/read', jsonParser, (_, response) => {
try {
const state = readSecretState();
return response.send(state);
@ -167,9 +163,9 @@ function registerEndpoints(app, jsonParser) {
console.error(error);
return response.send({});
}
});
});
app.post('/api/secrets/view', jsonParser, async (_, response) => {
router.post('/view', jsonParser, async (_, response) => {
const allowKeysExposure = getConfigValue('allowKeysExposure', false);
if (!allowKeysExposure) {
@ -189,9 +185,9 @@ function registerEndpoints(app, jsonParser) {
console.error(error);
return response.sendStatus(500);
}
});
});
app.post('/api/secrets/find', jsonParser, (request, response) => {
router.post('/find', jsonParser, (request, response) => {
const allowKeysExposure = getConfigValue('allowKeysExposure', false);
if (!allowKeysExposure) {
@ -213,8 +209,7 @@ function registerEndpoints(app, jsonParser) {
console.error(error);
return response.sendStatus(500);
}
});
}
});
module.exports = {
writeSecret,
@ -222,6 +217,6 @@ module.exports = {
readSecretState,
migrateSecrets,
getAllSecrets,
registerEndpoints,
SECRET_KEYS,
router,
};