occaecatidicta/tools/omelox-cli/lib/command.ts

import { consts } from './consts';
import * as util from './util';
require('cliff');
import * as fs from 'fs';
import { isFunction } from 'util';
import { AdminClient } from 'omelox-admin';
import { ReadLine } from 'readline';

export interface ICommand {
    handle(agent: AgentCommand, comd: string, argv: string, msg: {[key: string]: string}, rl: ReadLine, client: AdminClient): void;
}

export class AgentCommand {
    commands: {[key: string]: () => ICommand} = {};
    Context = 'all';

    constructor() {
        this.init();
    }
    init() {
        let self: AgentCommand = this;
        fs.readdirSync(__dirname + '/commands').forEach(function (filename) {
            if (/\.js$/.test(filename)) {
                let name = filename.substr(0, filename.lastIndexOf('.'));
                let _command = require('./commands/' + name).default;
                if (isFunction(_command)) {
                    self.commands[name] = _command;
                }
            }
        });
    }

    handle(argv: string, msg: any, rl: ReadLine, client: AdminClient): void {
        let self = this;
        let argvs = util.argsFilter(argv);
        let comd = argvs[0];
        let comd1 = argvs[1] || '';

        comd1 = comd1.trim();
        let m = this.commands[comd];
        if (m) {
            let _command = m();
            _command.handle(self, comd1, argv, msg, rl, client);
        } else {
            util.errorHandle(argv, rl);
        }
    }

    quit(rl: ReadLine) {
        rl.emit('close');
    }

    kill(rl: ReadLine, client: AdminClient) {
        rl.question(consts.KILL_QUESTION_INFO, function (answer) {
            if (answer === 'yes') {
                client.request(consts.CONSOLE_MODULE, {
                    signal: 'kill'
                }, function (err: Error, data: any) {
                        if (err) console.log(err);
                        rl.prompt();
                    });
            } else {
                rl.prompt();
            }
        });
    }

    getContext() {
        return this.Context;
    }

    setContext(context: string) {
        this.Context = context;
    }
}


export default function () {
    return new AgentCommand();
}