assistant-ui 0.0.121 → 0.0.123

package/src/commands/mcp.ts

@@ -2,6 +2,14 @@ import { Command } from "commander";
 import fs from "node:fs";
 import path from "node:path";
 import os from "node:os";
+import {
+  applyEdits,
+  parse as parseJsonc,
+  parseTree,
+  stripComments,
+  type Node,
+  type ParseError,
+} from "jsonc-parser";
 import { logger } from "../lib/utils/logger";
 import { runSpawn, SpawnExitError, SpawnSignalError } from "../lib/run-spawn";
 import * as p from "@clack/prompts";
@@ -21,22 +29,17 @@ const MCP_CONFIGS: Record<
   {
     name: string;
     getPath: () => string;
-    config: object;
-    replaceServerKey?: string;
+    serverKey: string;
+    server: object;
+    jsonc?: boolean;
     postInstall?: string;
   }
 > = {
   cursor: {
     name: "Cursor",
     getPath: () => path.join(process.cwd(), ".cursor", "mcp.json"),
-    config: {
-      mcpServers: {
-        "assistant-ui": {
-          url: HOSTED_MCP_URL,
-        },
-      },
-    },
-    replaceServerKey: "mcpServers",
+    serverKey: "mcpServers",
+    server: { url: HOSTED_MCP_URL },
     postInstall:
       "Open Cursor Settings → MCP → find 'assistant-ui' and click enable.",
   },
@@ -44,47 +47,33 @@ const MCP_CONFIGS: Record<
     name: "Windsurf",
     getPath: () =>
       path.join(os.homedir(), ".codeium", "windsurf", "mcp_config.json"),
-    config: {
-      mcpServers: {
-        "assistant-ui": {
-          serverUrl: HOSTED_MCP_URL,
-        },
-      },
-    },
-    replaceServerKey: "mcpServers",
+    serverKey: "mcpServers",
+    server: { serverUrl: HOSTED_MCP_URL },
     postInstall: "Fully quit and re-open Windsurf to activate.",
   },
   vscode: {
     name: "VSCode",
     getPath: () => path.join(process.cwd(), ".vscode", "mcp.json"),
-    config: {
-      servers: {
-        "assistant-ui": {
-          type: "http",
-          url: HOSTED_MCP_URL,
-        },
-      },
-    },
-    replaceServerKey: "servers",
+    serverKey: "servers",
+    server: { type: "http", url: HOSTED_MCP_URL },
+    jsonc: true,
     postInstall:
       "Enable MCP in Settings → search 'MCP' → enable 'Chat > MCP'. Use Copilot Chat in Agent mode.",
   },
   zed: {
     name: "Zed",
+    serverKey: "context_servers",
+    jsonc: true,
     getPath: () => {
       if (process.platform === "win32") {
         return path.join(process.env.APPDATA || "", "Zed", "settings.json");
       }
       return path.join(os.homedir(), ".config", "zed", "settings.json");
     },
-    config: {
-      context_servers: {
-        "assistant-ui": {
-          command: {
-            path: "npx",
-            args: ["-y", "@assistant-ui/mcp-docs-server"],
-          },
-        },
+    server: {
+      command: {
+        path: "npx",
+        args: ["-y", "@assistant-ui/mcp-docs-server"],
       },
     },
     postInstall: "The server starts automatically with the Assistant Panel.",
@@ -107,13 +96,10 @@ const MCP_CONFIGS: Record<
         "claude_desktop_config.json",
       );
     },
-    config: {
-      mcpServers: {
-        "assistant-ui": {
-          command: "npx",
-          args: ["-y", "@assistant-ui/mcp-docs-server"],
-        },
-      },
+    serverKey: "mcpServers",
+    server: {
+      command: "npx",
+      args: ["-y", "@assistant-ui/mcp-docs-server"],
     },
     postInstall: "Restart Claude Desktop to activate.",
   },
@@ -147,6 +133,162 @@ class McpConfigParseError extends Error {
 const getTargetFlag = (target: Exclude<MCPTarget, "claude-code">) =>
   `--${target}`;
 
+const lastPropertyValue = (node: Node, key: string) =>
+  node.children?.findLast((property) => property.children?.[0]?.value === key)
+    ?.children?.[1];
+
+const lineStartAt = (content: string, offset: number) =>
+  content.lastIndexOf("\n", offset - 1) + 1;
+
+const lineIndentAt = (content: string, offset: number) =>
+  content.slice(lineStartAt(content, offset), offset).match(/^[\t ]*/)?.[0] ??
+  "";
+
+const getIndentUnit = (content: string, object: Node) => {
+  const objectLine = lineStartAt(content, object.offset);
+  const objectIndent = lineIndentAt(content, object.offset);
+  const multilineChild = object.children?.find(
+    (child) => lineStartAt(content, child.offset) > objectLine,
+  );
+  if (multilineChild) {
+    const childIndent = lineIndentAt(content, multilineChild.offset);
+    if (
+      childIndent.startsWith(objectIndent) &&
+      childIndent.length > objectIndent.length
+    ) {
+      return childIndent.slice(objectIndent.length);
+    }
+  }
+
+  const documentIndent = content.match(/(?:^|\r?\n)([\t ]+)(?=")/)?.[1];
+  if (documentIndent?.includes("\t")) return "\t";
+  return documentIndent || "  ";
+};
+
+const stringifyValue = (
+  value: object,
+  baseIndent: string,
+  indentUnit: string,
+  eol: string,
+) =>
+  JSON.stringify(value, null, indentUnit).replaceAll(
+    "\n",
+    `${eol}${baseIndent}`,
+  );
+
+const appendObjectProperty = (
+  content: string,
+  object: Node,
+  key: string,
+  value: object,
+  eol: string,
+) => {
+  const objectIndent = lineIndentAt(content, object.offset);
+  const indentUnit = getIndentUnit(content, object);
+  const propertyIndent = objectIndent + indentUnit;
+  const property = `${propertyIndent}${JSON.stringify(key)}: ${stringifyValue(
+    value,
+    propertyIndent,
+    indentUnit,
+    eol,
+  )}`;
+  const closeOffset = object.offset + object.length - 1;
+  const closeLineStart = lineStartAt(content, closeOffset);
+  const closeOnOwnLine = /^[\t ]*$/.test(
+    content.slice(closeLineStart, closeOffset),
+  );
+  const edits = [
+    {
+      offset: closeOnOwnLine ? closeLineStart : closeOffset,
+      length: 0,
+      content: closeOnOwnLine
+        ? `${property}${eol}`
+        : `${eol}${property}${eol}${objectIndent}`,
+    },
+  ];
+
+  const lastProperty = object.children?.at(-1);
+  if (lastProperty) {
+    const propertyEnd = lastProperty.offset + lastProperty.length;
+    const trailingContent = stripComments(
+      content.slice(propertyEnd, closeOffset),
+    ).trimStart();
+    if (!trailingContent.startsWith(",")) {
+      edits.unshift({ offset: propertyEnd, length: 0, content: "," });
+    }
+  }
+
+  return applyEdits(content, edits);
+};
+
+const replaceNodeValue = (
+  content: string,
+  node: Node,
+  container: Node,
+  value: object,
+  eol: string,
+) =>
+  applyEdits(content, [
+    {
+      offset: node.offset,
+      length: node.length,
+      content: stringifyValue(
+        value,
+        lineIndentAt(content, node.offset),
+        getIndentUnit(content, container),
+        eol,
+      ),
+    },
+  ]);
+
+function updateJsoncConfig(
+  content: string,
+  serverKey: string,
+  server: object,
+): string {
+  const eol = content.includes("\r\n") ? "\r\n" : "\n";
+  const root = parseTree(content, [], {
+    allowTrailingComma: true,
+    allowEmptyContent: true,
+  });
+  if (!root) {
+    const initialized = JSON.stringify(
+      { [serverKey]: { "assistant-ui": server } },
+      null,
+      2,
+    ).replaceAll("\n", eol);
+    if (!content.trim()) return `${initialized}${eol}`;
+    return `${content}${content.endsWith(eol) ? "" : eol}${initialized}${eol}`;
+  }
+
+  // JSONC resolves duplicate keys to the last value.
+  const servers = lastPropertyValue(root, serverKey);
+  if (!servers) {
+    return appendObjectProperty(
+      content,
+      root,
+      serverKey,
+      { "assistant-ui": server },
+      eol,
+    );
+  }
+
+  if (servers.type !== "object") {
+    return replaceNodeValue(
+      content,
+      servers,
+      root,
+      { "assistant-ui": server },
+      eol,
+    );
+  }
+
+  const existing = lastPropertyValue(servers, "assistant-ui");
+  return existing
+    ? replaceNodeValue(content, existing, servers, server, eol)
+    : appendObjectProperty(content, servers, "assistant-ui", server, eol);
+}
+
 async function installForTarget(target: MCPTarget): Promise<void> {
   if (target === "claude-code") {
     logger.info("Installing MCP server for Claude Code...");
@@ -209,10 +351,35 @@ async function installForTarget(target: MCPTarget): Promise<void> {
   }
 
   let existingConfig: any = {};
+  let content = "{}\n";
   if (fs.existsSync(configPath)) {
-    const content = fs.readFileSync(configPath, "utf-8");
+    content = fs.readFileSync(configPath, "utf-8");
     try {
-      existingConfig = JSON.parse(content);
+      if (targetConfig.jsonc) {
+        const errors: ParseError[] = [];
+        existingConfig = parseJsonc(content, errors, {
+          allowTrailingComma: true,
+          allowEmptyContent: true,
+        });
+        if (existingConfig === undefined) existingConfig = {};
+        if (
+          errors.length > 0 ||
+          existingConfig === null ||
+          typeof existingConfig !== "object" ||
+          Array.isArray(existingConfig)
+        ) {
+          throw new SyntaxError("Invalid MCP configuration");
+        }
+      } else {
+        existingConfig = JSON.parse(content);
+        if (
+          existingConfig === null ||
+          typeof existingConfig !== "object" ||
+          Array.isArray(existingConfig)
+        ) {
+          throw new SyntaxError("Invalid MCP configuration");
+        }
+      }
     } catch {
       const flag = getTargetFlag(target);
       logger.error(`Could not parse ${targetConfig.name} MCP config.`);
@@ -225,23 +392,23 @@ async function installForTarget(target: MCPTarget): Promise<void> {
     }
   }
 
-  const newConfig = deepMerge(existingConfig, targetConfig.config);
-
-  if (targetConfig.replaceServerKey) {
-    const key = targetConfig.replaceServerKey;
-    newConfig[key] = {
-      ...newConfig[key],
-      "assistant-ui": (targetConfig.config as any)[key]["assistant-ui"],
-    };
-  }
-
-  fs.writeFileSync(configPath, `${JSON.stringify(newConfig, null, 2)}\n`);
+  const updatedContent = updateJsoncConfig(
+    content,
+    targetConfig.serverKey,
+    target === "zed" || target === "claude-desktop"
+      ? deepMerge(
+          existingConfig[targetConfig.serverKey]?.["assistant-ui"] ?? {},
+          targetConfig.server,
+        )
+      : targetConfig.server,
+  );
+  fs.writeFileSync(configPath, updatedContent);
 
   logger.break();
   logger.success(`MCP server installed for ${targetConfig.name}!`);
   logger.info(`Config written to: ${configPath}`);
 
-  if (targetConfig.replaceServerKey) {
+  if (target === "cursor" || target === "windsurf" || target === "vscode") {
     logger.break();
     logger.info(
       `Connects to the hosted assistant-ui MCP server at ${HOSTED_MCP_URL}.`,

package/src/lib/agent-skill.ts

@@ -1,16 +1,11 @@
 import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
-import { downloadTemplate } from "giget";
-import {
-  DOWNLOAD_TIMEOUT_MS,
-  dlxCommand,
-  resolveGitHubAuthToken,
-} from "./create-project";
-import { type PackageManagerName } from "./utils/package-manager";
+import { withStagedDownload } from "./utils/download";
+import { dlxCommand, type PackageManagerName } from "./utils/package-manager";
 
 export const SKILLS_PACKAGE = "assistant-ui/skills";
-export const SKILLS_COMMIT = "139674dc888ee076982b6726e8e6f5d0fe0b5f67";
+export const SKILLS_COMMIT = "1ba8e9105c504fe4ac2591e432783872d6b3c8ba";
 export const SKILLS_PLUGIN_SOURCE = `gh:${SKILLS_PACKAGE}/assistant-ui#${SKILLS_COMMIT}`;
 
 export function resolveSkillsInstall(params: {
@@ -70,42 +65,21 @@ export async function ensureSkillsPlugin(): Promise<string> {
   process.once("SIGINT", removeStagingOnSignal);
   process.once("SIGTERM", removeStagingOnSignal);
 
-  // giget logs to console.debug whenever DEBUG is set, which the `debug`
-  // package does at module load for an unrelated namespace.
-  const origDebug = process.env.DEBUG;
-  delete process.env.DEBUG;
-  let timer: ReturnType<typeof setTimeout> | undefined;
-  let downloadSettled = false;
-  let download: Promise<unknown> | undefined;
-  try {
-    const authToken = resolveGitHubAuthToken();
-    download = downloadTemplate(SKILLS_PLUGIN_SOURCE, {
-      dir: staging,
-      preferOffline: true,
-      silent: true,
-      ...(authToken ? { auth: authToken } : {}),
-    }).finally(() => {
-      downloadSettled = true;
-    });
-    const timeout = new Promise<never>((_, reject) => {
-      timer = setTimeout(
-        () => reject(new Error("Download timed out.")),
-        DOWNLOAD_TIMEOUT_MS,
-      );
-    });
-    await Promise.race([download, timeout]).catch((error: unknown) => {
-      throw new Error(
-        "Could not fetch the assistant-ui skills from GitHub. Check your network, or set GITHUB_TOKEN if you are rate limited.",
-        { cause: error },
-      );
-    });
-    await publish(staging, dir, marker);
-  } finally {
-    clearTimeout(timer);
-    if (origDebug !== undefined) process.env.DEBUG = origDebug;
-    if (!download || downloadSettled) removeStaging();
-    else void download.then(removeStaging, removeStaging);
-  }
+  await withStagedDownload(
+    SKILLS_PLUGIN_SOURCE,
+    { dir: staging, preferOffline: true, silent: true },
+    "Download timed out.",
+    removeStaging,
+    async (download) => {
+      await download.catch((error: unknown) => {
+        throw new Error(
+          "Could not fetch the assistant-ui skills from GitHub. Check your network, or set GITHUB_TOKEN if you are rate limited.",
+          { cause: error },
+        );
+      });
+      await publish(staging, dir, marker);
+    },
+  );
 
   return dir;
 }

package/src/lib/create-project.test.ts

@@ -24,23 +24,6 @@ describe("reconcileAssistantUIImportLayout", () => {
   const read = (file: string) =>
     fs.readFileSync(path.join(projectDir, file), "utf8");
 
-  it("rewrites a legacy import when only the elements layout exists", async () => {
-    write(
-      "app/page.tsx",
-      'import { Thread } from "@/components/assistant-ui/thread";\n' +
-        'import { ThreadList } from "@/components/assistant-ui/thread-list";\n',
-    );
-    write("components/assistant-ui/elements/thread.aui.tsx", "export {};");
-    write("components/assistant-ui/elements/thread-list.aui.tsx", "export {};");
-
-    await reconcileAssistantUIImportLayout(projectDir);
-
-    expect(read("app/page.tsx")).toBe(
-      'import { Thread } from "@/components/assistant-ui/elements/thread.aui";\n' +
-        'import { ThreadList } from "@/components/assistant-ui/elements/thread-list.aui";\n',
-    );
-  });
-
   it("rewrites module declarations without changing import-like source text", async () => {
     write(
       "app/page.tsx",

package/src/lib/create-project.ts

@@ -1,7 +1,6 @@
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
-import { downloadTemplate } from "giget";
 import {
   parse as parseJsonc,
   printParseErrorCode,
@@ -9,26 +8,14 @@ import {
 } from "jsonc-parser";
 import { logger } from "./utils/logger";
 import { runSpawn, SpawnExitError, SpawnSignalError } from "./run-spawn";
-import { type PackageManagerName } from "./utils/package-manager";
+import { dlxCommand, type PackageManagerName } from "./utils/package-manager";
+import { resolveGitHubAuthToken, withStagedDownload } from "./utils/download";
 import { readProjectFiles } from "./utils/file-scanner";
 import {
   detectRegistryPlatform,
   resolveRegistryItemUrl,
 } from "./utils/registry";
 
-export function dlxCommand(pm: PackageManagerName): [string, string[]] {
-  switch (pm) {
-    case "pnpm":
-      return ["pnpm", ["dlx"]];
-    case "yarn":
-      return ["yarn", ["dlx"]];
-    case "bun":
-      return ["bunx", []];
-    case "npm":
-      return ["npx", ["--yes"]];
-  }
-}
-
 export interface TransformOptions {
   hasLocalComponents: boolean;
   skipInstall?: boolean;
@@ -56,26 +43,6 @@ const LOCAL_PROJECT_ARTIFACT_GLOB_IGNORES = LOCAL_PROJECT_ARTIFACT_DIRS.map(
   (dir) => `**/${dir}/**`,
 );
 
-export function resolvePackageManager(opts: {
-  useNpm?: boolean;
-  usePnpm?: boolean;
-  useYarn?: boolean;
-  useBun?: boolean;
-}): PackageManagerName | undefined {
-  if (opts.useNpm) return "npm";
-  if (opts.usePnpm) return "pnpm";
-  if (opts.useYarn) return "yarn";
-  if (opts.useBun) return "bun";
-  return undefined;
-}
-
-export function resolveGitHubAuthToken(): string | undefined {
-  const token =
-    process.env.GIGET_AUTH ?? process.env.GITHUB_TOKEN ?? process.env.GH_TOKEN;
-  const trimmed = token?.trim();
-  return trimmed || undefined;
-}
-
 function toBearerAuthHeader(token: string): string {
   return token.toLowerCase().startsWith("bearer ") ? token : `Bearer ${token}`;
 }
@@ -97,7 +64,6 @@ export async function resolveLatestReleaseRef(): Promise<string | undefined> {
   }
 }
 
-export const DOWNLOAD_TIMEOUT_MS = 30_000;
 const pendingDownloadCleanups = new Set<() => void>();
 
 export function cleanupPendingProjectDownloads(): void {
@@ -117,16 +83,9 @@ export async function downloadProject(
     ? `gh:assistant-ui/assistant-ui/${repoPath}#${ref}`
     : `gh:assistant-ui/assistant-ui/${repoPath}`;
 
-  // Suppress giget's debug output. The `debug` package (used by the upgrade
-  // command) sets process.env.DEBUG at module-load time, and giget logs to
-  // console.debug whenever that env var is truthy — even for unrelated
-  // namespaces. Temporarily unsetting it targets the root cause.
-  const origDebug = process.env.DEBUG;
-  delete process.env.DEBUG;
   let destinationCreated = false;
   let stagingDir: string | undefined;
-  let downloadPromise: Promise<unknown> | undefined;
-  let downloadFinished = false;
+  let downloadStarted = false;
   let downloadCommitted = false;
   let cleanupRequested = false;
   const removeCleanupListeners = () => {
@@ -183,62 +142,38 @@ export async function downloadProject(
     process.once("exit", cleanupOnExit);
     pendingDownloadCleanups.add(cleanupOnExit);
 
-    const authToken = resolveGitHubAuthToken();
-    downloadPromise = downloadTemplate(source, {
-      dir: stagingDir,
-      force: true,
-      silent: true,
-      ...(authToken ? { auth: authToken } : {}),
-    }).finally(() => {
-      downloadFinished = true;
-    });
-
-    let timer: ReturnType<typeof setTimeout>;
-    const timeoutPromise = new Promise<never>((_, reject) => {
-      timer = setTimeout(
-        () =>
-          reject(
-            new Error(
-              "Download timed out. This may be due to GitHub rate limiting or a network issue. Try again in a few minutes.",
-            ),
-          ),
-        DOWNLOAD_TIMEOUT_MS,
-      );
-    });
-
-    try {
-      await Promise.race([downloadPromise, timeoutPromise]);
-      if (cleanupRequested) throw new Error("Download was interrupted.");
-      destinationCreated = !fs.existsSync(destDir);
-      await fs.promises.mkdir(destDir, { recursive: true });
-      for (const entry of await fs.promises.readdir(stagingDir)) {
-        const target = path.join(destDir, entry);
-        const sourceEntry = path.join(stagingDir, entry);
-        await fs.promises.rm(target, { recursive: true, force: true });
-        try {
-          await fs.promises.rename(sourceEntry, target);
-        } catch (error) {
-          if ((error as NodeJS.ErrnoException).code !== "EXDEV") throw error;
-          await fs.promises.cp(sourceEntry, target, {
-            recursive: true,
-            force: true,
-          });
-          await fs.promises.rm(sourceEntry, { recursive: true, force: true });
+    const currentStagingDir = stagingDir;
+    downloadStarted = true;
+    await withStagedDownload(
+      source,
+      { dir: currentStagingDir, force: true, silent: true },
+      "Download timed out. This may be due to GitHub rate limiting or a network issue. Try again in a few minutes.",
+      removeStagingDir,
+      async (download) => {
+        await download;
+        if (cleanupRequested) throw new Error("Download was interrupted.");
+        destinationCreated = !fs.existsSync(destDir);
+        await fs.promises.mkdir(destDir, { recursive: true });
+        for (const entry of await fs.promises.readdir(currentStagingDir)) {
+          const target = path.join(destDir, entry);
+          const sourceEntry = path.join(currentStagingDir, entry);
+          await fs.promises.rm(target, { recursive: true, force: true });
+          try {
+            await fs.promises.rename(sourceEntry, target);
+          } catch (error) {
+            if ((error as NodeJS.ErrnoException).code !== "EXDEV") throw error;
+            await fs.promises.cp(sourceEntry, target, {
+              recursive: true,
+              force: true,
+            });
+            await fs.promises.rm(sourceEntry, { recursive: true, force: true });
+          }
         }
-      }
-      downloadCommitted = true;
-    } finally {
-      clearTimeout(timer!);
-    }
+        downloadCommitted = true;
+      },
+    );
   } finally {
-    if (!downloadPromise || downloadFinished) {
-      await removeStagingDir();
-    } else {
-      void downloadPromise.then(removeStagingDir, removeStagingDir);
-    }
-    if (origDebug !== undefined) {
-      process.env.DEBUG = origDebug;
-    }
+    if (!downloadStarted) await removeStagingDir();
   }
 }
 

package/src/lib/transform-files.test.ts

@@ -2,7 +2,7 @@ import { describe, expect, it, onTestFinished } from "vitest";
 import { mkdirSync, mkdtempSync, writeFileSync, rmSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { dirname, join, relative } from "node:path";
-import { getRelevantFiles, countFilesToProcess } from "./transform";
+import { getRelevantFiles } from "./transform";
 
 const fixture = () => {
   const directory = mkdtempSync(join(tmpdir(), "aui-transform-files-"));
@@ -26,7 +26,6 @@ describe("codemod file discovery", () => {
       const { write } = fixture();
       const file = write(`my app.${extension}`);
       expect(getRelevantFiles(file)).toEqual([file]);
-      expect(countFilesToProcess(file)).toBe(1);
       expect(getRelevantFiles(relative(process.cwd(), file))).toEqual([file]);
     },
   );
@@ -54,7 +53,6 @@ describe("codemod file discovery", () => {
     const { write } = fixture();
     const file = write("app.tsx", source);
     expect(getRelevantFiles(file)).toEqual([file]);
-    expect(countFilesToProcess(file)).toBe(1);
   });
 
   it("skips unsupported explicit files", () => {