u1s1-cli 1.14.6 → 1.15.0

package/dist/launcher.d.ts

@@ -0,0 +1,48 @@
+import { type DefaultAgent } from "./config.js";
+export type AgentChoice = DefaultAgent;
+/** 启动器的保留字:不会再当 prompt 交给默认客户端。 */
+export declare const LAUNCHER_WORDS: readonly ["pi", "claude", "codex", "use", "agents", "config"];
+export declare function isLauncherWord(word: string): boolean;
+export type Dispatch = {
+    kind: "agent";
+    agent: AgentChoice;
+    args: string[];
+    via: "explicit" | "default" | "legacy" | "fallback";
+} | {
+    kind: "select";
+    args: string[];
+} | {
+    kind: "use";
+    args: string[];
+} | {
+    kind: "agents";
+    args: string[];
+} | {
+    kind: "config";
+    args: string[];
+};
+export interface DispatchContext {
+    /** config.json 里的 defaultAgent */
+    defaultAgent?: AgentChoice;
+    /** 存量用户:已有配置或用过 Pi */
+    existingUser: boolean;
+    /** stdin/stdout 都是终端,能弹选择器 */
+    interactive: boolean;
+}
+/**
+ * 纯函数:参数 → 去哪。
+ * 全新用户只有在「不带参数 + 终端可交互」时才出选择器;带了参数(如 -p 脚本调用)或没有终端时
+ * 直接用 Pi、不记住,免得脚本卡在提问上。
+ */
+export declare function resolveDispatch(args: readonly string[], ctx: DispatchContext): Dispatch;
+/**
+ * 存量用户:config.json 里除 baseUrl 以外还有东西(登录凭据、模型、邮箱…),或者 Pi 的
+ * agent 目录已经有设置文件。只有 baseUrl 的配置是安装脚本写的站点信息,不算。
+ */
+export declare function isExistingUser(raw: Record<string, unknown> | undefined, agentSettingsExists: boolean): boolean;
+export declare function runLauncher(args: string[]): Promise<void>;
+export declare function parseSelection(answer: string): AgentChoice | undefined;
+/** `u1s1 --help` 中文版的启动器几行(中文按 CJK 双宽手工对齐)。 */
+export declare function launcherHelpLines(): string[];
+/** 英文版:与 index.ts 的命令表一起按 ASCII 宽度对齐。 */
+export declare function launcherHelpRows(): [string, string][];

package/dist/launcher.js

