tower-studio 0.1.2 → 0.1.3

package/src/lib/ai/__tests__/capability-resolver.test.ts

@@ -0,0 +1,71 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+
+vi.mock("@/lib/db", () => ({
+  db: {
+    aiCapabilityConfig: {
+      findUnique: vi.fn(),
+    },
+  },
+}));
+
+import { resolveCliAdapter, resolveQueryAdapter } from "../capability-resolver";
+import { db } from "@/lib/db";
+import { AiProviderError } from "../types";
+
+describe("capability-resolver", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  describe("resolveCliAdapter", () => {
+    it("returns Claude adapter when no config exists (default)", async () => {
+      vi.mocked(db.aiCapabilityConfig.findUnique).mockResolvedValue(null);
+      const result = await resolveCliAdapter("terminal");
+      expect(result.provider.name).toBe("claude");
+      expect(result.adapter).toBeDefined();
+    });
+
+    it("returns configured provider adapter with model", async () => {
+      vi.mocked(db.aiCapabilityConfig.findUnique).mockResolvedValue({
+        id: "1", slot: "terminal", provider: "claude", mode: "cli", model: "opus",
+        createdAt: new Date(), updatedAt: new Date(),
+      });
+      const result = await resolveCliAdapter("terminal");
+      expect(result.provider.name).toBe("claude");
+      expect(result.model).toBe("opus");
+    });
+
+    it("throws AiProviderError for unknown provider", async () => {
+      vi.mocked(db.aiCapabilityConfig.findUnique).mockResolvedValue({
+        id: "1", slot: "terminal", provider: "nonexistent", mode: "cli", model: null,
+        createdAt: new Date(), updatedAt: new Date(),
+      });
+      await expect(resolveCliAdapter("terminal")).rejects.toThrow(AiProviderError);
+    });
+
+    it("throws for provider without CLI support", async () => {
+      // This tests the UNSUPPORTED_MODE path — would need a provider registered without cli
+      vi.mocked(db.aiCapabilityConfig.findUnique).mockResolvedValue({
+        id: "1", slot: "terminal", provider: "nonexistent", mode: "cli", model: null,
+        createdAt: new Date(), updatedAt: new Date(),
+      });
+      try {
+        await resolveCliAdapter("terminal");
+      } catch (e) {
+        expect(e).toBeInstanceOf(AiProviderError);
+        expect((e as AiProviderError).code).toBe("CLI_NOT_FOUND");
+      }
+    });
+  });
+
+  describe("resolveQueryAdapter", () => {
+    it("throws UNSUPPORTED_MODE when provider has no query adapter for mode", async () => {
+      vi.mocked(db.aiCapabilityConfig.findUnique).mockResolvedValue({
+        id: "1", slot: "summary", provider: "claude", mode: "api", model: null,
+        createdAt: new Date(), updatedAt: new Date(),
+      });
+      // Claude doesn't have api adapter registered yet
+      await expect(resolveQueryAdapter("summary")).rejects.toThrow(AiProviderError);
+    });
+  });
+});

package/src/lib/ai/__tests__/claude-cli-adapter.test.ts

