@cyanheads/mcp-ts-core 0.1.0-beta.12

package/dist/mcp-server/server.js

@@ -0,0 +1,62 @@
+/**
+ * @fileoverview Factory for creating configured MCP server instances.
+ * Creates an McpServer with identity, capabilities, and registered
+ * tools/resources/prompts/roots from the provided registries.
+ *
+ * MCP Specification References:
+ * - Lifecycle: https://modelcontextprotocol.io/specification/2025-06-18/basic/lifecycle
+ * - Overview (Capabilities): https://modelcontextprotocol.io/specification/2025-06-18/basic/index
+ * - Transports: https://modelcontextprotocol.io/specification/2025-06-18/basic/transports
+ * @module src/mcp-server/server
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { logger } from '../utils/internal/logger.js';
+import { requestContextService } from '../utils/internal/requestContext.js';
+/**
+ * Creates and configures a new instance of the `McpServer`.
+ * Registries are provided directly — no DI container resolution.
+ *
+ * @returns A promise resolving with the configured `McpServer` instance.
+ * @throws {McpError} If any resource or tool registration fails.
+ */
+export async function createMcpServerInstance(deps) {
+    const context = requestContextService.createRequestContext({
+        operation: 'createMcpServerInstance',
+    });
+    logger.info('Initializing MCP server instance', context);
+    const server = new McpServer({
+        name: deps.config.mcpServerName,
+        version: deps.config.mcpServerVersion,
+    }, {
+        capabilities: {
+            logging: {},
+            resources: { listChanged: true },
+            tools: { listChanged: true },
+            prompts: { listChanged: true },
+            // Experimental: Tasks API for long-running async operations
+            tasks: {
+                list: {},
+                cancel: {},
+                requests: {
+                    tools: { call: {} },
+                },
+            },
+        },
+    });
+    try {
+        logger.debug('Registering all MCP capabilities via registries...', context);
+        await Promise.all([
+            deps.toolRegistry.registerAll(server),
+            deps.resourceRegistry.registerAll(server),
+            deps.promptRegistry.registerAll(server),
+        ]);
+        deps.rootsRegistry.registerAll(server);
+        logger.info('All MCP capabilities registered successfully', context);
+    }
+    catch (err) {
+        logger.error('Failed to register MCP capabilities', err instanceof Error ? err : new Error(String(err)), context);
+        throw err;
+    }
+    return server;
+}
+//# sourceMappingURL=server.js.map

package/dist/mcp-server/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/mcp-server/server.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAOpE,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAW3E;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,IAAmB;IAC/D,MAAM,OAAO,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;QACzD,SAAS,EAAE,yBAAyB;KACrC,CAAC,CAAC;IACH,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,OAAO,CAAC,CAAC;IAEzD,MAAM,MAAM,GAAG,IAAI,SAAS,CAC1B;QACE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;QAC/B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;KACtC,EACD;QACE,YAAY,EAAE;YACZ,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;YAChC,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;YAC5B,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;YAC9B,4DAA4D;YAC5D,KAAK,EAAE;gBACL,IAAI,EAAE,EAAE;gBACR,MAAM,EAAE,EAAE;gBACV,QAAQ,EAAE;oBACR,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;iBACpB;aACF;SACF;KACF,CACF,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,oDAAoD,EAAE,OAAO,CAAC,CAAC;QAE5E,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC;YACrC,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,MAAM,CAAC;YACzC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAEvC,MAAM,CAAC,IAAI,CAAC,8CAA8C,EAAE,OAAO,CAAC,CAAC;IACvE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,KAAK,CACV,qCAAqC,EACrC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EACnD,OAAO,CACR,CAAC;QACF,MAAM,GAAG,CAAC;IACZ,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/mcp-server/tasks/core/sessionAwareTaskStore.d.ts

