@azteam/express 1.2.218 → 1.2.219

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@azteam/express",
-    "version": "1.2.218",
+    "version": "1.2.219",
     "main": "src/index.js",
     "engines": {
         "node": ">= 12.0.0",

package/src/Server.js

@@ -12,8 +12,6 @@ import _ from 'lodash';
 import 'express-async-errors';
 import {CORS, errorCatch, ErrorException, NOT_FOUND, UNKNOWN} from '@azteam/error';
 
-import {authMiddleware} from './middleware/authMiddleware';
-
 const RES_TYPE = {
     ARRAY: 'ARRAY',
     OBJECT: 'OBJECT',
@@ -273,8 +271,6 @@ class Server {
                 app.use(middleware);
             });
 
-            app.use(authMiddleware);
-
             const msg = [];
             _.map(this.controllers, (data) => {
                 const {controller} = data;

package/src/middleware/authMiddleware.js

@@ -11,37 +11,42 @@ function systemLogin(userData = null) {
     return user;
 }
 
-export function authMiddleware(req, res, next) {
-    const {headers} = req;
+export default function (cbLoginAPI) {
+    return async function (req, res, next) {
+        const {headers} = req;
 
-    if (headers['x-app-secret'] === process.env.SECRET_KEY) {
-        req.user = systemLogin(headers['x-app-user']);
-    } else {
-        let token = null;
+        if (headers['x-app-secret'] === process.env.SECRET_KEY) {
+            req.user = systemLogin(headers['x-app-user']);
+        } else {
+            let token = null;
 
-        if (headers.authorization) {
-            token = headers.authorization;
-        }
+            if (headers.authorization) {
+                token = headers.authorization;
+            }
 
-        if (token) {
-            if (token.startsWith('Bearer ')) {
-                token = token.replace('Bearer ', '');
+            if (token) {
+                if (token.startsWith('Bearer ')) {
+                    token = token.replace('Bearer ', '');
 
-                try {
-                    req.user = jwt.verify(token, process.env.SECRET_KEY);
-                    return next();
-                } catch (err) {
-                    if (err.name === 'TokenExpiredError') {
-                        throw new ErrorException(TOKEN_EXPIRED, err);
+                    try {
+                        req.user = jwt.verify(token, process.env.SECRET_KEY);
+                        return next();
+                    } catch (err) {
+                        if (err.name === 'TokenExpiredError') {
+                            throw new ErrorException(TOKEN_EXPIRED, err);
+                        }
+                        throw new ErrorException(TOKEN_FAILED, err);
                     }
-                    throw new ErrorException(TOKEN_FAILED, err);
+                } else {
+                    const data = await cbLoginAPI(token);
+                    if (data) {
+                        req.user = data;
+                    }
+                    return next();
                 }
             }
-            throw new ErrorException(TOKEN_FAILED, {
-                message: 'Token type invalid',
-            });
         }
-    }
 
-    return next();
+        return next();
+    };
 }

package/src/middleware/index.js

@@ -1,5 +1,6 @@
 export {default as signMiddleware} from './signMiddleware';
 export {default as etagMiddleware} from './etagMiddleware';
+export {default as authMiddleware} from './authMiddleware';
 export {default as roleMiddleware} from './roleMiddleware';
 export {default as adminRoleMiddleware} from './adminRoleMiddleware';
 export {default as systemRoleMiddleware} from './systemRoleMiddleware';