utils.js 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. const request = require('request');
  2. const qs = require('qs');
  3. /**
  4. * 公共请求
  5. * @param {Object} config
  6. * @return {Object} Promise
  7. */
  8. const createServer = (config) => {
  9. //console.log(config)
  10. return new Promise((resolve, reject) => {
  11. request(config, (err, ret, body) => {
  12. if (!err) {
  13. resolve(body)
  14. } else {
  15. reject(err)
  16. }
  17. })
  18. })
  19. }
  20. /**
  21. * 转换对象
  22. * @param {Object} req
  23. * @param {Object} res
  24. * @param {Object} next
  25. * @return {Array} [config,protocol,host,cb]
  26. */
  27. const convert = (req,res,next,url) => {
  28. let host = req.hostname;
  29. let protocol = req.protocol;
  30. let method = req.method.toUpperCase();
  31. //console.log(method)
  32. let ip = req.headers['x-real-ip'] ? req.headers['x-real-ip'] : req.ip.replace(/::ffff:/, '');
  33. let _params = method === 'GET' ? req.query : req.body;
  34. _params['url'] = req.body.url || req.query.url;
  35. let cb = _params.callback;
  36. let headers = req.headers;
  37. let config = {
  38. method: method,
  39. gzip: true
  40. };
  41. let params = {};
  42. for (let i in _params) {
  43. let temp = _params[i];
  44. if (i === 'url') {
  45. let o = temp.split('?');
  46. let uri = o[0]
  47. config['url'] = uri;
  48. headers['host'] = uri.replace(/^(http|https):\/\//g, '').split('/')[0];
  49. if (o.length > 1) {
  50. o[1].split('&').forEach(item => {
  51. let k = item.split('=');
  52. params[k[0]] = encodeURI(k[1]);
  53. })
  54. }
  55. }else if(i=='headers'){
  56. let _t = temp || '{}'
  57. let _chs = eval(`(${_t})`)
  58. for(let ch in _chs){
  59. headers[ch] = _chs[ch]
  60. }
  61. } else {
  62. params[i] = temp;
  63. }
  64. }
  65. let _hs = headers
  66. let hs = {}
  67. for(let _h in _hs){
  68. let h = _hs[_h]
  69. if(/(content-length)/.test(_h)) continue
  70. else hs[_h] = _hs[_h]
  71. }
  72. //console.log(hs)
  73. if (method === 'POST') config['form'] = params;
  74. else config['url'] = config['url'] ? `${config['url']}?${qs.stringify(params)}` : (url?`${url}?${qs.stringify(params)}`:null) ;
  75. config['headers'] = hs;
  76. return [config,protocol,host,cb,params]
  77. }
  78. module.exports = {
  79. convert,
  80. createServer
  81. }