@@ -0,0 +1,42 @@
+/**
+ * @fileoverview A TaskStore wrapper that enforces session ownership on top of
+ * the SDK's InMemoryTaskStore (which ignores sessionId parameters).
+ *
+ * This ensures that in HTTP mode, tasks created by one session cannot be
+ * accessed by another session, even when using the in-memory store.
+ *
+ * @experimental These APIs are experimental and may change without notice.
+ * @module src/mcp-server/tasks/core/sessionAwareTaskStore
+ */
+import type { Request, RequestId, Result } from '@modelcontextprotocol/sdk/types.js';
+import type { CreateTaskOptions, Task, TaskStore } from './taskTypes.js';
+/**
+ * Wraps an InMemoryTaskStore to add session ownership enforcement.
+ *
+ * Tracks which session created each task and rejects access from
+ * non-owning sessions. Tasks created without a sessionId are accessible
+ * by any session (backwards-compatible with stdio/unauth flows).
+ *
+ * @experimental
+ */
+export declare class SessionAwareTaskStore implements TaskStore {
+    private readonly inner;
+    /** Maps taskId -> sessionId that created it */
+    private readonly ownership;
+    constructor(inner: TaskStore);
+    createTask(taskParams: CreateTaskOptions, requestId: RequestId, request: Request, sessionId?: string): Promise<Task>;
+    getTask(taskId: string, sessionId?: string): Promise<Task | null>;
+    storeTaskResult(taskId: string, status: 'completed' | 'failed', result: Result, sessionId?: string): Promise<void>;
+    getTaskResult(taskId: string, sessionId?: string): Promise<Result>;
+    updateTaskStatus(taskId: string, status: Task['status'], statusMessage?: string, sessionId?: string): Promise<void>;
+    listTasks(cursor?: string, sessionId?: string): Promise<{
+        tasks: Task[];
+        nextCursor?: string;
+    }>;
+    /**
+     * Validates that the caller's session matches the task's owner.
+     * Tasks created without a sessionId are accessible by any session.
+     */
+    private assertOwnership;
+}
+//# sourceMappingURL=sessionAwareTaskStore.d.ts.map

package/dist/mcp-server/tasks/core/sessionAwareTaskStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionAwareTaskStore.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/sessionAwareTaskStore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAGrF,OAAO,KAAK,EAAE,iBAAiB,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAGzE;;;;;;;;GAQG;AACH,qBAAa,qBAAsB,YAAW,SAAS;IAIzC,OAAO,CAAC,QAAQ,CAAC,KAAK;IAHlC,+CAA+C;IAC/C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA6B;gBAE1B,KAAK,EAAE,SAAS;IAEvC,UAAU,CACd,UAAU,EAAE,iBAAiB,EAC7B,SAAS,EAAE,SAAS,EACpB,OAAO,EAAE,OAAO,EAChB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAQhB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAK3D,eAAe,CACnB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,WAAW,GAAG,QAAQ,EAC9B,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAKhB,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK5D,gBAAgB,CACpB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,EACtB,aAAa,CAAC,EAAE,MAAM,EACtB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IASV,SAAS,CACb,MAAM,CAAC,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC;QAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAWlD;;;OAGG;IACH,OAAO,CAAC,eAAe;CAOxB"}

package/dist/mcp-server/tasks/core/sessionAwareTaskStore.js

@@ -0,0 +1,70 @@
+import { JsonRpcErrorCode, McpError } from '../../../types-global/errors.js';
+import { isTerminal } from './taskTypes.js';
+/**
+ * Wraps an InMemoryTaskStore to add session ownership enforcement.
+ *
+ * Tracks which session created each task and rejects access from
+ * non-owning sessions. Tasks created without a sessionId are accessible
+ * by any session (backwards-compatible with stdio/unauth flows).
+ *
+ * @experimental
+ */
+export class SessionAwareTaskStore {
+    inner;
+    /** Maps taskId -> sessionId that created it */
+    ownership = new Map();
+    constructor(inner) {
+        this.inner = inner;
+    }
+    async createTask(taskParams, requestId, request, sessionId) {
+        const task = await this.inner.createTask(taskParams, requestId, request, sessionId);
+        if (sessionId) {
+            this.ownership.set(task.taskId, sessionId);
+        }
+        return task;
+    }
+    getTask(taskId, sessionId) {
+        this.assertOwnership(taskId, sessionId);
+        return this.inner.getTask(taskId, sessionId);
+    }
+    async storeTaskResult(taskId, status, result, sessionId) {
+        this.assertOwnership(taskId, sessionId);
+        await this.inner.storeTaskResult(taskId, status, result, sessionId);
+    }
+    getTaskResult(taskId, sessionId) {
+        this.assertOwnership(taskId, sessionId);
+        return this.inner.getTaskResult(taskId, sessionId);
+    }
+    async updateTaskStatus(taskId, status, statusMessage, sessionId) {
+        this.assertOwnership(taskId, sessionId);
+        await this.inner.updateTaskStatus(taskId, status, statusMessage, sessionId);
+        // Clean up ownership tracking when task reaches terminal state
+        if (isTerminal(status)) {
+            this.ownership.delete(taskId);
+        }
+    }
+    async listTasks(cursor, sessionId) {
+        const result = await this.inner.listTasks(cursor, sessionId);
+        if (!sessionId)
+            return result;
+        // Filter to only tasks owned by this session (or unowned tasks)
+        const filtered = result.tasks.filter((task) => {
+            const owner = this.ownership.get(task.taskId);
+            return !owner || owner === sessionId;
+        });
+        return { tasks: filtered, ...(result.nextCursor ? { nextCursor: result.nextCursor } : {}) };
+    }
+    /**
+     * Validates that the caller's session matches the task's owner.
+     * Tasks created without a sessionId are accessible by any session.
+     */
+    assertOwnership(taskId, callerSessionId) {
+        const owner = this.ownership.get(taskId);
+        if (!owner)
+            return; // No owner recorded — accessible by anyone
+        if (owner !== callerSessionId) {
+            throw new McpError(JsonRpcErrorCode.Forbidden, `Access denied to task ${taskId}`);
+        }
+    }
+}
+//# sourceMappingURL=sessionAwareTaskStore.js.map

