shadowsocks-manager/lib/plugins/webgui_ref/index.js

A shadowsocks manager tool for multi user and traffic control.

"use strict";

const knex = appRequire('init/knex').knex;

const getRefSetting = async () => {
  const setting = await knex('webguiSetting').select().where({
    key: 'webgui_ref'
  }).then(success => {
    if (!success.length) {
      return Promise.reject('settings not found');
    }

    success[0].value = JSON.parse(success[0].value);
    return success[0].value;
  });
  return setting;
};

const addRefCode = async (userId, max = 3) => {
  const code = Math.random().toString().substr(2, 10);
  await knex('webgui_ref_code').insert({
    code,
    sourceUserId: userId,
    maxUser: max,
    time: Date.now()
  });
};

const visitRefCode = async code => {
  const setting = await getRefSetting();

  if (!setting.useRef) {
    return false;
  }

  const codeInfo = (await knex('webgui_ref_code').where({
    code
  }))[0];

  if (!codeInfo) {
    return false;
  }

  const sourceUserInfo = (await knex('user').where({
    id: codeInfo.sourceUserId
  }))[0];

  if (!sourceUserInfo) {
    return;
  }

  const currentRefUser = await knex('webgui_ref').where({
    codeId: codeInfo.id
  });

  if (currentRefUser.length >= codeInfo.maxUser) {
    return false;
  }

  await knex('webgui_ref_code').where({
    code
  }).increment('visit', 1);
  return true;
};

const checkRefCodeForSignup = async code => {
  const setting = await getRefSetting();

  if (!setting.useRef) {
    return false;
  }

  if (!setting.useWhenSignupClose) {
    return false;
  }

  const codeInfo = (await knex('webgui_ref_code').where({
    code
  }))[0];

  if (!codeInfo) {
    return false;
  }

  const sourceUserInfo = (await knex('user').where({
    id: codeInfo.sourceUserId
  }))[0];

  if (!sourceUserInfo) {
    return;
  }

  const currentRefUser = await knex('webgui_ref').where({
    codeId: codeInfo.id
  });

  if (currentRefUser.length >= codeInfo.maxUser) {
    return false;
  }

  return true;
};

const addRefUser = async (code, userId) => {
  try {
    const setting = await getRefSetting();

    if (!setting.useRef) {
      return;
    }

    const codeInfo = (await knex('webgui_ref_code').where({
      code
    }))[0];

    if (!codeInfo) {
      return;
    }

    const sourceUserInfo = (await knex('user').where({
      id: codeInfo.sourceUserId
    }))[0];

    if (!sourceUserInfo) {
      return;
    }

    const currentRefUser = await knex('webgui_ref').where({
      codeId: codeInfo.id
    });

    if (currentRefUser.length >= codeInfo.maxUser) {
      return;
    }

    await knex('webgui_ref').insert({
      codeId: codeInfo.id,
      userId,
      time: Date.now()
    });
    await knex('user').update({
      group: sourceUserInfo.group
    }).where({
      id: userId
    });
  } catch (err) {
    console.error(err);
  }
};

exports.addRefCode = addRefCode;
exports.visitRefCode = visitRefCode;
exports.addRefUser = addRefUser;
exports.checkRefCodeForSignup = checkRefCodeForSignup;

const setDefaultValue = (key, value) => {
  knex('webguiSetting').select().where({
    key
  }).then(success => {
    if (success.length) {
      return;
    }

    return knex('webguiSetting').insert({
      key,
      value: JSON.stringify(value)
    });
  }).then();
};

setDefaultValue('webgui_ref', {
  useRef: false,
  useWhenSignupClose: false,
  refNumber: 1,
  refUserNumber: 1
});