@@ -0,0 +1,235 @@
+/**
+ * 启动器:选择器与分发。一个命令 `u1s1`,可以启动内置的 Pi,也可以启动用户机器上的官方原版客户端。
+ *
+ *   u1s1 pi|claude|codex [参数…] 指定客户端,后面的参数原样传给它
+ *   u1s1 [参数…]                 默认客户端;存量用户(已有配置、没有默认客户端字段)默认 Pi,
+ *                                全新用户首次运行出选择器并记住
+ *   u1s1 use <客户端>            改默认
+ *   u1s1 agents [pin|unpin|install …]  安装状态、版本、是否有更新;锁定版本;只下载不启动
+ *   u1s1 config [list|get|set|unset …]  用户可调的设置(默认客户端、各客户端默认模型、自动更新),见 config-command.ts
+ *
+ * claude / codex 是否可用完全由网关配置(/v1/agents/:agent/config 的 enabled)决定,这里不写死。
+ *
+ * 现有子命令(login、usage、model、update、deploy 等)在 index.ts 里先于这里处理,行为不变。
+ */
+import { existsSync } from "node:fs";
+import { createInterface } from "node:readline/promises";
+import { agentSettingsFile, isDefaultAgent, loadConfig, readRawConfigFile, saveDefaultAgent, } from "./config.js";
+import { tr } from "./i18n.js";
+/** 启动器的保留字:不会再当 prompt 交给默认客户端。 */
+export const LAUNCHER_WORDS = ["pi", "claude", "codex", "use", "agents", "config"];
+export function isLauncherWord(word) {
+    return LAUNCHER_WORDS.includes(word);
+}
+/**
+ * 纯函数:参数 → 去哪。
+ * 全新用户只有在「不带参数 + 终端可交互」时才出选择器;带了参数(如 -p 脚本调用)或没有终端时
+ * 直接用 Pi、不记住,免得脚本卡在提问上。
+ */
+export function resolveDispatch(args, ctx) {
+    const [first, ...rest] = args;
+    switch (first) {
+        case "pi":
+        case "claude":
+        case "codex":
+            return { kind: "agent", agent: first, args: rest, via: "explicit" };
+        case "use":
+            return { kind: "use", args: rest };
+        case "agents":
+            return { kind: "agents", args: rest };
+        case "config":
+            return { kind: "config", args: rest };
+    }
+    if (ctx.defaultAgent)
+        return { kind: "agent", agent: ctx.defaultAgent, args: [...args], via: "default" };
+    if (ctx.existingUser)
+        return { kind: "agent", agent: "pi", args: [...args], via: "legacy" };
+    if (args.length === 0 && ctx.interactive)
+        return { kind: "select", args: [] };
+    return { kind: "agent", agent: "pi", args: [...args], via: "fallback" };
+}
+/**
+ * 存量用户:config.json 里除 baseUrl 以外还有东西(登录凭据、模型、邮箱…),或者 Pi 的
+ * agent 目录已经有设置文件。只有 baseUrl 的配置是安装脚本写的站点信息,不算。
+ */
+export function isExistingUser(raw, agentSettingsExists) {
+    if (agentSettingsExists)
+        return true;
+    if (!raw)
+        return false;
+    return Object.entries(raw).some(([key, value]) => key !== "baseUrl" && value !== undefined && value !== null && value !== "");
+}
+// ---------------------------------------------------------------- run
+export async function runLauncher(args) {
+    const raw = readRawConfigFile();
+    const dispatch = resolveDispatch(args, {
+        defaultAgent: isDefaultAgent(raw?.["defaultAgent"]) ? raw["defaultAgent"] : undefined,
+        existingUser: isExistingUser(raw, existsSync(agentSettingsFile)),
+        interactive: !!process.stdin.isTTY && !!process.stdout.isTTY,
+    });
+    switch (dispatch.kind) {
+        case "use":
+            useCommand(dispatch.args);
+            return;
+        case "agents":
+            await agentsCommand(dispatch.args);
+            return;
+        case "config": {
+            const { configCommand } = await import("./config-command.js");
+            process.exitCode = await configCommand(dispatch.args);
+            return;
+        }
+        case "select": {
+            const agent = await selectAgent();
+            saveDefaultAgent(agent);
+            console.log(tr(`  已记住,以后运行 u1s1 直接进入 ${agent}。随时可以用 u1s1 use <客户端> 修改。`, `  Saved. Running u1s1 now opens ${agent}. Change it any time with u1s1 use <client>.`));
+            await launch(agent, dispatch.args);
+            return;
+        }
+        case "agent":
+            await launch(dispatch.agent, dispatch.args);
+            return;
+    }
+}
+async function launch(agent, args) {
+    if (agent === "pi") {
+        const { ensureAuth } = await import("./login.js");
+        const cfg = await ensureAuth();
+        const { installPendingUpdate, runAgent } = await import("./agents/pi.js");
+        await runAgent(cfg, args);
+        // autoUpdate 开着且启动时发现了新版:现在装(TUI 已退出,原地重铺安全)
+        installPendingUpdate();
+        return;
+    }
+    const code = agent === "codex"
+        ? await (await import("./agents/codex.js")).runCodex(args)
+        : await (await import("./agents/claude.js")).runClaude(args);
+    // 签名代理是 unref 的,子进程退出后不必等;直接带着子进程的退出码退出
+    process.exit(code);
+}
+const OPTIONS = [
+    {
+        agent: "pi",
+        line: () => tr("pi      u1s1 内置的编程助手,中文界面,开箱即用(推荐)", "pi      u1s1's built-in coding assistant, ready to go (recommended)"),
+    },
+    {
+        agent: "claude",
+        line: () => tr("claude  启动官方 Claude Code(首次使用会从官方渠道下载)", "claude  Launch the official Claude Code (downloaded from the official source on first use)"),
+    },
+    {
+        agent: "codex",
+        line: () => tr("codex   启动官方 Codex CLI(首次使用会从官方渠道下载)", "codex   Launch the official Codex CLI (downloaded from the official source on first use)"),
+    },
+];
+export function parseSelection(answer) {
+    const a = answer.trim().toLowerCase();
+    if (a === "")
+        return "pi";
+    const byIndex = Number(a);
+    if (Number.isInteger(byIndex) && byIndex >= 1 && byIndex <= OPTIONS.length)
+        return OPTIONS[byIndex - 1].agent;
+    return OPTIONS.find((o) => o.agent === a)?.agent;
+}
+async function selectAgent() {
+    console.log("");
+    console.log(tr("  选择要使用的客户端(以后可以用 u1s1 use <客户端> 修改):", "  Choose a client (you can change it later with u1s1 use <client>):"));
+    OPTIONS.forEach((o, i) => console.log(`    ${i + 1}) ${o.line()}`));
+    const rl = createInterface({ input: process.stdin, output: process.stdout });
+    try {
+        for (;;) {
+            const choice = parseSelection(await rl.question(tr("  输入序号 [1]: ", "  Enter a number [1]: ")));
+            if (choice)
+                return choice;
+            console.log(tr(`  请输入 1-${OPTIONS.length}`, `  Please enter 1-${OPTIONS.length}`));
+        }
+    }
+    finally {
+        rl.close();
+    }
+}
+// ---------------------------------------------------------------- use
+function useCommand(args) {
+    const target = args[0];
+    if (!isDefaultAgent(target)) {
+        console.error(tr("用法: u1s1 use <pi|claude|codex>", "Usage: u1s1 use <pi|claude|codex>"));
+        process.exitCode = target === undefined || target === "--help" || target === "-h" ? 0 : 1;
+        return;
+    }
+    saveDefaultAgent(target);
+    console.log(tr(`已把默认客户端设为 ${target}:运行 u1s1 直接进入它。`, `Default client set to ${target}: running u1s1 now opens it.`));
+}
+// ---------------------------------------------------------------- agents
+async function agentsCommand(args) {
+    const [sub, agent, version] = args;
+    if (sub === "--help" || sub === "-h" || sub === "help") {
+        for (const line of agentsHelpLines())
+            console.log(line);
+        return;
+    }
+    if (sub === "pin" || sub === "unpin") {
+        if (agent !== "claude" && agent !== "codex") {
+            console.error(sub === "pin"
+                ? tr("用法: u1s1 agents pin <claude|codex> <版本>", "Usage: u1s1 agents pin <claude|codex> <version>")
+                : tr("用法: u1s1 agents unpin <claude|codex>", "Usage: u1s1 agents unpin <claude|codex>"));
+            process.exitCode = 1;
+            return;
+        }
+        const { pinCommand, unpinCommand } = await import("./agents/status.js");
+        process.exitCode = sub === "pin" ? await pinCommand(version, agent) : unpinCommand(agent);
+        return;
+    }
+    if (sub === "install") {
+        if (agent === "pi" && version === undefined) {
+            console.log(tr("pi 随 u1s1 一起安装,不用单独下载。", "pi ships with u1s1; nothing to download."));
+            return;
+        }
+        if ((agent !== "claude" && agent !== "codex") || version !== undefined) {
+            console.error(tr("用法: u1s1 agents install <claude|codex>", "Usage: u1s1 agents install <claude|codex>"));
+            process.exitCode = 1;
+            return;
+        }
+        const { installCommand } = await import("./agents/status.js");
+        process.exitCode = await installCommand({}, agent);
+        return;
+    }
+    if (sub !== undefined) {
+        for (const line of agentsHelpLines())
+            console.error(line);
+        process.exitCode = 1;
+        return;
+    }
+    const { printAgentsStatus } = await import("./agents/status.js");
+    await printAgentsStatus(loadConfig().defaultAgent ?? (isExistingUser(readRawConfigFile(), existsSync(agentSettingsFile)) ? "pi" : undefined));
+}
+function agentsHelpLines() {
+    return [
+        tr("用法:", "Usage:"),
+        tr("  u1s1 agents                       查看各客户端的安装状态、版本、是否有更新", "  u1s1 agents                       Show each client's install status, version and updates"),
+        tr("  u1s1 agents install <客户端>      下载或更新到可用版本,不启动(claude / codex)", "  u1s1 agents install <client>      Download or update to a usable version without launching (claude / codex)"),
+        tr("  u1s1 agents pin <客户端> <版本>   锁定版本(不超出允许的范围)", "  u1s1 agents pin <client> <ver>    Pin a version (within the allowed range)"),
+        tr("  u1s1 agents unpin <客户端>        解除锁定,恢复自动更新", "  u1s1 agents unpin <client>        Remove the pin and resume automatic updates"),
+    ];
+}
+// ---------------------------------------------------------------- help
+/** `u1s1 --help` 中文版的启动器几行(中文按 CJK 双宽手工对齐)。 */
+export function launcherHelpLines() {
+    return [
+        "   u1s1 pi                进入内置的 Pi",
+        "   u1s1 claude [参数]     启动官方 Claude Code(首次使用会下载)",
+        "   u1s1 codex [参数]      启动官方 Codex CLI(首次使用会下载)",
+        "   u1s1 use <客户端>      设置默认客户端(pi / claude / codex)",
+        "   u1s1 agents            查看各客户端的安装状态、版本与更新",
+        "   u1s1 config            设置默认客户端、各客户端的默认模型、自动更新",
+    ];
+}
+/** 英文版:与 index.ts 的命令表一起按 ASCII 宽度对齐。 */
+export function launcherHelpRows() {
+    return [
+        ["u1s1 pi", "Open the built-in Pi"],
+        ["u1s1 claude [args]", "Launch the official Claude Code (downloaded on first use)"],
+        ["u1s1 codex [args]", "Launch the official Codex CLI (downloaded on first use)"],
+        ["u1s1 use <client>", "Set the default client (pi / claude / codex)"],
+        ["u1s1 agents", "Show each client's install status, version and updates"],
+        ["u1s1 config", "Set the default client, each client's default model and auto-update"],
+    ];
+}

