0xray 3.1.1 → 3.2.0

package/src/mcps/knowledge-skills/log-monitor.server.ts

@@ -6,14 +6,9 @@
  * and performance issues in real-time.
  */
 
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from "@modelcontextprotocol/sdk/types.js";
 import * as fs from "fs";
 import { fileURLToPath } from "url";
+import { XrayKnowledgeSkillBase } from "../shared/knowledge-skill-base.js";
 
 interface LogEntry {
   timestamp: string;
@@ -65,8 +60,7 @@ interface LogAnomaly {
   count: number;
 }
 
-class LogMonitorServer {
-  private server: Server;
+class LogMonitorServer extends XrayKnowledgeSkillBase {
   private patternLibrary: LogPattern[] = [
     {
       id: "memory-leak",
@@ -116,16 +110,8 @@ class LogMonitorServer {
   ];
 
   constructor() {
-    this.server = new Server(
-      { name: "log-monitor", version: "3.1.0" },
-      { capabilities: { tools: {} } },
-    );
-    this.setupToolHandlers();
-  }
-
-  private setupToolHandlers() {
-    this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
-      tools: [
+    super("log-monitor", "3.1.0");
+    this.tools = [
         {
           name: "analyze_logs",
           description:
@@ -238,82 +224,72 @@ class LogMonitorServer {
             required: ["analysis"],
           },
         },
-      ],
-    }));
-
-    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      const { name, arguments: args = {} } = request.params;
-
-      try {
+    ];
+    this.handlers = {
+      "analyze_logs": async (args) => {
         const params = args as Record<string, unknown>;
-        switch (name) {
-          case "analyze_logs": {
-            const result = this.analyzeLogs(
-              (params.logs as string[]) || [],
-              (params.options as Record<string, unknown>) || {},
-            );
-            return {
-              content: [
-                { type: "text", text: JSON.stringify(result, null, 2) },
-              ],
-            };
-          }
-          case "detect_patterns": {
-            const result = this.detectPatterns(
-              (params.logs as string[]) || [],
-              (params.customPatterns as string[]) || [],
-              (params.sensitivity as number) || 80,
-            );
-            return {
-              content: [
-                { type: "text", text: JSON.stringify(result, null, 2) },
-              ],
-            };
-          }
-          case "alert_on_issues": {
-            const result = this.alertOnIssues(
-              params.analysis as unknown as LogAnalysis,
-              (params.thresholds as Record<string, unknown>) || {},
-            );
-            return {
-              content: [
-                { type: "text", text: JSON.stringify(result, null, 2) },
-              ],
-            };
-          }
-          case "correlate_events": {
-            const result = this.correlateEvents(
-              (params.logSets as Array<{ source: string; logs: string[] }>) ||
-                [],
-              (params.timeWindow as number) || 60,
-            );
-            return {
-              content: [
-                { type: "text", text: JSON.stringify(result, null, 2) },
-              ],
-            };
-          }
-          case "generate_report": {
-            const result = this.generateReport(
-              params.analysis as unknown as LogAnalysis,
-              (params.format as string) || "json",
-            );
-            return {
-              content: [
-                { type: "text", text: JSON.stringify(result, null, 2) },
-              ],
-            };
-          }
-          default:
-            throw new Error(`Unknown tool: ${name}`);
-        }
-      } catch (error) {
+        const result = this.analyzeLogs(
+          (params.logs as string[]) || [],
+          (params.options as Record<string, unknown>) || {},
+        );
         return {
-          content: [{ type: "text", text: `Error: ${error}` }],
-          isError: true,
+          content: [
+            { type: "text", text: JSON.stringify(result, null, 2) },
+          ],
         };
-      }
-    });
+      },
+      "detect_patterns": async (args) => {
+        const params = args as Record<string, unknown>;
+        const result = this.detectPatterns(
+          (params.logs as string[]) || [],
+          (params.customPatterns as string[]) || [],
+          (params.sensitivity as number) || 80,
+        );
+        return {
+          content: [
+            { type: "text", text: JSON.stringify(result, null, 2) },
+          ],
+        };
+      },
+      "alert_on_issues": async (args) => {
+        const params = args as Record<string, unknown>;
+        const result = this.alertOnIssues(
+          params.analysis as unknown as LogAnalysis,
+          (params.thresholds as Record<string, unknown>) || {},
+        );
+        return {
+          content: [
+            { type: "text", text: JSON.stringify(result, null, 2) },
+          ],
+        };
+      },
+      "correlate_events": async (args) => {
+        const params = args as Record<string, unknown>;
+        const result = this.correlateEvents(
+          (params.logSets as Array<{ source: string; logs: string[] }>) ||
+            [],
+          (params.timeWindow as number) || 60,
+        );
+        return {
+          content: [
+            { type: "text", text: JSON.stringify(result, null, 2) },
+          ],
+        };
+      },
+      "generate_report": async (args) => {
+        const params = args as Record<string, unknown>;
+        const result = this.generateReport(
+          params.analysis as unknown as LogAnalysis,
+          (params.format as string) || "json",
+        );
+        return {
+          content: [
+            { type: "text", text: JSON.stringify(result, null, 2) },
+          ],
+        };
+      },
+    };
+    this.setupToolHandlers();
   }
 
   private analyzeLogs(logs: string[], options: Record<string, unknown>): LogAnalysis {
@@ -604,10 +580,7 @@ ${analysis.recommendations.map((r) => `- ${r}`).join("\n")}
     return recs;
   }
 
