rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use DIRECTORIES const for comfy workflow path

This commit is contained in:
LenAnderson 2023-11-20 18:26:13 +00:00
parent 46cc04c798
commit 8b5a56a99c
2 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/constants.js
View File

@ -23,6 +23,7 @@ const DIRECTORIES = {
backups: 'backups/',
quickreplies: 'public/QuickReplies',
assets: 'public/assets',
comfyWorkflows: 'public/user/workflows',
};
const UNSAFE_EXTENSIONS = [

17
src/stable-diffusion.js
View File

@ -2,6 +2,7 @@ const fetch = require('node-fetch').default;
const sanitize = require('sanitize-filename');
const { getBasicAuthHeader, delay } = require('./util');
const fs = require('fs');
const { DIRECTORIES } = require('./constants.js');
const writeFileAtomicSync = require('write-file-atomic').sync;
/**
@ -42,11 +43,8 @@ function removePattern(x, pattern) {
}
function getComfyWorkflows() {
if (!fs.existsSync('public/user/workflows')) {
return [];
}
return fs
.readdirSync('public/user/workflows')
.readdirSync(DIRECTORIES.comfyWorkflows)
.filter(file => file[0]!='.' && file.toLowerCase().endsWith('.json'))
.sort(Intl.Collator().compare);
}
@ -461,9 +459,9 @@ function registerEndpoints(app, jsonParser) {
app.post('/api/sd/comfy/workflow', jsonParser, async (request, response) => {
try {
let path = `public/user/workflows/${sanitize(String(request.body.file_name))}`;
let path = `${DIRECTORIES.comfyWorkflows}/${sanitize(String(request.body.file_name))}`;
if (!fs.existsSync(path)) {
path = 'public/user/workflows/Default_Comfy_Workflow.json';
path = `${DIRECTORIES.comfyWorkflows}/Default_Comfy_Workflow.json`;
}
const data = fs.readFileSync(
path,
@ -478,11 +476,8 @@ function registerEndpoints(app, jsonParser) {
app.post('/api/sd/comfy/saveWorkflow', jsonParser, async (request, response) => {
try {
if (!fs.existsSync('public/user/workflows')) {
fs.mkdirSync('public/user/workflows');
}
writeFileAtomicSync(
`public/user/workflows/${sanitize(String(request.body.file_name))}`,
`${DIRECTORIES.comfyWorkflows}/${sanitize(String(request.body.file_name))}`,
request.body.workflow,
'utf8'
);
@ -496,7 +491,7 @@ function registerEndpoints(app, jsonParser) {
app.post('/api/sd/comfy/deleteWorkflow', jsonParser, async (request, response) => {
try {
let path = `public/user/workflows/${sanitize(String(request.body.file_name))}`;
let path = `${DIRECTORIES.comfyWorkflows}/${sanitize(String(request.body.file_name))}`;
if (fs.existsSync(path)) {
fs.unlinkSync(path);
}