简易命令行控制的「微信机器人」及群聊信息过滤实现

// ./index.js
import tokenJSON  from './token.json'
import infoJSON  from './package.json'
console.log(`Running LazyBot ${infoJSON.version}...`)
console.log(`Trying to detect 'token' from './token.json'`)
if (!tokenJSON.token) {
    console.log(`Unable to acquire 'token' field from './token.json'. If you don't possess token of Wechaty, turn to https://github.com/juzibot/Welcome/wiki/Everything-about-Wechaty for more information.`)
    return
}
console.log(`Detect 'token' from './token.json': ${tokenJSON.token}`)
import main  from './main.js'
main(tokenJSON.token, './bot-settings.json')

// ./main.js
module.exports = function(token, botSettingFile) {
    // Import Wechaty
    // ...
    // Import Settings
    // ...
    import botSettings  from './bot-settings.json';
    import schedule  from 'node-schedule';
    // Import Command System
    import commandUnits  from './commands.js';
    import utils  from './utils.js';
    // Construct Puppet
    const puppet = new PuppetPadplus({ token });
    const name = "LazyBot";
    const bot = new Wechaty({ puppet, name });
    // Initialize Writing Settings
    // Writing Settings of Bot to local file every 10 min.
    // ...
    // Initialize Bot Settings
    // ...
    // Begin Bot
    bot
    .on('scan', function (qrcode, status) {
        if (status === ScanStatus.Waiting) {
            QrcodeTerminal.generate(qrcode);
        }
    })
    .on('login', function (user) {
        console.log(`${user.name()} Login`);
    })
    .on('logout', function (user) {
        console.log(`${user.name()} Logout`);
    })
    .on('room-invite', async roomInvitation => {
        // ...
    })
    .on('room-leave', async (room, leaverList, remover) => {
        // ...
    })
    .on('friendship', async (friendship) => {
        // ...
    })
    .on('message', async function(message){
        // ...
    });
    bot.start();
}

// ./utils.js
/**
 * Test whether message is a command
 * @param {string} message
 * @returns {boolean}
 */
function isCommand(message) {
    return message[0] === '.';
}
// ./main.js -> 'message' event function
if (message.room()) {
    const id = message.room().id;
    // ...
    // Initialize settings of group
    if (!botSettings["groups"][id]) botSettings["groups"][id] = { "switch": false, "monitors": {} };
    // ...
    if (utils.isCommand(text)) {
        // Enable or Disable LazyBot
        if (text === ".enable-lazybot") { botSettings["groups"][id]["switch"] = true; return;}
        else if (text === ".disable-lazybot") { botSettings["groups"][id]["switch"] = false;return; }
        // ... Parse Command if ("switch" === true) and Do Something
    }
}

// ./utils.js
/**
 * Parse message as command
 * ' ', which is not between '' or "", is perceived as separator
 * subCommand beginning with '-' is perceived as non-boolean flag
 * subCommand beginning with '--' is perceived as boolean flag
 * @param {string} message
 */
function parseCommand (message) {
    // ... function for checking boolean flag
    // ... function for checking non-boolean flag
    const ret = {
        mainCommand:"",
        flags:{},
        args:[],
        err: ""
    };
    const _commands = message.split(' ');
    const commands = [];
    // After Splitting message by ' ', merge items like ["a, b"], and delete quotes.
    // e.g. ".abc 'abc e'" => [".abc", "'abc", "e'"] => [".abc", "abc e"]
    // ...
    // Parse Commands
    // ...
    return ret;
    }

// ./commands.js
/**
 * Command ".help" is reserved for listing all possible commands with their descriptions.
 */
class CommandUnits{
    RegisterCommand(command, description) {
        this.commands[command.mainCommand] = command;
        this.descriptions[command.mainCommand] = description;
    }
    RegisterRegexCommand(regex, command, display, description) {
        this.regexCommands.push({
            regex, command, display, description
        });
    }
    CallCommand(parsedCommands, message, botSettings){
        if (parsedCommands.mainCommand === ".help") {
            // Display All Possible Commands
            // ...
            return ret;
        }
        // Test for complete Match first
        if (this.commands[parsedCommands.mainCommand]) return this.commands[parsedCommands.mainCommand].Call(parsedCommands, message, botSettings);
        for (const regexCommand of this.regexCommands) {
            if (regexCommand.regex.test(parsedCommands.mainCommand)) return regexCommand.command.Call(parsedCommands, message, botSettings);
        }
        return `Unrecognized Command: ${parsedCommands.mainCommand}.`;
    }
    Ready(){
        // Sort Regex Command to Display more beautifully
        // ...
    }
    constructor() {
        this.commands = {};
        this.descriptions = {};
        this.regexCommands = [];
    }
}

