@testomatio/mcp 2.0.0 → 2.0.1-beta.1

package/src/mcp/server.js

@@ -7,14 +7,30 @@ import { createLogger } from '../core/logger.js';
 import { getPackageVersion } from '../config/package-version.js';
 
 export class TestomatioMCPServer {
-  constructor({ config, apiClient, logger }) {
+  constructor({
+    config,
+    apiClient,
+    logger,
+    tools = TOOL_DEFINITIONS,
+    name = 'testomatio-mcp-server',
+    registryOptions = {},
+  }) {
     this.config = config;
+    this.apiClient = apiClient;
     this.logger = logger || createLogger();
-    this.toolRegistry = new ToolRegistry({ config, apiClient, logger: this.logger });
+    this.tools = tools;
+    this.toolRegistry = new ToolRegistry({
+      config,
+      apiClient,
+      logger: this.logger,
+      tools,
+      ...registryOptions,
+    });
+    this.cleanupStarted = false;
 
     this.server = new Server(
       {
-        name: 'testomatio-mcp-server',
+        name,
         version: getPackageVersion(),
       },
       {
@@ -29,7 +45,7 @@ export class TestomatioMCPServer {
 
   setupHandlers() {
     this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
-      tools: TOOL_DEFINITIONS,
+      tools: this.tools,
     }));
 
     this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
@@ -44,6 +60,33 @@ export class TestomatioMCPServer {
   async run() {
     const transport = new StdioServerTransport();
     await this.server.connect(transport);
+    this.installSessionCleanup();
     this.logger.info('Testomatio MCP server started');
   }
+
+  installSessionCleanup() {
+    const cleanup = async () => {
+      if (this.cleanupStarted) {
+        return;
+      }
+
+      this.cleanupStarted = true;
+      await this.apiClient?.stopSession?.();
+    };
+
+    this.server.onclose = () => {
+      void cleanup();
+    };
+
+    process.once('beforeExit', () => {
+      void cleanup();
+    });
+
+    for (const signal of ['SIGINT', 'SIGTERM']) {
+      process.once(signal, async () => {
+        await cleanup();
+        process.exit(0);
+      });
+    }
+  }
 }

package/src/mcp/tool-definitions.js

@@ -8,8 +8,10 @@ import { STEPS_TOOLS } from './definitions/steps.js';
 import { SNIPPETS_TOOLS } from './definitions/snippets.js';
 import { LABELS_TOOLS } from './definitions/labels.js';
 import { TAGS_TOOLS } from './definitions/tags.js';
+import { MILESTONES_TOOLS } from './definitions/milestones.js';
 import { ISSUES_TOOLS } from './definitions/issues.js';
 import { PLANS_TOOLS } from './definitions/plans.js';
+import { REQUIREMENTS_TOOLS } from './definitions/requirements.js';
 
 export const TOOL_DEFINITIONS = [
   ...SYSTEM_TOOLS,
@@ -22,6 +24,8 @@ export const TOOL_DEFINITIONS = [
   ...SNIPPETS_TOOLS,
   ...LABELS_TOOLS,
   ...TAGS_TOOLS,
+  ...MILESTONES_TOOLS,
   ...ISSUES_TOOLS,
   ...PLANS_TOOLS,
+  ...REQUIREMENTS_TOOLS,
 ];

package/src/mcp/tool-registry.js

@@ -12,10 +12,12 @@ function formatJson(payload) {
 }
 
 export class ToolRegistry {
-  constructor({ config, apiClient, logger }) {
+  constructor({ config, apiClient, logger, tools = TOOL_DEFINITIONS, handlerRegistrars = [] }) {
     this.config = config;
     this.apiClient = apiClient;
     this.logger = logger;
+    this.tools = tools;
+    this.handlerRegistrars = handlerRegistrars;
     this.handlers = this.buildHandlers();
   }
 
@@ -37,8 +39,11 @@ export class ToolRegistry {
     this.registerEntityCrudHandlers(handlers);
     this.registerScopedIssueHandlers(handlers);
     this.registerGlobalHandlers(handlers);
+    for (const registerHandlers of this.handlerRegistrars) {
+      registerHandlers.call(this, handlers);
+    }
 
-    for (const tool of TOOL_DEFINITIONS) {
+    for (const tool of this.tools) {
       if (tool.name === 'system_ping') continue;
       if (!handlers[tool.name]) {
         handlers[tool.name] = async () => textResponse(`${DEFAULT_TOOL_RESPONSE} (${tool.name})`);