@vellumai/cli 0.11.3 → 0.11.4-dev.202608190019.b94dbf2

package/src/lib/web-dist.ts

@@ -0,0 +1,48 @@
+import { existsSync } from "node:fs";
+import { createRequire } from "node:module";
+import { dirname, join } from "node:path";
+
+const loadModule = createRequire(import.meta.url);
+
+interface WebDistOptions {
+  execPath?: string;
+  platform?: NodeJS.Platform;
+  resolvePackage?: () => string;
+  startDir?: string;
+}
+
+/** Locate the pre-built web SPA for installed, packaged, or source CLIs. */
+export function findWebDistDir(options: WebDistOptions = {}): string | null {
+  try {
+    const packageJson = (
+      options.resolvePackage ??
+      (() => loadModule.resolve("@vellumai/web/package.json"))
+    )();
+    const distDir = join(dirname(packageJson), "dist");
+    if (existsSync(join(distDir, "index.html"))) {
+      return distDir;
+    }
+  } catch {}
+
+  if ((options.platform ?? process.platform) === "win32") {
+    const runtimeDir = dirname(options.execPath ?? process.execPath);
+    const packagedDist = join(runtimeDir, "web-dist");
+    if (existsSync(join(packagedDist, "index.html"))) {
+      return packagedDist;
+    }
+  }
+
+  let dir = options.startDir ?? import.meta.dir;
+  for (let depth = 0; depth < 8; depth++) {
+    const candidate = join(dir, "clients", "web", "dist", "index.html");
+    if (existsSync(candidate)) {
+      return dirname(candidate);
+    }
+    const parent = dirname(dir);
+    if (parent === dir) {
+      break;
+    }
+    dir = parent;
+  }
+  return null;
+}

package/src/lib/feature-flags.test.ts

