push-oppo/lib/push-oppo.js

push-oppo

const axios = require('axios');
const _ = require('lodash');
const querystring = require('querystring');
const crypto = require('crypto');

class Oppo {
  constructor(options = {}) {
    options.getTokenUrl = options.getTokenUrl || 'https://api.push.oppomobile.com/server/v1/auth';
    options.saveMessageUrl = options.saveMessageUrl || 'https://api.push.oppomobile.com/server/v1/message/notification/save_message_content';
    options.pushUrl = options.pushUrl || 'https://api.push.oppomobile.com/server/v1/message/notification/broadcast';
    options.maxLength = options.maxLength || 1000;
    options.timeout = options.timeout || 300000;

    if (!options.appKey) throw new Error('oppo appKey 不能为空');
    if (!options.masterSecret) throw new Error('oppo masterSecret 不能为空');

    this.options = options;
  }

  async sleep(time) {
    return new Promise((reslove) => {
      setTimeout(() => {
        reslove({});
      }, time);
    });
  }

  async saveMessage(data, auth_token) {
    const params = _.merge({
      app_message_id: +new Date()
    }, data);
    return axios({
      url: this.options.saveMessageUrl,
      method: 'POST',
      timeout: this.options.timeout,
      data: querystring.stringify(params),
      headers: {
        auth_token
      }
    });
  }

  async push(data) {
    let n = 0;
    let success_total = 0;
    let fail_total = 0;
    let message_id = 0;
    let auth_token = '';
    const target_value = _.chunk(data.list, this.options.maxLength);

    try {
      const tokenData = await this.getToken();
      auth_token = tokenData.data.auth_token;
      const msgData = await this.saveMessage(data, auth_token);
      message_id = msgData.data.data.message_id;
    } catch (err) {
      data.fail(err);
      data.finish({
        status: 'success',
        maxLength: this.options.maxLength,
        group: target_value.length,
        success_total,
        fail_total: data.list.length
      });
      return;
    }
    data.success = data.success || function () { };
    data.fail = data.fail || function () { };
    data.finish = data.finish || function () { };

    for (const i in target_value) {
      axios({
        url: this.options.pushUrl,
        method: 'POST',
        timeout: this.options.timeout,
        data: querystring.stringify({
          message_id,
          target_type: 2,
          target_value: target_value[i].join(';')
        }),
        headers: {
          auth_token
        },
      }).then(res => {
        data.success(res);
        if (res.data && res.data.code == 0) {
          const len = res.data.data['10000'] ? res.data.data['10000'].length : 0;
          fail_total += len;
          success_total += target_value[i].length - len;
        } else {
          // fail_total += target_value[i].length;
          throw new Error(JSON.stringify(res));
        }
      }).catch((err) => {
        fail_total += target_value[i].length;
        data.fail(err);
      }).then(() => {
        n++;
        if (n >= target_value.length) {
          data.finish({
            status: 'success',
            maxLength: this.options.maxLength,
            group: target_value.length,
            success_total,
            fail_total
          });
        }
      });

      await this.sleep(data.sleep);
    }

  }

  async getToken() {
    const timestamp = Date.now();
    const sign = crypto.createHash('SHA256').update(`${this.options.appKey}${timestamp}${this.options.masterSecret}`).digest('hex');

    const res = await axios({
      url: this.options.getTokenUrl,
      method: 'POST',
      timeout: this.options.timeout,
      data: querystring.stringify({
        app_key: this.options.appKey,
        sign,
        timestamp
      }),
    });

    return res.data;
  }

}

module.exports = Oppo;