@turingflow/agent-kit 0.0.0-stage → 0.1.0

package/dist/server/auth/config.d.ts

@@ -0,0 +1,34 @@
+/**
+ * agent-kit · 认证配置（从 env 解析）
+ *
+ * env 分块约定（具体见 packages/agent-kit/.env.example）：
+ *   - 认证方式开关：AUTH_PROVIDER / APP_PUBLIC_URL / AUTH_COOKIE_PATH /
+ *     AUTH_SESSION_SECRET / AUTH_SECURE_COOKIE
+ *   - 钉钉（AUTH_PROVIDER 含 dingtalk）：DD_CLIENT_ID / DD_CLIENT_SECRET / DD_CORP_ID /
+ *     DD_AGENT_ID（企业内部应用微应用 AgentId，分享到钉钉的 H5 JSAPI 前端鉴权必填）
+ *   - 飞书（AUTH_PROVIDER 含 feishu）：FS_APP_ID / FS_APP_SECRET /
+ *     FS_BASE_URL（可选，国际版）/ FS_REDIRECT_URI（可选，默认自动拼接）
+ *   - builtin（AUTH_PROVIDER 含 builtin）：自建账号体系，登录系统尚未实现，
+ *     isMethodConfigured 暂恒 false（详见该函数注释），配置层已认可该方式（留口子）。
+ *
+ * 已移除：guest 模式、dev 免登（AUTH_DEV_LOGIN）、数据库开关（AUTH_DB_ENABLED）。
+ * 登录强制真实 OAuth；会话/消息持久化强制 PostgreSQL（配好 POSTGRES_* 即启用）。
+ *
+ * AUTH_PROVIDER 支持多值（`,` 或 `|` 分隔），例如 `feishu` / `dingtalk|feishu`：
+ *   - 只有一个真实登录方式（OAuth 仅一个）→ 登录页直接跳转，不出现选择器；
+ *   - 有多个 → 前端渲染登录方式选择界面。
+ *
+ * adaptive（自适应）：AUTH_PROVIDER=adaptive 时按客户端环境自动决定登录方式——
+ *   钉钉 App 内 → 钉钉（JSAPI 免登优先，失败回退 OAuth）；飞书 App 内 → 飞书 OAuth；
+ *   其他浏览器 → 渲染钉钉/飞书选择器。实际可用的 provider 由 getLoginOptions 展开为
+ *   已配置完整的 OAuth 平台（钉钉 DD_* 与飞书 FS_* env 各自齐全者才会出现）。
+ */
+import type { AuthConfig, AuthMethod } from "./types.js";
+/** 解析 AUTH_PROVIDER：按 [,|] 分隔 → trim → 去重 → 过滤非法值；none 与其它值共存时忽略。 */
+export declare function parseAuthProviders(raw: string | undefined): AuthMethod[];
+/** 解析并校验认证配置。非法值一律忽略；无任何合法值 → none。 */
+export declare function loadAuthConfig(): AuthConfig;
+/** 判断某个方式是否已配置完整（OAuth 平台看对应 env）。 */
+export declare function isMethodConfigured(config: AuthConfig, method: AuthMethod): boolean;
+/** 判断当前配置是否至少有一个可用的真实登录方式。 */
+export declare function isAuthEnabled(config: AuthConfig): boolean;

package/dist/server/auth/config.js

