wave-code 1.6.0 → 1.6.1

package/src/stdio/agentBridge.ts

@@ -60,6 +60,7 @@ import {
   updateUserPreferenceSettings,
   type UserPreferenceSettings,
   type UserPreferenceSettingsView,
+  type AccountCredential,
   type SubagentConfiguration,
   type SkillMetadata,
   type PluginReloadResult,
@@ -838,6 +839,25 @@ export class AgentBridge {
       if (entry) {
         this.pendingPermissions.delete(p.requestId);
         entry.resolve(p.decision);
+        // 不带反馈的拒绝 = 用户关掉了确认卡（关闭键 / Esc），本轮在用户这边就
+        // 结束：与桌面端 / VS Code / JetBrains 的 `denied → abortMessage()` 是
+        // 同一条语义，也与 Claude Code 的 `cancelAndAbort`（无 feedback 直接
+        // abort 整轮）一致。**带 message 的拒绝不中断**——那是用户给出理由后
+        // 拒绝这一次工具调用，模型要照反馈继续或改问（spec
+        // ui/confirm-ui.md 场景 2 / 5、remote/remote-web.md 场景 24）。
+        // 中断本身不阻塞回包，失败（会话正被重建 / 已销毁）也不影响这次拒绝
+        // 已经生效，因此静默。
+        const rejectionFeedback = p.decision?.message?.trim();
+        if (
+          p.decision?.behavior === "deny" &&
+          !rejectionFeedback &&
+          entry.sessionId
+        ) {
+          const session = this.sessions.get(entry.sessionId);
+          if (session && !session.transitioning) {
+            void session.agent.abortMessage();
+          }
+        }
       }
     }
   }
