1
0
Fork 0
tt9/scripts/remove-foreign-words.js
sspanak ef2f92c0ac language processing scripts improvements
* remove-foreign-words script now supports blacklist and whitelist mode

  * fixed .domain sorting

  * clean'n'sort script
2024-02-09 20:07:05 +02:00

87 lines
2.2 KiB
JavaScript

const { basename } = require('path');
const { createReadStream, existsSync } = require('fs');
const { createInterface } = require('readline');
function printHelp() {
console.log(`Usage ${basename(process.argv[1])} --blacklist|whitelist DICTIONARY_LOCALE DICTIONARY.TXT FOREIGN_WORDS_LOCALE FOREIGN-WORD-DICTIONARY.txt`);
console.log('Removes foreign words from a dictionary. "blacklist" and "whitelist" determine how the FOREIGN-WORD-DICTIONARY.txt is used.');
}
function validateInput() {
if (process.argv.length < 7) {
printHelp();
process.exit(1);
}
if (process.argv[2] !== '--blacklist' && process.argv[2] !== '--whitelist') {
console.error(`Failure! You must specify whether to use the foreign words file as a blacklist or a whitelist."`);
process.exit(3);
}
if (!existsSync(process.argv[4])) {
console.error(`Failure! Could not find words file "${process.argv[4]}."`);
process.exit(2);
}
if (!existsSync(process.argv[6])) {
console.error(`Failure! Could not find foreign words file "${process.argv[6]}."`);
process.exit(2);
}
return {
isBlacklist: process.argv[2] === '--blacklist',
locale: process.argv[3],
fileName: process.argv[4],
foreignWordsLocale: process.argv[5],
foreignWordsFileName: process.argv[6]
};
}
async function work({ isBlacklist, locale, fileName, foreignWordsLocale, foreignWordsFileName }) {
const foreignWords = new Set();
let lineReader = createInterface({ input: createReadStream(foreignWordsFileName) });
for await (const line of lineReader) {
foreignWords.add(line.toLocaleLowerCase(foreignWordsLocale));
}
const goodWords = new Set();
lineReader = createInterface({ input: createReadStream(fileName) });
for await (const line of lineReader) {
if (typeof line !== 'string' || line.length === 0) {
continue;
}
const wordKey = line.toLocaleLowerCase(locale);
if (
(!isBlacklist && foreignWords.has(wordKey))
|| (isBlacklist && !foreignWords.has(wordKey))
) {
goodWords.add(line);
}
}
return goodWords;
}
function printWords(wordList) {
if (wordList instanceof Set) {
wordList.forEach(w => console.log(w));
}
}
/** main **/
work(validateInput())
.then(words => printWords(words))
.catch(e => console.error(e));