@@ -0,0 +1,82 @@
+const VALID_METHODS = ["dingtalk", "feishu", "builtin", "none", "adaptive"];
+/** 解析 AUTH_PROVIDER：按 [,|] 分隔 → trim → 去重 → 过滤非法值；none 与其它值共存时忽略。 */
+export function parseAuthProviders(raw) {
+    if (!raw)
+        return ["none"];
+    const parts = raw
+        .split(/[,|]/)
+        .map((s) => s.trim().toLowerCase())
+        .filter((s) => s.length > 0);
+    const unique = [];
+    for (const p of parts) {
+        if (!VALID_METHODS.includes(p))
+            continue; // 非法值忽略
+        if (unique.includes(p))
+            continue; // 去重
+        unique.push(p);
+    }
+    if (unique.length === 0)
+        return ["none"];
+    // none 表示"关闭登录"，只在其单独出现时生效
+    if (unique.includes("none") && unique.length > 1) {
+        return unique.filter((m) => m !== "none");
+    }
+    return unique;
+}
+/** 从完整列表取主认证方式（第一个非 none 的值；不存在 → none）。 */
+function primaryMethod(methods) {
+    const primary = methods.find((m) => m !== "none");
+    return primary ?? "none";
+}
+/** 解析并校验认证配置。非法值一律忽略；无任何合法值 → none。 */
+export function loadAuthConfig() {
+    const methods = parseAuthProviders(process.env.AUTH_PROVIDER);
+    const publicUrl = (process.env.APP_PUBLIC_URL || "").replace(/\/$/, "");
+    const cookiePath = process.env.AUTH_COOKIE_PATH || "/agent";
+    const sessionSecret = process.env.AUTH_SESSION_SECRET || process.env.SESSION_SECRET || "agent-kit-dev-secret";
+    const secureCookie = process.env.AUTH_SECURE_COOKIE === "true";
+    // 数据库持久化与开发免登为强制行为：数据必须落 PostgreSQL；登录必须是真实 OAuth。
+    return {
+        methods,
+        method: primaryMethod(methods),
+        publicUrl,
+        cookiePath,
+        sessionSecret,
+        secureCookie,
+        dingtalk: {
+            clientId: process.env.DD_CLIENT_ID || "",
+            clientSecret: process.env.DD_CLIENT_SECRET || "",
+            corpId: process.env.DD_CORP_ID || "",
+            agentId: process.env.DD_AGENT_ID || "",
+            robotCode: process.env.DD_ROBOT_CODE || "",
+            cardTemplateId: process.env.DD_CARD_TEMPLATE_ID || "",
+            redirectUri: process.env.DD_REDIRECT_URI || undefined,
+        },
+        feishu: {
+            appId: process.env.FS_APP_ID || "",
+            appSecret: process.env.FS_APP_SECRET || "",
+            baseUrl: (process.env.FS_BASE_URL || "https://open.feishu.cn").replace(/\/$/, ""),
+            redirectUri: process.env.FS_REDIRECT_URI || undefined,
+        },
+    };
+}
+/** 判断某个方式是否已配置完整（OAuth 平台看对应 env）。 */
+export function isMethodConfigured(config, method) {
+    if (method === "dingtalk") {
+        return Boolean(config.dingtalk.clientId && config.dingtalk.clientSecret && config.dingtalk.corpId);
+    }
+    if (method === "feishu") {
+        return Boolean(config.feishu.appId && config.feishu.appSecret);
+    }
+    // builtin（自建账号体系）留口子：登录系统尚未实现，暂恒 false——
+    // 即使 AUTH_PROVIDER 含 builtin 也不会出现在登录方式清单里（登录页不会渲染未实现的入口）。
+    // 后续实现自建登录时，改为读取 BUILTIN_USERS（用户名:密码哈希列表）等 env 即可启用。
+    if (method === "builtin") {
+        return false;
+    }
+    return false;
+}
+/** 判断当前配置是否至少有一个可用的真实登录方式。 */
+export function isAuthEnabled(config) {
+    return config.methods.some((m) => isMethodConfigured(config, m));
+}

package/dist/server/auth/db.d.ts

