shadowsocks-manager/plugins/telegram/flow.js

A shadowsocks manager tool for multi user and traffic control.

const telegram = appRequire('plugins/telegram/index').telegram;
const managerAddress = appRequire('plugins/telegram/managerAddress');
const flow = appRequire('plugins/telegram/flowSaver');

const log4js = require('log4js');
const logger = log4js.getLogger('telegram');

const flowNumber = (number) => {
  if(number < 1000) return number + ' B';
  else if(number < 1000 * 1000) return (number / 1000).toFixed(0) + ' KB';
  else if(number < 1000 * 1000 * 1000) return (number / 1000000).toFixed(1) + ' MB';
  else if(number < 1000 * 1000 * 1000 * 1000) return (number / 1000000000).toFixed(3) + ' GB';
};

const getFlow = (message, time) => {
  const start = Date.now() - time;
  const end = Date.now();
  flow.getFlow(managerAddress.get().host, managerAddress.get().port, start, end)
  .then(ports => {
    let str = '';
    if(ports.length === 0) {
      str = 'No flows.';
    } else {
      str += `${managerAddress.get().host}:${managerAddress.get().port}\n\n`;
      ports.forEach(port => {
        str += port.port + ', ' + flowNumber(port.sumFlow) + '\n';
      });
    }
    telegram.emit('send', message, str);
  }).catch(err => {
    logger.error(err);
  });
};

telegram.on('manager', message => {

  const minReg = new RegExp(/^flow(\d{0,2})min$/);
  const hourReg = new RegExp(/^flow(\d{0,2})hour$/);

  if(message.message.text === 'flow') {
    getFlow(message, 10 * 60 * 1000);
  } else if(message.message.text.match(minReg)) {
    const reg = message.message.text.match(minReg);
    const time = reg[1] * 60 * 1000;
    getFlow(message, time);
  } else if(message.message.text.match(hourReg)) {
    const reg = message.message.text.match(hourReg);
    const time = reg[1] * 60 * 60 * 1000;
    getFlow(message, time);
  }
});