@longzai-intelligence-tenant/elysia 0.0.1 → 0.0.2

package/dist/index.d.ts

@@ -12,7 +12,7 @@ type TenantValidateFn = (tenantId: string) => Promise<{
   error?: string;
 }>;
 /**
- * 租户上下文轻插件配置
+ * 租户上下文插件配置
  */
 type TenantContextOptions = {
   /**
@@ -21,6 +21,19 @@ type TenantContextOptions = {
    * 请求头未携带 X-Tenant-ID 时使用此值。
    */
   defaultTenantId?: string;
+  /**
+   * 租户 ID 来源
+   *
+   * - `'header'`：从请求头解析（默认）
+   * - `'url'`：从 URL 路径参数 :tenantId 解析
+   */
+  tenantIdSource?: "header" | "url";
+  /**
+   * 请求头名称（当 tenantIdSource 为 header 时使用）
+   *
+   * 默认 `'x-tenant-id'`。
+   */
+  headerName?: string;
   /**
    * 租户验证函数（可选）
    *
@@ -28,12 +41,32 @@ type TenantContextOptions = {
    * 不提供时跳过验证（信任上游已校验）。
    */
   validateFn?: TenantValidateFn;
+  /**
+   * 验证租户成员资格的回调（可选）
+   *
+   * 需要请求上下文中已存在 user 对象（userId）。
+   */
+  validateMembership?: (userId: string, tenantId: string) => Promise<boolean>;
+  /**
+   * 是否启用数据库行级安全上下文设置
+   *
+   * 启用后，在验证租户后调用 setRLSContext，请求结束时调用 resetRLSContext。
+   */
+  enableRLS?: boolean;
+  /**
+   * 设置 RLS 上下文的回调
+   */
+  setRLSContext?: (tenantId: string) => Promise<void>;
+  /**
+   * 重置 RLS 上下文的回调
+   */
+  resetRLSContext?: () => Promise<void>;
 };
 /**
- * 创建租户上下文 Elysia 轻插件
+ * 创建租户上下文 Elysia 插件
  *
- * 从请求头 X-Tenant-ID 解析当前租户 ID，注入到 request context。
- * 可选地调用 validateFn 校验租户有效性。
+ * 从请求头或 URL 参数解析当前租户 ID，注入到 request context。
+ * 支持可选的租户验证、成员资格校验、RLS 上下文管理。
  *
  * 不挂载路由，不注册适配器，不依赖 tenant 域内部包。
  */

package/dist/index.js

@@ -1 +1 @@
-import{Elysia as e}from"elysia";function t(t={}){let{defaultTenantId:n=`default`,validateFn:r}=t;return new e({name:`@longzai-intelligence-tenant/elysia`}).derive(({request:e})=>({tenantId:e.headers.get(`x-tenant-id`)??n})).as(`scoped`).onBeforeHandle(async({tenantId:e})=>{if(r&&e!==n&&!(await r(e)).valid){let{PermissionDeniedError:e}=await import(`@longzai-intelligence/error`);throw new e(`tenant`,`access`)}}).as(`global`)}export{t as createTenantContextPlugin};
+import{Elysia as e}from"elysia";function t(t={}){let{defaultTenantId:n=`default`,tenantIdSource:r=`header`,headerName:i=`x-tenant-id`,validateFn:a,validateMembership:o,enableRLS:s,setRLSContext:c,resetRLSContext:l}=t;return new e({name:`@longzai-intelligence-tenant/elysia`}).derive(({request:e,params:t})=>{let a;return r===`url`&&(a=t?.tenantId),!a&&r===`header`&&(a=e.headers.get(i)??void 0),{tenantId:a??n}}).as(`scoped`).onBeforeHandle(async({tenantId:e,request:t})=>{if(a&&e!==n&&!(await a(e)).valid){let{PermissionDeniedError:e}=await import(`@longzai-intelligence/error`);throw new e(`tenant`,`access`)}if(o){let n=t;if(n.user?.userId&&!await o(n.user.userId,e)){let{PermissionDeniedError:e}=await import(`@longzai-intelligence/error`);throw new e(`tenant`,`access`)}}s&&c&&await c(e)}).as(`global`).onAfterHandle(async()=>{s&&l&&await l()}).as(`global`)}export{t as createTenantContextPlugin};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@longzai-intelligence-tenant/elysia",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "private": false,
   "license": "UNLICENSED",
   "type": "module",