@@ -1345,11 +1365,12 @@ export class AgentBridge {
    */
   private emitContextUsage(agent: Agent): void {
     const tokens = agent.latestTotalTokens;
-    const max = agent.getMaxInputTokens();
+    const max = agent.getContextWindow();
+    const maxOutput = agent.getMaxOutputTokens?.() ?? 32_000;
     if (tokens > 0 && max > 0) {
       this.emit(
         "contextUsage",
-        remoteWire.contextUsage(tokens, max),
+        remoteWire.contextUsage(tokens, max, maxOutput),
         agent.sessionId,
       );
     }
@@ -1364,9 +1385,10 @@ export class AgentBridge {
    */
   private contextUsagePercentOf(agent: Agent): number | undefined {
     const tokens = agent.latestTotalTokens;
-    const max = agent.getMaxInputTokens();
+    const max = agent.getContextWindow();
+    const maxOutput = agent.getMaxOutputTokens?.() ?? 32_000;
     if (tokens > 0 && max > 0) {
-      return percentOf(tokens, max);
+      return percentOf(tokens, max, maxOutput);
     }
     return undefined;
   }
@@ -2790,20 +2812,77 @@ export class AgentBridge {
   }
 
   // ── Auth (global) ────────────────────────────────────────────
+  /**
+   * 认证状态 + **凭据形态**（spec desktop-account.md「账户卡片 · 认证方式」）。
+   *
+   * `isAuthenticated` 只回答「有没有 SSO token」；宿主据此渲染卡片与发送入口会
+   * 把「key 写在 settings.json 里」的人判成没有凭据（卡片显示登录按钮、输入框
+   * 被锁）。故同一次查询里一并给出 `credential`，优先级与 SDK 的
+   * `resolveGatewayConfig` 对齐：企业账号 > 自定义网关 > 无凭据。
+   */
   private async getAuthStatus(): Promise<{
     isAuthenticated: boolean;
+    credential: AccountCredential;
     user: { id: string; email?: string } | undefined;
     serverUrl: string;
   }> {
     const authService = AuthService.getInstance();
     await authService.checkAndRefreshTokenIfNeeded();
+    const isAuthenticated = authService.isSSOAuthenticated();
     return {
-      isAuthenticated: authService.isSSOAuthenticated(),
+      isAuthenticated,
+      credential: this.resolveCredential(isAuthenticated),
       user: authService.getAuthUser(),
       serverUrl: authService.getServerUrl(),
     };
   }
 
+  /**
+   * 凭据形态。**判据只有这一处**——消费方（宿主 / webview）不得自己读配置推断。
+   *
+   * 网关层的取值顺序与 `resolveGatewayConfig` 一致：
+   * - 无 SSO：先问一个活着的会话的 `getGatewayConfig()`（与真实请求同源，已合并
+   *   用户 / 项目 / 远端三层 `env`），没有会话时退回「用户级 settings.json 的
+   *   `env`（`loadUserConfigEnv`）→ OS 环境变量」。
+   * - 有 SSO：**不能**用 `getGatewayConfig()`（它此时返回 SSO 凭据），回落的那份
+   *   网关只能从配置层读——卡片要靠它给「退出登录」项写后果。
+   *
+   * 注意 `setEnvironmentVars` 只把 `WAVE_SERVER_URL` 镜像进 `process.env`，网关
+   * 凭据留在会话快照里，所以只看 `process.env` 会把配好的人判成无凭据。
+   */
+  private resolveCredential(isAuthenticated: boolean): AccountCredential {
+    const gateway = isAuthenticated
+      ? this.configuredGatewayCredential()
+      : (this.sessionGatewayCredential() ?? this.configuredGatewayCredential());
+    if (isAuthenticated) {
+      // 企业账号生效中；`gateway` 仅表示「退出登录后会落到的那份」。
+      return gateway ? { kind: "sso", gateway } : { kind: "sso" };
+    }
+    return gateway ? { kind: "gateway", gateway } : { kind: "none" };
+  }
+
+  /** 活着的会话真正解析出来的网关配置（无 SSO 时它就是要发出去的那一份）。 */
+  private sessionGatewayCredential(): { host: string | null } | null {
+    for (const entry of this.sessions.values()) {
+      const gateway = entry.agent.getGatewayConfig();
+      if (gateway.apiKey) return { host: hostOfBaseUrl(gateway.baseURL) };
+    }
+    return null;
+  }
+
+  /**
+   * 配置层的网关凭据：用户级 settings.json 的 `env` 优先于 OS 环境变量（与
+   * `resolveGatewayConfig` 的 `env（settings.json）> process.env` 同序）。
+   */
+  private configuredGatewayCredential(): { host: string | null } | null {
+    const userEnv = loadUserConfigEnv();
+    const pick = (key: "WAVE_API_KEY" | "WAVE_BASE_URL"): string | undefined =>
+      userEnv[key]?.trim() || process.env[key]?.trim() || undefined;
+    const apiKey = pick("WAVE_API_KEY");
+    if (!apiKey) return null;
+    return { host: hostOfBaseUrl(pick("WAVE_BASE_URL")) };
+  }
+
   private async getAccountInfo(): Promise<{
     /**
      * 计费结论（套餐两根条 + 到期日 + 降级原因），形状见 codechat `GET /api/v1/account` → `billing`
@@ -3285,11 +3364,12 @@ export class AgentBridge {
         // request (fresh total) and on session restore (the persisted total),
         // so the webview button shows real usage without waiting for a turn —
         // aligned with the CLI, which reads the same callback (spec 场景 5/6).
-        const max = ctx.agent?.getMaxInputTokens() ?? 0;
+        const max = ctx.agent?.getContextWindow() ?? 0;
+        const maxOutput = ctx.agent?.getMaxOutputTokens?.() ?? 32_000;
         if (max > 0) {
           this.emit(
             "contextUsage",
-            remoteWire.contextUsage(tokens, max),
+            remoteWire.contextUsage(tokens, max, maxOutput),
             ctx.registeredSessionId,
           );
         }
@@ -3355,6 +3435,13 @@ export class AgentBridge {
       onMcpServersChange: (servers: McpServerStatus[]) => {
         this.emit("mcpServersChange", { servers }, ctx.registeredSessionId);
       },
+      onQuotaHeaders: (headers: Record<string, string>) => {
+        // 额度响应头（codechat proxy 每次模型调用都发，spec:
+        // desktop/desktop-account.md「额度响应头」）：桌面端宿主据此刷新账户卡片的
+        // 用量读数，是 60s 轮询删除后的主要刷新源。原样透传 —— 字段解析与映射都
+        // 在消费方。不进 remoteWire：远程页不渲染账户卡片。
+        this.emit("quotaHeaders", { headers }, ctx.registeredSessionId);
+      },
       onNotificationMessageAdded: (params) => {
         // The LAST matching message, not the first: one task can notify more
         // than once under the same task id (a workflow warns about its size
@@ -3488,6 +3575,19 @@ function isSessionRecoveryError(error: unknown): boolean {
   );
 }
 
+/**
+ * `baseURL` → 网关主机名（`api.example.com`）。缺省 / 不可解析一律 null（卡片读
+ * 「默认端点」）——这里只做展示，不做任何可用性判断。
+ */
+function hostOfBaseUrl(baseURL: string | undefined): string | null {
+  if (!baseURL) return null;
+  try {
+    return new URL(baseURL).host || null;
+  } catch {
+    return null;
+  }
+}
+
 /** Read a text file, returning "" when it does not exist. */
 async function readTextFileSafe(filePath: string): Promise<string> {
   try {

package/src/stdio/startupStage.ts

@@ -0,0 +1,22 @@
+/**
+ * 首次启动的进度上报（spec `ui/stdio-transport.md`「首次启动的进度文案」）。
+ *
+ * 走的是同一条 stdout 上的 JSON-RPC 通知通道（一条一行），只是没有 `sessionId`
+ * ——这些阶段发生在任何会话存在之前。宿主侧 `NotificationRouter` 的全局处理器
+ * 收下它，把它变成加载动画下方那一行文案。
+ *
+ * **旁路**：通知发不出去（stdout 已关、宿主已经退出）不是启动错误，写失败一律
+ * 吞掉；上报也不参与任何完成条件，没有宿主消费时启动照旧。
+ */
+
+import type { StartupStage } from "wave-agent-sdk";
+
+export function reportStartupStage(stage: StartupStage): void {
+  try {
+    process.stdout.write(
+      `${JSON.stringify({ method: "startupStage", params: { stage } })}\n`,
+    );
+  } catch {
+    // 见文件头：上报失败不影响启动。
+  }
+}

package/src/stdio-cli.ts

@@ -8,6 +8,7 @@
 
 import { ensureRuntimeDeps, setTelemetryStdoutProtected } from "wave-agent-sdk";
 import { StdioServer } from "./stdio/stdioServer.js";
+import { reportStartupStage } from "./stdio/startupStage.js";
 import { createRemoteControlSinkFactory } from "./remote/remoteControlSink.js";
 import { logger } from "./utils/logger.js";
 
@@ -64,7 +65,11 @@ export async function startStdioCli(): Promise<void> {
   // answering any RPC (see cli.tsx). Blocking here is what closes the race: the
   // host's `initialize` — and with it the webview's initial state — only lands
   // once a pasted image can be handled. Failures are non-fatal.
-  await ensureRuntimeDeps();
+  //
+  // The ~25MB this can pull is the longest silent stretch of a first launch, so
+  // it is reported (a warm cache reports nothing: `ensureRuntimeDeps` only
+  // fires `onStage` when something is actually missing).
+  await ensureRuntimeDeps({ onStage: reportStartupStage });
   // Registered here (stdio mode only): the interactive CLI keeps Node's
   // default behavior so the terminal shows the crash stack directly.
   process.on("uncaughtException", (error) =>
@@ -79,6 +84,9 @@ export async function startStdioCli(): Promise<void> {
     // same machine — is what serves a paired phone.
     createRemoteSink: createRemoteControlSinkFactory(process.cwd()),
   });
+  // Last startup stage: from here the process is up and what the user is waiting
+  // for is `initialize` (and, on the restore path, the transcript replay).
+  reportStartupStage("startingSession");
   server.start();
 
   // Keep the process alive — readline on stdin handles the event loop.