package/dist/mcp-server/tasks/core/sessionAwareTaskStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionAwareTaskStore.js","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/sessionAwareTaskStore.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5C;;;;;;;;GAQG;AACH,MAAM,OAAO,qBAAqB;IAIH;IAH7B,+CAA+C;IAC9B,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEvD,YAA6B,KAAgB;QAAhB,UAAK,GAAL,KAAK,CAAW;IAAG,CAAC;IAEjD,KAAK,CAAC,UAAU,CACd,UAA6B,EAC7B,SAAoB,EACpB,OAAgB,EAChB,SAAkB;QAElB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QACpF,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CAAC,MAAc,EAAE,SAAkB;QACxC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,MAA8B,EAC9B,MAAc,EACd,SAAkB;QAElB,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACxC,MAAM,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IACtE,CAAC;IAED,aAAa,CAAC,MAAc,EAAE,SAAkB;QAC9C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,MAAc,EACd,MAAsB,EACtB,aAAsB,EACtB,SAAkB;QAElB,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACxC,MAAM,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QAC5E,+DAA+D;QAC/D,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CACb,MAAe,EACf,SAAkB;QAElB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC7D,IAAI,CAAC,SAAS;YAAE,OAAO,MAAM,CAAC;QAC9B,gEAAgE;QAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9C,OAAO,CAAC,KAAK,IAAI,KAAK,KAAK,SAAS,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;IAC9F,CAAC;IAED;;;OAGG;IACK,eAAe,CAAC,MAAc,EAAE,eAAmC;QACzE,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,2CAA2C;QAC/D,IAAI,KAAK,KAAK,eAAe,EAAE,CAAC;YAC9B,MAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,SAAS,EAAE,yBAAyB,MAAM,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;CACF"}

package/dist/mcp-server/tasks/core/storageBackedTaskStore.d.ts