@@ -0,0 +1,159 @@
+import { Pool, type PoolClient } from "pg";
+import type { AuthUser } from "./types.js";
+export interface DbUser {
+    id: number;
+    provider: string;
+    unionId: string;
+    openId: string | null;
+    userCode: string | null;
+    name: string;
+    avatar: string | null;
+    email: string | null;
+    mobile: string | null;
+    deptName: string | null;
+    createdAt: Date;
+    updatedAt: Date;
+    lastLoginAt: Date | null;
+}
+export declare function ensureMigrated(): Promise<void>;
+/** 进程级连接池单例；未配置 Postgres 时返回 null。 */
+export declare function getPool(): Pool | null;
+/** 是否已启用数据库（配置了 Postgres 即启用；无开关，数据强制落库）。 */
+export declare function isDbEnabled(): boolean;
+/** 幂等建表；无数据库配置时静默跳过。 */
+export declare function migrate(): Promise<void>;
+/**
+ * 将认证用户 upsert 到 users 表：按 (provider, union_id) 匹配，存在则更新档案与
+ * last_login_at，不存在则插入。返回数据库行（含内部 id）。
+ */
+export declare function upsertUser(user: AuthUser, client?: PoolClient): Promise<DbUser | null>;
+/** 按 (provider, union_id) 查用户；不存在返回 null。 */
+export declare function findUserByProvider(provider: string, unionId: string, client?: PoolClient): Promise<DbUser | null>;
+/** 仅刷新 last_login_at（登录成功时调用）。 */
+export declare function touchLastLogin(id: number, client?: PoolClient): Promise<void>;
+/** 取飞书 OAuth 令牌（用户身份发消息用）；无令牌或非飞书用户返回 null。 */
+export declare function getUserOauthTokenByKey(userId: string): Promise<{
+    accessToken: string | null;
+    refreshToken: string | null;
+} | null>;
+/** 回写飞书 OAuth 令牌（刷新续期后调用；refreshToken 为空时保留原值）。 */
+export declare function updateUserOauthToken(userId: string, token: {
+    accessToken: string;
+    refreshToken?: string;
+}): Promise<void>;
+export interface AgentSessionRow {
+    id: string;
+    /** 用户主体键 `${provider}:${unionId}` */
+    userId: string;
+    title: string;
+    /** 本会话工作目录（沙箱范围；NULL=未限定，本地 app 选择工作区后写入） */
+    workspaceDir: string | null;
+    createdAt: Date;
+    updatedAt: Date;
+    messageCount: number;
+    lastPreview: string | null;
+}
+export interface AgentMessageRow {
+    id: number;
+    sessionId: string;
+    role: string;
+    content: string;
+    images: unknown;
+    /** 助手消息的分步思考过程（thinking/text/tool/ask/secs 数组），历史恢复时前端据此渲染思考胶囊 */
+    steps: unknown;
+    /** 助手消息总用时（秒） */
+    thinkingSecs: number | null;
+    /** 共用时（秒）：用户发送该消息 → 全部答复完成/停止，历史恢复时前端据此显示「共用时 xx」 */
+    elapsedSecs: number | null;
+    /** 本轮模型信息 { usedFallback, fallbackModel, models }，历史恢复时前端据此显示积分提示 */
+    modelInfo: unknown;
+    /** 本轮产物清单 [{id,name,type,size}]，历史恢复时前端据此渲染文件卡片与「查看所有产物」按钮 */
+    artifacts: unknown;
+    /** 用户消息的上传附件 { url, name, size }，历史恢复时前端据此恢复附件条 */
+    attachment: unknown;
+    /** 本轮消耗 { costUsd, usage }，历史恢复时前端据此展示 token 用量 */
+    costInfo: unknown;
+    createdAt: Date;
+}
+/** 注册/复用会话：已存在则返回现有行，不存在则插入。workspaceDir 仅首建时生效（复用不改）。 */
+export declare function createAgentSession(userId: string, sessionId: string, title?: string, workspaceDir?: string | null, client?: PoolClient): Promise<AgentSessionRow | null>;
+/** 当前用户的会话列表（按最近更新倒序），附带消息数与最后一条消息预览。 */
+export declare function listAgentSessions(userId: string, client?: PoolClient): Promise<AgentSessionRow[]>;
+/** 校验会话归属并返回该会话；不属于当前用户或不存在时返回 null。 */
+export declare function getAgentSession(userId: string, sessionId: string, client?: PoolClient): Promise<AgentSessionRow | null>;
+/** 重命名会话（仅限归属用户）。 */
+export declare function renameAgentSession(userId: string, sessionId: string, title: string, userRenamed?: boolean, client?: PoolClient): Promise<boolean>;
+/** SDK 自动命名（topic / ai-title）落库：仅当用户从未手动改过标题时覆盖（user_renamed=false），
+ *  否则跳过，保护用户手动编辑的标题不被自动命名替换。 */
+export declare function setSessionTitleIfNotUserRenamed(userId: string, sessionId: string, title: string, client?: PoolClient): Promise<boolean>;
+/** 更新会话工作目录（沙箱范围；传 null 表示清除限定）。仅限归属用户。 */
+export declare function updateAgentSessionWorkspace(userId: string, sessionId: string, workspaceDir: string | null, client?: PoolClient): Promise<boolean>;
+/** 删除会话（消息级联删除）。 */
+export declare function deleteAgentSession(userId: string, sessionId: string, client?: PoolClient): Promise<boolean>;
+/** admin 用量聚合（暂不鉴权）：summary 总量 + 按模型 / 按会话 / 单条消息 Top 消耗。
+ *  数据源：agent_messages.cost_info JSONB { costUsd, usage: { input_tokens, output_tokens } }。 */
+export declare function queryAdminUsage(limit?: number): Promise<{
+    summary: {
+        inputTokens: number;
+        outputTokens: number;
+        costUsd: number;
+        costedMsgs: number;
+        sessions: number;
+    };
+    byModel: {
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        msgs: number;
+    }[];
+    bySession: {
+        sessionId: string;
+        title: string;
+        createdAt: Date;
+        inputTokens: number;
+        outputTokens: number;
+        costUsd: number;
+        costedMsgs: number;
+    }[];
+    details: {
+        id: number;
+        sessionId: string;
+        sessionTitle: string;
+        createdAt: Date;
+        inputTokens: number;
+        outputTokens: number;
+        costUsd: number;
+        model: string;
+        contentPreview: string;
+        userContent: string;
+        user: string;
+    }[];
+} | null>;
+/** 追加一条消息并刷新会话更新时间。 */
+export declare function addAgentMessage(sessionId: string, role: string, content: string, images?: unknown, steps?: unknown, thinkingSecs?: number | null, modelInfo?: unknown, artifacts?: unknown, attachment?: unknown, elapsedSecs?: number | null, costInfo?: unknown, client?: PoolClient): Promise<boolean>;
+/** 会话内全部消息（按时间正序）；不属于当前用户返回空数组。 */
+export declare function listAgentMessages(userId: string, sessionId: string, client?: PoolClient): Promise<AgentMessageRow[]>;
+/** Copy a bounded history in one transaction; serialize suffix allocation per owner. */
+export declare function forkAgentSession(userId: string, sourceId: string, messageIndex: number, messageId?: number, snapshot?: unknown): Promise<AgentSessionRow>;
+/** 会话内所有用户消息的附件 url 列表（删会话清理磁盘文件前收集，须在删除会话前调用）。 */
+export declare function listSessionAttachmentUrls(userId: string, sessionId: string, client?: PoolClient): Promise<string[]>;
+/** 截断会话：删除指定 id（含）之后的所有消息（编辑重发时调用，让刷新后不出现旧消息）。 */
+export declare function truncateAgentMessages(userId: string, sessionId: string, upToId: number): Promise<void>;
+/** 分享记录行。 */
+export interface ShareRow {
+    token: string;
+    sessionId: string;
+    /** 创建者主体键 `${provider}:${unionId}` */
+    createdBy: string;
+    createdAt: Date;
+}
+/** 按主键查分享记录；不存在返回 null。 */
+export declare function getShareByToken(token: string): Promise<ShareRow | null>;
+/** 创建分享记录；会话必须已存在（外键保证归属校验由调用方完成）。 */
+export declare function createShareRecord(token: string, sessionId: string, userId: string): Promise<boolean>;
+/** 按 session_id 查会话标题（公开分享页展示用；不校验归属）。 */
+export declare function getSessionTitleBySessionId(sessionId: string): Promise<string>;
+/** 公开读取会话消息（分享页用；不校验用户归属，依赖分享 token 不可枚举）。 */
+export declare function listAgentMessagesPublic(sessionId: string): Promise<AgentMessageRow[]>;
+/** 按主体键 `${provider}:${unionId}` 查钉钉 userid / 飞书 employee_id（OA 发送消息时作为 sender 使用）。 */
+export declare function getUserCodeByKey(userId: string): Promise<string>;