@adhd/agent-store-tools 2.1.0

package/src/store/mcp-server-store.js

@@ -0,0 +1,89 @@
+import { eq } from 'drizzle-orm';
+import { mcpServersTable } from '../db/schema.js';
+export class McpServerStoreError extends Error {
+    code;
+    constructor(code, message) {
+        super(message);
+        this.code = code;
+        this.name = 'McpServerStoreError';
+    }
+}
+// ──────────────────────────────────────────────
+// McpServerStore
+//
+// Thin Drizzle queries over mcp_servers.
+// Mirrors the pattern in entrypoint/agent-mcp's AgentStore (agent-cache-store pattern).
+// Constructor accepts a BetterSQLite3Database so tests can inject their own
+// connection without touching the production singleton in client.ts.
+// ──────────────────────────────────────────────
+export class McpServerStore {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    /**
+     * Register an MCP server.
+     * Throws MCP_SERVER_ALREADY_EXISTS if the id PK already exists.
+     */
+    create(input) {
+        const existing = this.db
+            .select()
+            .from(mcpServersTable)
+            .where(eq(mcpServersTable.id, input.id))
+            .get();
+        if (existing) {
+            throw new McpServerStoreError('MCP_SERVER_ALREADY_EXISTS', `MCP server '${input.id}' already exists`);
+        }
+        const server = {
+            id: input.id,
+            transport: input.transport,
+            name: input.name,
+            providedToolIds: input.providedToolIds ?? [],
+            configSchema: input.configSchema ?? {},
+        };
+        this.db
+            .insert(mcpServersTable)
+            .values({
+            id: server.id,
+            transport: server.transport,
+            name: server.name,
+            providedToolIds: server.providedToolIds,
+            configSchema: server.configSchema,
+        })
+            .run();
+        return server;
+    }
+    /**
+     * Read an MCP server by id.
+     * Throws MCP_SERVER_NOT_FOUND if absent.
+     */
+    read(id) {
+        const row = this.db
+            .select()
+            .from(mcpServersTable)
+            .where(eq(mcpServersTable.id, id))
+            .get();
+        if (!row) {
+            throw new McpServerStoreError('MCP_SERVER_NOT_FOUND', `MCP server '${id}' not found`);
+        }
+        return {
+            id: row.id,
+            transport: row.transport,
+            name: row.name,
+            providedToolIds: row.providedToolIds,
+            configSchema: row.configSchema,
+        };
+    }
+    /** List all registered MCP servers. */
+    list() {
+        const rows = this.db.select().from(mcpServersTable).all();
+        return rows.map((row) => ({
+            id: row.id,
+            transport: row.transport,
+            name: row.name,
+            providedToolIds: row.providedToolIds,
+            configSchema: row.configSchema,
+        }));
+    }
+}
+//# sourceMappingURL=mcp-server-store.js.map