@@ -0,0 +1,109 @@
+/**
+ * @fileoverview A TaskStore implementation backed by the template's StorageService.
+ * This provides persistent task storage that survives server restarts, using
+ * whatever storage backend is configured (filesystem, Supabase, SurrealDB, etc.).
+ *
+ * Use this instead of InMemoryTaskStore when you need task durability.
+ *
+ * @experimental These APIs are experimental and may change without notice.
+ * @module src/mcp-server/tasks/core/storageBackedTaskStore
+ */
+import type { Request, RequestId, Result } from '@modelcontextprotocol/sdk/types.js';
+import type { StorageService } from '../../../storage/core/StorageService.js';
+import type { CreateTaskOptions, Task, TaskStore } from './taskTypes.js';
+/**
+ * Configuration options for StorageBackedTaskStore.
+ */
+export interface StorageBackedTaskStoreOptions {
+    /**
+     * Default TTL in milliseconds if not specified in task creation.
+     * Set to null for unlimited lifetime.
+     * @default null
+     */
+    defaultTtl?: number | null;
+    /**
+     * Prefix for storage keys.
+     * @default 'tasks'
+     */
+    keyPrefix?: string;
+    /**
+     * Page size for listTasks pagination.
+     * @default 10
+     */
+    pageSize?: number;
+    /**
+     * The tenant ID to use for storage operations.
+     * Tasks are stored under this tenant for isolation.
+     * @default 'system-tasks'
+     */
+    tenantId?: string;
+}
+/**
+ * A TaskStore implementation backed by StorageService for persistent task storage.
+ *
+ * Unlike InMemoryTaskStore, this implementation:
+ * - Persists tasks across server restarts
+ * - Works with any configured storage backend (filesystem, Supabase, SurrealDB, etc.)
+ * - Supports TTL via storage provider options
+ *
+ * @example
+ * ```typescript
+ * const taskStore = new StorageBackedTaskStore(storageService, {
+ *   tenantId: 'my-app-tasks',
+ *   defaultTtl: 3600000, // 1 hour
+ * });
+ * ```
+ *
+ * @experimental
+ */
+export declare class StorageBackedTaskStore implements TaskStore {
+    private readonly storage;
+    private readonly tenantId;
+    private readonly keyPrefix;
+    private readonly defaultTtl;
+    private readonly pageSize;
+    constructor(storage: StorageService, options?: StorageBackedTaskStoreOptions);
+    /**
+     * Creates a request context for storage operations.
+     */
+    private createContext;
+    /**
+     * Generates the storage key for a task.
+     */
+    private getTaskKey;
+    /**
+     * Validates that the caller's sessionId matches the stored task's sessionId.
+     * Tasks created without a sessionId (legacy/pre-ownership) are accessible by any session.
+     * @throws {McpError} Forbidden if sessionId mismatch
+     */
+    private assertOwnership;
+    /**
+     * Generates a unique task ID.
+     * Uses the template's idGenerator for consistent ID format.
+     */
+    private generateTaskId;
+    createTask(taskParams: CreateTaskOptions, requestId: RequestId, request: Request, sessionId?: string): Promise<Task>;
+    getTask(taskId: string, sessionId?: string): Promise<Task | null>;
+    storeTaskResult(taskId: string, status: 'completed' | 'failed', result: Result, sessionId?: string): Promise<void>;
+    getTaskResult(taskId: string, sessionId?: string): Promise<Result>;
+    updateTaskStatus(taskId: string, status: Task['status'], statusMessage?: string, sessionId?: string): Promise<void>;
+    listTasks(cursor?: string, sessionId?: string): Promise<{
+        tasks: Task[];
+        nextCursor?: string;
+    }>;
+    /**
+     * Deletes a task from storage.
+     * Useful for manual cleanup or testing.
+     *
+     * @param taskId - The task ID to delete
+     */
+    deleteTask(taskId: string): Promise<void>;
+    /**
+     * Clears all tasks from storage.
+     * Useful for testing or administrative purposes.
+     *
+     * WARNING: This deletes ALL tasks for the configured tenant.
+     */
+    clearAllTasks(): Promise<void>;
+}
+//# sourceMappingURL=storageBackedTaskStore.d.ts.map

package/dist/mcp-server/tasks/core/storageBackedTaskStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storageBackedTaskStore.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/storageBackedTaskStore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAErF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAIvE,OAAO,KAAK,EAAE,iBAAiB,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAgBzE;;GAEG;AACH,MAAM,WAAW,6BAA6B;IAC5C;;;;OAIG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE3B;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,sBAAuB,YAAW,SAAS;IAOpD,OAAO,CAAC,QAAQ,CAAC,OAAO;IAN1B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAgB;IAC3C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;gBAGf,OAAO,EAAE,cAAc,EACxC,OAAO,GAAE,6BAAkC;IAQ7C;;OAEG;IACH,OAAO,CAAC,aAAa;IASrB;;OAEG;IACH,OAAO,CAAC,UAAU;IAIlB;;;;OAIG;IACH,OAAO,CAAC,eAAe;IAWvB;;;OAGG;IACH,OAAO,CAAC,cAAc;IAKhB,UAAU,CACd,UAAU,EAAE,iBAAiB,EAC7B,SAAS,EAAE,SAAS,EACpB,OAAO,EAAE,OAAO,EAChB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAsCV,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAQjE,eAAe,CACnB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,WAAW,GAAG,QAAQ,EAC9B,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IA+BV,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAgBlE,gBAAgB,CACpB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,EACtB,aAAa,CAAC,EAAE,MAAM,EACtB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAkCV,SAAS,CACb,MAAM,CAAC,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC;QAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAiClD;;;;;OAKG;IACG,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK/C;;;;;OAKG;IACG,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;CAiBrC"}

package/dist/mcp-server/tasks/core/storageBackedTaskStore.js

