shadowsocks-manager/plugins/flowSaver/server.js

A shadowsocks manager tool for multi user and traffic control.

const knex = appRequire('init/knex').knex;
const manager = appRequire('services/manager');
const accountFlow = appRequire('plugins/account/accountFlow');

const add = async options => {
  const { name, host, port, password, method, scale = 1, comment = '', shift = 0 } = options;
  const [ serverId ] = await knex('server').insert({
    name,
    comment,
    host,
    port,
    password,
    method,
    scale,
    shift,
  });
  accountFlow.addServer(serverId);
  return [ serverId ];
};

const del = (id) => {
  return knex.transaction(trx => {
    return knex('server').transacting(trx).where({ id }).delete()
    .then(() => knex('saveFlow').transacting(trx).where({ id }).delete())
    .then(trx.commit)
    .catch(trx.rollback);
  });
};

const edit = async options => {
  const { id, name, host, port, password, method, scale = 1, comment = '', shift = 0, check } = options;
  const serverInfo = await knex('server').where({ id }).then(s => s[0]);
  if(serverInfo.shift !== shift) {
    const accounts = await knex('account_plugin').where({});
    (async server => {
      for(account of accounts) {
        await manager.send({
          command: 'del',
          port: account.port + server.shift,
        }, {
          host: server.host,
          port: server.port,
          password: server.password,
        }).catch();
      }
    })(serverInfo);
  }
  if(check) { accountFlow.editServer(id); }
  return knex('server').where({ id }).update({
    name,
    comment,
    host,
    port,
    password,
    method,
    scale,
    shift,
  });
};

const list = async (options = {}) => {
  const serverList = await knex('server').select([
    'id',
    'name',
    'host',
    'port',
    'password',
    'method',
    'scale',
    'comment',
    'shift',
  ]).orderBy('name');
  if(options.status) {
    const serverStatus = [];
    const getServerStatus = (server, index) => {
      return manager.send({
        command: 'version',
      }, {
        host: server.host,
        port: server.port,
        password: server.password,
      }).then(success => {
        return { status: success.version, isGfw: success.isGfw, index };
      }).catch(error => {
        return { status: -1, index };
      });
    };
    serverList.forEach((server, index) => {
      serverStatus.push(getServerStatus(server, index));
    });
    const status = await Promise.all(serverStatus);
    status.forEach(f => {
      serverList[f.index].status = f.status;
      serverList[f.index].isGfw = !!f.isGfw;
    });
  }
  return serverList;
};

exports.add = add;
exports.del = del;
exports.edit = edit;
exports.list = list;