@@ -1,157 +0,0 @@
-import {
-  afterAll,
-  afterEach,
-  beforeEach,
-  describe,
-  expect,
-  test,
-} from "bun:test";
-import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-
-import {
-  isAssistantFeatureFlagEnabled,
-  WEB_REMOTE_INGRESS_FLAG,
-} from "./feature-flags.js";
-
-const testDir = mkdtempSync(join(tmpdir(), "cli-feature-flags-test-"));
-const originalFetch = globalThis.fetch;
-const originalLockfileDir = process.env.VELLUM_LOCKFILE_DIR;
-
-function writeLockfile(): void {
-  mkdirSync(testDir, { recursive: true });
-  writeFileSync(
-    join(testDir, ".vellum.lock.json"),
-    JSON.stringify(
-      {
-        activeAssistant: "assistant-1",
-        assistants: [
-          {
-            assistantId: "assistant-1",
-            runtimeUrl: "http://127.0.0.1:7830",
-            cloud: "local",
-          },
-        ],
-      },
-      null,
-      2,
-    ),
-  );
-}
-
-function jsonResponse(body: unknown, status = 200): Response {
-  return new Response(JSON.stringify(body), {
-    status,
-    headers: { "content-type": "application/json" },
-  });
-}
-
-function mockFetch(response: Response): void {
-  const fetchMock = async () => response;
-  globalThis.fetch = fetchMock as unknown as typeof globalThis.fetch;
-}
-
-function mockFetchWithUrls(response: Response): string[] {
-  const urls: string[] = [];
-  const fetchMock = async (input: RequestInfo | URL) => {
-    urls.push(String(input));
-    return response;
-  };
-  globalThis.fetch = fetchMock as unknown as typeof globalThis.fetch;
-  return urls;
-}
-
-describe("isAssistantFeatureFlagEnabled", () => {
-  beforeEach(() => {
-    process.env.VELLUM_LOCKFILE_DIR = testDir;
-    rmSync(join(testDir, ".vellum.lock.json"), { force: true });
-    writeLockfile();
-  });
-
-  afterEach(() => {
-    globalThis.fetch = originalFetch;
-  });
-
-  afterAll(() => {
-    if (originalLockfileDir === undefined) {
-      delete process.env.VELLUM_LOCKFILE_DIR;
-    } else {
-      process.env.VELLUM_LOCKFILE_DIR = originalLockfileDir;
-    }
-    rmSync(testDir, { recursive: true, force: true });
-  });
-
-  test("returns true when the assistant flag is enabled", async () => {
-    mockFetch(
-      jsonResponse({
-        flags: [{ key: WEB_REMOTE_INGRESS_FLAG, enabled: true }],
-      }),
-    );
-
-    await expect(
-      isAssistantFeatureFlagEnabled("assistant-1", WEB_REMOTE_INGRESS_FLAG),
-    ).resolves.toBe(true);
-  });
-
-  test("uses the supplied runtime URL instead of a stale lockfile runtimeUrl", async () => {
-    writeFileSync(
-      join(testDir, ".vellum.lock.json"),
-      JSON.stringify(
-        {
-          activeAssistant: "assistant-1",
-          assistants: [
-            {
-              assistantId: "assistant-1",
-              runtimeUrl: "https://stale-tunnel.ngrok-free.dev",
-              cloud: "local",
-            },
-          ],
-        },
-        null,
-        2,
-      ),
-    );
-    const urls = mockFetchWithUrls(
-      jsonResponse({
-        flags: [{ key: WEB_REMOTE_INGRESS_FLAG, enabled: true }],
-      }),
-    );
-
-    await expect(
-      isAssistantFeatureFlagEnabled("assistant-1", WEB_REMOTE_INGRESS_FLAG, {
-        runtimeUrl: "http://127.0.0.1:9123",
-      }),
-    ).resolves.toBe(true);
-
-    expect(urls).toEqual([
-      "http://127.0.0.1:9123/v1/assistants/assistant-1/feature-flags",
-    ]);
-  });
-
-  test("returns false when the assistant flag is disabled or missing", async () => {
-    mockFetch(
-      jsonResponse({
-        flags: [{ key: WEB_REMOTE_INGRESS_FLAG, enabled: false }],
-      }),
-    );
-
-    await expect(
-      isAssistantFeatureFlagEnabled("assistant-1", WEB_REMOTE_INGRESS_FLAG),
-    ).resolves.toBe(false);
-
-    mockFetch(jsonResponse({ flags: [] }));
-
-    await expect(
-      isAssistantFeatureFlagEnabled("assistant-1", WEB_REMOTE_INGRESS_FLAG),
-    ).resolves.toBe(false);
-  });
-
-  test("throws when the gateway rejects the flag request", async () => {
-    mockFetch(jsonResponse({ error: "nope" }, 500));
-
-    await expect(
-      isAssistantFeatureFlagEnabled("assistant-1", WEB_REMOTE_INGRESS_FLAG),
-    ).rejects.toThrow("Failed to fetch feature flags");
-  });
-});

package/src/lib/feature-flags.ts

@@ -1,38 +0,0 @@
-import { AssistantClient } from "./assistant-client.js";
-
-export const WEB_REMOTE_INGRESS_FLAG = "web-remote-ingress";
-
-type FeatureFlagEntry = {
-  key?: unknown;
-  enabled?: unknown;
-};
-
-type FeatureFlagsResponse = {
-  flags?: FeatureFlagEntry[];
-};
-
-export async function isAssistantFeatureFlagEnabled(
-  assistantId: string,
-  key: string,
-  opts: { runtimeUrl?: string } = {},
-): Promise<boolean> {
-  const client = new AssistantClient({
-    assistantId,
-    runtimeUrl: opts.runtimeUrl,
-  });
-  const res = await client.get("/feature-flags");
-  if (!res.ok) {
-    const body = await res.text().catch(() => "");
-    throw new Error(
-      `Failed to fetch feature flags: HTTP ${res.status} ${body}`.trim(),
-    );
-  }
-
-  const data = (await res.json()) as FeatureFlagsResponse;
-  const flag = data.flags?.find((entry) => entry.key === key);
-  return flag?.enabled === true;
-}
-
-export function formatFeatureFlagGateMessage(flagKey: string): string {
-  return `This command is behind the \`${flagKey}\` feature flag. Enable it with \`vellum flags set ${flagKey} true\` and try again.`;
-}