shadowsocks-manager/lib/plugins/telegram/flow.js

A shadowsocks manager tool for multi user and traffic control.

"use strict";

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);
  }
});