@@ -0,0 +1,209 @@
+import { JsonRpcErrorCode, McpError } from '../../../types-global/errors.js';
+import { idGenerator } from '../../../utils/security/idGenerator.js';
+import { isTerminal } from './taskTypes.js';
+/**
+ * A TaskStore implementation backed by StorageService for persistent task storage.
+ *
+ * Unlike InMemoryTaskStore, this implementation:
+ * - Persists tasks across server restarts
+ * - Works with any configured storage backend (filesystem, Supabase, SurrealDB, etc.)
+ * - Supports TTL via storage provider options
+ *
+ * @example
+ * ```typescript
+ * const taskStore = new StorageBackedTaskStore(storageService, {
+ *   tenantId: 'my-app-tasks',
+ *   defaultTtl: 3600000, // 1 hour
+ * });
+ * ```
+ *
+ * @experimental
+ */
+export class StorageBackedTaskStore {
+    storage;
+    tenantId;
+    keyPrefix;
+    defaultTtl;
+    pageSize;
+    constructor(storage, options = {}) {
+        this.storage = storage;
+        this.tenantId = options.tenantId ?? 'system-tasks';
+        this.keyPrefix = options.keyPrefix ?? 'tasks';
+        this.defaultTtl = options.defaultTtl ?? null;
+        this.pageSize = options.pageSize ?? 10;
+    }
+    /**
+     * Creates a request context for storage operations.
+     */
+    createContext(operation) {
+        return {
+            operation: `StorageBackedTaskStore.${operation}`,
+            requestId: idGenerator.generate('req'),
+            timestamp: new Date().toISOString(),
+            tenantId: this.tenantId,
+        };
+    }
+    /**
+     * Generates the storage key for a task.
+     */
+    getTaskKey(taskId) {
+        return `${this.keyPrefix}/${taskId}`;
+    }
+    /**
+     * Validates that the caller's sessionId matches the stored task's sessionId.
+     * Tasks created without a sessionId (legacy/pre-ownership) are accessible by any session.
+     * @throws {McpError} Forbidden if sessionId mismatch
+     */
+    assertOwnership(stored, callerSessionId, taskId) {
+        if (!stored.sessionId)
+            return;
+        if (stored.sessionId !== callerSessionId) {
+            throw new McpError(JsonRpcErrorCode.Forbidden, `Access denied to task ${taskId}`);
+        }
+    }
+    /**
+     * Generates a unique task ID.
+     * Uses the template's idGenerator for consistent ID format.
+     */
+    generateTaskId() {
+        // Generate a longer ID for global uniqueness (16 chars)
+        return idGenerator.generate('task', { length: 16 });
+    }
+    async createTask(taskParams, requestId, request, sessionId) {
+        const context = this.createContext('createTask');
+        const taskId = this.generateTaskId();
+        const actualTtl = taskParams.ttl ?? this.defaultTtl;
+        const createdAt = new Date().toISOString();
+        const task = {
+            taskId,
+            status: 'working',
+            ttl: actualTtl,
+            createdAt,
+            lastUpdatedAt: createdAt,
+            pollInterval: taskParams.pollInterval ?? 1000,
+        };
+        const storedTask = {
+            task,
+            request,
+            requestId,
+        };
+        // Bind session ownership if provided
+        if (sessionId) {
+            storedTask.sessionId = sessionId;
+        }
+        // Store with TTL if specified (convert ms to seconds for StorageService)
+        await this.storage.set(this.getTaskKey(taskId), storedTask, context, actualTtl ? { ttl: Math.ceil(actualTtl / 1000) } : undefined);
+        return task;
+    }
+    async getTask(taskId, sessionId) {
+        const context = this.createContext('getTask');
+        const stored = await this.storage.get(this.getTaskKey(taskId), context);
+        if (!stored)
+            return null;
+        this.assertOwnership(stored, sessionId, taskId);
+        return { ...stored.task };
+    }
+    async storeTaskResult(taskId, status, result, sessionId) {
+        const context = this.createContext('storeTaskResult');
+        const key = this.getTaskKey(taskId);
+        const stored = await this.storage.get(key, context);
+        if (!stored) {
+            throw new McpError(JsonRpcErrorCode.InvalidRequest, `Task with ID ${taskId} not found`);
+        }
+        this.assertOwnership(stored, sessionId, taskId);
+        // Don't allow storing results for tasks already in terminal state
+        if (isTerminal(stored.task.status)) {
+            throw new McpError(JsonRpcErrorCode.InvalidRequest, `Cannot store result for task ${taskId} in terminal status '${stored.task.status}'. Task results can only be stored once.`);
+        }
+        stored.result = result;
+        stored.task.status = status;
+        stored.task.lastUpdatedAt = new Date().toISOString();
+        // Re-store with TTL reset (if ttl is set, convert ms to seconds)
+        await this.storage.set(key, stored, context, stored.task.ttl ? { ttl: Math.ceil(stored.task.ttl / 1000) } : undefined);
+    }
+    async getTaskResult(taskId, sessionId) {
+        const context = this.createContext('getTaskResult');
+        const stored = await this.storage.get(this.getTaskKey(taskId), context);
+        if (!stored) {
+            throw new McpError(JsonRpcErrorCode.InvalidRequest, `Task with ID ${taskId} not found`);
+        }
+        this.assertOwnership(stored, sessionId, taskId);
+        if (!stored.result) {
+            throw new McpError(JsonRpcErrorCode.InvalidRequest, `Task ${taskId} has no result stored`);
+        }
+        return stored.result;
+    }
+    async updateTaskStatus(taskId, status, statusMessage, sessionId) {
+        const context = this.createContext('updateTaskStatus');
+        const key = this.getTaskKey(taskId);
+        const stored = await this.storage.get(key, context);
+        if (!stored) {
+            throw new McpError(JsonRpcErrorCode.InvalidRequest, `Task with ID ${taskId} not found`);
+        }
+        this.assertOwnership(stored, sessionId, taskId);
+        // Don't allow transitions from terminal states
+        if (isTerminal(stored.task.status)) {
+            throw new McpError(JsonRpcErrorCode.InvalidRequest, `Cannot update task ${taskId} from terminal status '${stored.task.status}' to '${status}'. Terminal states (completed, failed, cancelled) cannot transition to other states.`);
+        }
+        stored.task.status = status;
+        if (statusMessage) {
+            stored.task.statusMessage = statusMessage;
+        }
+        stored.task.lastUpdatedAt = new Date().toISOString();
+        // Re-store, reset TTL if transitioning to terminal state (convert ms to seconds)
+        const shouldResetTtl = isTerminal(status) && stored.task.ttl;
+        await this.storage.set(key, stored, context, shouldResetTtl ? { ttl: Math.ceil(stored.task.ttl / 1000) } : undefined);
+    }
+    async listTasks(cursor, sessionId) {
+        const context = this.createContext('listTasks');
+        // List all task keys with pagination
+        const listResult = await this.storage.list(this.keyPrefix, context, cursor ? { cursor, limit: this.pageSize } : { limit: this.pageSize });
+        // Fetch all tasks in parallel, filtering by session ownership
+        const taskPromises = listResult.keys.map(async (key) => {
+            const stored = await this.storage.get(key, context);
+            if (!stored)
+                return null;
+            // Filter: hide tasks whose sessionId doesn't match the caller's.
+            // Consistent with assertOwnership: session-bound tasks are only
+            // visible to the session that created them.
+            if (stored.sessionId && stored.sessionId !== sessionId) {
+                return null;
+            }
+            return { ...stored.task };
+        });
+        const tasksOrNull = await Promise.all(taskPromises);
+        const tasks = tasksOrNull.filter((t) => t !== null);
+        // Handle exactOptionalPropertyTypes - only include nextCursor if defined
+        if (listResult.nextCursor !== undefined) {
+            return { tasks, nextCursor: listResult.nextCursor };
+        }
+        return { tasks };
+    }
+    /**
+     * Deletes a task from storage.
+     * Useful for manual cleanup or testing.
+     *
+     * @param taskId - The task ID to delete
+     */
+    async deleteTask(taskId) {
+        const context = this.createContext('deleteTask');
+        await this.storage.delete(this.getTaskKey(taskId), context);
+    }
+    /**
+     * Clears all tasks from storage.
+     * Useful for testing or administrative purposes.
+     *
+     * WARNING: This deletes ALL tasks for the configured tenant.
+     */
+    async clearAllTasks() {
+        const context = this.createContext('clearAllTasks');
+        // List and delete all tasks
+        let cursor;
+        do {
+            const result = await this.storage.list(this.keyPrefix, context, cursor ? { cursor, limit: 100 } : { limit: 100 });
+            await Promise.all(result.keys.map((key) => this.storage.delete(key, context)));
+            cursor = result.nextCursor;
+        } while (cursor);
+    }
+}
+//# sourceMappingURL=storageBackedTaskStore.js.map

