api.js 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. 'use strict';
  2. const fs = require('fs');
  3. const http = require('http');
  4. const path = require('path');
  5. const _ = require('lodash');
  6. const storage = require('./storage');
  7. const apis = {};
  8. exports.setup = async function () {
  9. ELog.info('[api] [setup]');
  10. setApi();
  11. // use api server
  12. let port = await storage.setApiDynamicPort();
  13. console.log('api port:', port);
  14. const listen = 'localhost';
  15. port = port ? port : 7069;
  16. const server = http.createServer(function(req, res) {
  17. ELog.info('[ api ] command received', { method: req.method, url: req.url });
  18. if ((req.method === 'POST' && req.url === '/send')) {
  19. let body = '';
  20. req.setEncoding('utf8');
  21. req
  22. .on('data', function(data) {
  23. body += data;
  24. })
  25. .on('end', function() {
  26. let message;
  27. try {
  28. message = JSON.parse(body);
  29. } catch (e) {
  30. res.statusCode = 400;
  31. return res.end('request body parse failure.');
  32. }
  33. if (!apis[message.cmd]) {
  34. ELog.info('[ api ] invalid command called:', message.cmd);
  35. res.statusCode = 404;
  36. return res.end('NG');
  37. }
  38. const start = Date.now();
  39. const data = apis[message.cmd]();
  40. const elapsed = Date.now() - start;
  41. ELog.info(`api [${message.cmd}] success. elapsed: ${elapsed}ms`, data);
  42. res.statusCode = 200;
  43. const result = {
  44. code: 0,
  45. data,
  46. };
  47. res.end(JSON.stringify(result));
  48. });
  49. } else {
  50. res.statusCode = 404;
  51. res.end('NOT FOUND');
  52. }
  53. });
  54. server.listen(port, listen, function() {
  55. ELog.info('[ api ] server is listening on', `${listen}:${port}`);
  56. });
  57. };
  58. function setApi() {
  59. const apiDir = path.normalize(__dirname + '/apis');
  60. // console.log('apiDir:', apiDir);
  61. fs.readdirSync(apiDir).forEach(function(filename) {
  62. if (path.extname(filename) === '.js' && filename !== 'index.js') {
  63. const name = path.basename(filename, '.js');
  64. const fileObj = require(`./apis/${filename}`);
  65. _.map(fileObj, function(fn, method) {
  66. let methodName = getApiName(name, method);
  67. apis[methodName] = fn;
  68. // ELog.info('[ Monitor ]', methodName);
  69. });
  70. }
  71. });
  72. };
  73. /**
  74. * get api method name for proteus-web
  75. * ex.) jsname='user' method='get' => 'userGet'
  76. * @param {String} jsname
  77. * @param {String} method
  78. */
  79. function getApiName (jsname, method) {
  80. return jsname + method.charAt(0).toUpperCase() + method.slice(1);
  81. };