// ./commands.js
class Command{
    _parse(commands) {
        const parsedCommands = {
            mainCommand: commands.mainCommand,
            err: "",
            args: [],
            flags:{}
        };
        // Parse parsed Commands to suit needs of this specific instance
        // ...
        return parsedCommands;
    }
    Call(commands, message, botSettings) {
        const parsedCommands = this._parse(commands);
        if (parsedCommands.err !== "") return parsedCommands.err;
        if (parsedCommands.flags["help"] || parsedCommands.flags["h"] !== undefined) return this.help();
        return this.caller(parsedCommands, message , botSettings);
    }
    help() {
        // Return Help Information for this instance
        // ...
    }
    /**
     *
     * @param {string} mainCommand
     * @param {Array<{flag: string, description: string}>} booleanFlags
     * @param {Array<{flag: string, description: string}>} nonBooleanFlags
     * @param {(commands, message, botSettings) => string} caller
     */
    constructor(mainCommand, booleanFlags, nonBooleanFlags, caller) {
        this.mainCommand = mainCommand;
        this.booleanFlags = { flags: [], descriptions:{}};
        this.nonBooleanFlags = { flags: [], descriptions:{} };
        for (const flag of booleanFlags) {
            this.booleanFlags.flags.push(flag["flag"]);
            this.booleanFlags.descriptions[flag["flag"]] = flag["description"] || "";
        }
        // Add `--help` command
        // ...
        for (const flag of nonBooleanFlags) {
            this.nonBooleanFlags.flags.push(flag["flag"]);
            this.nonBooleanFlags.descriptions[flag["flag"]] = flag["description"] || "";
        }
        // Add `-h` command
        // ...
        this.caller = caller;
    }
}

// 让 Bot 识别 .hi 指令，对注册过的用户回复“欢迎”的消息。
commandUnits.RegisterCommand(new Command(".hi",[],[],
function (commands, message, botSettings) {
    if (!botSettings["users"][message.from().id]) {
        console.log(`Invalid .hi command from unregistered user ${message.from().id}`);
        return registerPrompt;
    }
    return `Hello, ${message.from().name()}! What a nice day!`;
}), "Say Hi to Bot");

// 让 Bot 识别 .register 指令， 以接受用户的注册。对注册过的用户，提示已经注册过，除非显式加入`--force`参数。
commandUnits.RegisterCommand(new Command(".register",
[
    {
        flag:"force",
        description:"Force to register account. In this case, the original account will be deleted, if have registered."
    }
], [],
function (commands, message, botSettings) {
    if (message.room()) return "";
    const id = message.from().id;
    if (botSettings["users"][id] && !commands.flags["force"]) {
        console.log("Account",id,"Tried to register again");
        return `Have registered Account ${id}`;
    }
    botSettings["users"][id] = {};
    console.log("Account Registered",id);
    return `Hello, ${message.from().name()}!`;
}), "Register Account");

// ./command.js
commandUnits.RegisterCommand(new Command(".monitor",
[
    {flag: "off", description: "Turn Off The Monitor"},
    {flag: "whitelist", description: "Toggle the WhiteList"},
    {flag: "blacklist", description: "Toggle the BlackList"}
],
[],
async function (commands, message, botSettings) {
    // Check whether `.monitor` is valid in current environment
    // ...
    const id = message.room().id;
    const userId = message.from().id;
    if (!botSettings["groups"][id]["monitors"][userId]) {
        // Initialize User Settings in Monitors
        // ...
    }
    // Parse Options
    // ...
    return "";
}), "Monitor Group Chat");

commandUnits.RegisterCommand(new Command(".whitelist",
[
    {flag:"delete", description: "Whether to delete rules instead of adding"},
    {flag:"show", description: "Display WhiteList"},
],
[
    {flag:"u", description: "Add/Delete User to/from White List"},
    {flag:"m", description: "Add/Delete Message Filter (Regex) to/from White List"}
],
async function (commands, message, botSettings) {
    // Check whether `.whitelist` is valid in current environment
    // ...
    const topic = await message.room().topic();
    const id = message.room().id;
    const userId = message.from().id;
    if (commands.flags["show"]) {
        // Send Setting Information of White List to User
        // ...
        return "";
    }
    if (commands.flags["u"]) {
        // Get User's Id from its name and Add / Delete it to / from settings if appropriate.
        // ...
        return "";
    }
    if (commands.flags["m"]) {
        // Add / Delete filtering rule to / from settings if appropriate.
        // ...
        return "";
    }
}), "Manipulate WhiteList of Message Filter of Group Chat");