package/dist/mcp-server.d.ts

@@ -0,0 +1,105 @@
+import type { ApiFeatures } from "./api.js";
+export declare const MCP_SERVER_NAME = "u1s1";
+export declare const MCP_PROXY_ENV = "U1S1_MCP_PROXY";
+export declare const MCP_TOKEN_ENV = "U1S1_MCP_TOKEN";
+export declare const MCP_TOOLS_ENV = "U1S1_MCP_TOOLS";
+/** 关闭开关:U1S1_MCP=0(或 off / false / no)时启动器不注入 MCP 服务。 */
+export declare const MCP_SWITCH_ENV = "U1S1_MCP";
+/** 排查用:设成一个文件路径时,服务把收发的每条协议消息和启动信息(不含 token)追加写进去。 */
+export declare const MCP_TRACE_ENV = "U1S1_MCP_TRACE";
+/** 启动器写进客户端环境、再由客户端转给 MCP 服务的变量(客户端配置里只引用名字,不写值)。 */
+export declare const MCP_ENV_NAMES: readonly ["U1S1_MCP_PROXY", "U1S1_MCP_TOKEN", "U1S1_MCP_TOOLS", "U1S1_LANG"];
+/** 我们支持的协议版本,新的在前。客户端要的版本在列表里就原样回,否则回第一个(规范的版本协商)。 */
+export declare const MCP_PROTOCOL_VERSIONS: readonly ["2025-11-25", "2025-06-18", "2025-03-26", "2024-11-05"];
+export type McpToolName = "web_search" | "web_fetch" | "generate_image";
+/** 用户用环境变量关掉了注入。 */
+export declare function mcpInjectionDisabled(env?: NodeJS.ProcessEnv): boolean;
+/**
+ * 网关能力 → U1S1_MCP_TOOLS。缺省值与 Pi 相同(web.ts / agents/pi.ts):搜索缺字段按开、
+ * 云端渲染和生图缺字段按关;拿不到 /v1/models 时(features 为 undefined)同样处理。
+ */
+export declare function mcpToolFlags(features: ApiFeatures | undefined): string;
+/** 客户端拉起服务用的命令:同一个 node、同一份 CLI(dist/index.js),子命令 mcp-server。 */
+export declare function mcpServerCommand(): {
+    command: string;
+    args: string[];
+};
+/** 注入时写进客户端 MCP 配置的环境变量(值由启动器放进客户端自己的环境,见各客户端模块)。 */
+export interface McpLaunchEnv {
+    [MCP_PROXY_ENV]: string;
+    [MCP_TOKEN_ENV]: string;
+    [MCP_TOOLS_ENV]: string;
+    U1S1_LANG: "zh" | "en";
+}
+export declare function mcpLaunchEnv(input: {
+    proxyBaseUrl: string;
+    localToken: string;
+    features: ApiFeatures | undefined;
+    lang: "zh" | "en";
+}): McpLaunchEnv;
+export declare function isLoopbackProxyUrl(url: string): boolean;
+export interface McpContent {
+    type: "text";
+    text: string;
+}
+export interface McpToolResult {
+    content: McpContent[];
+    isError?: boolean;
+}
+export interface McpTool {
+    name: McpToolName;
+    title: string;
+    description: string;
+    inputSchema: Record<string, unknown>;
+    annotations: Record<string, unknown>;
+    /** 参数已按 inputSchema 校验过;失败抛 Error,由服务转成 isError 结果。 */
+    run(args: Record<string, unknown>, signal: AbortSignal): Promise<McpToolResult>;
+}
+export interface McpToolsConfig {
+    /** 签名代理地址;没有(或不是回环地址)时只提供直连的 web_fetch */
+    proxyBaseUrl?: string;
+    localToken?: string;
+    /** U1S1_MCP_TOOLS 里的能力 */
+    flags: ReadonlySet<string>;
+}
+export declare function parseMcpToolsConfig(env: NodeJS.ProcessEnv): McpToolsConfig;
+/**
+ * 按配置建工具。执行逻辑来自 Pi 的 tools.ts(动态引入:它会带进 Pi 的模块,只在这里用)。
+ * cfg 里只有代理地址和本地 token;没有设备凭证,api.ts 会以 Bearer <本地 token> 访问代理。
+ */
+export declare function buildMcpTools(config: McpToolsConfig): Promise<McpTool[]>;
+/** 只做 inputSchema 里用到的那几种检查(必填、类型、数值范围、数组元素为字符串),出错给模型能看懂的一句话。 */
+export declare function checkArguments(schema: Record<string, unknown>, args: unknown): string | null;
+export interface McpServerOptions {
+    tools: McpTool[] | (() => Promise<McpTool[]>);
+    /** 写一条协议消息(不含换行) */
+    send: (line: string) => void;
+    log?: (line: string) => void;
+}
+/**
+ * 协议核心:喂进一行,按需回一行。与传输无关,测试直接驱动它。
+ * 请求并发处理;notifications/cancelled 中止对应的 tools/call,之后不再回结果(规范要求)。
+ */
+export declare class McpServer {
+    private readonly options;
+    private readonly inflight;
+    private toolsPromise;
+    constructor(options: McpServerOptions);
+    private tools;
+    private reply;
+    /** 处理一行输入。返回的 Promise 在这条消息处理完(含工具执行)后落定。 */
+    handleLine(line: string): Promise<void>;
+    /** 关闭:中止所有在执行的工具调用。 */
+    abortAll(): void;
+    private handleMessage;
+    private handleNotification;
+    private initializeResult;
+    private callTool;
+}
+/**
+ * 本进程的全局 dispatcher:发往回环地址(签名代理)的请求直连,其余照旧交给原来的 dispatcher
+ * (按用户的 HTTP(S)_PROXY 走)。tools.ts / api.ts 用全局 fetch,这样不用改它们。
+ */
+export declare function installLoopbackDirectDispatcher(): void;
+/** `u1s1 mcp-server` 入口:stdin 一行一条消息,stdout 只写协议消息。stdin 关闭即退出。 */
+export declare function runMcpServer(): Promise<void>;