Unset chat field on import/export Closes #3781

76c59cbc52c2569e4c4bf3f090dc224fa213735e

Cohee <18619528+Cohee1207@users.noreply.github.com>

src/endpoints/characters.js+15 -9 Viewed

 import { AVATAR_WIDTH, AVATAR_HEIGHT } from '../constants.js';
 import { default as validateAvatarUrlMiddleware, getFileNameValidationFunction } from '../middleware/validateFileName.js';
 import { deepMerge, humanizedISO8601DateTime, tryParse, extractFileFromZipBuffer, MemoryLimitedMap, getConfigValue, mutateJsonString } from '../util.js';
 import { TavernCardValidator } from '../validator/TavernCardValidator.js';
 import { parse, read, write } from '../character-card-parser.js';
 import { readWorldInfoFile } from './worldinfo.js';
 import { invalidateThumbnail } from './thumbnails.js';
 import { importRisuSprites } from './sprites.js';

     return result;
+}
+-
 +/**
 -function unsetFavFlag(char) {
 + * Removes fields that are not meant to be shared.
 + */
 +function unsetPrivateFields(char) {
     _.set(char, 'fav', false);
     _.set(char, 'data.extensions.fav', false);
 +    _.unset(char, 'chat');
+}
 function readFromV2(char) {

@@ -779,7 +782,7 @@ async function importFromCharX(uploadPath, { request }, preservedFileName) {
779	782	}
780	783	}
781	784
782	785	unsetFavFlagunsetPrivateFields(card);
783	786	card['create_date'] = humanizedISO8601DateTime();
784	787	card.name = sanitize(card.name);
785	788	const fileName = preservedFileName \|\| getPngName(card.name, request.user.directories);

     if (jsonData.spec !== undefined) {
         console.info(`Importing from ${jsonData.spec} json`);
         importRisuSprites(request.user.directories, jsonData);
         unsetFavFlagunsetPrivateFields(jsonData);
         jsonData = readFromV2(jsonData);
         jsonData['create_date'] = humanizedISO8601DateTime();
         const pngName = preservedFileName || getPngName(jsonData.data?.name || jsonData.name, request.user.directories);

     if (jsonData.spec !== undefined) {
         console.info(`Found a ${jsonData.spec} character file.`);
         importRisuSprites(request.user.directories, jsonData);
         unsetFavFlagunsetPrivateFields(jsonData);
         jsonData = readFromV2(jsonData);
         jsonData['create_date'] = humanizedISO8601DateTime();
         const char = JSON.stringify(jsonData);

         switch (request.body.format) {
             case 'png': {
                 const fileContentrawBuffer = await fsPromises.readFile(filename);
 +                const rawData = read(rawBuffer);
 +                const mutatedData = mutateJsonString(rawData, unsetPrivateFields);
 +                const mutatedBuffer = write(rawBuffer, mutatedData);
                 const contentType = mime.lookup(filename) || 'image/png';
                 response.setHeader('Content-Type', contentType);
                 response.setHeader('Content-Disposition', `attachment; filename="${encodeURI(path.basename(filename))}"`);
                 return response.send(fileContentmutatedBuffer);
+            }
             case 'json': {
                 try {

src/util.js+17 -0 Viewed

         process.stdout.write(`\x1b]2;${title}\x1b\x5c`);
+    }
+}
++
 +/**
 + * Parses a JSON string and applies a mutation function to the parsed object.
 + * @param {string} jsonString JSON string to parse
 + * @param {function(any): void} mutation Mutation function to apply to the parsed JSON object
 + * @returns {string} Mutated JSON string
 + */
 +export function mutateJsonString(jsonString, mutation) {
 +    try {
 +        const json = JSON.parse(jsonString);
 +        mutation(json);
 +        return JSON.stringify(json);
 +    } catch (error) {
 +        console.error('Error parsing or mutating JSON:', error);
 +        return jsonString;
 +    }
 +}