SOP/sop-sdk/sdk-nodejs/common/OpenClient.js

const needle = require('needle')
const moment = require('moment')

const {Class} = require("./Class");
const {RequestType} = require('./RequestType')
const {SignUtil} = require('./SignUtil')
const {BaseRequest} = require('../request/BaseRequest')

const HEADERS = {'Accept-Encoding': 'identity'}

const getHeaders = function (headers) {
    if (!headers) {
        return HEADERS
    }
    for (const key in HEADERS) {
        headers[key] = HEADERS[key];
    }
    return headers
}

const OpenClient = Class.create({
    /**
     * 初始化客户端
     * @param appId 应用ID
     * @param privateKey 应用私钥，2048位，PKCS8
     * @param url 请求url
     */
    init: function (appId, privateKey, url) {
        this.appId = appId;
        this.privateKey = privateKey;
        this.url = url;
    },
    /**
     * 发送请求
     * @param request 请求类
     * @param callback 回调函数，参数json
     */
    execute: function (request, callback) {
      this.executeToken(request, null, callback)
    },
    /**
     * 发送请求
     * @param request 请求类
     * @param token token
     * @param callback 回调函数，参数json
     */
    executeToken: function (request, token, callback) {
        if (!(request instanceof BaseRequest)) {
            throw 'request类未继承BaseRequest'
        }
        const requestType = request.getRequestType();
        if (request.files) {
            this._postFile(request, callback);
        } else {
            switch (requestType) {
                case RequestType.GET:
                    this._get(request, callback);
                    break
                case RequestType.POST_FORM:
                    this._postForm(request, callback);
                    break
                case RequestType.POST_JSON:
                    this._postJson(request, callback);
                    break
                case RequestType.POST_FILE:
                    this._postFile(request, callback);
                    break
                default :{
                    throw 'request.getRequestType()类型不正确'
                }
            }
        }
    },
    _get: function(request, callback) {
        const allParams = this._buildParams(request)
        const that = this
        // needle.request(method, url, data[, options][, callback])
        needle.request('GET',this.url, allParams, {
            headers: getHeaders()
        }, function(error, response) {
            callback(that._parseResponse(error, response, request))
        });
    },
    _postForm: function (request, callback) {
        const allParams = this._buildParams(request)
        const that = this
        needle.request('POST',this.url, allParams, {
            headers: getHeaders({
                'Content-Type': 'application/x-www-form-urlencoded'
            })
        }, function(error, response) {
            callback(that._parseResponse(error, response, request))
        });
    },
    _postJson: function (request, callback) {
        const allParams = this._buildParams(request)
        const that = this
        needle.request('POST',this.url, allParams, {
            headers: getHeaders(), json: true
        }, function(error, response) {
            callback(that._parseResponse(error, response, request))
        });
    },
    _postFile: function (request, callback) {
        const allParams = this._buildParams(request)
        const files = request.files;
        files.forEach(row => {
            // 设置成{ file: row.path, content_type: 'application/octet-stream' }格式
            // needle会认为是上传文件
            allParams[row.name] = { file: row.path, content_type: 'application/octet-stream' }
        })
        const that = this
        needle.request('POST',this.url, allParams, {
            headers: getHeaders(), multipart: true
        }, function(error, response) {
            callback(that._parseResponse(error, response, request))
        });
    },
    _parseResponse: function (error, response, request) {
        if (!error && response.statusCode === 200) {
            return request.parseResponse(response.body)
        } else {
            throw '请求异常：' + error
        }
    },
    _buildParams: function (request, token) {
        const allParams = {
            'app_id': this.appId,
            'method': request.getMethod(),
            'charset': 'UTF-8',
            'sign_type': 'RSA2',
            'timestamp': moment().format('YYYY-MM-DD HH:mm:ss'),
            'version': request.getVersion()
        }
        const bizModel = request.bizModel

        for (const key in bizModel) {
            allParams[key] = bizModel[key]
        }

        if (token) {
            allParams['app_auth_token'] = token
        }
        // 创建签名
        const sign = SignUtil.createSign(allParams, this.privateKey, 'RSA2')
        allParams.sign = sign
        return allParams;
    }
})

module.exports = OpenClient