-  async run() {
-    const transport = new StdioServerTransport();
-    await this.server.connect(transport);
-  }
+
 }
 
 const entryPoint = fs.realpathSync(process.argv[1] ?? "");

package/src/mcps/knowledge-skills/mobile-development.server.ts

@@ -5,14 +5,9 @@
  * React Native, Flutter, and mobile performance optimization
  */
 
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from "@modelcontextprotocol/sdk/types.js";
 import * as fs from "fs";
 import { fileURLToPath } from "url";
+import { XrayKnowledgeSkillBase } from "../shared/knowledge-skill-base.js";
 
 interface IOSBlueprint {
   projectName: string;
@@ -107,28 +102,10 @@ interface AppStoreMetadataArgs {
   features?: string[];
 }
 
-class XrayMobileDevelopmentServer {
-  private server: Server;
-
+class XrayMobileDevelopmentServer extends XrayKnowledgeSkillBase {
   constructor() {
-    this.server = new Server(
-      {
-        name: "mobile-development", version: "3.1.0",
-      },
-      {
-        capabilities: {
-          tools: {},
-        },
-      },
-    );
-
-    this.setupToolHandlers();
-  }
-
-  private setupToolHandlers() {
-    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
-      return {
-        tools: [
+    super("mobile-development", "3.1.0");
+    this.tools = [
           {
             name: "ios_blueprint",
             description:
@@ -305,30 +282,16 @@ class XrayMobileDevelopmentServer {
               required: ["appName", "platform"],
             },
           },
-        ],
-      };
-    });
-
-    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      const { name, arguments: args } = request.params;
-
-      switch (name) {
-        case "ios_blueprint":
-          return await this.generateIOSBlueprint(args ?? {});
-        case "android_blueprint":
-          return await this.generateAndroidBlueprint(args ?? {});
-        case "react_native_boilerplate":
-          return await this.generateReactNativeBlueprint(args ?? {});
-        case "flutter_boilerplate":
-          return await this.generateFlutterBlueprint(args ?? {});
-        case "mobile_performance_profile":
-          return await this.analyzeMobilePerformance(args ?? {});
-        case "app_store_metadata":
-          return await this.generateAppStoreMetadata(args ?? {});
-        default:
-          throw new Error(`Unknown tool: ${name}`);
-      }
-    });
+    ];
+    this.handlers = {
+      "ios_blueprint": async (args) => this.generateIOSBlueprint((args ?? {}) as Record<string, unknown>),
+      "android_blueprint": async (args) => this.generateAndroidBlueprint((args ?? {}) as Record<string, unknown>),
+      "react_native_boilerplate": async (args) => this.generateReactNativeBlueprint((args ?? {}) as Record<string, unknown>),
+      "flutter_boilerplate": async (args) => this.generateFlutterBlueprint((args ?? {}) as Record<string, unknown>),
+      "mobile_performance_profile": async (args) => this.analyzeMobilePerformance((args ?? {}) as Record<string, unknown>),
+      "app_store_metadata": async (args) => this.generateAppStoreMetadata((args ?? {}) as Record<string, unknown>),
+    };
+    this.setupToolHandlers();
   }
 
   private async generateIOSBlueprint(args: Record<string, unknown>) {
@@ -657,16 +620,13 @@ class HomePage extends StatelessWidget {
     };
   }
 