@@ -0,0 +1,103 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { ClaudeCliAdapter } from "../adapters/cli/claude-cli-adapter";
+import type { CliSpawnOptions } from "../types";
+
+describe("ClaudeCliAdapter", () => {
+  let adapter: ClaudeCliAdapter;
+
+  beforeEach(() => {
+    adapter = new ClaudeCliAdapter();
+  });
+
+  describe("buildSpawnArgs", () => {
+    const baseOpts: CliSpawnOptions = {
+      taskId: "ctask123456789012345678",
+      prompt: "Fix the bug",
+      cwd: "/project",
+    };
+
+    it("builds fresh start args with prompt as last argument", () => {
+      const result = adapter.buildSpawnArgs(baseOpts);
+      expect(result.command).toBe("claude");
+      expect(result.args[result.args.length - 1]).toBe("Fix the bug");
+      expect(result.initialInput).toBeUndefined();
+    });
+
+    it("builds resume args with --resume flag", () => {
+      const result = adapter.buildSpawnArgs({
+        ...baseOpts,
+        resumeSessionId: "session-abc-123",
+      });
+      expect(result.args).toContain("--resume");
+      expect(result.args).toContain("session-abc-123");
+      expect(result.args[result.args.length - 1]).not.toBe("Fix the bug");
+    });
+
+    it("builds continue args with --continue flag and no prompt", () => {
+      const result = adapter.buildSpawnArgs({
+        ...baseOpts,
+        continueLatest: true,
+      });
+      expect(result.args).toContain("--continue");
+      expect(result.args).not.toContain("Fix the bug");
+    });
+
+    it("merges profileArgs into args", () => {
+      const result = adapter.buildSpawnArgs({
+        ...baseOpts,
+        profileArgs: ["--model", "opus"],
+      });
+      expect(result.args).toContain("--model");
+      expect(result.args).toContain("opus");
+    });
+
+    it("merges profileEnvVars into env", () => {
+      const result = adapter.buildSpawnArgs({
+        ...baseOpts,
+        profileEnvVars: { CUSTOM_VAR: "value" },
+      });
+      expect(result.env.CUSTOM_VAR).toBe("value");
+    });
+  });
+
+  describe("buildEnvOverrides", () => {
+    it("returns TOWER_* env vars", () => {
+      const env = adapter.buildEnvOverrides({
+        taskId: "ctask123",
+        taskTitle: "Test task",
+        apiUrl: "http://localhost:3000",
+      });
+      expect(env.TOWER_TASK_ID).toBe("ctask123");
+      expect(env.TOWER_TASK_TITLE).toBe("Test task");
+      expect(env.TOWER_API_URL).toBe("http://localhost:3000");
+      expect(env.TOWER_STARTED_AT).toBeDefined();
+    });
+
+    it("includes CALLBACK_URL when provided", () => {
+      const env = adapter.buildEnvOverrides({
+        taskId: "ctask123",
+        taskTitle: "Test",
+        apiUrl: "http://localhost:3000",
+        callbackUrl: "http://external/callback",
+      });
+      expect(env.CALLBACK_URL).toBe("http://external/callback");
+    });
+
+    it("omits CALLBACK_URL when not provided", () => {
+      const env = adapter.buildEnvOverrides({
+        taskId: "ctask123",
+        taskTitle: "Test",
+        apiUrl: "http://localhost:3000",
+      });
+      expect(env.CALLBACK_URL).toBeUndefined();
+    });
+  });
+
+  describe("metadata", () => {
+    it("returns correct config paths", () => {
+      expect(adapter.getConfigDir()).toContain(".claude");
+      expect(adapter.getSettingsPath()).toContain("settings.json");
+      expect(adapter.getSessionsDir()).toContain("projects");
+    });
+  });
+});

package/src/lib/ai/__tests__/provider-registry.test.ts

@@ -0,0 +1,74 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { ProviderRegistry } from "../provider-registry";
+import type { ProviderDefinition } from "../types";
+import { ClaudeCliAdapter } from "../adapters/cli/claude-cli-adapter";
+
+function makeClaudeProvider(): ProviderDefinition {
+  return {
+    name: "claude",
+    displayName: "Claude Code",
+    agentFieldValue: "CLAUDE_CODE",
+    cli: {
+      command: "claude",
+      adapter: new ClaudeCliAdapter(),
+    },
+    models: {
+      cli: ["sonnet", "opus", "haiku"],
+      api: [],
+    },
+  };
+}
+
+describe("ProviderRegistry", () => {
+  let registry: ProviderRegistry;
+
+  beforeEach(() => {
+    registry = new ProviderRegistry();
+  });
+
+  it("registers and retrieves a provider", () => {
+    registry.register(makeClaudeProvider());
+    const provider = registry.get("claude");
+    expect(provider).toBeDefined();
+    expect(provider!.displayName).toBe("Claude Code");
+  });
+
+  it("returns undefined for unknown provider", () => {
+    expect(registry.get("unknown")).toBeUndefined();
+  });
+
+  it("returns CLI adapter for registered provider", () => {
+    registry.register(makeClaudeProvider());
+    const adapter = registry.getCliAdapter("claude");
+    expect(adapter).toBeDefined();
+  });
+
+  it("returns null CLI adapter for provider without CLI", () => {
+    registry.register({
+      ...makeClaudeProvider(),
+      name: "api-only",
+      cli: undefined,
+    });
+    expect(registry.getCliAdapter("api-only")).toBeNull();
+  });
+
+  it("returns all allowed commands from registered CLI providers", () => {
+    registry.register(makeClaudeProvider());
+    expect(registry.getAllowedCommands()).toContain("claude");
+  });
+
+  it("lists all providers", () => {
+    registry.register(makeClaudeProvider());
+    expect(registry.getAll()).toHaveLength(1);
+  });
+
+  it("returns null query adapter for unregistered provider", () => {
+    expect(registry.getQueryAdapter("nonexistent", "api")).toBeNull();
+  });
+
+  it("returns null query adapter when mode not supported", () => {
+    registry.register(makeClaudeProvider());
+    // Claude provider has no api adapter registered yet
+    expect(registry.getQueryAdapter("claude", "api")).toBeNull();
+  });
+});