package/dist/mcp-server/tasks/core/storageBackedTaskStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storageBackedTaskStore.js","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/storageBackedTaskStore.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAE9D,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AA6C5C;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,sBAAsB;IAOd;IANF,QAAQ,CAAS;IACjB,SAAS,CAAS;IAClB,UAAU,CAAgB;IAC1B,QAAQ,CAAS;IAElC,YACmB,OAAuB,EACxC,UAAyC,EAAE;QAD1B,YAAO,GAAP,OAAO,CAAgB;QAGxC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,cAAc,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC;QAC9C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,IAAI,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,SAAiB;QACrC,OAAO;YACL,SAAS,EAAE,0BAA0B,SAAS,EAAE;YAChD,SAAS,EAAE,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC;YACtC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,MAAc;QAC/B,OAAO,GAAG,IAAI,CAAC,SAAS,IAAI,MAAM,EAAE,CAAC;IACvC,CAAC;IAED;;;;OAIG;IACK,eAAe,CACrB,MAAkB,EAClB,eAAmC,EACnC,MAAc;QAEd,IAAI,CAAC,MAAM,CAAC,SAAS;YAAE,OAAO;QAC9B,IAAI,MAAM,CAAC,SAAS,KAAK,eAAe,EAAE,CAAC;YACzC,MAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,SAAS,EAAE,yBAAyB,MAAM,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,cAAc;QACpB,wDAAwD;QACxD,OAAO,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,UAAU,CACd,UAA6B,EAC7B,SAAoB,EACpB,OAAgB,EAChB,SAAkB;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAErC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC;QACpD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAE3C,MAAM,IAAI,GAAS;YACjB,MAAM;YACN,MAAM,EAAE,SAAS;YACjB,GAAG,EAAE,SAAS;YACd,SAAS;YACT,aAAa,EAAE,SAAS;YACxB,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,IAAI;SAC9C,CAAC;QAEF,MAAM,UAAU,GAAe;YAC7B,IAAI;YACJ,OAAO;YACP,SAAS;SACV,CAAC;QAEF,qCAAqC;QACrC,IAAI,SAAS,EAAE,CAAC;YACd,UAAU,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,CAAC;QAED,yEAAyE;QACzE,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CACpB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EACvB,UAAU,EACV,OAAO,EACP,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAC7D,CAAC;QAEF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,SAAkB;QAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAa,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;QACpF,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QACzB,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAChD,OAAO,EAAE,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,MAA8B,EAC9B,MAAc,EACd,SAAkB;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;QACtD,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAEpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAa,GAAG,EAAE,OAAO,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB,MAAM,YAAY,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAEhD,kEAAkE;QAClE,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,QAAQ,CAChB,gBAAgB,CAAC,cAAc,EAC/B,gCAAgC,MAAM,wBAAwB,MAAM,CAAC,IAAI,CAAC,MAAM,0CAA0C,CAC3H,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErD,iEAAiE;QACjE,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CACpB,GAAG,EACH,MAAM,EACN,OAAO,EACP,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CACzE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,SAAkB;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAa,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;QAEpF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB,MAAM,YAAY,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAEhD,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,cAAc,EAAE,QAAQ,MAAM,uBAAuB,CAAC,CAAC;QAC7F,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,MAAc,EACd,MAAsB,EACtB,aAAsB,EACtB,SAAkB;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;QACvD,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAEpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAa,GAAG,EAAE,OAAO,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB,MAAM,YAAY,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAEhD,+CAA+C;QAC/C,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,QAAQ,CAChB,gBAAgB,CAAC,cAAc,EAC/B,sBAAsB,MAAM,0BAA0B,MAAM,CAAC,IAAI,CAAC,MAAM,SAAS,MAAM,sFAAsF,CAC9K,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAC5B,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QAC5C,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErD,iFAAiF;QACjF,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;QAC7D,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CACpB,GAAG,EACH,MAAM,EACN,OAAO,EACP,cAAc,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,CAAE,MAAM,CAAC,IAAI,CAAC,GAAc,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CACpF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS,CACb,MAAe,EACf,SAAkB;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;QAEhD,qCAAqC;QACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CACxC,IAAI,CAAC,SAAS,EACd,OAAO,EACP,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,CACrE,CAAC;QAEF,8DAA8D;QAC9D,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAa,GAAG,EAAE,OAAO,CAAC,CAAC;YAChE,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC;YACzB,iEAAiE;YACjE,gEAAgE;YAChE,4CAA4C;YAC5C,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACvD,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,EAAE,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAa,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;QAE/D,yEAAyE;QACzE,IAAI,UAAU,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACxC,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;QACtD,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACjD,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa;QACjB,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;QAEpD,4BAA4B;QAC5B,IAAI,MAA0B,CAAC;QAC/B,GAAG,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CACpC,IAAI,CAAC,SAAS,EACd,OAAO,EACP,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CACjD,CAAC;YAEF,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;YAE/E,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC;QAC7B,CAAC,QAAQ,MAAM,EAAE;IACnB,CAAC;CACF"}