-  async start() {
-    const transport = new StdioServerTransport();
-    await this.server.connect(transport);
-  }
+
 }
 
 const entryPoint = fs.realpathSync(process.argv[1] ?? "");
 if (entryPoint && fileURLToPath(import.meta.url) === entryPoint) {
   const server = new XrayMobileDevelopmentServer();
-  server.start().catch(() => {});
+  server.run().catch(() => {});
 }
 
 export { XrayMobileDevelopmentServer };

package/src/mcps/knowledge-skills/multimodal-looker.server.ts

@@ -8,15 +8,10 @@
  * Production-ready implementation with real analysis logic.
  */
 
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from "@modelcontextprotocol/sdk/types.js";
 import * as fs from "fs";
 import { fileURLToPath } from "url";
 import { frameworkLogger } from "../../core/framework-logger.js";
+import { XrayKnowledgeSkillBase } from "../shared/knowledge-skill-base.js";
 
 /* ============================================================================
  * Type Definitions
@@ -162,24 +157,10 @@ interface VisualComparison {
  * Analysis Engine
  * ============================================================================ */
 
-class MultimodalLookerServer {
-  private server: Server;
-
+class MultimodalLookerServer extends XrayKnowledgeSkillBase {
   constructor() {
-    this.server = new Server(
-      { name: "multimodal-looker", version: "3.1.0" },
-      { capabilities: { tools: {} } },
-    );
-    this.setupToolHandlers();
-  }
-
-  /* --------------------------------------------------------------------------
-   * Tool Registration
-   * -------------------------------------------------------------------------- */
-
-  private setupToolHandlers() {
-    this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
-      tools: [
+    super("multimodal-looker", "3.1.0");
+    this.tools = [
         {
           name: "analyze-screenshot",
           description:
@@ -309,35 +290,15 @@ class MultimodalLookerServer {
             required: ["imageA", "imageB"],
           },
         },
-      ],
-    }));
-
-    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      const { name, arguments: args = {} } = request.params;
-      const params = args as Record<string, unknown>;
-
-      try {
-        switch (name) {
-          case "analyze-screenshot":
-            return this.handleAnalyzeScreenshot(params);
-          case "describe-diagram":
-            return this.handleDescribeDiagram(params);
-          case "review-ui-mockup":
-            return this.handleReviewUIMockup(params);
-          case "extract-visual-specs":
-            return this.handleExtractVisualSpecs(params);
-          case "compare-designs":
-            return this.handleCompareDesigns(params);
-          default:
-            throw new Error(`Unknown tool: ${name}`);
-        }
-      } catch (error) {
-        return {
-          content: [{ type: "text", text: `Error: ${error}` }],
-          isError: true,
-        };
-      }
-    });
+    ];
+    this.handlers = {
+      "analyze-screenshot": async (args) => this.handleAnalyzeScreenshot(args as Record<string, unknown>),
+      "describe-diagram": async (args) => this.handleDescribeDiagram(args as Record<string, unknown>),
+      "review-ui-mockup": async (args) => this.handleReviewUIMockup(args as Record<string, unknown>),
+      "extract-visual-specs": async (args) => this.handleExtractVisualSpecs(args as Record<string, unknown>),
+      "compare-designs": async (args) => this.handleCompareDesigns(args as Record<string, unknown>),
+    };
+    this.setupToolHandlers();
   }
 
   /* --------------------------------------------------------------------------
@@ -1482,14 +1443,7 @@ class MultimodalLookerServer {
     return recommendations;
   }
 
-  /* --------------------------------------------------------------------------
-   * Server Lifecycle
-   * -------------------------------------------------------------------------- */
 
-  async run(): Promise<void> {
-    const transport = new StdioServerTransport();
-    await this.server.connect(transport);
-  }
 }
 
 const entryPoint = fs.realpathSync(process.argv[1] ?? "");

package/src/mcps/knowledge-skills/performance-optimization.server.ts

