Files
revanced-bots/server/events/addTrainData.js
2022-11-18 15:55:17 +03:00

23 lines
586 B
JavaScript

import { readFileSync, writeFileSync } from 'node:fs';
import { join } from 'node:path';
export default function addTrainData(eventData, __dirname, config) {
const file = readFileSync(
join(__dirname, config.fasttext.trainFile),
'utf-8'
);
const { label, text } = eventData;
if (file.includes(text)) return;
const data = file.split('\n');
const labelIndex = data.findIndex((data) => data.startsWith(label));
data.splice(labelIndex === -1 ? 0 : labelIndex, 0, `${label} ${text}`);
writeFileSync(join(__dirname, config.fasttext.trainFile), data.join('\n'));
return;
}