weaver-work-cli 0.1.8 → 0.1.21

package/dist/internal/e10/auth/session.js

@@ -2,6 +2,7 @@ import { constants, createDecipheriv, createHash, publicEncrypt } from 'node:cry
 import { existsSync, mkdirSync, readFileSync, readdirSync, unlinkSync } from 'node:fs';
 import { dirname, resolve } from 'node:path';
 import { CliError, ExitCode } from '../../../core/errors.js';
+import { isConnectorMode } from '../../../core/connector.js';
 import { writeWithRetry } from '../../../core/write-with-retry.js';
 import { encryptAuthData, parseAuthFile } from './crypto.js';
 import { getE10ActiveProfilePath, getE10ProfilesRoot } from './paths.js';
@@ -58,7 +59,20 @@ export function getActiveProfile() {
     }
     return 'default';
 }
+/**
+ * 写活跃 profile 指针（并确保该 profile 目录存在）。
+ *
+ * 连接器模式（全局参数 `--is-connector`）下**直接跳过**：连接器授权只新增自己的
+ * `profiles/<CONNECTOR_PROFILE>/` 凭据，绝不改动用户既有的活跃 profile 选择。
+ * 需要无条件切换（`auth profile use` 这类显式操作）请用 `switchActiveProfile`。
+ */
 export function setActiveProfile(name) {
+    if (isConnectorMode())
+        return;
+    switchActiveProfile(name);
+}
+/** 无条件写活跃 profile 指针，用于 `auth profile use` 等显式切换场景。 */
+export function switchActiveProfile(name) {
     mkdirSync(getE10ProfilesRoot(), { recursive: true });
     mkdirSync(resolve(getE10ProfilesRoot(), name), { recursive: true });
     writeWithRetry(getE10ActiveProfilePath(), name, 'utf-8');
@@ -226,6 +240,52 @@ export function extractRawSetCookie(resp) {
     const setCookies = headers.getSetCookie?.() || splitSetCookie(headers.get('set-cookie') || '');
     return setCookies.map((cookie) => cookie.split(';')[0]).filter(Boolean).join('; ');
 }
+/** 从 `set-cookie` 或已存的 raw cookie 串里取出网关会话票据 `ZT_SESSION` 的值。
+ *
+ *  按参数顺序逐个来源查找，取第一个命中；都取不到返回空串。名称按 cookie 边界匹配，
+ *  以免把 `XZT_SESSION` 之类的相似名当成命中。 */
+export function extractSessionTicket(...sources) {
+    for (const source of sources) {
+        if (!source)
+            continue;
+        const match = source.match(/(?:^|;\s*)ZT_SESSION=([^;]+)/u);
+        if (match?.[1])
+            return match[1];
+    }
+    return '';
+}
+/** 读网关会话票据的 `exp`（Unix 秒）。票据是 3 段结构、**第 2 段**是 base64url 的 JSON 载荷。
+ *
+ *  结构不符 / 没有 `exp` / 解析失败一律返回 `null`，表示**本地无法判断是否过期**——调用方据此
+ *  不要把票判成过期（票据格式演进时不至于把有效票误丢）。 */
+export function readSessionTicketExpiry(ticket) {
+    const payload = ticket.split('.')[1];
+    if (!payload)
+        return null;
+    try {
+        const data = JSON.parse(Buffer.from(payload, 'base64url').toString('utf-8'));
+        const exp = Number(data?.exp);
+        return Number.isFinite(exp) && exp > 0 ? exp : null;
+    }
+    catch {
+        return null;
+    }
+}
+/** 票是否可用：缺省／空／纯空白／已过期 ⇒ 不可用（返回空串）；`exp` 解不出 ⇒ 按可用处理。 */
+function usableSessionTicket(ticket) {
+    const trimmed = (ticket || '').trim();
+    if (!trimmed)
+        return '';
+    const exp = readSessionTicketExpiry(trimmed);
+    if (exp !== null && exp * 1000 <= Date.now())
+        return '';
+    return trimmed;
+}
+/** auth 文件里的 `userId` 是否可当操作人使用（空值与登录降级占位 `'unknown'` 不算）。 */
+function hasUsableUserId(session) {
+    const id = session.userId;
+    return Boolean(id) && id !== 'unknown';
+}
 export class E10AuthSession {
     filePath;
     data;
@@ -287,6 +347,16 @@ export class E10AuthSession {
         const raw = this.cookies.raw || '';
         return buildE10CookieHeader(raw, this.eteamsId, this.agentType);
     }
+    /** 在 `cookieHeader()` 基础上附加网关会话票据（`ZT_SESSION`）。
+     *
+     *  **只在票存在、非空且未过期时才追加**：没有票（老 auth 文件／网关未下发 set-cookie／set-cookie 为空）
+     *  或票已过期时，返回值与 `cookieHeader()` **逐字节相同** ⇒ 未受惠也算不上回退。
+     *  是否使用本方法由调用方决定（目前只有流程侧，见 `E10RequestRuntimeDependencies.attachSessionTicket`）。 */
+    cookieHeaderWithSessionTicket() {
+        const base = this.cookieHeader();
+        const ticket = usableSessionTicket(this.data.ztSession);
+        return ticket ? `${base}; ZT_SESSION=${ticket}` : base;
+    }
     uaHeader() {
         return this.agentType ? `AgentType=${this.agentType},IsAgent=true` : 'weaver-work-cli';
     }
@@ -328,7 +398,7 @@ export class E10AuthSession {
             return null;
         }
     }
-    static async fromOptions(options, startDir = process.cwd()) {
+    static async fromOptions(options, startDir = process.cwd(), load = {}) {
         let baseUrl = options.baseUrl;
         let passportUrl = options.passportUrl;
         if (options.profile) {
@@ -340,8 +410,10 @@ export class E10AuthSession {
         if (!authPath)
             return null;
         const session = E10AuthSession.fromFile(authPath, baseUrl, passportUrl);
-        if (session)
+        // 富化只是"刷新/纠偏"：登录时这些字段已落盘。调用方可显式跳过（见 E10SessionLoadOptions）。
+        if (session && !(load.skipProfileEnrichment && hasUsableUserId(session))) {
             await session.fetchEmployeeId();
+        }
         return session;
     }
     async fetchEmployeeId() {