shadowsocks-manager/plugins/cli/menu/main.js

A shadowsocks manager tool for multi user and traffic control.

'use strict';

const _ = require('lodash');
const config = appRequire('services/config').all();
const isFlowSaverUse = _.get(config, 'plugins.flowSaver.use');
const inquirer = require('inquirer');
const listPort = appRequire('plugins/cli/menu/listPort');
const addPort = appRequire('plugins/cli/menu/addPort');
const listServer = appRequire('plugins/cli/menu/listServer');
const addServer = appRequire('plugins/cli/menu/addServer');
const flow = appRequire('plugins/cli/menu/flow');
const index = appRequire('plugins/cli/index');

const main = [
  {
    type: 'list',
    name: 'mainMeun',
    message: () => {
      return 'Main menu [' + index.getManagerAddress().host + ':' + index.getManagerAddress().port + ']';
    },
    choices: ['add port', 'list port'],
  }
];

if(isFlowSaverUse) {
  main[0].choices.push('add server', 'list server', 'flow');
}

const mainMeun = () => {
  console.log();
  inquirer.prompt(main)
  .then(success => {
    if(success.mainMeun === 'list port') {
      return listPort.list();
    } else if(success.mainMeun === 'add port') {
      return addPort.add();
    } else if(success.mainMeun === 'list server') {
      return listServer.list();
    } else if(success.mainMeun === 'add server') {
      return addServer.add();
    } else if(success.mainMeun === 'flow') {
      return flow.getFlow();
    }
  }).then(() => {
    return mainMeun();
  })
  .catch(() => {
    return mainMeun();
  });
};

mainMeun();