TuKouKouClient/electron/api.js

'use strict';

const fs = require('fs');
const http = require('http');
const path = require('path');
const _ = require('lodash');
const storage = require('./lib/storage');
const socketIo = require('socket.io');
const eLogger = require('./lib/eLogger').get();
// const {app} = require('electron');

const apis = {};

exports.setup = async function () {
  eLogger.info('[api] [setup] start');
  setApi();

  // use api server
  let port = await storage.setIpcDynamicPort();
  eLogger.info('[api] [setup] dynamic ipc port:', port);
  const listen = 'localhost';
  port = port ? port : 7069;

  const server = http.createServer(function(req, res) {
    eLogger.info('[ api ] [setup] command received', { method: req.method, url: req.url });
    if ((req.method === 'POST' && req.url === '/send')) {
      let body = '';
      req.setEncoding('utf8');
      req
      .on('data', function(data) {
        body += data;
      })
      .on('end', function() {
        let message;
        try {
          message = JSON.parse(body);
        } catch (e) {
          res.statusCode = 400;
          return res.end('request body parse failure.');
        }
        if (!apis[message.cmd]) {
          eLogger.info('[ api ] [setup] invalid command called:', message.cmd);
          res.statusCode = 404;
          return res.end('NG');
        }

        eLogger.info('[ api ] [setup] command received message:', message);
        const data = apis[message.cmd](...message.params);
        res.statusCode = 200;
        const result = {
          err: null,
          data: data,
        };
        res.end(JSON.stringify(result));
      });
    } else {
      res.statusCode = 404;
      res.end('NOT FOUND');
    }
  });

  // socket io
  const io = socketIo(server);
  io.on('connection', (socket) => {
    socket.on('ipc', (message, callback) => {
      eLogger.info('[ api ] [setup] socket id:' + socket.id + ' message cmd: ' + message.cmd);
      
      // const filepath = message;
      // const fileObj = require(`./apis/${filepath}`);
      const data = apis[message.cmd](...message.params);
      if (data && typeof data.then === 'function') { // 判断是否是异步
        data.then((data) => {
          const result = {
            err: null,
            data: data,
          };
          callback(result)
        });
      } else {
        const result = {
          err: null,
          data: data,
        };
        callback(result);
      }
    });
  });

  server.listen(port, listen, function() {
    eLogger.info('[ api ] [setup] server is listening on', `${listen}:${port}`);
  });

  return true;
};

// function setApi() {
//   // fs读文件的时候，用path正规化 [打包读文件问题]
//   const apiDir = path.normalize(__dirname + '/apis');
//   eLogger.info('[setApi] apiDir: ', apiDir);
//   const fileArr = fs.readdirSync(apiDir);
//   eLogger.info('[setApi] fileArr: ', fileArr);
//   for (let i = 0; i < fileArr.length; i++) {
//     let filename = fileArr[i];
//     if (path.extname(filename) === '.js' && filename !== 'index.js') {
//       const name = path.basename(filename, '.js');
//       const fileObj = require(`./apis/${filename}`);
//       _.map(fileObj, function(fn, method) {
//         let methodName = getApiName(name, method);
//         apis[methodName] = fn;
//         eLogger.info('[setApi] method Name', methodName);
//       });
//     }
//   }
  
//   return true;
// }

// function setApi() {
//   // fs读文件的时候，用path正规化 [打包读文件问题]
//   const apiDir = path.normalize(__dirname + '/apis');
//   eLogger.info('[setApi] apiDir: ', apiDir);
//   // const baseDir = app.getAppPath();
//   // const apiDir = path.join(baseDir, 'electron/apis');
//   fs.readdirSync(apiDir).forEach(function(filename) {
//     if (path.extname(filename) === '.js' && filename !== 'index.js') {
//       const name = path.basename(filename, '.js');
//       // require文件的时候，用相对路径并且不能path处理 [打包读文件问题]
//       const fileObj = require(`./apis/${filename}`);
//       _.map(fileObj, function(fn, method) {
//         let methodName = getApiName(name, method);
//         apis[methodName] = fn;
//         //eLogger.info('[setApi] method Name', methodName);
//       });
//     }
//   });
// }

function setApi() {
  const apiDir = path.normalize(__dirname + '/apis');
  fs.readdirSync(apiDir).forEach(function(filename) {
    if (path.extname(filename) === '.js' && filename !== 'index.js') {
      const name = path.basename(filename, '.js');
      const fileObj = require(`./apis/${filename}`);
      _.map(fileObj, function(fn, method) {
        let methodName = getApiName(name, method);
        apis[methodName] = fn;
      });
    }
  });
};

/**
 * get api method name
 * ex.) jsname='user' method='get' => 'user.get'
 * @param {String} jsname
 * @param {String} method
 */
function getApiName (jsname, method) {
  return jsname + '.' + method;
  //return jsname + method.charAt(0).toUpperCase() + method.slice(1);
}