package/src/lib/pty/__tests__/ws-server-assistant.test.ts

@@ -0,0 +1,7 @@
+import { describe, it } from "vitest";
+
+describe("ws-server assistant handling", () => {
+  it.todo("immediately destroys __assistant__ session on WS close (BE-05)");
+  it.todo("does not set keepalive timer for __assistant__ sessions");
+  it.todo("still applies keepalive for non-assistant sessions");
+});

package/.next/standalone/tower/.claude/rules/process-lifecycle.md

@@ -1,31 +0,0 @@
-# Process & Resource Lifecycle
-
-## PTY Sessions
-
-- Sessions keyed by `taskId` — one active session per task.
-- Concurrency limited by `system.maxConcurrentExecutions` config (default 20).
-- Disconnect keepalive: 2h (running) / 5min (exited), then auto-destroy.
-- SIGTERM handler in `session-store.ts` kills all sessions on exit.
-
-## Preview Processes
-
-- Auto-stop on component unmount (page navigation).
-- Auto-remove from registry on process exit.
-- SIGTERM/SIGINT handlers kill all preview processes on app exit.
-- No `child.unref()` — parent must track child lifecycle.
-
-## WebSocket Server
-
-- Flush batched sender on WebSocket close.
-- Clear polling timer on WebSocket error.
-- Port read from `terminal.wsPort` config (default 3001).
-
-## Timers
-
-- All `setTimeout`/`setInterval` must be cleared in cleanup (useEffect return, onClose, onExit).
-- Resize debounce timers must be cleared on component unmount.
-
-## Database
-
-- Prisma `$disconnect()` called on SIGTERM/SIGINT.
-- SQLite WAL mode + busy_timeout=5000 set in `initDb()`.

package/.next/standalone/tower/.claude/rules/security.md

@@ -1,29 +0,0 @@
----
-paths:
-  - "src/actions/**/*.ts"
-  - "src/app/api/**/*.ts"
-  - "src/lib/**/*.ts"
----
-
-# Security Rules
-
-## CLI Profile
-
-- `command` field must be validated against allowlist (`claude`, `claude-code`).
-- `baseArgs` elements must be strings with no shell metacharacters (`;&|` etc.).
-- `envVars` must block dangerous keys: PATH, LD_PRELOAD, DYLD_INSERT_LIBRARIES, NODE_OPTIONS, HOME, SHELL.
-
-## Environment Injection
-
-- Never mutate `process.env` for per-session vars — pass via `envOverrides` to `pty.spawn()`.
-- Signal files use `$TMPDIR/tower-signals/` with 0700 dir + 0600 file permissions.
-
-## Internal API Routes
-
-- All `/api/internal/` routes must call `requireLocalhost(request)` + `validateTaskId(taskId)`.
-- Task IDs must match CUID format (`/^c[a-z0-9]{20,30}$/`).
-
-## Server Actions
-
-- Catch Prisma `P2025` (record not found) — return user-friendly error, never leak schema details.
-- Validate JSON fields (`JSON.parse`) with try/catch before use.

package/.next/standalone/tower/.claude/rules/ui.md

