assistant-ui 0.0.78 → 0.0.80

package/src/commands/init.ts

@@ -1,41 +1,97 @@
-import { Command } from "commander";
+import { Command, Option } from "commander";
 import { spawn } from "cross-spawn";
 import fs from "node:fs";
 import path from "node:path";
 import { logger } from "../lib/utils/logger";
-import { hasConfig } from "../lib/utils/config";
+import { create } from "./create";
 
 const DEFAULT_REGISTRY_URL =
   "https://r.assistant-ui.com/chat/b/ai-sdk-quick-start/json";
 
-// Keep in sync with packages/create-assistant-ui/src/index.ts
-const templates = {
-  default: "https://github.com/assistant-ui/assistant-ui-starter",
-  minimal: "https://github.com/assistant-ui/assistant-ui-starter-minimal",
-  cloud: "https://github.com/assistant-ui/assistant-ui-starter-cloud",
-  langgraph: "https://github.com/assistant-ui/assistant-ui-starter-langgraph",
-  mcp: "https://github.com/assistant-ui/assistant-ui-starter-mcp",
-};
-
-const templateNames = Object.keys(templates);
+class SpawnExitError extends Error {
+  code: number;
+
+  constructor(code: number) {
+    super(`Process exited with code ${code}`);
+    this.code = code;
+  }
+}
+
+interface ExistingProjectInitPlan {
+  initArgs: string[] | null;
+  addArgs: string[];
+}
+
+export function createExistingProjectInitPlan(params: {
+  yes: boolean;
+  overwrite: boolean;
+  registryUrl: string;
+}): ExistingProjectInitPlan {
+  const { yes, overwrite, registryUrl } = params;
+
+  if (!yes) {
+    const addArgs = [`shadcn@latest`, "add"];
+    if (overwrite) addArgs.push("--overwrite");
+    addArgs.push(registryUrl);
+    return { initArgs: null, addArgs };
+  }
+
+  const initArgs = [`shadcn@latest`, "init", "--defaults", "--yes"];
+
+  const addArgs = [`shadcn@latest`, "add", "--yes"];
+  if (overwrite) addArgs.push("--overwrite");
+  addArgs.push(registryUrl);
+
+  return { initArgs, addArgs };
+}
+
+export function isNonInteractiveShell(
+  stdinIsTTY = process.stdin.isTTY,
+): boolean {
+  return !stdinIsTTY;
+}
+
+async function runSpawn(
+  command: string,
+  args: string[],
+  cwd: string,
+): Promise<void> {
+  return new Promise((resolve, reject) => {
+    const child = spawn(command, args, {
+      stdio: "inherit",
+      cwd,
+    });
+
+    child.on("error", (error) => {
+      reject(error);
+    });
+
+    child.on("close", (code) => {
+      if (code !== 0) {
+        reject(new SpawnExitError(code || 1));
+      } else {
+        resolve();
+      }
+    });
+  });
+}
 
 export const init = new Command()
   .name("init")
-  .description("initialize assistant-ui in a new or existing project")
+  .description("initialize assistant-ui in an existing project")
   .argument("[project-directory]", "directory for the new project")
+  .option("-y, --yes", "skip confirmation prompt.", false)
+  .option("-o, --overwrite", "overwrite existing files.", false)
   .option(
     "-c, --cwd <cwd>",
     "the working directory. defaults to the current directory.",
     process.cwd(),
   )