package/dist/mcp-server/tasks/core/taskManager.d.ts

@@ -0,0 +1,103 @@
+/**
+ * @fileoverview Singleton service for managing MCP task state and message queues.
+ * Supports both in-memory and storage-backed task stores based on configuration.
+ *
+ * Configure via environment variables:
+ * - TASK_STORE_TYPE: 'in-memory' (default) or 'storage'
+ * - TASK_STORE_TENANT_ID: Tenant ID for storage isolation (default: 'system-tasks')
+ * - TASK_STORE_DEFAULT_TTL_MS: Default TTL in milliseconds (optional)
+ *
+ * @experimental These APIs are experimental and may change without notice.
+ * @module src/mcp-server/tasks/core/taskManager
+ */
+import type { config as configType } from '../../../config/index.js';
+import type { StorageService } from '../../../storage/core/StorageService.js';
+import type { RequestContext } from '../../../utils/internal/requestContext.js';
+import { type TaskMessageQueue, type TaskStore } from './taskTypes.js';
+/**
+ * Singleton service that manages task state and message queues for the MCP server.
+ *
+ * The TaskManager provides:
+ * - A shared TaskStore for creating, tracking, and completing tasks
+ * - A shared TaskMessageQueue for side-channel message delivery
+ * - Cleanup methods for graceful shutdown
+ *
+ * The store type is determined by configuration:
+ * - `in-memory`: Fast, suitable for development (data lost on restart)
+ * - `storage`: Persistent, uses configured StorageService backend
+ *
+ * @example
+ * ```typescript
+ * // Inject via DI
+ * constructor(@inject(TaskManagerToken) private taskManager: TaskManager) {}
+ *
+ * // Access stores
+ * const taskStore = this.taskManager.getTaskStore();
+ * const messageQueue = this.taskManager.getMessageQueue();
+ * ```
+ *
+ * @experimental
+ */
+export declare class TaskManager {
+    private readonly taskStore;
+    private readonly inMemoryTaskStore;
+    private readonly messageQueue;
+    private readonly storeType;
+    private isShuttingDown;
+    constructor(config: typeof configType, storageService: StorageService);
+    /**
+     * Returns the TaskStore instance for managing task lifecycle.
+     *
+     * The TaskStore handles:
+     * - Task creation with TTL and poll intervals
+     * - Status updates (working, input_required, completed, failed, cancelled)
+     * - Result storage and retrieval
+     * - Task listing with pagination
+     *
+     * @returns The singleton TaskStore instance
+     */
+    getTaskStore(): TaskStore;
+    /**
+     * Returns the TaskMessageQueue instance for side-channel messaging.
+     *
+     * The message queue enables:
+     * - Queuing requests/notifications for delivery via tasks/result
+     * - FIFO ordering per task
+     * - Atomic enqueue with size limits
+     *
+     * @returns The singleton TaskMessageQueue instance
+     */
+    getMessageQueue(): TaskMessageQueue;
+    /**
+     * Returns the store type currently in use.
+     *
+     * @returns 'in-memory' or 'storage'
+     */
+    getStoreType(): 'in-memory' | 'storage';
+    /**
+     * Performs cleanup of task resources.
+     *
+     * Should be called during graceful server shutdown to:
+     * - Cancel cleanup timers in the in-memory task store
+     * - Clear any pending message queues
+     *
+     * Note: Storage-backed task stores don't require cleanup as data persists.
+     *
+     * @param context - Request context for logging
+     */
+    cleanup(context?: RequestContext): void;
+    /**
+     * Returns the current task count (for debugging/monitoring).
+     * Only available for in-memory store; returns `null` for storage-backed store.
+     *
+     * @returns The number of tasks currently tracked, or `null` if unavailable
+     */
+    getTaskCount(): number | null;
+    /**
+     * Checks if the TaskManager is shutting down.
+     *
+     * @returns True if cleanup has been initiated
+     */
+    isCleaningUp(): boolean;
+}
+//# sourceMappingURL=taskManager.d.ts.map

package/dist/mcp-server/tasks/core/taskManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"taskManager.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/taskManager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAEvE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAIzE,OAAO,EAGL,KAAK,gBAAgB,EACrB,KAAK,SAAS,EACf,MAAM,gBAAgB,CAAC;AAExB;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAkC;IACpE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA2B;IACxD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA0B;IACpD,OAAO,CAAC,cAAc,CAAS;gBAEnB,MAAM,EAAE,OAAO,UAAU,EAAE,cAAc,EAAE,cAAc;IAyBrE;;;;;;;;;;OAUG;IACI,YAAY,IAAI,SAAS;IAIhC;;;;;;;;;OASG;IACI,eAAe,IAAI,gBAAgB;IAI1C;;;;OAIG;IACI,YAAY,IAAI,WAAW,GAAG,SAAS;IAI9C;;;;;;;;;;OAUG;IACI,OAAO,CAAC,OAAO,CAAC,EAAE,cAAc,GAAG,IAAI;IAuB9C;;;;;OAKG;IACI,YAAY,IAAI,MAAM,GAAG,IAAI;IAQpC;;;;OAIG;IACI,YAAY,IAAI,OAAO;CAG/B"}