@@ -5,18 +5,12 @@
  * profiling, benchmarking, memory analysis, and Core Web Vitals measurement
  */
 
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from "@modelcontextprotocol/sdk/types.js";
 import { frameworkLogger, generateJobId } from "../../core/framework-logger.js";
-import { createGracefulShutdown } from "../../utils/shutdown-handler.js";
 import fs from "fs";
 import path from "path";
 import os from "os";
 import { execSync } from "child_process";
+import { XrayKnowledgeSkillBase } from "../shared/knowledge-skill-base.js";
 
 interface ProfileApplicationArgs {
   projectRoot: string;
@@ -217,30 +211,10 @@ interface McpToolResponse {
   data?: Record<string, unknown>;
 }
 
-class XrayPerformanceOptimizationServer {
-  private server: Server;
-  private startTime: number;
-
+class XrayPerformanceOptimizationServer extends XrayKnowledgeSkillBase {
   constructor() {
-    this.server = new Server(
-      {
-        name: "performance-optimization", version: "3.1.0",
-      },
-      {
-        capabilities: {
-          tools: {},
-        },
-      },
-    );
-
-    this.startTime = Date.now();
-    this.setupToolHandlers();
-  }
-
-  private setupToolHandlers() {
-    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
-      return {
-        tools: [
+    super("performance-optimization", "3.1.0");
+    this.tools = [
           {
             name: "profile-application",
             description: "Run comprehensive performance profiling on the codebase to identify bottlenecks and hot paths",
@@ -323,38 +297,15 @@ class XrayPerformanceOptimizationServer {
               required: ["projectRoot"]
             }
           }
-        ]
-      };
-    });
-
-    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      const { name, arguments: args } = request.params;
-
-      try {
-        switch (name) {
-          case "profile-application":
-            return await this.handleProfileApplication(args as unknown as ProfileApplicationArgs) as never;
-          case "analyze-memory":
-            return await this.handleAnalyzeMemory(args as unknown as AnalyzeMemoryArgs) as never;
-          case "benchmark-code":
-            return await this.handleBenchmarkCode(args as unknown as BenchmarkCodeArgs) as never;
-          case "suggest-optimizations":
-            return await this.handleSuggestOptimizations(args as unknown as SuggestOptimizationsArgs) as never;
-          case "measure-core-web-vitals":
-            return await this.handleMeasureCoreWebVitals(args as unknown as MeasureCoreWebVitalsArgs) as never;
-          default:
-            throw new Error(`Unknown tool: ${name}`);
-        }
-      } catch (error) {
-        frameworkLogger.log("mcp/performance-optimization", "tool-handler", "error", { tool: name, error: String(error) });
-        return {
-          content: [{
-            type: "text",
-            text: `Error executing tool "${name}": ${error instanceof Error ? error.message : String(error)}`
-          }]
-        };
-      }
-    });
+    ];
+    this.handlers = {
+      "profile-application": async (args) => this.handleProfileApplication(args as unknown as ProfileApplicationArgs),
+      "analyze-memory": async (args) => this.handleAnalyzeMemory(args as unknown as AnalyzeMemoryArgs),
+      "benchmark-code": async (args) => this.handleBenchmarkCode(args as unknown as BenchmarkCodeArgs),
+      "suggest-optimizations": async (args) => this.handleSuggestOptimizations(args as unknown as SuggestOptimizationsArgs),
+      "measure-core-web-vitals": async (args) => this.handleMeasureCoreWebVitals(args as unknown as MeasureCoreWebVitalsArgs),
+    };
+    this.setupToolHandlers();
   }
 
   private async handleProfileApplication(args: ProfileApplicationArgs): Promise<McpToolResponse> {
@@ -2040,19 +1991,7 @@ ${result.recommendations.slice(0, 5).map((r, i) => `${i + 1}. ${r}`).join("\n")
     return files.slice(0, 100);
   }
 
-  async run(): Promise<void> {
-    const transport = new StdioServerTransport();
-    await this.server.connect(transport);
-
-    createGracefulShutdown({
-      serverName: "performance-optimization.server",
-      server: this.server,
-    });
 
-    frameworkLogger.log("mcp/performance-optimization", "server-started", "info", {
-      uptime: Date.now() - this.startTime
-    });
-  }
 }
 
 if (import.meta.url === `file://${fs.realpathSync(process.argv[1]!)}`) {