@sex-pomelo/sex-pomelo-cli/lib/cli.js

Pomelo command line client for managing pomelo

let readline = require('readline');
let adminClient = require('@sex-pomelo/sex-pomelo-admin').adminClient;
let command = require('./command')();
let consts = require('./consts');
let util = require('./util');
const argv = require('minimist')(process.argv);

let username = argv['u'] = argv['u'] || 'monitor';
let password = argv['p'] = argv['p'] || 'monitor';
let host = argv['h'] = argv['h'] || 'localhost';
let port = argv['P'] = argv['P'] || 3005;
let context = 'all';
let client = null;
let g_bCli = false; // 控制台是否已经启动
let g_rl = null;
module.exports = doConnect;

function doConnect() {
  client = new adminClient({
    username: username,
    password: password,
    md5: true
  });
  let id = 'pomelo_cli_' + Date.now();
  client.connect(id, host, port, function(err) {
    if (err) {
      util.log('\n' + err + '\n');
      process.exit(0);
    } else {
      if( g_bCli === false){
        let ASCII_LOGO = consts.ASCII_LOGO;
        for (let i = 0; i < ASCII_LOGO.length; i++) {
          util.log(ASCII_LOGO[i]);
        }
  
        let WELCOME_INFO = consts.WELCOME_INFO;
        for (let i = 0, l = WELCOME_INFO.length; i < l; i++) {
          util.log(WELCOME_INFO[i]);
        }
        startCli();
      }else{
        g_rl.prompt();
      }
    }
  });
  client.on('close', function() {
    client.socket.disconnect();
    util.log('\ndisconnect from master');
    process.exit(0);
  });
}

function startCli() {
  g_bCli = true;
  let rl = readline.createInterface(process.stdin, process.stdout, completer);
  g_rl = rl;
  let PROMPT = username + consts.PROMPT + context + '>';
  rl.setPrompt(PROMPT);
  rl.prompt();

  rl.on('line', function(line) {
    let key = line.trim();
    if (!key) {
      util.help();
      rl.prompt();
      return;
    }
    switch (key) {
      case 'help':
        util.help();
        rl.prompt();
        break;
      case '?':
        util.help();
        rl.prompt();
        break;
      case 'quit':
        command.quit(rl);
        break;
      case 'kill':
        command.kill(rl, client);
        break;
      default:
        command.handle(key, {
          user: username
        }, rl, client);
        break;
    }
  }).on('close', function() {
    util.log('bye ' + username);
    process.exit(0);
  });
}

function completer(line) {
  line = line.trim();
  let completions = consts.COMANDS_COMPLETE;
  let hits = [];
  // commands tab for infos 
  if (consts.COMPLETE_TWO[line]) {
    if (line === "show") {
      for (let k in consts.SHOW_COMMAND) {
        hits.push(k);
      }
    } else if (line === "help") {
      for (let k in consts.COMANDS_COMPLETE_INFO) {
        hits.push(k);
      }
    } else if (line === "enable" || line === "disable") {
      hits.push("app");
      hits.push("module");
    } else if (line === "dump") {
      hits.push("memory");
      hits.push("cpu");
    }
  }

  hits = util.tabComplete(hits, line, consts.COMANDS_COMPLETE_INFO, "complete");
  hits = util.tabComplete(hits, line, consts.COMANDS_COMPLETE_INFO, "help");
  hits = util.tabComplete(hits, line, consts.SHOW_COMMAND, "show");
  hits = util.tabComplete(hits, line, null, "enable");
  hits = util.tabComplete(hits, line, null, "disable");
  hits = util.tabComplete(hits, line, null, "disable");
  hits = util.tabComplete(hits, line, null, "dump");
  hits = util.tabComplete(hits, line, null, "use");
  hits = util.tabComplete(hits, line, null, "stop");
   
  // show all completions if none found
  return [hits.length ? hits : completions, line];
}