src/wdetect.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

const fs = require('fs');

function total(pc) {
    t = 0;
    keys = Object.keys(pc);
    keys.forEach((e) => {
        if (e !== "_words") {
            t = t + pc[e];
        }
    })

    return t;
}

module.exports = (pc, text, plangs) => {
    plgkeys = Object.keys(plangs);
    pc["_words"] = {};
    plgkeys.forEach(lk => {
        pc["_words"][lk] = 0;
        if (fs.existsSync("./data/" + lk + ".dat")) {
            langwords = JSON.parse(fs.readFileSync("./data/" + lk + ".dat").toString())["words"];
            textwords = text.split(/[, ]+/);
            textwords.forEach((word) => {
                if (langwords.includes(word)) {
                    pc[lk] = pc["_words"][lk] * 1000;
                    pc["_words"][lk]++;
                }
            })
        }
    });

    return pc;
}