package/src/store/mcp-server-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server-store.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-tools/src/store/mcp-server-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAIjC,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAuClD,MAAM,OAAO,mBAAoB,SAAQ,KAAK;IAChB;IAA5B,YAA4B,IAA6B,EAAE,OAAe;QACxE,KAAK,CAAC,OAAO,CAAC,CAAC;QADW,SAAI,GAAJ,IAAI,CAAyB;QAEvD,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC;IACpC,CAAC;CACF;AAED,iDAAiD;AACjD,iBAAiB;AACjB,EAAE;AACF,yCAAyC;AACzC,wFAAwF;AACxF,4EAA4E;AAC5E,qEAAqE;AACrE,iDAAiD;AAEjD,MAAM,OAAO,cAAc;IAEN;IADnB,YACmB,EAAgD;QAAhD,OAAE,GAAF,EAAE,CAA8C;IAChE,CAAC;IAEJ;;;OAGG;IACH,MAAM,CAAC,KAA2B;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE;aACrB,MAAM,EAAE;aACR,IAAI,CAAC,eAAe,CAAC;aACrB,KAAK,CAAC,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;aACvC,GAAG,EAAE,CAAC;QAET,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,mBAAmB,CAC3B,2BAA2B,EAC3B,eAAe,KAAK,CAAC,EAAE,kBAAkB,CAC1C,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAc;YACxB,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,eAAe,EAAE,KAAK,CAAC,eAAe,IAAI,EAAE;YAC5C,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,EAAE;SACvC,CAAC;QAEF,IAAI,CAAC,EAAE;aACJ,MAAM,CAAC,eAAe,CAAC;aACvB,MAAM,CAAC;YACN,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,YAAY,EAAE,MAAM,CAAC,YAAY;SAClC,CAAC;aACD,GAAG,EAAE,CAAC;QAET,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,IAAI,CAAC,EAAU;QACb,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,MAAM,EAAE;aACR,IAAI,CAAC,eAAe,CAAC;aACrB,KAAK,CAAC,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;aACjC,GAAG,EAAE,CAAC;QAET,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,mBAAmB,CAC3B,sBAAsB,EACtB,eAAe,EAAE,aAAa,CAC/B,CAAC;QACJ,CAAC;QAED,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,eAAe,EAAE,GAAG,CAAC,eAA2B;YAChD,YAAY,EAAE,GAAG,CAAC,YAAuC;SAC1D,CAAC;IACJ,CAAC;IAED,uCAAuC;IACvC,IAAI;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,GAAG,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,eAAe,EAAE,GAAG,CAAC,eAA2B;YAChD,YAAY,EAAE,GAAG,CAAC,YAAuC;SAC1D,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}

package/src/store/tool-store.d.ts

