This commit is contained in:
SillyLossy
2023-05-28 13:42:09 +03:00
6 changed files with 105 additions and 6 deletions

View File

@@ -1593,6 +1593,63 @@ app.post("/dupecharacter", jsonParser, async function (request, response) {
}
});
app.post("/exportchat", jsonParser, async function (request, response) {
if (!request.body.file || (!request.body.avatar_url && request.body.is_group === false)) {
return response.sendStatus(400);
}
const pathToFolder = request.body.is_group
? directories.groupChats
: path.join(directories.chats, String(request.body.avatar_url).replace('.png', ''));
//let charname = String(sanitize(request.body.avatar_url)).replace('.png', '');
let filename = path.join(pathToFolder, request.body.file);
let exportfilename = path.join(pathToFolder, request.body.exportfilename)
if (!fs.existsSync(filename)) {
const errorMessage = {
message: `Could not find JSONL file to export. Source chat file: ${filename}. Intended destination file: ${exportfilename}.`
}
console.log(errorMessage.message);
return response.status(404).json(errorMessage);
}
if (fs.existsSync(exportfilename)) {
const errorMessage = {
message: `File by that name already exists. Export chat aborted.`
}
console.log(errorMessage.message);
return response.status(400).json(errorMessage);
}
try {
const readline = require('readline');
const fs = require('fs');
const readStream = fs.createReadStream(filename);
const writeStream = fs.createWriteStream(exportfilename);
const rl = readline.createInterface({
input: readStream,
});
rl.on('line', (line) => {
const data = JSON.parse(line);
if (data.mes) {
const name = data.name;
const message = data.mes.replace(/\r?\n/g, '\n');
writeStream.write(`${name}: ${message}\n\n`);
}
});
rl.on('close', () => {
writeStream.end();
});
//fs.promises.copyFile(filename, exportfilename)
const successMessage = {
message: `Chat exported as ${exportfilename}`
}
console.log(`Chat exported as ${exportfilename}`);
return response.status(200).json(successMessage);
}
catch (err) {
console.log("chat export failed.")
console.log(err);
return response.sendStatus(400);
}
})
app.post("/exportcharacter", jsonParser, async function (request, response) {
if (!request.body.format || !request.body.avatar_url) {
return response.sendStatus(400);