Move preset management into a separate file

2025-06-05 21:59:27 +02:00 · 2023-09-16 17:36:54 +03:00
parent 38b63b07f5
commit 61995bb33f
4 changed files with 101 additions and 74 deletions
--- a/src/presets.js
+++ b/src/presets.js
@@ -0,0 +1,94 @@
+const fs = require('fs');
+const path = require('path');
+const sanitize = require('sanitize-filename');
+const writeFileAtomicSync = require('write-file-atomic').sync;
+const { DIRECTORIES } = require('./constants');
+
+/**
+ * Gets the folder and extension for the preset settings based on the API source ID.
+ * @param {string} apiId API source ID
+ * @returns {object} Object containing the folder and extension for the preset settings
+ */
+function getPresetSettingsByAPI(apiId) {
+    switch (apiId) {
+        case 'kobold':
+        case 'koboldhorde':
+            return { folder: DIRECTORIES.koboldAI_Settings, extension: '.settings' };
+        case 'novel':
+            return { folder: DIRECTORIES.novelAI_Settings, extension: '.settings' };
+        case 'textgenerationwebui':
+            return { folder: DIRECTORIES.textGen_Settings, extension: '.settings' };
+        case 'openai':
+            return { folder: DIRECTORIES.openAI_Settings, extension: '.settings' };
+        case 'instruct':
+            return { folder: DIRECTORIES.instruct, extension: '.json' };
+        case 'context':
+            return { folder: DIRECTORIES.context, extension: '.json' };
+        default:
+            return { folder: null, extension: null };
+    }
+}
+
+/**
+ * Registers the preset management endpoints.
+ * @param {import('express').Express} app Express app
+ * @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);
+        if (!request.body.preset || !name) {
+            return response.sendStatus(400);
+        }
+
+        const settings = getPresetSettingsByAPI(request.body.apiId);
+        const filename = name + settings.extension;
+
+        if (!settings.folder) {
+            return response.sendStatus(400);
+        }
+
+        const fullpath = path.join(settings.folder, filename);
+        writeFileAtomicSync(fullpath, JSON.stringify(request.body.preset, null, 4), 'utf-8');
+        return response.send({ name });
+    });
+
+    app.post("/api/presets/delete", jsonParser, function (request, response) {
+        const name = sanitize(request.body.name);
+        if (!name) {
+            return response.sendStatus(400);
+        }
+
+        const settings = getPresetSettingsByAPI(request.body.apiId);
+        const filename = name + settings.extension;
+
+        if (!settings.folder) {
+            return response.sendStatus(400);
+        }
+
+        const fullpath = path.join(settings.folder, filename);
+
+        if (fs.existsSync(fullpath)) {
+            fs.unlinkSync(fullpath);
+            return response.sendStatus(200);
+        } else {
+            return response.sendStatus(404);
+        }
+    });
+
+    // TODO: Merge with /api/presets/save
+    app.post("/api/presets/save-openai", jsonParser, function (request, response) {
+        if (!request.body || typeof request.query.name !== 'string') return response.sendStatus(400);
+        const name = sanitize(request.query.name);
+        if (!name) return response.sendStatus(400);
+
+        const filename = `${name}.settings`;
+        const fullpath = path.join(DIRECTORIES.openAI_Settings, filename);
+        writeFileAtomicSync(fullpath, JSON.stringify(request.body, null, 4), 'utf-8');
+        return response.send({ name });
+    });
+}
+
+module.exports = {
+    registerEndpoints,
+}