@@ -0,0 +1,55 @@
+import type { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
+/** A seeded tool-type classification row (text PK, never an enum). */
+export interface ToolType {
+    slug: string;
+    description: string;
+}
+/** A canonical, platform-independent agent capability. */
+export interface Tool {
+    name: string;
+    type: string;
+    description: string;
+    version: number;
+    requiresApproval: boolean;
+    isDestructive: boolean;
+    dependencyToolIds: string[];
+    capabilities: string[];
+}
+/** Input for creating a canonical tool. */
+export interface ToolCreateInput {
+    name: string;
+    type: string;
+    description: string;
+    /** Defaults to 1 if omitted. */
+    version?: number;
+    requiresApproval?: boolean;
+    isDestructive?: boolean;
+    dependencyToolIds?: string[];
+    capabilities?: string[];
+}
+export type ToolStoreErrorCode = 'TOOL_ALREADY_EXISTS' | 'TOOL_NOT_FOUND' | 'TOOL_TYPE_NOT_FOUND';
+export declare class ToolStoreError extends Error {
+    readonly code: ToolStoreErrorCode;
+    constructor(code: ToolStoreErrorCode, message: string);
+}
+export declare class ToolStore {
+    private readonly db;
+    constructor(db: BetterSQLite3Database<Record<string, never>>);
+    /** Seed or upsert a tool-type lookup row. */
+    seedToolType(input: ToolType): void;
+    /** Return all seeded tool types. */
+    listToolTypes(): ToolType[];
+    /**
+     * Create a canonical tool row.
+     * Throws TOOL_ALREADY_EXISTS if the name PK already exists.
+     */
+    create(input: ToolCreateInput): Tool;
+    /**
+     * Read a canonical tool by name.
+     * Throws TOOL_NOT_FOUND if absent.
+     */
+    read(name: string): Tool;
+    /** List all canonical tools. */
+    list(): Tool[];
+}
+//# sourceMappingURL=tool-store.d.ts.map

package/src/store/tool-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-store.d.ts","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-tools/src/store/tool-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAQxE,sEAAsE;AACtE,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,0DAA0D;AAC1D,MAAM,WAAW,IAAI;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,aAAa,EAAE,OAAO,CAAC;IACvB,iBAAiB,EAAE,MAAM,EAAE,CAAC;IAC5B,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB;AAED,2CAA2C;AAC3C,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,gCAAgC;IAChC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB;AAMD,MAAM,MAAM,kBAAkB,GAC1B,qBAAqB,GACrB,gBAAgB,GAChB,qBAAqB,CAAC;AAE1B,qBAAa,cAAe,SAAQ,KAAK;aACX,IAAI,EAAE,kBAAkB;gBAAxB,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,MAAM;CAItE;AAWD,qBAAa,SAAS;IAElB,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAAF,EAAE,EAAE,qBAAqB,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAKnE,6CAA6C;IAC7C,YAAY,CAAC,KAAK,EAAE,QAAQ,GAAG,IAAI;IAQnC,oCAAoC;IACpC,aAAa,IAAI,QAAQ,EAAE;IAM3B;;;OAGG;IACH,MAAM,CAAC,KAAK,EAAE,eAAe,GAAG,IAAI;IA0CpC;;;OAGG;IACH,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAuBxB,gCAAgC;IAChC,IAAI,IAAI,IAAI,EAAE;CAaf"}

package/src/store/tool-store.js

@@ -0,0 +1,115 @@
+import { eq } from 'drizzle-orm';
+import { toolTypesTable, toolsTable } from '../db/schema.js';
+export class ToolStoreError extends Error {
+    code;
+    constructor(code, message) {
+        super(message);
+        this.code = code;
+        this.name = 'ToolStoreError';
+    }
+}
+// ──────────────────────────────────────────────
+// ToolStore
+//
+// Thin Drizzle queries over tool_types and tools.
+// Mirrors the pattern in entrypoint/agent-mcp's AgentStore (agent-cache-store pattern).
+// Constructor accepts a BetterSQLite3Database so tests can inject their own
+// connection without touching the production singleton in client.ts.
+// ──────────────────────────────────────────────
+export class ToolStore {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    // ── tool_types ────────────────────────────
+    /** Seed or upsert a tool-type lookup row. */
+    seedToolType(input) {
+        this.db
+            .insert(toolTypesTable)
+            .values({ slug: input.slug, description: input.description })
+            .onConflictDoNothing()
+            .run();
+    }
+    /** Return all seeded tool types. */
+    listToolTypes() {
+        return this.db.select().from(toolTypesTable).all();
+    }
+    // ── tools ─────────────────────────────────
+    /**
+     * Create a canonical tool row.
+     * Throws TOOL_ALREADY_EXISTS if the name PK already exists.
+     */
+    create(input) {
+        const existing = this.db
+            .select()
+            .from(toolsTable)
+            .where(eq(toolsTable.name, input.name))
+            .get();
+        if (existing) {
+            throw new ToolStoreError('TOOL_ALREADY_EXISTS', `Tool '${input.name}' already exists`);
+        }
+        const tool = {
+            name: input.name,
+            type: input.type,
+            description: input.description,
+            version: input.version ?? 1,
+            requiresApproval: input.requiresApproval ?? false,
+            isDestructive: input.isDestructive ?? false,
+            dependencyToolIds: input.dependencyToolIds ?? [],
+            capabilities: input.capabilities ?? [],
+        };
+        this.db
+            .insert(toolsTable)
+            .values({
+            name: tool.name,
+            type: tool.type,
+            description: tool.description,
+            version: tool.version,
+            requiresApproval: tool.requiresApproval,
+            isDestructive: tool.isDestructive,
+            dependencyToolIds: tool.dependencyToolIds,
+            capabilities: tool.capabilities,
+        })
+            .run();
+        return tool;
+    }
+    /**
+     * Read a canonical tool by name.
+     * Throws TOOL_NOT_FOUND if absent.
+     */
+    read(name) {
+        const row = this.db
+            .select()
+            .from(toolsTable)
+            .where(eq(toolsTable.name, name))
+            .get();
+        if (!row) {
+            throw new ToolStoreError('TOOL_NOT_FOUND', `Tool '${name}' not found`);
+        }
+        return {
+            name: row.name,
+            type: row.type,
+            description: row.description,
+            version: row.version,
+            requiresApproval: row.requiresApproval,
+            isDestructive: row.isDestructive,
+            dependencyToolIds: row.dependencyToolIds,
+            capabilities: row.capabilities,
+        };
+    }
+    /** List all canonical tools. */
+    list() {
+        const rows = this.db.select().from(toolsTable).all();
+        return rows.map((row) => ({
+            name: row.name,
+            type: row.type,
+            description: row.description,
+            version: row.version,
+            requiresApproval: row.requiresApproval,
+            isDestructive: row.isDestructive,
+            dependencyToolIds: row.dependencyToolIds,
+            capabilities: row.capabilities,
+        }));
+    }
+}
+//# sourceMappingURL=tool-store.js.map

package/src/store/tool-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-store.js","sourceRoot":"","sources":["../../../../../../packages/agent/agent-store-tools/src/store/tool-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAIjC,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AA8C7D,MAAM,OAAO,cAAe,SAAQ,KAAK;IACX;IAA5B,YAA4B,IAAwB,EAAE,OAAe;QACnE,KAAK,CAAC,OAAO,CAAC,CAAC;QADW,SAAI,GAAJ,IAAI,CAAoB;QAElD,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;IAC/B,CAAC;CACF;AAED,iDAAiD;AACjD,YAAY;AACZ,EAAE;AACF,kDAAkD;AAClD,wFAAwF;AACxF,4EAA4E;AAC5E,qEAAqE;AACrE,iDAAiD;AAEjD,MAAM,OAAO,SAAS;IAED;IADnB,YACmB,EAAgD;QAAhD,OAAE,GAAF,EAAE,CAA8C;IAChE,CAAC;IAEJ,6CAA6C;IAE7C,6CAA6C;IAC7C,YAAY,CAAC,KAAe;QAC1B,IAAI,CAAC,EAAE;aACJ,MAAM,CAAC,cAAc,CAAC;aACtB,MAAM,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;aAC5D,mBAAmB,EAAE;aACrB,GAAG,EAAE,CAAC;IACX,CAAC;IAED,oCAAoC;IACpC,aAAa;QACX,OAAO,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,GAAG,EAAE,CAAC;IACrD,CAAC;IAED,6CAA6C;IAE7C;;;OAGG;IACH,MAAM,CAAC,KAAsB;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE;aACrB,MAAM,EAAE;aACR,IAAI,CAAC,UAAU,CAAC;aAChB,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;aACtC,GAAG,EAAE,CAAC;QAET,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,IAAI,cAAc,CACtB,qBAAqB,EACrB,SAAS,KAAK,CAAC,IAAI,kBAAkB,CACtC,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAS;YACjB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,CAAC;YAC3B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB,IAAI,KAAK;YACjD,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,KAAK;YAC3C,iBAAiB,EAAE,KAAK,CAAC,iBAAiB,IAAI,EAAE;YAChD,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,EAAE;SACvC,CAAC;QAEF,IAAI,CAAC,EAAE;aACJ,MAAM,CAAC,UAAU,CAAC;aAClB,MAAM,CAAC;YACN,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,YAAY,EAAE,IAAI,CAAC,YAAY;SAChC,CAAC;aACD,GAAG,EAAE,CAAC;QAET,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;OAGG;IACH,IAAI,CAAC,IAAY;QACf,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE;aAChB,MAAM,EAAE;aACR,IAAI,CAAC,UAAU,CAAC;aAChB,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;aAChC,GAAG,EAAE,CAAC;QAET,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,cAAc,CAAC,gBAAgB,EAAE,SAAS,IAAI,aAAa,CAAC,CAAC;QACzE,CAAC;QAED,OAAO;YACL,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;YACtC,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,iBAAiB,EAAE,GAAG,CAAC,iBAA6B;YACpD,YAAY,EAAE,GAAG,CAAC,YAAwB;SAC3C,CAAC;IACJ,CAAC;IAED,gCAAgC;IAChC,IAAI;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,GAAG,EAAE,CAAC;QACrD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;YACtC,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,iBAAiB,EAAE,GAAG,CAAC,iBAA6B;YACpD,YAAY,EAAE,GAAG,CAAC,YAAwB;SAC3C,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}