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（undefined则使用executeSync）
     */
    execute: function (request, callback) {
        if (typeof callback == 'function') {
            this.executeToken(request, null, callback);
        } else {
            return this.executeSync(request);
        }
    },
    /**
     * 发送同步请求
     * @param request 请求类
     * */
    executeSync: function (request) {
        return new Promise((resolve) => {
            this.execute(request, res => {
                resolve(res);
            });
        });
    },
    /**
     * 发送请求
     * @param request 请求类
     * @param token token
     * @param callback 回调函数，参数json（undefined则使用executeTokenSync）
     */
    executeToken: function (request, token, callback) {
        if (!(request instanceof BaseRequest)) {
            throw 'request类未继承BaseRequest';
        }
        if (typeof callback == 'function') {
            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()类型不正确';
                    }
                }
            }
        } else {
            return this.executeTokenSync(request, token);
        }
    },
    /**
     * 发送同步请求
     * @param request 请求类
     * @param token token
     */
    executeTokenSync: function (request, token) {
        return new Promise((resolve) => {
            this.executeToken(request, token, res => {
                resolve(res);
            });
        });
    },
    _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
            return { // 重新封装请求异常回调，以防中断
                msg: '请求异常',
                code: '502',
                sub_msg: `${error}`,
                sub_code: 'isv.invalid-server'
            };
        }
    },
    _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(),
            'biz_content': JSON.stringify(request.bizModel)
        };

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

module.exports = OpenClient;