diff --git a/server.js b/server.js index a0b24bb5b..7571f1cad 100644 --- a/server.js +++ b/server.js @@ -3603,7 +3603,7 @@ require('./src/endpoints/assets').registerEndpoints(app, jsonParser); require('./src/endpoints/sprites').registerEndpoints(app, jsonParser, urlencodedParser); // Custom content management -require('./src/endpoints/content-manager').registerEndpoints(app, jsonParser); +app.use('/api/content', require('./src/endpoints/content-manager').router); // Stable Diffusion generation require('./src/endpoints/stable-diffusion').registerEndpoints(app, jsonParser); diff --git a/src/endpoints/content-manager.js b/src/endpoints/content-manager.js index cc4796980..191b4f4ce 100644 --- a/src/endpoints/content-manager.js +++ b/src/endpoints/content-manager.js @@ -1,8 +1,10 @@ const fs = require('fs'); const path = require('path'); +const express = require('express'); const fetch = require('node-fetch').default; const sanitize = require('sanitize-filename'); const { getConfigValue } = require('../util'); +const { jsonParser } = require('../express-common'); const contentDirectory = path.join(process.cwd(), 'default/content'); const contentLogPath = path.join(contentDirectory, 'content.log'); const contentIndexPath = path.join(contentDirectory, 'index.json'); @@ -302,62 +304,57 @@ function parseJannyUrl(url) { return uuid; } -/** - * Registers endpoints for custom content management - * @param {import('express').Express} app Express app - * @param {any} jsonParser JSON parser middleware - */ -function registerEndpoints(app, jsonParser) { - app.post('/api/content/import', jsonParser, async (request, response) => { - if (!request.body.url) { - return response.sendStatus(400); - } +const router = express.Router(); - try { - const url = request.body.url; - let result; - let type; +router.post('/import', jsonParser, async (request, response) => { + if (!request.body.url) { + return response.sendStatus(400); + } - const isJannnyContent = url.includes('janitorai'); - if (isJannnyContent) { - const uuid = parseJannyUrl(url); - if (!uuid) { - return response.sendStatus(404); - } + try { + const url = request.body.url; + let result; + let type; - type = 'character'; - result = await downloadJannyCharacter(uuid); - } else { - const chubParsed = parseChubUrl(url); - type = chubParsed?.type; - - if (chubParsed?.type === 'character') { - console.log('Downloading chub character:', chubParsed.id); - result = await downloadChubCharacter(chubParsed.id); - } - else if (chubParsed?.type === 'lorebook') { - console.log('Downloading chub lorebook:', chubParsed.id); - result = await downloadChubLorebook(chubParsed.id); - } - else { - return response.sendStatus(404); - } + const isJannnyContent = url.includes('janitorai'); + if (isJannnyContent) { + const uuid = parseJannyUrl(url); + if (!uuid) { + return response.sendStatus(404); } - if (result.fileType) response.set('Content-Type', result.fileType); - response.set('Content-Disposition', `attachment; filename="${result.fileName}"`); - response.set('X-Custom-Content-Type', type); - return response.send(result.buffer); - } catch (error) { - console.log('Importing custom content failed', error); - return response.sendStatus(500); + type = 'character'; + result = await downloadJannyCharacter(uuid); + } else { + const chubParsed = parseChubUrl(url); + type = chubParsed?.type; + + if (chubParsed?.type === 'character') { + console.log('Downloading chub character:', chubParsed.id); + result = await downloadChubCharacter(chubParsed.id); + } + else if (chubParsed?.type === 'lorebook') { + console.log('Downloading chub lorebook:', chubParsed.id); + result = await downloadChubLorebook(chubParsed.id); + } + else { + return response.sendStatus(404); + } } - }); -} + + if (result.fileType) response.set('Content-Type', result.fileType); + response.set('Content-Disposition', `attachment; filename="${result.fileName}"`); + response.set('X-Custom-Content-Type', type); + return response.send(result.buffer); + } catch (error) { + console.log('Importing custom content failed', error); + return response.sendStatus(500); + } +}); module.exports = { checkForNewContent, - registerEndpoints, getDefaultPresets, getDefaultPresetFile, + router, };