@app-vitals/agent-admin 0.0.0-stage → 0.1.0

package/src/agent-plugins.ts

@@ -0,0 +1,95 @@
+/**
+ * agent/src/agent-plugins.ts
+ * AgentPluginService — CRUD for Claude Code plugins per agent.
+ *
+ * Each plugin is identified by its canonical Claude install spec — the string
+ * passed to `claude plugin install`: "<plugin>@<marketplace>" (e.g.
+ * "shipwright@shipwright"), or a bare "<plugin>" that defaults to the bundled
+ * "shipwright" marketplace.
+ * The unique constraint on [agentId, name] prevents duplicates; add() uses
+ * upsert so re-adding an existing plugin updates its version and re-enables it.
+ */
+
+import type { AgentPlugin, PrismaClient } from "../prisma/client/client.ts";
+import { NotFoundError } from "./errors.ts";
+import type { PrismaTransactionClient } from "./prisma-tx.ts";
+
+export type { AgentPlugin };
+
+export class AgentPluginService {
+  constructor(private prisma: PrismaClient) {}
+
+  /**
+   * List all plugins for a given agent, ordered by createdAt.
+   */
+  async list(agentId: string): Promise<AgentPlugin[]> {
+    return this.prisma.agentPlugin.findMany({
+      where: { agentId },
+      orderBy: { createdAt: "asc" },
+    });
+  }
+
+  /**
+   * List enabled plugins for a given agent, ordered by createdAt.
+   * Filters `enabled: true` server-side.
+   */
+  async listEnabled(agentId: string): Promise<AgentPlugin[]> {
+    return this.prisma.agentPlugin.findMany({
+      where: { agentId, enabled: true },
+      orderBy: { createdAt: "asc" },
+    });
+  }
+
+  /**
+   * Add a plugin for the given agent.
+   * Uses upsert so re-adding an existing plugin updates its version and re-enables it.
+   *
+   * @param client - defaults to `this.prisma`; pass the `tx` argument from a
+   *   caller's `prisma.$transaction(async (tx) => ...)` to make this write
+   *   participate in that transaction (see createAgent() in agents.ts).
+   */
+  async add(
+    agentId: string,
+    name: string,
+    version?: string | null,
+    client: PrismaTransactionClient = this.prisma,
+  ): Promise<AgentPlugin> {
+    return client.agentPlugin.upsert({
+      where: { agentId_name: { agentId, name } },
+      create: { agentId, name, version: version ?? null, enabled: true },
+      update: { version: version ?? null, enabled: true },
+    });
+  }
+
+  /**
+   * Remove a plugin by ID. Verifies agentId ownership before deleting.
+   * Throws NotFoundError if the pluginId doesn't exist or belongs to a different agent.
+   */
+  async remove(agentId: string, pluginId: string): Promise<void> {
+    const existing = await this.prisma.agentPlugin.findUnique({
+      where: { id: pluginId },
+    });
+
+    if (!existing || existing.agentId !== agentId) {
+      throw new NotFoundError(`plugin ${pluginId} not found`);
+    }
+
+    await this.prisma.agentPlugin.delete({ where: { id: pluginId } });
+  }
+
+  /**
+   * Remove a plugin by name. Verifies agentId ownership before deleting.
+   * Throws NotFoundError if no plugin with that name exists for the agent.
+   */
+  async removeByName(agentId: string, name: string): Promise<void> {
+    const existing = await this.prisma.agentPlugin.findUnique({
+      where: { agentId_name: { agentId, name } },
+    });
+
+    if (!existing) {
+      throw new NotFoundError(`plugin ${name} not found`);
+    }
+
+    await this.prisma.agentPlugin.delete({ where: { id: existing.id } });
+  }
+}

package/src/agent-plugins.unit.test.ts

@@ -0,0 +1,43 @@
+/**
+ * agent/src/agent-plugins.unit.test.ts
+ * Unit tests for AgentPluginService.listEnabled — pure logic over an injected
+ * Prisma double, no real DB.
+ */
+
+import { describe, expect, it, mock } from "bun:test";
+import type { PrismaClient } from "../prisma/client/client.ts";
+import { AgentPluginService } from "./agent-plugins.ts";
+
+describe("AgentPluginService.listEnabled", () => {
+  it("filters by enabled: true server-side", async () => {
+    const findMany = mock(async () => []);
+    const prisma = { agentPlugin: { findMany } } as unknown as PrismaClient;
+    const service = new AgentPluginService(prisma);
+
+    await service.listEnabled("agent-1");
+
+    expect(findMany).toHaveBeenCalledWith({
+      where: { agentId: "agent-1", enabled: true },
+      orderBy: { createdAt: "asc" },
+    });
+  });
+
+  it("returns only the enabled plugins the query resolves", async () => {
+    const enabledPlugin = {
+      id: "p1",
+      agentId: "agent-1",
+      name: "shipwright@shipwright",
+      version: null,
+      enabled: true,
+      createdAt: new Date(),
+      updatedAt: new Date(),
+    };
+    const findMany = mock(async () => [enabledPlugin]);
+    const prisma = { agentPlugin: { findMany } } as unknown as PrismaClient;
+    const service = new AgentPluginService(prisma);
+
+    const result = await service.listEnabled("agent-1");
+
+    expect(result).toEqual([enabledPlugin]);
+  });
+});