assistant-ui 0.0.112 → 0.0.114

package/src/codemods/v0-12/primitive-if-to-aui-if.ts

@@ -120,12 +120,18 @@ const fixedConditionMap: Record<string, Record<string, string>> = {
   },
 };
 
+// A prop value the maps cannot faithfully express as a static condition
+// (dynamic expressions, `{undefined}`); elements carrying one are skipped
+// so runtime behavior is never silently changed.
+const UNSUPPORTED_VALUE: unique symbol = Symbol("unsupported");
+
 /**
  * Extract the value of a JSX attribute.
  * - Boolean prop (no value): `<X.If user>` → `true`
  * - `{true}` / `{false}`: → `true` / `false`
  * - `{"positive"}`: → `"positive"`
  * - `{null}`: → `null`
+ * - anything else (dynamic expressions): → UNSUPPORTED_VALUE
  */
 const getAttrValue = (j: any, attr: any): unknown => {
   // Boolean attribute (no value), e.g. `<X.If user>`
@@ -137,12 +143,14 @@ const getAttrValue = (j: any, attr: any): unknown => {
   if (j.JSXExpressionContainer.check(attr.value)) {
     const expr = attr.value.expression;
     if (j.BooleanLiteral.check(expr)) return expr.value;
+    // NullLiteral bases Literal in ast-types but carries no `value` field,
+    // so it must be recognized before the generic Literal branch.
+    if (j.NullLiteral.check(expr)) return null;
     if (j.Literal.check(expr)) {
       if (expr.value === null) return null;
       return expr.value;
     }
-    if (j.NullLiteral.check(expr)) return null;
-    if (j.Identifier.check(expr) && expr.name === "undefined") return undefined;
+    return UNSUPPORTED_VALUE;
   }
 
   // String literal
@@ -150,7 +158,7 @@ const getAttrValue = (j: any, attr: any): unknown => {
     return attr.value.value;
   }
 
-  return undefined;
+  return UNSUPPORTED_VALUE;
 };
 
 const buildConditionString = (fragments: ConditionFragment[]): string => {
@@ -185,61 +193,52 @@ const migratePrimitiveIfToAuiIf = createTransformer(
 
     if (importedPrimitives.size === 0) return;
 
-    // Process fixed-condition components: <ThreadPrimitive.Empty> → <AuiIf condition={...}>
-    root.find(j.JSXOpeningElement).forEach((path: any) => {
-      const name = path.value.name;
-      if (!j.JSXMemberExpression.check(name)) return;
-      if (!j.JSXIdentifier.check(name.object)) return;
-      if (!j.JSXIdentifier.check(name.property)) return;
-
-      const primitiveName = name.object.name as string;
-      const propertyName = name.property.name as string;
-      const fixedMap = fixedConditionMap[primitiveName];
-      if (!fixedMap) return;
-      const conditionBody = fixedMap[propertyName];
-      if (!conditionBody) return;
-      if (!importedPrimitives.has(primitiveName)) return;
-
-      // Only transform if there are no props (other than children, which are implicit)
-      const attrs: any[] = path.value.attributes || [];
-      if (attrs.length > 0) return;
-
+    // Opening and closing tags are rewritten together per element, so a
+    // skipped element can never be left with a mismatched closing tag.
+    const convertElementToAuiIf = (elementPath: any, conditionBody: string) => {
       const arrowFnAst = j(`(s) => ${conditionBody}`)
         .find(j.ArrowFunctionExpression)
         .paths()[0]!.value;
 
-      path.value.name = j.jsxIdentifier("AuiIf");
-      path.value.attributes = [
+      const opening = elementPath.value.openingElement;
+      opening.name = j.jsxIdentifier("AuiIf");
+      opening.attributes = [
         j.jsxAttribute(
           j.jsxIdentifier("condition"),
           j.jsxExpressionContainer(arrowFnAst),
         ),
       ];
+      if (elementPath.value.closingElement) {
+        elementPath.value.closingElement.name = j.jsxIdentifier("AuiIf");
+      }
 
       needsAuiIfImport = true;
       markAsChanged();
-    });
+    };
 
-    // Update closing elements for fixed-condition components
-    root.find(j.JSXClosingElement).forEach((path: any) => {
-      const name = path.value.name;
+    // Process fixed-condition components: <ThreadPrimitive.Empty> → <AuiIf condition={...}>
+    root.find(j.JSXElement).forEach((path: any) => {
+      const name = path.value.openingElement.name;
       if (!j.JSXMemberExpression.check(name)) return;
       if (!j.JSXIdentifier.check(name.object)) return;
       if (!j.JSXIdentifier.check(name.property)) return;
 
       const primitiveName = name.object.name as string;
       const propertyName = name.property.name as string;
-      const fixedMap = fixedConditionMap[primitiveName];
-      if (!fixedMap?.[propertyName]) return;
+      const conditionBody = fixedConditionMap[primitiveName]?.[propertyName];
+      if (!conditionBody) return;
       if (!importedPrimitives.has(primitiveName)) return;
 
-      path.value.name = j.jsxIdentifier("AuiIf");
-      markAsChanged();
+      // Only transform if there are no props (other than children, which are implicit)
+      const attrs: any[] = path.value.openingElement.attributes || [];
+      if (attrs.length > 0) return;
+
+      convertElementToAuiIf(path, conditionBody);
     });
 
     // Process JSX elements: <ThreadPrimitive.If ...> → <AuiIf condition={...}>
-    root.find(j.JSXOpeningElement).forEach((path: any) => {
-      const name = path.value.name;
+    root.find(j.JSXElement).forEach((path: any) => {
+      const name = path.value.openingElement.name;
 
       // Check for `<XPrimitive.If ...>`
       if (!j.JSXMemberExpression.check(name)) return;
@@ -253,7 +252,7 @@ const migratePrimitiveIfToAuiIf = createTransformer(
       if (!importedPrimitives.has(primitiveName)) return;
 
       // Extract props
-      const attrs: any[] = path.value.attributes || [];
+      const attrs: any[] = path.value.openingElement.attributes || [];
       const fragments: ConditionFragment[] = [];
       let hasUnknownProp = false;
 
@@ -265,7 +264,11 @@ const migratePrimitiveIfToAuiIf = createTransformer(
         }
         const propName =
           typeof attr.name.name === "string" ? attr.name.name : null;
-        if (!propName) continue;
+        if (!propName) {
+          // e.g. JSXNamespacedName — not expressible as a condition
+          hasUnknownProp = true;
+          continue;
+        }
 
         const mapper = propMap[propName];
         if (!mapper) {
@@ -274,6 +277,10 @@ const migratePrimitiveIfToAuiIf = createTransformer(
         }
 
         const value = getAttrValue(j, attr);
+        if (value === UNSUPPORTED_VALUE) {
+          hasUnknownProp = true;
+          continue;
+        }
         const fragment = mapper(value);
         if (fragment) {
           fragments.push(fragment);
@@ -283,42 +290,7 @@ const migratePrimitiveIfToAuiIf = createTransformer(
       // If we couldn't map all props, skip this element
       if (hasUnknownProp || fragments.length === 0) return;
 
-      const conditionBody = buildConditionString(fragments);
-
-      // Parse the arrow function as an expression to get a proper AST node
-      const arrowFnAst = j(`(s) => ${conditionBody}`)
-        .find(j.ArrowFunctionExpression)
-        .paths()[0]!.value;
-
-      // Replace <XPrimitive.If ...> with <AuiIf condition={...}>
-      path.value.name = j.jsxIdentifier("AuiIf");
-
-      // Replace all attributes with a single condition prop
-      path.value.attributes = [
-        j.jsxAttribute(
-          j.jsxIdentifier("condition"),
-          j.jsxExpressionContainer(arrowFnAst),
-        ),
-      ];
-
-      needsAuiIfImport = true;
-      markAsChanged();
-    });
-
-    // Update closing elements to match
-    root.find(j.JSXClosingElement).forEach((path: any) => {
-      const name = path.value.name;
-      if (!j.JSXMemberExpression.check(name)) return;
-      if (!j.JSXIdentifier.check(name.object)) return;
-      if (!j.JSXIdentifier.check(name.property)) return;
-      if (name.property.name !== "If") return;
-
-      const primitiveName = name.object.name;
-      if (!primitiveMap[primitiveName]) return;
-      if (!importedPrimitives.has(primitiveName)) return;
-
-      path.value.name = j.jsxIdentifier("AuiIf");
-      markAsChanged();
+      convertElementToAuiIf(path, buildConditionString(fragments));
     });
 
     // Add AuiIf import if needed

package/src/commands/add.ts

@@ -5,13 +5,12 @@ import {
   getComponentsJsonStyle,
   resolveRegistryItemUrl,
 } from "../lib/utils/registry";
+import { dlxCommand, resolvePackageManager } from "../lib/create-project";
+import { runSpawn, SpawnExitError, SpawnSignalError } from "../lib/run-spawn";
 import {
-  dlxCommand,
-  resolvePackageManager,
-  resolvePackageManagerForCwd,
   type PackageManagerName,
-} from "../lib/create-project";
-import { runSpawn, SpawnExitError } from "../lib/run-spawn";
+  resolvePackageManagerForCwd,
+} from "../lib/utils/package-manager";
 
 export interface AddComponentsPlan {
   command: string;
@@ -92,6 +91,7 @@ export const add = new Command()
     try {
       await runSpawn(command, args, opts.cwd);
     } catch (error) {
+      if (error instanceof SpawnSignalError) throw error;
       if (error instanceof SpawnExitError) {
         logger.error(`Process exited with code ${error.code}`);
         process.exit(error.code);

package/src/commands/create.ts

@@ -9,12 +9,17 @@ import {
   downloadProject,
   resolveLatestReleaseRef,
   resolvePackageManager,
-  resolvePackageManagerForCwd,
   scaffoldProject,
   transformProject,
   type TransformResult,
 } from "../lib/create-project";
-import { runSpawn, SpawnExitError } from "../lib/run-spawn";
+import {
+  hasActiveSpawn,
+  runSpawn,
+  SpawnExitError,
+  SpawnSignalError,
+} from "../lib/run-spawn";
+import { resolvePackageManagerForCwd } from "../lib/utils/package-manager";
 import {
   buildSkillsAddCommand,
   resolveSkillsInstall,
@@ -597,10 +602,31 @@ export const create = new Command()
     );
 
     // Clean up partial project directory on unexpected exit (e.g. Ctrl+C)
+    let cleanupArmed = true;
     const cleanupOnExit = () => {
+      if (!cleanupArmed) return;
+      cleanupArmed = false;
       fs.rmSync(absoluteProjectDir, { recursive: true, force: true });
     };
+    const disarmCleanup = () => {
+      cleanupArmed = false;
+      process.removeListener("exit", cleanupOnExit);
+      process.removeListener("SIGINT", cleanupOnSignal);
+      process.removeListener("SIGTERM", cleanupOnSignal);
+    };
+    // Node emits no "exit" when a signal kills the process. An in-flight
+    // runSpawn forwards the signal itself, so the directory is removed on the
+    // error path once the child is reaped rather than while it is still writing.
+    const cleanupOnSignal = (signal: NodeJS.Signals) => {
+      if (hasActiveSpawn()) return;
+      cleanupOnExit();
+      disarmCleanup();
+      process.kill(process.pid, signal);
+    };
+
     process.once("exit", cleanupOnExit);
+    process.on("SIGINT", cleanupOnSignal);
+    process.on("SIGTERM", cleanupOnSignal);
 
     try {
       // 3. Resolve latest release ref (started before prompts)
@@ -655,7 +681,8 @@ export const create = new Command()
           });
           try {
             await runSpawn(skillsCmd, skillsArgs, absoluteProjectDir);
-          } catch {
+          } catch (error) {
+            if (error instanceof SpawnSignalError) throw error;
             logger.warn(
               `Could not add assistant-ui agent skills. You can add them later with:\n  ${skillsCmd} ${skillsArgs.join(" ")}`,
             );
@@ -663,12 +690,13 @@ export const create = new Command()
         }
       } catch (err) {
         // Clean up partially created project directory
-        fs.rmSync(absoluteProjectDir, { recursive: true, force: true });
+        cleanupOnExit();
+        disarmCleanup();
         throw err;
       }
 
       if (transformResult.registryInstallFailure) {
-        process.removeListener("exit", cleanupOnExit);
+        disarmCleanup();
         logger.break();
         logger.error("Project created with missing components.");
         logger.info("Retry the component install with:");
@@ -696,14 +724,15 @@ export const create = new Command()
             ],
             absoluteProjectDir,
           );
-        } catch {
+        } catch (error) {
+          if (error instanceof SpawnSignalError) throw error;
           logger.warn(
             `Preset application failed. You can retry manually with:\n  ${dlxCmd} ${[...dlxArgs, "shadcn@latest", "add", presetUrl].join(" ")}`,
           );
         }
       }
 
-      process.removeListener("exit", cleanupOnExit);
+      disarmCleanup();
 
       logger.break();
       logger.success("Project created successfully!");
@@ -736,6 +765,11 @@ export const create = new Command()
       logger.info(`  # Set up your environment variables in ${envFile}`);
       logger.info(`  ${runCmd} ${devScript}`);
     } catch (error) {
+      if (error instanceof SpawnSignalError) {
+        cleanupOnExit();
+        disarmCleanup();
+        throw error;
+      }
       if (error instanceof SpawnExitError) {
         logger.error(`Project creation failed with code ${error.code}`);
         process.exit(error.code);

package/src/commands/info.ts

@@ -26,6 +26,7 @@ const ASSISTANT_UI_PACKAGES = [
   "@assistant-ui/cloud-ai-sdk",
   // Adapters
   "@assistant-ui/eve",
+  "@assistant-ui/ai-sdk",
   "@assistant-ui/react-ai-sdk",
   "@assistant-ui/react-langgraph",
   "@assistant-ui/react-ag-ui",

package/src/commands/init.ts

@@ -1,13 +1,10 @@
 import { Command, Option } from "commander";
 import fs from "node:fs";
 import path from "node:path";
-import {
-  dlxCommand,
-  resolvePackageManager,
-  resolvePackageManagerForCwd,
-} from "../lib/create-project";
-import { runSpawn, SpawnExitError } from "../lib/run-spawn";
+import { dlxCommand, resolvePackageManager } from "../lib/create-project";
+import { runSpawn, SpawnExitError, SpawnSignalError } from "../lib/run-spawn";
 import { logger } from "../lib/utils/logger";
+import { resolvePackageManagerForCwd } from "../lib/utils/package-manager";
 import {
   getComponentsJsonStyle,
   resolveQuickStartRegistryUrl,
@@ -135,6 +132,7 @@ export const init = new Command()
       logger.success("Project initialized successfully!");
       logger.info("You can now add more components with 'assistant-ui add'");
     } catch (error) {
+      if (error instanceof SpawnSignalError) throw error;
       if (error instanceof SpawnExitError) {
         logger.error(`Initialization failed with code ${error.code}`);
         process.exit(error.code);

package/src/commands/mcp.ts

@@ -2,8 +2,8 @@ import { Command } from "commander";
 import fs from "node:fs";
 import path from "node:path";
 import os from "node:os";
-import { spawn } from "cross-spawn";
 import { logger } from "../lib/utils/logger";
+import { runSpawn, SpawnExitError, SpawnSignalError } from "../lib/run-spawn";
 import * as p from "@clack/prompts";
 
 type MCPTarget =
@@ -14,12 +14,15 @@ type MCPTarget =
   | "claude-code"
   | "claude-desktop";
 
+const HOSTED_MCP_URL = "https://www.assistant-ui.com/mcp";
+
 const MCP_CONFIGS: Record<
   Exclude<MCPTarget, "claude-code">,
   {
     name: string;
     getPath: () => string;
     config: object;
+    replaceServerKey?: string;
     postInstall?: string;
   }
 > = {
@@ -29,11 +32,11 @@ const MCP_CONFIGS: Record<
     config: {
       mcpServers: {
         "assistant-ui": {
-          command: "npx",
-          args: ["-y", "@assistant-ui/mcp-docs-server"],
+          url: HOSTED_MCP_URL,
         },
       },
     },
+    replaceServerKey: "mcpServers",
     postInstall:
       "Open Cursor Settings → MCP → find 'assistant-ui' and click enable.",
   },
@@ -44,11 +47,11 @@ const MCP_CONFIGS: Record<
     config: {
       mcpServers: {
         "assistant-ui": {
-          command: "npx",
-          args: ["-y", "@assistant-ui/mcp-docs-server"],
+          serverUrl: HOSTED_MCP_URL,
         },
       },
     },
+    replaceServerKey: "mcpServers",
     postInstall: "Fully quit and re-open Windsurf to activate.",
   },
   vscode: {
@@ -57,12 +60,12 @@ const MCP_CONFIGS: Record<
     config: {
       servers: {
         "assistant-ui": {
-          command: "npx",
-          args: ["-y", "@assistant-ui/mcp-docs-server"],
-          type: "stdio",
+          type: "http",
+          url: HOSTED_MCP_URL,
         },
       },
     },
+    replaceServerKey: "servers",
     postInstall:
       "Enable MCP in Settings → search 'MCP' → enable 'Chat > MCP'. Use Copilot Chat in Agent mode.",
   },
@@ -152,45 +155,42 @@ async function installForTarget(target: MCPTarget): Promise<void> {
     logger.info("Installing MCP server for Claude Code...");
     logger.break();
 
-    const child = spawn(
-      "claude",
-      [
+    try {
+      await runSpawn("claude", ["mcp", "remove", "assistant-ui"]).catch(
+        (error: unknown) => {
+          if (error instanceof SpawnSignalError) throw error;
+        },
+      );
+      await runSpawn("claude", [
         "mcp",
         "add",
+        "--transport",
+        "http",
         "assistant-ui",
-        "--",
-        "npx",
-        "-y",
-        "@assistant-ui/mcp-docs-server",
-      ],
-      {
-        stdio: "inherit",
-      },
-    );
-
-    return new Promise((resolve, reject) => {
-      child.on("error", (error) => {
+        HOSTED_MCP_URL,
+      ]);
+    } catch (error) {
+      if (error instanceof SpawnSignalError) throw error;
+      if (error instanceof SpawnExitError) {
+        logger.error(`Installation failed with code ${error.code}`);
+      } else if (error instanceof Error) {
         logger.error(`Failed to install: ${error.message}`);
         logger.info(
           "Make sure Claude Code CLI is installed: https://docs.anthropic.com/en/docs/claude-code",
         );
-        reject(error);
-      });
+      }
+      throw error;
+    }
 
-      child.on("close", (code) => {
-        if (code !== 0) {
-          logger.error(`Installation failed with code ${code}`);
-          reject(new Error(`Exit code ${code}`));
-        } else {
-          logger.break();
-          logger.success("MCP server installed for Claude Code!");
-          logger.info(
-            "The server starts automatically. Try asking about assistant-ui!",
-          );
-          resolve();
-        }
-      });
-    });
+    logger.break();
+    logger.success("MCP server installed for Claude Code!");
+    logger.info(
+      `Connects to the hosted assistant-ui MCP server at ${HOSTED_MCP_URL}.`,
+    );
+    logger.info(
+      "The server starts automatically. Try asking about assistant-ui!",
+    );
+    return;
   }
 
   if (target === "claude-desktop" && process.platform === "linux") {
@@ -230,12 +230,27 @@ 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`);
 
   logger.break();
   logger.success(`MCP server installed for ${targetConfig.name}!`);
   logger.info(`Config written to: ${configPath}`);
 
+  if (targetConfig.replaceServerKey) {
+    logger.break();
+    logger.info(
+      `Connects to the hosted assistant-ui MCP server at ${HOSTED_MCP_URL}.`,
+    );
+  }
+
   if (targetConfig.postInstall) {
     logger.break();
     logger.info(targetConfig.postInstall);
@@ -244,7 +259,7 @@ async function installForTarget(target: MCPTarget): Promise<void> {
 
 export const mcp = new Command()
   .name("mcp")
-  .description("install assistant-ui MCP docs server for your IDE")
+  .description("connect your IDE to the assistant-ui MCP server")
   .option("--cursor", "install for Cursor")
   .option("--windsurf", "install for Windsurf")
   .option("--vscode", "install for VSCode")
@@ -288,7 +303,8 @@ export const mcp = new Command()
     for (const target of targets) {
       try {
         await installForTarget(target);
-      } catch {
+      } catch (error) {
+        if (error instanceof SpawnSignalError) throw error;
         process.exit(1);
       }
     }

package/src/commands/update.ts

@@ -1,10 +1,10 @@
 import { Command } from "commander";
 import fs from "node:fs";
 import path from "node:path";
-import { sync as spawnSync } from "cross-spawn";
 import { logger } from "../lib/utils/logger";
 import { logPackageJsonParseError } from "../lib/utils/package-json";
 import { getInstallCommand } from "../lib/utils/package-manager";
+import { runSpawn, SpawnExitError, SpawnSignalError } from "../lib/run-spawn";
 
 export const update = new Command()
   .name("update")
@@ -73,14 +73,12 @@ export const update = new Command()
     }
 
     logger.step("Updating packages...");
-    const result = spawnSync(installCmd.command, installCmd.args, {
-      stdio: "inherit",
-      cwd: opts.cwd,
-    });
-
-    if (result.status !== 0) {
+    try {
+      await runSpawn(installCmd.command, installCmd.args, opts.cwd);
+    } catch (error) {
+      if (error instanceof SpawnSignalError) throw error;
       logger.error("Package manager update failed.");
-      process.exit(result.status || 1);
+      process.exit(error instanceof SpawnExitError ? error.code : 1);
     }
 
     logger.break();

package/src/commands/upgrade.ts

@@ -1,6 +1,7 @@
 import { Command } from "commander";
 import { transform } from "../lib/transform";
 import { upgrade } from "../lib/upgrade";
+import { SpawnSignalError } from "../lib/run-spawn";
 import debug from "debug";
 
 export interface TransformOptions {
@@ -30,10 +31,11 @@ export const codemodCommand = addTransformOptions(
     .description("CLI tool for running codemods")
     .argument("<codemod>", "Codemod to run (e.g., rewrite-framework-imports)")
     .argument("<source>", "Path to source files or directory to transform"),
-).action((codemod, source, options: TransformOptions) => {
+).action(async (codemod, source, options: TransformOptions) => {
   try {
-    transform(codemod, source, options);
+    await transform(codemod, source, options);
   } catch (err) {
+    if (err instanceof SpawnSignalError) throw err;
     const errorMessage = err instanceof Error ? err.message : String(err);
     const errorStack = err instanceof Error ? err.stack : undefined;
     error(`Error transforming: ${errorMessage}`);
@@ -52,6 +54,7 @@ export const upgradeCommand = addTransformOptions(
   try {
     await upgrade(options);
   } catch (err) {
+    if (err instanceof SpawnSignalError) throw err;
     const errorMessage = err instanceof Error ? err.message : String(err);
     const errorStack = err instanceof Error ? err.stack : undefined;
     error(`Error upgrading: ${errorMessage}`);

package/src/index.ts

@@ -1,11 +1,6 @@
 #!/usr/bin/env node
 
+import { handleCliError } from "./lib/handle-cli-error";
 import { runCli } from "./run";
 
-process.on("SIGINT", () => process.exit(0));
-process.on("SIGTERM", () => process.exit(0));
-
-void runCli().catch((error: unknown) => {
-  console.error(error);
-  process.exitCode = 1;
-});
+void runCli().catch(handleCliError);

package/src/lib/agent-skill.ts

@@ -1,4 +1,5 @@
-import { dlxCommand, type PackageManagerName } from "./create-project";
+import { dlxCommand } from "./create-project";
+import { type PackageManagerName } from "./utils/package-manager";
 
 export const SKILLS_PACKAGE = "assistant-ui/skills";