@dreamor/atlas-cli 0.7.23 → 0.7.24

package/dist/adapters/atlas/auth/session.js

@@ -1,167 +0,0 @@
-import { readFile, unlink } from 'fs/promises';
-import { existsSync } from 'fs';
-import { secureMkdir, secureWriteFile } from '../util/secure-fs.js';
-import { getAtlasHome, getCookieFile } from '../util/paths.js';
-const SESSION_DIR = getAtlasHome();
-const COOKIE_FILE = getCookieFile();
-let cachedCookies = null;
-/**
- * 读取持久化的 cookies（从本地文件）
- * Playwright SSO 登录后将 cookies 写入此文件
- *
- * 沙盒免命令登录：本地文件缺失时若设置了 ATLAS_COOKIES_B64，则内存中一次性
- * import（不落盘），让 agent 只注入 secret 就能用上。这条路径在 startAuthRefresh
- * 之外的成本几乎为零，避免每次调用都重新解 base64/gzip。
- */
-export async function readCookies() {
-    if (cachedCookies)
-        return cachedCookies;
-    try {
-        if (!existsSync(COOKIE_FILE)) {
-            // 沙盒自动导入：ATLAS_COOKIES_B64 内存注入（不落盘）
-            const b64 = process.env.ATLAS_COOKIES_B64;
-            if (b64 && b64.trim().length > 0) {
-                const imported = await importFromEnvBase64(b64);
-                if (imported && imported.length > 0) {
-                    cachedCookies = imported;
-                    return imported;
-                }
-            }
-            return null;
-        }
-        const data = await readFile(COOKIE_FILE, 'utf-8');
-        const cookies = JSON.parse(data);
-        cachedCookies = cookies;
-        return cookies;
-    }
-    catch {
-        return null;
-    }
-}
-/**
- * 解 ATLAS_COOKIES_B64 环境变量为新版 portable bundle（gzip+base64 JSON）。
- * 兼容旧格式：纯 JSON 数组 cookies（无 manifest 包装）。
- * 失败返回 null，绝不抛——避免阻塞主流程。
- */
-async function importFromEnvBase64(b64) {
-    try {
-        const { parseBundleBase64 } = await import('../util/portable-store.js');
-        const { bundle } = await parseBundleBase64(b64);
-        return bundle.cookies;
-    }
-    catch {
-        // 兼容：直接是纯 JSON 数组字符串（非 gzip）
-        try {
-            const parsed = JSON.parse(b64);
-            if (Array.isArray(parsed))
-                return parsed;
-        }
-        catch {
-            // fallthrough
-        }
-        return null;
-    }
-}
-/** 仅内存导入（importCmd --persist=false 用），不落盘 */
-export function __setCachedCookiesForImport(cookies) {
-    cachedCookies = cookies;
-}
-export async function writeCookies(cookies) {
-    cachedCookies = cookies;
-    await secureMkdir(SESSION_DIR, { recursive: true });
-    await secureWriteFile(COOKIE_FILE, JSON.stringify(cookies, null, 2));
-}
-export async function clearCookies() {
-    cachedCookies = null;
-    try {
-        await unlink(COOKIE_FILE);
-    }
-    catch {
-        // ignore
-    }
-}
-/**
- * 从 daemon 获取 cookies（附带 Bearer token 鉴权）。
- *
- * 连接配置走 resolveDaemonConfig()：优先 ATLAS_DAEMON_URL + ATLAS_DAEMON_TOKEN
- * （沙盒连宿主机 daemon），回退 localhost + 本机 ~/.atlas/daemon.token（本机复用）。
- * 失败返回 null，绝不抛——调用方据此降级到本地 cookies。
- */
-export async function fetchCookiesFromDaemon() {
-    try {
-        const { resolveDaemonConfig } = await import('../util/environment.js');
-        const cfg = resolveDaemonConfig();
-        if (!cfg.url)
-            return null;
-        // token：env 优先（ATLAS_DAEMON_TOKEN），回退本地 token 文件（default 分支）
-        let token = cfg.token ?? '';
-        if (!token) {
-            const { getDaemonTokenFile } = await import('../util/paths.js');
-            const tokenFile = getDaemonTokenFile();
-            if (existsSync(tokenFile)) {
-                token = (await readFile(tokenFile, 'utf-8')).trim();
-            }
-        }
-        const { request } = await import('undici');
-        const headers = {};
-        if (token)
-            headers['Authorization'] = `Bearer ${token}`;
-        const resp = await request(`${cfg.url.replace(/\/$/, '')}/api/cookies`, {
-            headers,
-            headersTimeout: 5000,
-            bodyTimeout: 5000,
-        });
-        const text = await resp.body.text();
-        const json = JSON.parse(text);
-        if (json?.cookies && Array.isArray(json.cookies)) {
-            return json.cookies;
-        }
-        return null;
-    }
-    catch {
-        return null;
-    }
-}
-/** 兼容旧接口：从 cookies 中提取第一个可用的 token 类值 */
-export async function getSessionToken() {
-    const cookies = await readCookies();
-    if (!cookies || cookies.length === 0)
-        return null;
-    // Try to find an auth token cookie
-    const tokenCookie = cookies.find((c) => c.name.includes('token') ||
-        c.name.includes('sid') ||
-        c.name.includes('session') ||
-        c.name.includes('SESSION'));
-    // 没找到 token cookie 时返回 null，避免返回无效 cookie 导致上层误判为"已登录"
-    return tokenCookie?.value ?? null;
-}
-const findCookie = (cookies, name) => cookies.find((c) => c.name === name);
-/**
- * 从已有 cookies 派生 Banma 身份。纯函数，无副作用，便于测试。
- * 任一来源缺失对应字段返回空字符串；cookies 为空数组返回 null。
- */
-export function getBanmaIdentity(cookies) {
-    if (!cookies || cookies.length === 0)
-        return null;
-    const token = findCookie(cookies, 'access_token')?.value ?? '';
-    const user = findCookie(cookies, 'buc_username')?.value ?? '';
-    // buc_userinfo 是 base64 编码的 JSON，含 emp_id / account / name 等
-    let staffId = '';
-    const userinfoValue = findCookie(cookies, 'buc_userinfo')?.value;
-    if (userinfoValue) {
-        try {
-            const decoded = JSON.parse(Buffer.from(userinfoValue, 'base64').toString('utf-8'));
-            if (decoded && decoded.emp_id != null)
-                staffId = String(decoded.emp_id);
-        }
-        catch {
-            // 解码或解析失败：保持空字符串，不阻断请求
-        }
-    }
-    return { token, user, staffId };
-}
-/** 读取持久化 cookies 并派生 Banma 身份，供 HttpClient 使用 */
-export async function readBanmaIdentity() {
-    const cookies = await readCookies();
-    return getBanmaIdentity(cookies);
-}