build.ts

import * as fs from 'fs';
import * as marked from 'marked';
import * as program from 'commander';
import {findIndex} from 'lodash';

interface RowItem {
    id:string;
    title:string;
    url:string;
    difficulty:string;
    have_md:boolean;
}
function insert_to_table(token:any, item:RowItem):RowItem[]{
    const cells: RowItem[] = token.cells.map((row:string[]) => {
        const title = row[1].match(/\[.*\]/)[0];
        const url = row[1].match(/\(https.*\)/)[0];
        return {
            id:row[0],
            title:title.substring(1, title.length - 1),
            url:url.substring(1, url.length - 1),
            difficulty:row[3],
            have_md:/md/.test(row[2]),
        };
    });

    if (findIndex(cells, (row) => {
        return row.id == item.id;
    }) !== -1) {
        console.log('this item had insert');
        return cells;
    }
    cells.push(item);
    cells.sort((item1, item2) => {
        return parseInt(item1.id) - parseInt(item2.id);
    })
    return cells;
}

program.option('-i, --id <id>','The leetcode No', (id) => {
    const readMe = fs.readFileSync('./README.md').toString();
    const tokens = marked.lexer(readMe);
    const table_token = tokens[6];
    const item:RowItem = require('./src/' + id + '/index.js');
    const table_data = insert_to_table(table_token, item);
    const md_table_str = table_data.reduce((res, cur) => {
        let row = `|${cur.id}|[${cur.title}](${cur.url})|[link](/src/${cur.id}/index.js)|${cur.difficulty}|\n`;
        if (cur.have_md) {
            row = `|${cur.id}|[${cur.title}](${cur.url})|[link](/src/${cur.id}/index.md)|${cur.difficulty}|\n`;
        }
        return res + row;
    }, '');

    const classificationData = table_data.reduce((res:any, cur) => {
        res[cur.difficulty.toLowerCase()] ++;
        return res;
    }, {hard:0, easy:0, medium:0});

    let md_tempalte = fs.readFileSync('./README_TEMPLATE.md').toString();
    md_tempalte = md_tempalte.replace('[easy]', classificationData['easy']);
    md_tempalte = md_tempalte.replace('[medium]', classificationData['medium']);
    md_tempalte = md_tempalte.replace('[hard]', classificationData['hard']);
    md_tempalte = md_tempalte.replace('[table_data]', md_table_str);
    fs.writeFile('./README.md', md_tempalte, (err) => {
        if (err) {
            console.error(err);
            return;
        }
        console.log('saved success')
    });
});

program.parse(process.argv);