kiokuko-dsh 0.1.0

package/dist/mcp/runtime-owner.js

@@ -0,0 +1,129 @@
+import { initializeDatabase } from '../commands/init.js';
+import { getGlobalDatabasePath } from '../config/paths.js';
+import { openConnection } from '../db/connection.js';
+import { KiokukoError } from '../errors.js';
+import { openEmbeddingDatabase } from '../embedding/backend.js';
+import { createEmbeddingRuntime } from '../embedding/runtime.js';
+import { createEmbeddingWorker } from '../embedding/worker.js';
+import { WriteQueue } from '../server/write-queue.js';
+/** Own one MCP-process database/runtime instead of opening a connection per tool call. */
+export class McpRuntimeOwner {
+    #options;
+    #initialization;
+    #state;
+    #closing = false;
+    #closed = false;
+    #closePromise;
+    constructor(options = {}) {
+        this.#options = options;
+    }
+    async #initialize() {
+        const databasePath = this.#options.databasePath ?? getGlobalDatabasePath(this.#options);
+        const initialize = this.#options.initializeDatabase ?? initializeDatabase;
+        await initialize({
+            databasePath,
+            ...(this.#options.migrationsDirectory === undefined ? {} : { migrationsDirectory: this.#options.migrationsDirectory }),
+        });
+        const config = this.#options.embeddingConfig;
+        const opened = await openEmbeddingDatabase(databasePath, {
+            ...(config === undefined ? {} : { config }),
+            openDatabase: this.#options.openDatabase ?? openConnection,
+            ...(this.#options.embeddingBackend === undefined ? {} : { backend: this.#options.embeddingBackend }),
+        });
+        const database = opened.database;
+        const queue = new WriteQueue(64);
+        try {
+            const runtime = createEmbeddingRuntime(database, config, {
+                ...(this.#options.embeddingProvider === undefined ? {} : { provider: this.#options.embeddingProvider }),
+                ...(opened.backend === undefined ? {} : { backend: opened.backend }),
+                enqueueWrite: (operation) => queue.enqueue(operation),
+            });
+            const worker = runtime.profileId === null ? undefined : createEmbeddingWorker({ runtime });
+            const state = { database, runtime, worker, queue };
+            this.#state = state;
+            worker?.start();
+            return state;
+        }
+        catch (error) {
+            try {
+                await queue.close();
+            }
+            catch (closeError) {
+                throw new AggregateError([error, closeError], 'MCP runtime initialization failed and its write queue could not be closed');
+            }
+            try {
+                database.close();
+            }
+            catch (closeError) {
+                throw new AggregateError([error, closeError], 'MCP runtime initialization failed and its database could not be closed');
+            }
+            throw error;
+        }
+    }
+    async #ensureState() {
+        if (this.#closed || this.#closing)
+            throw new KiokukoError('SERVICE_UNAVAILABLE', 'MCP runtime is closed');
+        if (this.#state !== undefined)
+            return this.#state;
+        if (this.#initialization === undefined) {
+            const initialization = this.#initialize();
+            this.#initialization = initialization;
+            void initialization.then(undefined, () => {
+                if (this.#initialization === initialization)
+                    this.#initialization = undefined;
+            });
+        }
+        return this.#initialization;
+    }
+    async withDatabase(operation) {
+        const state = await this.#ensureState();
+        return operation(state.database, state.runtime);
+    }
+    async close() {
+        if (this.#closePromise !== undefined)
+            return this.#closePromise;
+        this.#closing = true;
+        const attempt = (async () => {
+            const state = this.#state ?? await this.#initialization;
+            if (state === undefined) {
+                this.#closed = true;
+                return;
+            }
+            const errors = [];
+            try {
+                await state.worker?.close();
+            }
+            catch (error) {
+                errors.push(error);
+            }
+            if (state.worker === undefined) {
+                try {
+                    await state.runtime.close();
+                }
+                catch (error) {
+                    errors.push(error);
+                }
+            }
+            try {
+                await state.queue.close();
+            }
+            catch (error) {
+                errors.push(error);
+            }
+            try {
+                state.database.close();
+            }
+            catch (error) {
+                errors.push(error);
+            }
+            this.#closed = true;
+            if (errors.length === 1)
+                throw errors[0];
+            if (errors.length > 1)
+                throw new AggregateError(errors, 'MCP runtime cleanup failed');
+        })();
+        this.#closePromise = attempt;
+        return attempt;
+    }
+}
+//# sourceMappingURL=runtime-owner.js.map

package/dist/mcp/runtime-owner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime-owner.js","sourceRoot":"","sources":["../../src/mcp/runtime-owner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAoB,MAAM,qBAAqB,CAAC;AAC3E,OAAO,EAAE,qBAAqB,EAAwB,MAAM,oBAAoB,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,qBAAqB,EAAgC,MAAM,yBAAyB,CAAC;AAC9F,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,OAAO,EAAE,qBAAqB,EAAwB,MAAM,wBAAwB,CAAC;AACrF,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AA6BtD,0FAA0F;AAC1F,MAAM,OAAO,eAAe;IACjB,QAAQ,CAAyB;IAC1C,eAAe,CAAkC;IACjD,MAAM,CAAyB;IAC/B,QAAQ,GAAG,KAAK,CAAC;IACjB,OAAO,GAAG,KAAK,CAAC;IAChB,aAAa,CAA4B;IAEzC,YAAY,UAAkC,EAAE;QAC9C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxF,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,IAAI,kBAAkB,CAAC;QAC1E,MAAM,UAAU,CAAC;YACf,YAAY;YACZ,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC;SACvH,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC;QAC7C,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,YAAY,EAAE;YACvD,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YAC3C,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,cAAc;YAC1D,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC;SACrG,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAU,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,MAAM,EAAE;gBACvD,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC;gBACvG,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpE,YAAY,EAAE,CAAI,SAAmC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAe;aACjG,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;YAC3F,MAAM,KAAK,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAuB,CAAC;YACxE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,MAAM,EAAE,KAAK,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;YACtB,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,MAAM,IAAI,cAAc,CAAC,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,2EAA2E,CAAC,CAAC;YAC7H,CAAC;YACD,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,MAAM,IAAI,cAAc,CAAC,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,wEAAwE,CAAC,CAAC;YAC1H,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,YAAY,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,CAAC;QAC1G,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC;QAClD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACvC,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC1C,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;YACtC,KAAK,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;gBACvC,IAAI,IAAI,CAAC,eAAe,KAAK,cAAc;oBAAE,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;YAChF,CAAC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,YAAY,CAAI,SAAkC;QACtD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,aAAa,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,aAAa,CAAC;QAChE,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;YAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC;YACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;gBACpB,OAAO;YACT,CAAC;YACD,MAAM,MAAM,GAAc,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;YAC9B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC9B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,CAAC;gBACH,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACzB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,MAAM,MAAM,CAAC,CAAC,CAAC,CAAC;YACzC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,IAAI,cAAc,CAAC,MAAM,EAAE,4BAA4B,CAAC,CAAC;QACxF,CAAC,CAAC,EAAE,CAAC;QACL,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/mcp/server.d.ts

@@ -0,0 +1,24 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { type InitOptions } from '../commands/init.js';
+import { openConnection } from '../db/connection.js';
+import type { SqliteDatabase } from '../db/adapter.js';
+import type { EmbeddingProvider, EmbeddingRuntime, VectorSearchBackend } from '../embedding/types.js';
+import { type McpDatabaseOwner } from './runtime-owner.js';
+import { type McpDeadlinePolicyOverrides } from './request-deadline.js';
+export interface McpServerDependencies {
+    databasePath?: string;
+    migrationsDirectory?: string;
+    cwd?: () => string;
+    openConnection?: typeof openConnection;
+    initializeDatabase?: (options: InitOptions) => unknown | PromiseLike<unknown>;
+    fetchImpl?: typeof fetch;
+    embeddingEnvironment?: NodeJS.ProcessEnv;
+    embeddingProvider?: EmbeddingProvider;
+    embeddingBackend?: VectorSearchBackend;
+    databaseOwner?: McpDatabaseOwner;
+    deadlinePolicy?: McpDeadlinePolicyOverrides;
+}
+export declare function withDatabase<T>(dependencies: McpServerDependencies, operation: (database: SqliteDatabase, runtime?: EmbeddingRuntime) => Promise<T> | T): Promise<T>;
+export declare function createKiokukoMcpServer(dependencies?: McpServerDependencies): McpServer;
+export declare function runMcpServer(dependencies?: McpServerDependencies): Promise<void>;
+//# sourceMappingURL=server.d.ts.map

package/dist/mcp/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAsB,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AA8CvD,OAAO,KAAK,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AACtG,OAAO,EAAmB,KAAK,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC5E,OAAO,EAML,KAAK,0BAA0B,EAEhC,MAAM,uBAAuB,CAAC;AAE/B,MAAM,WAAW,qBAAqB;IACpC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,GAAG,CAAC,EAAE,MAAM,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,OAAO,cAAc,CAAC;IACvC,kBAAkB,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IAC9E,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;IACzB,oBAAoB,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC;IACzC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC;IACvC,aAAa,CAAC,EAAE,gBAAgB,CAAC;IACjC,cAAc,CAAC,EAAE,0BAA0B,CAAC;CAC7C;AAED,wBAAsB,YAAY,CAAC,CAAC,EAClC,YAAY,EAAE,qBAAqB,EACnC,SAAS,EAAE,CAAC,QAAQ,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,gBAAgB,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GAClF,OAAO,CAAC,CAAC,CAAC,CAoCZ;AAyPD,wBAAgB,sBAAsB,CAAC,YAAY,GAAE,qBAA0B,GAAG,SAAS,CA0L1F;AAED,wBAAsB,YAAY,CAAC,YAAY,GAAE,qBAA0B,GAAG,OAAO,CAAC,IAAI,CAAC,CA2B1F"}