-  .option(
-    "-p, --preset <url>",
-    "preset URL from playground (e.g., https://www.assistant-ui.com/playground/init?preset=chatgpt)",
-  )
-  .option(
-    "-t, --template <template>",
-    `template to use (${templateNames.join(", ")})`,
-    "minimal",
+  .addOption(
+    new Option(
+      "-p, --preset <url>",
+      "preset URL from playground (forwarded to 'assistant-ui create')",
+    ).hideHelp(),
   )
   .option("--use-npm", "explicitly use npm")
   .option("--use-pnpm", "explicitly use pnpm")
@@ -44,12 +100,14 @@ export const init = new Command()
   .option("--skip-install", "skip installing packages")
   .action(async (projectDirectory, opts) => {
     const cwd = opts.cwd;
+    const presetUrl = opts.preset as string | undefined;
     const targetDir = projectDirectory
       ? path.resolve(cwd, projectDirectory)
       : cwd;
-    const presetUrl = opts.preset;
 
-    if (hasConfig(targetDir)) {
+    const componentsConfigPath = path.join(targetDir, "components.json");
+
+    if (!presetUrl && fs.existsSync(componentsConfigPath)) {
       logger.warn("Project is already initialized.");
       logger.info("Use 'assistant-ui add' to add more components.");
       return;
@@ -58,78 +116,62 @@ export const init = new Command()
     const packageJsonPath = path.join(targetDir, "package.json");
     const packageJsonExists = fs.existsSync(packageJsonPath);
 
-    if (packageJsonExists) {
-      const registryUrl = presetUrl ?? DEFAULT_REGISTRY_URL;
-
-      if (presetUrl) {
-        logger.info("Initializing assistant-ui with preset configuration...");
-      } else {
-        logger.info("Initializing assistant-ui in existing project...");
+    if (presetUrl || !packageJsonExists) {
+      if (!presetUrl) {
+        logger.info("No existing project found. Running 'create' instead...");
+        logger.break();
       }
-      logger.break();
 
-      const child = spawn("npx", [`shadcn@latest`, "add", registryUrl], {
-        stdio: "inherit",
-        cwd: targetDir,
-      });
-
-      child.on("error", (error) => {
-        logger.error(`Failed to initialize: ${error.message}`);
-        process.exit(1);
-      });
+      const createArgs: string[] = [];
+      if (projectDirectory) createArgs.push(projectDirectory);
+      if (presetUrl) createArgs.push("--preset", presetUrl);
+      if (opts.useNpm) createArgs.push("--use-npm");
+      if (opts.usePnpm) createArgs.push("--use-pnpm");
+      if (opts.useYarn) createArgs.push("--use-yarn");
+      if (opts.useBun) createArgs.push("--use-bun");
+      if (opts.skipInstall) createArgs.push("--skip-install");
 
-      child.on("close", (code) => {
-        if (code !== 0) {
-          logger.error(`Initialization failed with code ${code}`);
-          process.exit(code || 1);
-        } else {
-          logger.break();
-          logger.success("Project initialized successfully!");
-          logger.info(
-            "You can now add more components with 'assistant-ui add'",
-          );
-        }
-      });
-    } else {
-      const templateName = opts.template as keyof typeof templates;
-      const templateUrl = templates[templateName];
-
-      if (!templateUrl) {
-        logger.error(`Unknown template: ${opts.template}`);
-        logger.info(`Available templates: ${templateNames.join(", ")}`);
-        process.exit(1);
-      }
+      await create.parseAsync(createArgs, { from: "user" });
+      return;
+    }
 
-      logger.info(
-        `Creating a new assistant-ui project (template: ${templateName})...`,
+    const registryUrl = DEFAULT_REGISTRY_URL;
+    logger.info("Initializing assistant-ui in existing project...");
+    logger.break();
+
+    if (!opts.yes && isNonInteractiveShell()) {
+      logger.error(
+        [
+          "Detected a non-interactive shell, but 'assistant-ui init' needs interactive prompts by default.",
+          "To run this in CI/agent mode, re-run with '--yes' so shadcn initialization and component install run non-interactively.",
+          "Example: assistant-ui init --yes",
+        ].join("\n"),
       );
-      logger.break();
-
-      const cnaArgs: string[] = ["create-next-app@latest"];
-      cnaArgs.push(projectDirectory || ".");
-      cnaArgs.push("-e", templateUrl);
-
-      if (opts.useNpm) cnaArgs.push("--use-npm");
-      if (opts.usePnpm) cnaArgs.push("--use-pnpm");
-      if (opts.useYarn) cnaArgs.push("--use-yarn");
-      if (opts.useBun) cnaArgs.push("--use-bun");
-      if (opts.skipInstall) cnaArgs.push("--skip-install");
-
-      const child = spawn("npx", cnaArgs, { stdio: "inherit", cwd });
+      process.exit(1);
+    }
 
-      child.on("error", (error) => {
-        logger.error(`Failed to create project: ${error.message}`);
-        process.exit(1);
+    try {
+      const { initArgs, addArgs } = createExistingProjectInitPlan({
+        yes: opts.yes,
+        overwrite: opts.overwrite,
+        registryUrl,
       });
 
-      child.on("close", (code) => {
-        if (code !== 0) {
-          logger.error(`Project creation failed with code ${code}`);
-          process.exit(code || 1);
-        } else {
-          logger.break();
-          logger.success("Project created successfully!");
-        }
-      });
+      if (initArgs) {
+        await runSpawn("npx", initArgs, targetDir);
+      }
+      await runSpawn("npx", addArgs, targetDir);
+
+      logger.break();
+      logger.success("Project initialized successfully!");
+      logger.info("You can now add more components with 'assistant-ui add'");
+    } catch (error) {
+      if (error instanceof SpawnExitError) {
+        logger.error(`Initialization failed with code ${error.code}`);
+        process.exit(error.code);
+      }
+      const message = error instanceof Error ? error.message : String(error);
+      logger.error(`Failed to initialize: ${message}`);
+      process.exit(1);
     }
   });

package/src/lib/create-from-example.ts

@@ -16,7 +16,9 @@ export interface CreateFromExampleOptions {
 const VALID_EXAMPLES = [
   "with-ag-ui",
   "with-ai-sdk-v6",
+  "with-artifacts",
   "with-assistant-transport",
+  "with-chain-of-thought",
   "with-cloud",
   "with-custom-thread-list",
   "with-elevenlabs-scribe",
@@ -87,10 +89,13 @@ export async function createFromExample(
   }
 
   // 9. Install shadcn UI components (standard shadcn components like button, tooltip, etc.)
-  if (shadcnUI.length > 0) {
-    logger.step(`Installing shadcn UI components: ${shadcnUI.join(", ")}...`);
-    await installShadcnComponents(absoluteProjectDir, shadcnUI);
+  //    Always include "utils" since assistant-ui components import cn from @/lib/utils
+  //    and shadcn does not declare it as a registryDependency of button/tooltip/etc.
+  if (!shadcnUI.includes("utils")) {
+    shadcnUI.push("utils");
   }
+  logger.step(`Installing shadcn UI components: ${shadcnUI.join(", ")}...`);
+  await installShadcnComponents(absoluteProjectDir, shadcnUI);
 
   // 10. Install assistant-ui components
   if (assistantUI.length > 0) {
@@ -211,6 +216,7 @@ async function transformTsConfig(projectDir: string): Promise<void> {
   if (tsconfig.compilerOptions?.paths) {
     delete tsconfig.compilerOptions.paths["@/components/assistant-ui/*"];
     delete tsconfig.compilerOptions.paths["@/components/ui/*"];
+    delete tsconfig.compilerOptions.paths["@/lib/utils"];
     delete tsconfig.compilerOptions.paths["@assistant-ui/ui/*"];
 
     // If paths is empty, remove it

package/src/lib/upgrade.ts

@@ -13,6 +13,7 @@ const bundle = [
   "v0-11/content-part-to-message-part",
   "v0-12/assistant-api-to-aui",
   "v0-12/event-names-to-camelcase",
+  "v0-12/primitive-if-to-aui-if",
 ];
 
 const log = debug("codemod:upgrade");