api.js 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  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 socketIo = require('socket.io');
  8. const apis = {};
  9. exports.setup = async function () {
  10. setApi();
  11. // use api server
  12. let port = await storage.setIpcDynamicPort();
  13. ELog.info('[api] [setup] dynamic port:', port);
  14. const listen = 'localhost';
  15. port = port ? port : 7069;
  16. const server = http.createServer(function(req, res) {
  17. ELog.info('[ api ] [setup] 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 ] [setup] invalid command called:', message.cmd);
  35. res.statusCode = 404;
  36. return res.end('NG');
  37. }
  38. ELog.info('[ api ] [setup] command received message:', message);
  39. const data = apis[message.cmd](...message.params);
  40. res.statusCode = 200;
  41. const result = {
  42. err: null,
  43. data: data,
  44. };
  45. res.end(JSON.stringify(result));
  46. });
  47. } else {
  48. res.statusCode = 404;
  49. res.end('NOT FOUND');
  50. }
  51. });
  52. // socket io
  53. const io = socketIo(server);
  54. io.on('connection', (socket) => {
  55. socket.on('ipc', (message, callback) => {
  56. ELog.info('[ api ] [setup] socket id:' + socket.id + ' message cmd: ' + message.cmd);
  57. const data = apis[message.cmd](...message.params);
  58. const result = {
  59. err: null,
  60. data: data,
  61. };
  62. callback(result);
  63. });
  64. });
  65. server.listen(port, listen, function() {
  66. ELog.info('[ api ] [setup] server is listening on', `${listen}:${port}`);
  67. });
  68. };
  69. function setApi() {
  70. const apiDir = path.normalize(__dirname + '/apis');
  71. // console.log('apiDir:', apiDir);
  72. fs.readdirSync(apiDir).forEach(function(filename) {
  73. if (path.extname(filename) === '.js' && filename !== 'index.js') {
  74. const name = path.basename(filename, '.js');
  75. const fileObj = require(`./apis/${filename}`);
  76. _.map(fileObj, function(fn, method) {
  77. let methodName = getApiName(name, method);
  78. apis[methodName] = fn;
  79. // ELog.info('[ Monitor ]', methodName);
  80. });
  81. }
  82. });
  83. };
  84. /**
  85. * get api method name
  86. * ex.) jsname='user' method='get' => 'user.get'
  87. * @param {String} jsname
  88. * @param {String} method
  89. */
  90. function getApiName (jsname, method) {
  91. return jsname + '.' + method;
  92. //return jsname + method.charAt(0).toUpperCase() + method.slice(1);
  93. };