@gingkoo/base-server 0.0.2-alpha.9 → 0.0.3-alpha.2

package/app.js

@@ -15,9 +15,11 @@ const router = require('./backend/router');
 class App {
   /**
    * @param {string} optinos.timeout 超时时间 默认 30s
+   * @param {boolean} optinos.closeSocket 是否关闭socket 默认 false 不关闭
    * @param {(app: any, express: any) => void} optinos.middleware 注册中间件
    * @param {(app: any, express: any) => void} optinos.beforeRoutes 在前面注册路由
    * @param {(app: any, express: any) => void} optinos.afterRoutes 在后面注册路由
+   * @param {(server: any, app: any, express: any) => void} optinos.expandServer 扩展 server
    */
   constructor(options = {}) {
     this.options = options;
@@ -27,10 +29,17 @@ class App {
 
     this.connectDB();
     this.middleware();
+
+    if (!options.closeSocket) {
+      this.listenSocket();
+    }
+
     this.routes();
     this.schedule();
 
-    this.listenSocket();
+    if (this.options.expandServer) {
+      this.options.expandServer(this.httpServer, this.app, express);
+    }
   }
 
   // 连接数据库
@@ -50,7 +59,11 @@ class App {
     this.app.use(cors);
     this.app.use(response);
     this.app.use((req, res, next) => {
-      if (~req.path.indexOf('/sse')) {
+      let ignoreCompressionPath = config.ignoreCompressionPath;
+      if (
+        Array.isArray(ignoreCompressionPath) &&
+        ignoreCompressionPath.find((v) => ~req.path.indexOf(v))
+      ) {
         return next();
       }
 

package/backend/common/logger/index.js

@@ -23,6 +23,13 @@ const appenders = {
     compress: false,
     numBackups: 7,
   },
+  userlog: {
+    type: 'dateFile',
+    filename: path.join(config.app.home, 'logs', 'user.log'),
+    pattern: 'yyyy-MM-dd.log',
+    compress: false,
+    numBackups: 7,
+  },
 
   infolog: {
     type: 'dateFile',
@@ -59,6 +66,10 @@ const categories = {
     appenders: ['console', 'wslog'],
     level: 'debug',
   },
+  userlog: {
+    appenders: ['console', 'userlog'],
+    level: 'debug',
+  },
 
   schedulelog: {
     appenders: ['console', 'schedulelog'],

package/backend/common/middleware/auth.js

@@ -1,6 +1,7 @@
 'use strict';
 const jwt = require('../../utils/jwt');
 const config = require('../../config/index');
+const { userlog } = require('../logger');
 const { loginType, getLoginMethod } = require('../services/login');
 
 // 后端登录
@@ -102,8 +103,10 @@ function setTokenCookies(res, token) {
   expires.setDate(expires.getDate() + config.jwt.exp);
 
   if (config.jwt.domain) {
+    userlog.info(`domain: ${config.jwt.domain}`);
     return res.cookie('X-Token', token, { expires, domain: config.jwt.domain });
   } else {
+    userlog.info('不用 domain');
     return res.cookie('X-Token', token, { expires });
   }
 }

package/backend/common/socket.js

@@ -1,7 +1,7 @@
 const config = require('../config/index');
 
 class Socket {
-  scokets = config.scoket.files;
+  scokets = config.scoket?.files?.() || [];
   constructor(io) {
     this.io = io;
     this.dispenseNamespace();

package/backend/common/wechat/services/auth.js

@@ -1,6 +1,7 @@
 var httpclient = require('node-httpclient');
 var util = require('../../../utils/util');
 const jwt = require('../../../utils/jwt');
+const config = require('../../../config/index');
 const { logger } = require('../../logger');
 const userService = require('../../services/user');
 const configService = require('../../services/generalConfig');
@@ -16,7 +17,10 @@ const ginfoServices = require('../../ginfo/index');
 const getRedirectUrl = async (userAgent) => {
   let auth_redirect_url = '';
   let data = await configService.getFromConfig({ GROUP_ID: 'qywx' });
-  let { qywx_AgentID, qywx_APPID, qywx_REDIRECT } = data;
+  let { qywx_AgentID, qywx_APPID } = data;
+
+  let qywx_REDIRECT = config.app.qywx_redirect_uri || data.qywx_REDIRECT;
+
   // qywx_REDIRECT = http://gingkoo.com/teams
   let redirect_uri = `${qywx_REDIRECT}/wechat/auth/oauth/qywx/callback`;
 

package/backend/config/index.js

@@ -29,6 +29,7 @@ function gConfig() {
       app: {
         home: path.resolve('app_home'), // 资源存放的磁盘位置 （绝对路径）
         redirect_uri: '',
+        qywx_redirect_uri: '',
       },
 
       jwt: {
@@ -42,7 +43,9 @@ function gConfig() {
       },
 
       scoket: {
-        files: [],
+        files: () => {
+          return [];
+        },
         config: {},
       },
 
@@ -54,6 +57,17 @@ function gConfig() {
       site: {
         title: '',
       },
+      ignoreCompressionPath: [],
+
+      frontend_config: {
+        load_type: 'iframe', // iframe | micro app
+        home_mode: 'menu', // menu 菜单模式 ｜ page 单页面模式
+        home_url: '', // page 模式地址
+      },
+
+      oauth: {
+        account_login_url: '',
+      },
     },
     serverConfig,
   );

package/backend/routers/app.js

@@ -3,9 +3,11 @@ var express = require('express');
 const nodemailer = require('nodemailer');
 const bcrypt = require('bcryptjs');
 const config = require('../config');
+const { userlog } = require('../../backend/common/logger');
 const jwt = require('../utils/jwt');
 const { checkLogin, setTokenCookies } = require('../common/middleware/auth');
 const { join, readFileContent } = require('../utils/path');
+const api = require('../utils/http');
 const userService = require('../common/services/user');
 const generalConfig = require('../common/services/generalConfig');
 const emailService = require('../common/services/email');
@@ -24,25 +26,66 @@ router.post('/login', async (req, res) => {
     res.sendErr('请输入用户名');
     return;
   }
-
-  let userinfo = await userService.getUserInfo(userId);
-  if (!userinfo?.USER_ID) {
-    res.sendErr('用户不存在');
+  if (!password) {
+    res.sendErr('请输入密码');
     return;
   }
 
-  // 判断密码
-  if (userinfo.PASSWD) {
-    let flag = await bcrypt.compare(password, userinfo.PASSWD);
-    if (!flag) {
-      return res.sendErr('密码错误');
+  let userid = null;
+  if (config?.oauth?.account_login_url) {
+    // 统一账号登录
+    let redirect_uri = config.oauth.account_login_url.format({
+      username: userId,
+      password,
+    });
+
+    let resp = await api.get({
+      url: redirect_uri,
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJkYXRhIjp7ImFwcGlkIjoiOThndmRscm8ifSwiZXhwIjoyMDQ2NzYwNTM5LCJpYXQiOjE3MzE0MDA1Mzl9.EHZi_kvZw7CGGkQ7BoGirdRtkT6nG0pgFBCuF9wXw6I`,
+      },
+    });
+
+    // {
+    //   "access_token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdWQiOiJjb2RlbWFnaWMiLCJleHAiOjE3MzQwNjI2NDMsImlhdCI6MTczNDA1NTQ0MywiaXNzIjoiaWRtIiwic3ViIjoieGlsaW4uZ2FvIn0.6cZQwz2sawEFT5vaMtU_eoGx0CBBllgWhDBSNknmEZM",
+    //   "expires_in": 7200,
+    //   "token_type": "Bearer",
+    //   "user_id": "xilin.gao"
+    // }
+
+    if (resp.isError) {
+      return res.sendErr('账号或密码错误');
+    }
+
+    let { access_token, expires_in, token_type, user_id } = resp;
+    userid = user_id;
+  } else {
+    let userinfo = await userService.getUserInfo(userId);
+    if (!userinfo?.USER_ID) {
+      res.sendErr('用户不存在');
+      return;
+    }
+
+    // 判断密码
+    if (userinfo.PASSWD) {
+      let flag = await bcrypt.compare(password, userinfo.PASSWD);
+      if (!flag) {
+        return res.sendErr('密码错误');
+      }
     }
+    userid = userinfo?.USER_ID;
   }
 
+  if (!userid) {
+    return res.sendErr('用户不存在');
+  }
   let token = jwt.generateToken({
-    userid: userinfo?.USER_ID,
+    userid,
   });
 
+  userlog.info(`用户：${userid} 登录成功，token: ${token}`);
+
   setTokenCookies(res, token);
   res.sendOk({
     token,

package/backend/routers/module.js

@@ -46,16 +46,17 @@ router.get('/getMenu', async function (req, res) {
   if (!Array.isArray(modules)) {
     return res.sendOk([]);
   }
-  let member = await servicesServices.getSystemPower();
+  // TODO. 这里用写死成员过滤 teams-system 感觉不合适 先注释掉
+  // let member = await servicesServices.getSystemPower();
 
-  if (!Array.isArray(member)) member = [];
+  // if (!Array.isArray(member)) member = [];
 
-  member = member.filter((v) => v?.userid);
+  // member = member.filter((v) => v?.userid);
   // modules = modules.filter((v) => member.includes(userid) || v.MODULE_ID !== 'teams-system');
 
   // 先过滤掉权限
-  // let permissionModules = await userPermissionModules(userid);
-  // modules = modules.filter((v) => permissionModules.includes(v.MODULE_ID));
+  let permissionModules = await userPermissionModules(userid);
+  modules = modules.filter((v) => permissionModules.includes(v.MODULE_ID));
 
   let data = await Promise.all(
     modules.map(async (v) => {

package/backend/space.js

@@ -2,6 +2,7 @@
 var express = require('express');
 const { checkLogin, authCheck } = require('./common/middleware/auth');
 const { loginType, getLoginMethod } = require('./common/services/login');
+const config = require('./config');
 const automateRouter = require('./routers/automate');
 const app = require('./routers/app');
 const users = require('./routers/user');
@@ -46,6 +47,7 @@ router.get('/config', async function (req, res) {
   res.sendOk({
     loginUrl,
     logo,
+    ...config.frontend_config,
   });
 });
 

package/backend/utils/http.js

@@ -0,0 +1,117 @@
+/**
+ * axios封装
+ * 请求拦截、相应拦截、错误统一处理
+ */
+const axios = require('axios');
+const QS = require('qs');
+
+const SuccessCode = 'OK'; // 业务code 成功的状态码
+
+axios.defaults.timeout = 60000; //请求超时时间
+axios.defaults.withCredentials = false;
+
+const HttpCodeMessage = {
+  200: '服务器成功返回请求的数据。',
+  201: '新建或修改数据成功。',
+  202: '一个请求已经进入后台排队（异步任务）。',
+  204: '删除数据成功。',
+  400: '发出的请求有错误，服务器没有进行新建或修改数据的操作。',
+  401: '用户没有权限（令牌、用户名、密码错误）。',
+  403: '用户得到授权，但是访问是被禁止的。',
+  404: '发出的请求针对的是不存在的记录，服务器没有进行操作。',
+  406: '请求的格式不可得。',
+  410: '请求的资源被永久删除，且不会再得到的。',
+  422: '当创建一个对象时，发生一个验证错误。',
+  500: '服务器发生错误，请检查服务器。',
+  502: '网关错误。',
+  503: '服务不可用，服务器暂时过载或维护。',
+  504: '网关超时。',
+};
+
+// 请求拦截器
+axios.interceptors.request.use(
+  (options) => {
+    options.headers['Content-Type'] = options.headers['Content-Type']
+      ? options.headers['Content-Type']
+      : 'application/json';
+
+    return options;
+  },
+  (error) => {
+    return Promise.reject(error);
+  },
+);
+
+// 响应拦截器
+axios.interceptors.response.use(
+  (response) => {
+    return response;
+  },
+  (error) => {
+    let errorRes = error.response || {};
+    let { status, statusText = '' } = errorRes;
+
+    return {
+      data: {
+        isError: true,
+        code: status,
+        message: HttpCodeMessage[status] || statusText,
+      },
+    };
+  },
+);
+
+const sendRequest = ({ url, method, params, headers, timeout }) => {
+  let options = {
+    url,
+    method,
+    paramsSerializer: function (params) {
+      return QS.stringify(params);
+    },
+  };
+
+  if (method == 'GET') {
+    options.params = params;
+  } else {
+    options.data = params;
+  }
+  if (headers) {
+    options.headers = headers;
+  }
+  if (timeout) {
+    options.timeout = timeout;
+  }
+
+  return new Promise((resolve, reject) => {
+    axios
+      .request(options)
+      .then((result) => {
+        resolve(result.data);
+      })
+      .catch((error) => {
+        return resolve(null);
+      })
+      .finally(() => {});
+  });
+};
+
+class BaseRequest {
+  get(options) {
+    options.method = 'GET';
+    return sendRequest(options);
+  }
+  post(options) {
+    options.method = 'POST';
+    return sendRequest(options);
+  }
+  put(options) {
+    options.method = 'PUT';
+    return sendRequest(options);
+  }
+  del(options) {
+    options.method = 'DELETE';
+    return sendRequest(options);
+  }
+}
+
+module.exports = new BaseRequest();