Use Express router for presets endpoint

This commit is contained in:
valadaptive
2023-12-04 12:54:18 -05:00
parent 414c9bd5fb
commit ba74288e4a
2 changed files with 86 additions and 91 deletions

View File

@@ -3582,7 +3582,7 @@ require('./src/endpoints/openai').registerEndpoints(app, jsonParser, urlencodedP
require('./src/endpoints/tokenizers').registerEndpoints(app, jsonParser); require('./src/endpoints/tokenizers').registerEndpoints(app, jsonParser);
// Preset management // Preset management
require('./src/endpoints/presets').registerEndpoints(app, jsonParser); app.use('/api/presets', require('./src/endpoints/presets').router);
// Secrets managemenet // Secrets managemenet
require('./src/endpoints/secrets').registerEndpoints(app, jsonParser); require('./src/endpoints/secrets').registerEndpoints(app, jsonParser);

View File

@@ -1,9 +1,11 @@
const fs = require('fs'); const fs = require('fs');
const path = require('path'); const path = require('path');
const express = require('express');
const sanitize = require('sanitize-filename'); const sanitize = require('sanitize-filename');
const writeFileAtomicSync = require('write-file-atomic').sync; const writeFileAtomicSync = require('write-file-atomic').sync;
const { DIRECTORIES } = require('../constants'); const { DIRECTORIES } = require('../constants');
const { getDefaultPresetFile, getDefaultPresets } = require('./content-manager'); const { getDefaultPresetFile, getDefaultPresets } = require('./content-manager');
const { jsonParser } = require('../express-common');
/** /**
* Gets the folder and extension for the preset settings based on the API source ID. * Gets the folder and extension for the preset settings based on the API source ID.
@@ -30,13 +32,9 @@ function getPresetSettingsByAPI(apiId) {
} }
} }
/** const router = express.Router();
* Registers the preset management endpoints.
* @param {import('express').Express} app Express app router.post('/save', jsonParser, function (request, response) {
* @param {any} jsonParser JSON parser middleware
*/
function registerEndpoints(app, jsonParser) {
app.post('/api/presets/save', jsonParser, function (request, response) {
const name = sanitize(request.body.name); const name = sanitize(request.body.name);
if (!request.body.preset || !name) { if (!request.body.preset || !name) {
return response.sendStatus(400); return response.sendStatus(400);
@@ -54,7 +52,7 @@ function registerEndpoints(app, jsonParser) {
return response.send({ name }); return response.send({ name });
}); });
app.post('/api/presets/delete', jsonParser, function (request, response) { router.post('/delete', jsonParser, function (request, response) {
const name = sanitize(request.body.name); const name = sanitize(request.body.name);
if (!name) { if (!name) {
return response.sendStatus(400); return response.sendStatus(400);
@@ -77,7 +75,7 @@ function registerEndpoints(app, jsonParser) {
} }
}); });
app.post('/api/presets/restore', jsonParser, function (request, response) { router.post('/restore', jsonParser, function (request, response) {
try { try {
const settings = getPresetSettingsByAPI(request.body.apiId); const settings = getPresetSettingsByAPI(request.body.apiId);
const name = sanitize(request.body.name); const name = sanitize(request.body.name);
@@ -100,7 +98,7 @@ function registerEndpoints(app, jsonParser) {
}); });
// TODO: Merge with /api/presets/save // TODO: Merge with /api/presets/save
app.post('/api/presets/save-openai', jsonParser, function (request, response) { router.post('/save-openai', jsonParser, function (request, response) {
if (!request.body || typeof request.query.name !== 'string') return response.sendStatus(400); if (!request.body || typeof request.query.name !== 'string') return response.sendStatus(400);
const name = sanitize(request.query.name); const name = sanitize(request.query.name);
if (!name) return response.sendStatus(400); if (!name) return response.sendStatus(400);
@@ -112,7 +110,7 @@ function registerEndpoints(app, jsonParser) {
}); });
// TODO: Merge with /api/presets/delete // TODO: Merge with /api/presets/delete
app.post('/api/presets/delete-openai', jsonParser, function (request, response) { router.post('/delete-openai', jsonParser, function (request, response) {
if (!request.body || !request.body.name) { if (!request.body || !request.body.name) {
return response.sendStatus(400); return response.sendStatus(400);
} }
@@ -127,8 +125,5 @@ function registerEndpoints(app, jsonParser) {
return response.send({ error: true }); return response.send({ error: true });
}); });
}
module.exports = { module.exports = { router };
registerEndpoints,
};