gramjs/gramjs_generator/parsers/methods.js

const fs = require('fs');
const csvParse = require('csv-parse/lib/sync');

const Usability = {
    UNKNOWN: 0,
    USER: 1,
    BOT: 2,
    BOTH: 4,
};

class MethodInfo {
    constructor(name, usability, errors, friendly) {
        this.name = name;
        this.errors = errors;
        this.friendly = friendly;

        if (usability.toUpperCase() in Usability) {
            this.usability = Usability[usability.toUpperCase()];
        } else {
            throw new Error(
                `Usability must be either user, bot, both or unknown, not ${usability}`
            );
        }
    }
}

/**
 * Parses the input CSV file with columns (method, usability, errors)
 * and yields `MethodInfo` instances as a result.
 */
const parseMethods = function*(csvFile, friendlyCsvFile, errorsDict) {
    const rawToFriendly = {};
    const f1 = csvParse(
        fs.readFileSync(friendlyCsvFile, { encoding: 'utf-8' })
    );

    for (const [ns, friendly, rawList] of f1.slice(1)) {
        for (const raw of rawList.split(' ')) {
            rawToFriendly[raw] = [ns, friendly];
        }
    }

    const f2 = csvParse(fs.readFileSync(csvFile, { encoding: 'utf-8' })).slice(
        1
    );

    for (let line = 0; line < f2.length; line++) {
        let [method, usability, errors] = f2[line];

        errors = errors
            .split(' ')
            .filter(Boolean)
            .map(x => {
                if (x && !(x in errorsDict)) {
                    throw new Error(
                        `Method ${method} references unknown errors ${errors}`
                    );
                }

                return errorsDict[x];
            });

        const friendly = rawToFriendly[method];
        delete rawToFriendly[method];
        yield new MethodInfo(method, usability, errors, friendly);
    }
};

module.exports = {
    Usability,
    MethodInfo,
    parseMethods,
};