koa2-wechat2/lib/wx/weReply.js

koa2 middleware for wechat

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

exports.default = function (opts, handler) {
  var _this = this;

  var wechat = (0, _config.getWechat)(opts);

  return function () {
    var _ref = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(ctx, next) {
      var token, signature, nonce, timestamp, echostr, str, sha, data, content, message;
      return regeneratorRuntime.wrap(function _callee$(_context) {
        while (1) {
          switch (_context.prev = _context.next) {
            case 0:
              token = opts.token;
              signature = ctx.query.signature;
              nonce = ctx.query.nonce;
              timestamp = ctx.query.timestamp;
              echostr = ctx.query.echostr;
              str = [token, timestamp, nonce].sort().join('');
              sha = (0, _sha2.default)(str);

              if (!(ctx.method === 'GET')) {
                _context.next = 11;
                break;
              }

              if (sha === signature) {
                ctx.body = echostr + '';
              } else {
                ctx.status = 401;
                ctx.body = 'Invalid signature';
              }
              _context.next = 29;
              break;

            case 11:
              if (!(ctx.method === 'POST')) {
                _context.next = 29;
                break;
              }

              if (!(sha !== signature)) {
                _context.next = 16;
                break;
              }

              ctx.status = 401;
              ctx.body = 'Invalid signature';

              return _context.abrupt('return', false);

            case 16:
              _context.next = 18;
              return (0, _rawBody2.default)(ctx.req, {
                length: ctx.length,
                limit: '1mb',
                encoding: true
              });

            case 18:
              data = _context.sent;
              _context.next = 21;
              return util.parseXMLAsync(data);

            case 21:
              content = _context.sent;

              console.log('content', content);

              message = util.formatMessage(content.xml);


              ctx.weixin = message;

              if (!handler) {
                handler = _reply2.default;
              }

              _context.next = 28;
              return handler.call(_this, ctx, next);

            case 28:

              wechat.reply.call(ctx);

            case 29:
            case 'end':
              return _context.stop();
          }
        }
      }, _callee, _this);
    }));

    return function (_x, _x2) {
      return _ref.apply(this, arguments);
    };
  }();
};

var _sha = require('sha1');

var _sha2 = _interopRequireDefault(_sha);

var _rawBody = require('raw-body');

var _rawBody2 = _interopRequireDefault(_rawBody);

var _util = require('./util');

var util = _interopRequireWildcard(_util);

var _config = require('../config');

var _reply = require('./reply');

var _reply2 = _interopRequireDefault(_reply);

function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _asyncToGenerator(fn) { return function () { var gen = fn.apply(this, arguments); return new Promise(function (resolve, reject) { function step(key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then(function (value) { step("next", value); }, function (err) { step("throw", err); }); } } return step("next"); }); }; }
/**
 * koa2 middleware
 * opts 
 * {
 *  appId: xx,
 *  appSecret: xx,
 *  token: xx
 * }
 * @param  {object} opts Wechat config opts
 * @param  {async fn}   handle weixin reply logic
 * @return {async fn}      koa2 middleware
 */