assistant-ui 0.0.114 → 0.0.116

package/src/lib/create-project.ts

@@ -10,6 +10,10 @@ import { logger } from "./utils/logger";
 import { runSpawn, SpawnExitError, SpawnSignalError } from "./run-spawn";
 import { type PackageManagerName } from "./utils/package-manager";
 import { readProjectFiles } from "./utils/file-scanner";
+import {
+  detectRegistryPlatform,
+  resolveRegistryItemUrl,
+} from "./utils/registry";
 
 export function dlxCommand(pm: PackageManagerName): [string, string[]] {
   switch (pm) {
@@ -186,6 +190,7 @@ export async function scaffoldProject(
 
 export interface TransformResult {
   registryInstallFailure?: { retryCommand: string };
+  registryInstallCommand?: string;
 }
 
 export async function transformProject(
@@ -199,29 +204,23 @@ export async function transformProject(
   transformTsConfig(projectDir);
   transformCssFiles(projectDir);
 
-  let assistantUI: string[] | undefined;
-  let shadcnUI: string[] | undefined;
-
-  if (!opts.hasLocalComponents) {
-    const components = scanRequiredComponents(projectDir);
-    assistantUI = components.assistantUI;
-    shadcnUI = components.shadcnUI;
-  }
+  const components = opts.hasLocalComponents
+    ? undefined
+    : resolveRegistryComponents(projectDir, scanRequiredComponents(projectDir));
 
   const pm = opts.packageManager;
-  if (!opts.skipInstall) {
-    logger.step("Installing dependencies...");
-    await installDependencies(projectDir, pm);
+  if (opts.skipInstall) {
+    if (!components) return {};
+    const [cmd, dlxArgs] = dlxCommand(pm);
+    return {
+      registryInstallCommand: `${cmd} ${[...dlxArgs, "shadcn@latest", "add", ...components].join(" ")}`,
+    };
   }
 
-  if (
-    !opts.skipInstall &&
-    !opts.hasLocalComponents &&
-    shadcnUI &&
-    assistantUI
-  ) {
-    const auiComponents = assistantUI.map((c) => `@assistant-ui/${c}`);
-    const components = ["@assistant-ui/utils", ...shadcnUI, ...auiComponents];
+  logger.step("Installing dependencies...");
+  await installDependencies(projectDir, pm);
+
+  if (components) {
     logger.step(`Installing components: ${components.join(", ")}...`);
     const failure = await installShadcnRegistry(
       projectDir,
@@ -235,6 +234,22 @@ export async function transformProject(
   return {};
 }
 
+function resolveRegistryComponents(
+  projectDir: string,
+  { assistantUI, shadcnUI }: RequiredComponents,
+): string[] {
+  if (detectRegistryPlatform(projectDir) === "native") {
+    return ["utils", ...shadcnUI, ...assistantUI].map((component) =>
+      resolveRegistryItemUrl(component, undefined, "native"),
+    );
+  }
+  return [
+    "@assistant-ui/utils",
+    ...shadcnUI,
+    ...assistantUI.map((component) => `@assistant-ui/${component}`),
+  ];
+}
+
 function transformPackageJson(projectDir: string): void {
   const pkgPath = path.join(projectDir, "package.json");
   const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
@@ -543,7 +558,7 @@ export async function reconcileAssistantUIImportLayout(
   }
 }
 
-function scanRequiredComponents(projectDir: string): RequiredComponents {
+export function scanRequiredComponents(projectDir: string): RequiredComponents {
   const assistantUIComponents = new Set<string>();
   const shadcnUIComponents = new Set<string>();
 

package/src/lib/run-spawn.test.ts

@@ -36,7 +36,6 @@ const trackSignal = (signal: NodeJS.Signals) => {
 describe("runSpawn", () => {
   afterEach(() => {
     vi.restoreAllMocks();
-    vi.clearAllMocks();
   });
 
   it("forwards the first termination signal and cleans up listeners", async () => {

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

@@ -3,6 +3,7 @@ import * as os from "node:os";
 import * as path from "node:path";
 import { afterEach, beforeEach, describe, expect, it } from "vitest";
 import {
+  detectRegistryPlatform,
   getComponentsJsonStyle,
   resolveQuickStartRegistryUrl,
   resolveRegistryItemUrl,
@@ -61,6 +62,49 @@ describe("getComponentsJsonStyle", () => {
   });
 });
 
+describe("detectRegistryPlatform", () => {
+  let cwd: string;
+
+  beforeEach(() => {
+    cwd = fs.mkdtempSync(path.join(os.tmpdir(), "assistant-ui-cli-"));
+  });
+
+  afterEach(() => {
+    fs.rmSync(cwd, { recursive: true, force: true });
+  });
+
+  it("detects react-native in dependencies", () => {
+    fs.writeFileSync(
+      path.join(cwd, "package.json"),
+      JSON.stringify({ dependencies: { "react-native": "0.86.3" } }),
+    );
+
+    expect(detectRegistryPlatform(cwd)).toBe("native");
+  });
+
+  it("detects react-native in devDependencies", () => {
+    fs.writeFileSync(
+      path.join(cwd, "package.json"),
+      JSON.stringify({ devDependencies: { "react-native": "0.86.3" } }),
+    );
+
+    expect(detectRegistryPlatform(cwd)).toBe("native");
+  });
+
+  it("uses web without react-native", () => {
+    fs.writeFileSync(
+      path.join(cwd, "package.json"),
+      JSON.stringify({ dependencies: { react: "19.2.3" } }),
+    );
+
+    expect(detectRegistryPlatform(cwd)).toBe("web");
+  });
+
+  it("uses web when package.json is missing", () => {
+    expect(detectRegistryPlatform(cwd)).toBe("web");
+  });
+});
+
 describe("resolveQuickStartRegistryUrl", () => {
   it("uses the base quick start for base styles", () => {
     expect(resolveQuickStartRegistryUrl("base-nova")).toBe(
@@ -111,4 +155,16 @@ describe("resolveRegistryItemUrl", () => {
       "https://r.assistant-ui.com/styles/base-nova%3Fpreview%3D1/thread.json",
     );
   });
+
+  it("uses the native URL regardless of style", () => {
+    expect(resolveRegistryItemUrl("thread", "base-nova", "native")).toBe(
+      "https://r.assistant-ui.com/native/thread.json",
+    );
+  });
+
+  it("keeps shared items at the root URL on native", () => {
+    expect(resolveRegistryItemUrl("utils", undefined, "native")).toBe(
+      "https://r.assistant-ui.com/utils.json",
+    );
+  });
 });

package/src/lib/utils/registry.ts

@@ -2,6 +2,34 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 
 const REGISTRY_BASE_URL = "https://r.assistant-ui.com";
+export const SHARED_REGISTRY_ITEMS = new Set(["utils"]);
+
+export function detectRegistryPlatform(cwd: string): "web" | "native" {
+  try {
+    const packageJsonPath = path.join(cwd, "package.json");
+    const packageJson = JSON.parse(
+      fs.readFileSync(packageJsonPath, "utf8"),
+    ) as {
+      dependencies?: unknown;
+      devDependencies?: unknown;
+    };
+    const dependencyGroups = [
+      packageJson.dependencies,
+      packageJson.devDependencies,
+    ];
+
+    return dependencyGroups.some(
+      (dependencies) =>
+        typeof dependencies === "object" &&
+        dependencies !== null &&
+        Object.hasOwn(dependencies, "react-native"),
+    )
+      ? "native"
+      : "web";
+  } catch {
+    return "web";
+  }
+}
 
 export function getComponentsJsonStyle(cwd: string): string | undefined {
   try {
@@ -30,7 +58,14 @@ export function resolveQuickStartRegistryUrl(style?: string): string {
 export function resolveRegistryItemUrl(
   component: string,
   style?: string,
+  platform: "web" | "native" = "web",
 ): string {
+  if (platform === "native") {
+    return SHARED_REGISTRY_ITEMS.has(component)
+      ? `${REGISTRY_BASE_URL}/${encodeURIComponent(component)}.json`
+      : `${REGISTRY_BASE_URL}/native/${encodeURIComponent(component)}.json`;
+  }
+
   if (style === undefined) {
     return `${REGISTRY_BASE_URL}/base/${encodeURIComponent(component)}.json`;
   }

package/src/run.test.ts

@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 
 const mocks = vi.hoisted(() => ({
   parseAsync: vi.fn(),
@@ -13,10 +13,6 @@ vi.mock("./program", () => ({
 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);