@@ -1,105 +0,0 @@
----
-paths:
-  - "src/components/**/*.tsx"
-  - "src/app/**/*.tsx"
----
-
-# UI Rules
-
-## Component Sizing
-
-- All Button and SelectTrigger use **default size** (`h-8` = 32px). Never use `size="sm"`.
-- Inputs, selects, and buttons that sit side by side must use the same height.
-
-## Select (Base UI)
-
-- **Never use `<SelectValue />`** — it renders the raw `value` (ID), not the display name.
-- Use a manual `<span>` inside `<SelectTrigger>` to show the selected item's name:
-  ```tsx
-  <SelectTrigger>
-    <span className="truncate">{items.find(x => x.id === value)?.name ?? value}</span>
-  </SelectTrigger>
-  ```
-- `SelectContent` defaults: `alignItemWithTrigger=false`, `align="start"` — dropdown below trigger, left-aligned.
-- Never use native `<select>/<option>` — always use the shadcn Select component.
-
-## Toast
-
-- Use **Sonner** (`import { toast } from "sonner"`).
-- API: `toast.success("msg")` / `toast.error("msg")` / `toast.info("msg")`
-- `<Toaster richColors position="top-right" />` is in layout.tsx.
-
-## Loading States
-
-- Never insert/remove loading text that causes layout shift.
-- Use opacity overlay + spinner on the content area:
-  ```tsx
-  <div className={`relative ${isPending ? "opacity-40 pointer-events-none" : ""}`}>
-    {isPending && <Loader2 className="absolute inset-0 m-auto size-5 animate-spin" />}
-    {children}
-  </div>
-  ```
-
-## i18n
-
-- `I18nProvider` starts with `"zh"` on both server and client (avoids hydration mismatch).
-- Locale is read from localStorage in `useEffect` after mount.
-- All user-facing strings must use `t("key")` from `useI18n()`.
-
-## Buttons — 统一使用 `<Button>` 组件
-
-**禁止在可交互按钮场景使用原生 `<button>` + 手写 hover 样式。** 必须使用 `<Button>` 组件，通过 `variant` 和 `size` 控制样式。
-
-### Icon Buttons（工具栏 / 头部 / 面板）
-
-```tsx
-<Button variant="ghost" size="icon" className="text-muted-foreground">
-  <Icon className="h-4 w-4" />
-</Button>
-```
-
-### Text Buttons（带文字的操作按钮）
-
-```tsx
-<Button variant="ghost" className="text-muted-foreground">
-  <Icon className="h-3.5 w-3.5" />
-  <span className="text-xs">{label}</span>
-</Button>
-```
-
-### 规则
-
-- **Always use `<Button>`** — ghost 变体的 hover 样式（`hover:bg-accent`）在 light/dark 下统一生效
-- **Never use `<button>` + 手写 `hover:bg-*`** — 容易和 Button 组件不一致，dark 模式下可能看不见
-- `variant="ghost"` 用于工具栏、头部、面板内的非主要操作
-- `variant="outline"` 用于次要操作（取消、返回等）
-- `variant="default"` 用于主要操作（创建、提交等）
-- 只在特殊场景用原生 `<button>`：右键菜单项、自定义交互组件内部等
-
-## Header Heights
-
-Standardized header/toolbar heights. **Panels on the same horizontal line MUST use the same level.**
-
-| Level | Utility Class | Height | Usage |
-|-------|--------------|--------|-------|
-| 1 | `header-xs` | 36px | Inner sub-tabs (Files/Search/Git) |
-| 2 | `header-sm` | 44px | Action bars, simple toolbars |
-| 3 | `header-md` | 56px | Main tab bars (Files/Changes/Preview), page nav |
-| 4 | `header-lg` | 72px | Page headers (title + status, two lines) |
-| 5 | `header-xl` | 88px | Page headers with actions (title + status + buttons) |
-
-Each utility sets `min-height` + `border-bottom`. Combine with padding classes as needed:
-
-```tsx
-{/* Left panel header — level 4 */}
-<div className="header-lg px-4 py-3 flex items-center">...</div>
-
-{/* Right panel tab bar — must also be level 4 to align */}
-<div className="header-lg px-3 py-3 flex items-center">...</div>
-```
-
-CSS variables available: `var(--header-xs)` through `var(--header-xl)`.
-
-## DnD Kit
-
-- Always pass a stable `id` prop to `<DndContext>` to prevent hydration mismatch.