assistant-ui 0.0.106 → 0.0.108

package/src/lib/create-project.ts

@@ -210,22 +210,25 @@ export async function resolvePackageManagerForCwd(
   }
 }
 
+export interface TransformResult {
+  registryInstallFailure?: { retryCommand: string };
+}
+
 export async function transformProject(
   projectDir: string,
   opts: TransformOptions,
-): Promise<void> {
+): Promise<TransformResult> {
   logger.step("Transforming package.json...");
   transformPackageJson(projectDir);
 
+  logger.step("Transforming project files...");
+  transformTsConfig(projectDir);
+  transformCssFiles(projectDir);
+
   let assistantUI: string[] | undefined;
   let shadcnUI: string[] | undefined;
 
   if (!opts.hasLocalComponents) {
-    logger.step("Transforming project files...");
-
-    transformTsConfig(projectDir);
-    transformCssFiles(projectDir);
-
     const components = scanRequiredComponents(projectDir);
     assistantUI = components.assistantUI;
     shadcnUI = components.shadcnUI;
@@ -249,8 +252,15 @@ export async function transformProject(
     const auiComponents = assistantUI.map((c) => `@assistant-ui/${c}`);
     const components = [...allShadcn, ...auiComponents];
     logger.step(`Installing components: ${components.join(", ")}...`);
-    await installShadcnRegistry(projectDir, components, "components", pm);
+    const failure = await installShadcnRegistry(
+      projectDir,
+      components,
+      "components",
+      pm,
+    );
+    if (failure) return { registryInstallFailure: failure };
   }
+  return {};
 }
 
 function transformPackageJson(projectDir: string): void {
@@ -326,7 +336,8 @@ function transformTsConfig(projectDir: string): void {
         Array.isArray(targets) &&
         targets.some(
           (target) =>
-            typeof target === "string" && target.includes("packages/ui/"),
+            typeof target === "string" &&
+            (target.includes("packages/ui/") || target.startsWith("../")),
         );
       if (workspaceKeys.has(key) || targetsWorkspace) {
         delete tsconfig.compilerOptions.paths[key];
@@ -461,20 +472,20 @@ async function installShadcnRegistry(
   components: string[],
   label: string,
   pm: PackageManagerName,
-): Promise<void> {
+): Promise<{ retryCommand: string } | undefined> {
   const [cmd, dlxArgs] = dlxCommand(pm);
   // For npm, dlxArgs may already include `--yes` for npx auto-install.
   // The trailing `--yes` is for shadcn's own confirmation prompt.
-  const addArgs = [...dlxArgs, "shadcn@latest", "add", ...components, "--yes"];
+  const retryArgs = [...dlxArgs, "shadcn@latest", "add", ...components];
+  const addArgs = [...retryArgs, "--yes"];
 
   try {
     await runSpawn(cmd, addArgs, projectDir);
+    return undefined;
   } catch (error) {
     if (error instanceof SpawnExitError) {
-      logger.warn(
-        `shadcn exited with code ${error.code}. Run the following to retry:\n  ${cmd} ${addArgs.slice(0, -1).join(" ")}`,
-      );
-      return;
+      logger.warn(`shadcn exited with code ${error.code}.`);
+      return { retryCommand: `${cmd} ${retryArgs.join(" ")}` };
     }
 
     const message = error instanceof Error ? error.message : String(error);

package/src/lib/upgrade.ts

@@ -14,6 +14,7 @@ const bundle = [
   "v0-12/assistant-api-to-aui",
   "v0-12/event-names-to-camelcase",
   "v0-12/primitive-if-to-aui-if",
+  "v0-15/aui-accessor-calls-to-properties",
 ];
 
 const log = debug("codemod:upgrade");

package/src/lib/utils/registry.test.ts

@@ -94,9 +94,15 @@ describe("resolveRegistryItemUrl", () => {
     );
   });
 
-  it("uses the plain URL without a style", () => {
+  it("uses the base URL without a style", () => {
     expect(resolveRegistryItemUrl("thread")).toBe(
-      "https://r.assistant-ui.com/thread.json",
+      "https://r.assistant-ui.com/base/thread.json",
+    );
+  });
+
+  it("uses the base URL for an explicit undefined style", () => {
+    expect(resolveRegistryItemUrl("thread", undefined)).toBe(
+      "https://r.assistant-ui.com/base/thread.json",
     );
   });
 

package/src/lib/utils/registry.ts

@@ -31,7 +31,11 @@ export function resolveRegistryItemUrl(
   component: string,
   style?: string,
 ): string {
-  const registryUrl = style?.startsWith("base-")
+  if (style === undefined) {
+    return `${REGISTRY_BASE_URL}/base/${encodeURIComponent(component)}.json`;
+  }
+
+  const registryUrl = style.startsWith("base-")
     ? `${REGISTRY_BASE_URL}/styles/${encodeURIComponent(style)}`
     : REGISTRY_BASE_URL;
 

package/src/lib/utils/workspace.ts

@@ -0,0 +1,34 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+
+export function resolveRealPath(inputPath: string): string {
+  const resolved = path.resolve(inputPath);
+
+  try {
+    return fs.realpathSync(resolved);
+  } catch {
+    return resolved;
+  }
+}
+
+export function findWorkspaceRoot(cwd: string): string | null {
+  let dir = path.resolve(cwd);
+  const root = path.parse(dir).root;
+
+  while (true) {
+    if (fs.existsSync(path.join(dir, "pnpm-workspace.yaml"))) return dir;
+
+    const pkgPath = path.join(dir, "package.json");
+    if (fs.existsSync(pkgPath)) {
+      try {
+        const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf8"));
+        if (pkg.workspaces) return dir;
+      } catch {
+        // Malformed package manifests are not workspace markers.
+      }
+    }
+
+    if (dir === root) return null;
+    dir = path.dirname(dir);
+  }
+}

package/src/run.test.ts

@@ -0,0 +1,27 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mocks = vi.hoisted(() => ({
+  parseAsync: vi.fn(),
+}));
+
+vi.mock("./program", () => ({
+  buildProgram: () => ({
+    parseAsync: mocks.parseAsync,
+  }),
+}));
+
+import { runCli } from "./run";
+
+describe("runCli", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("awaits and propagates asynchronous command failures", async () => {
+    const error = new Error("command failed");
+    mocks.parseAsync.mockRejectedValue(error);
+
+    await expect(runCli()).rejects.toBe(error);
+    expect(mocks.parseAsync).toHaveBeenCalledOnce();
+  });
+});

package/src/run.ts

@@ -0,0 +1,5 @@
+import { buildProgram } from "./program";
+
+export async function runCli(): Promise<void> {
+  await buildProgram().parseAsync();
+}