mm_api 1.1.4 → 1.1.5

package/lib/api.js

@@ -24,6 +24,9 @@ const Param = loadOptional("mm_param/lib/param.js"); // 是MM自带的参数机
 const mm_dao = loadOptional("mm_dao");
 const Dao = mm_dao ? mm_dao.Dao : null;
 const Sql = loadOptional("mm_sql/lib/sql.js"); // 是MM自带的sql机制，可以不使用
+// 登录态存储：token 解析优先走 mm_state_store（用户状态）+ mm_dict（token→user_id 映射）
+const mm_state_store = loadOptional("mm_state_store");
+const mm_dict = loadOptional("mm_dict");
 
 /** param 实例缓存（模块级，替代原全局 param 模块列表） */
 const PARAM_CACHE = new Map();
@@ -528,15 +531,37 @@ Api.prototype._getUserFromSession = function (ctx) {
   return null;
 };
 
+/**
+ * 从上下文中提取 auth token（协议头优先，Cookie 兜底）
+ * @param {object} ctx 请求上下文
+ * @returns {string} token，无则返回空串
+ * @private
+ */
+Api.prototype._getToken = function (ctx) {
+  var headers = (ctx && (ctx.headers || (ctx.request && ctx.request.headers))) || {};
+  var token = headers[this.config.token_key] || "";
+  if (!token && ctx) {
+    var cookies = ctx.cookies || (ctx.request && ctx.request.cookies);
+    if (cookies) {
+      token =
+        typeof cookies.get === "function"
+          ? cookies.get(this.config.token_key) || ""
+          : cookies[this.config.token_key] || "";
+    }
+  }
+  return String(token || "");
+};
+
 /**
  * 从 token 中获取用户信息
+ * 解析顺序：缓存 → 用户状态存储（mm_dict 映射 / 解码 → mm_state_store）→ 数据库
  * @param {object} ctx 请求上下文
  * @param {object} db 数据管理器
  * @returns {object|null} 用户信息
  * @private
  */
 Api.prototype._getUserFromToken = async function (params, ctx, options) {
-  var token = ctx.headers[this.config.token_key];
+  var token = this._getToken(ctx);
   if (!token) {
     return null;
   }
@@ -547,7 +572,76 @@ Api.prototype._getUserFromToken = async function (params, ctx, options) {
       return await this._processUserFromCache(u, ctx, options, token);
     }
   }
-  return await this._getUserFromDb(token, ctx, options);
+  var o = await this._getUserFromState(token, ctx, options);
+  if (o) {
+    return o;
+  }
+  // 数据库兜底：仅当传入的是表模型管理器（有 new 方法）时才可查询
+  if (options && typeof options.new === "function") {
+    return await this._getUserFromDb(token, ctx, options);
+  }
+  return null;
+};
+
+/**
+ * 从用户状态存储获取用户信息
+ * token → user_id：mm_dict 映射，或 base64(JSON) 直接解码；
+ * user_id → 用户：mm_state_store 的 stateAdmin('user') 中 user_{id} 状态的 user 字段
+ * @param {string} token token
+ * @param {object} ctx 请求上下文
+ * @param {object} options 数据管理器
+ * @returns {object|null} 用户信息，未识别或状态中无用户资料返回 null
+ * @private
+ */
+Api.prototype._getUserFromState = async function (token, ctx, options) {
+  if (!mm_state_store || !mm_dict) {
+    return null;
+  }
+  try {
+    // 1. token → user_id：字典映射，或 base64(JSON) 解码
+    var user_id = null;
+    try {
+      user_id = mm_dict.dictAdmin("token").get(token);
+    } catch (err) {
+      user_id = null;
+    }
+    if (!user_id) {
+      try {
+        var json = Buffer.from(token, "base64").toString("utf8");
+        if (/^\s*\{[\s\S]*\}\s*$/.test(json)) {
+          var obj = JSON.parse(json);
+          if (obj && obj.user_id !== undefined && obj.user_id !== null && obj.user_id !== "") {
+            user_id = String(obj.user_id);
+          }
+        }
+      } catch (err) {
+        user_id = null;
+      }
+    }
+    if (!user_id) {
+      return null;
+    }
+    // 2. user_id → 用户状态
+    var state = null;
+    try {
+      state = await mm_state_store.stateAdmin("user").get("user_" + user_id);
+    } catch (err) {
+      state = null;
+    }
+    if (!state || !state.user) {
+      return null;
+    }
+    var o = { ...state.user };
+    o.token = token;
+    o.user_id = String(user_id);
+    if (ctx && ctx.session) {
+      ctx.session.uuid = ctx.session.uuid || token;
+      ctx.session.user = o;
+    }
+    return o;
+  } catch (err) {
+    return null;
+  }
 };
 
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "mm_api",
-    "version": "1.1.4",
+    "version": "1.1.5",
     "description": "API框架，用于实现web api和mvc模式路由请求，负责配置路由路径、访问权限、校验机制；基于 mm_corejs，内置管理器池，无全局/原型扩展依赖",
     "main": "index.js",
     "keywords": [
@@ -24,11 +24,13 @@
     "dependencies": {
         "mm_corejs": "^1.1.4",
         "mm_dao": "^1.8.2",
+        "mm_dict": "^1.0.3",
         "mm_helper": "^1.0.0",
         "mm_machine": "^3.5.4",
         "mm_param": "^1.1.0",
         "mm_ret": "^1.4.8",
-        "mm_sql": "^1.8.0"
+        "mm_sql": "^1.8.0",
+        "mm_state_store": "^1.0.3"
     },
     "scripts": {
         "test": "node test.js && node --test tests/*.test.js"