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