0xray 3.1.1 → 3.2.0

package/dist/mcps/enforcer-tools.server.js

@@ -4,259 +4,243 @@
  * Converts enforcer-tools.ts functions into MCP server tools
  * Provides rule enforcement and validation capabilities via MCP protocol
  */
-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 { CallToolRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { XrayKnowledgeSkillBase } from "./shared/knowledge-skill-base.js";
 import * as fs from "fs";
 import { frameworkLogger } from "../core/framework-logger.js";
 import { getCodexPolicyService } from "../governance/codex-policy.service.js";
 // Import actual enforcer-tools functions
 import { ruleValidation as runRuleValidation } from "../enforcement/enforcer-tools.js";
-class XrayEnforcerToolsServer {
-    server;
+class XrayEnforcerToolsServer extends XrayKnowledgeSkillBase {
     constructor() {
-        this.server = new Server({
-            name: "enforcer", version: "3.1.0",
-        }, {
-            capabilities: {
-                tools: {},
-            },
-        });
-        this.setupToolHandlers();
-    }
-    setupToolHandlers() {
-        this.server.setRequestHandler(ListToolsRequestSchema, async () => {
-            return {
-                tools: [
-                    {
-                        name: "rule-validation",
-                        description: "Validate operations against the comprehensive rule hierarchy including duplicate code prevention, test requirements, and architectural constraints",
-                        inputSchema: {
+        super("enforcer", "3.1.0");
+        this.tools = [
+            {
+                name: "rule-validation",
+                description: "Validate operations against the comprehensive rule hierarchy including duplicate code prevention, test requirements, and architectural constraints",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        operation: {
+                            type: "string",
+                            description: "Operation to validate (create, modify, refactor, etc.)",
+                        },
+                        files: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "Files affected by the operation",
+                        },
+                        newCode: {
+                            type: "string",
+                            description: "New code being added (optional)",
+                        },
+                        existingCode: {
                             type: "object",
-                            properties: {
-                                operation: {
-                                    type: "string",
-                                    description: "Operation to validate (create, modify, refactor, etc.)",
-                                },
-                                files: {
-                                    type: "array",
-                                    items: { type: "string" },
-                                    description: "Files affected by the operation",
-                                },
-                                newCode: {
-                                    type: "string",
-                                    description: "New code being added (optional)",
-                                },
-                                existingCode: {
-                                    type: "object",
-                                    description: "Map of existing code for comparison (optional)",
-                                },
-                                dependencies: {
-                                    type: "array",
-                                    items: { type: "string" },
-                                    description: "Dependencies being declared (optional)",
-                                },
-                                tests: {
-                                    type: "array",
-                                    items: { type: "string" },
-                                    description: "Test files for validation (optional)",
-                                },
-                            },
-                            required: ["operation"],
+                            description: "Map of existing code for comparison (optional)",
+                        },
+                        dependencies: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "Dependencies being declared (optional)",
+                        },
+                        tests: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "Test files for validation (optional)",
                         },
                     },
-                    {
-                        name: "codex-enforcement",
-                        description: "Enforce all Universal Development Codex terms with comprehensive compliance validation and actionable remediation",
-                        inputSchema: {
-                            type: "object",
-                            properties: {
-                                operation: {
-                                    type: "string",
-                                    description: "Operation to validate against codex",
-                                },
-                                files: {
-                                    type: "array",
-                                    items: { type: "string" },
-                                    description: "Files to check for codex compliance",
-                                },
-                                newCode: {
-                                    type: "string",
-                                    description: "New code to validate against codex terms",
-                                },
-                                focusAreas: {
-                                    type: "array",
-                                    items: {
-                                        type: "string",
-                                        enum: [
-                                            "error-handling",
-                                            "type-safety",
-                                            "performance",
-                                            "security",
-                                            "architecture",
-                                        ],
-                                    },
-                                    description: "Specific codex areas to focus validation on",
-                                },
+                    required: ["operation"],
+                },
+            },
+            {
+                name: "codex-enforcement",
+                description: "Enforce all Universal Development Codex terms with comprehensive compliance validation and actionable remediation",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        operation: {
+                            type: "string",
+                            description: "Operation to validate against codex",
+                        },
+                        files: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "Files to check for codex compliance",
+                        },
+                        newCode: {
+                            type: "string",
+                            description: "New code to validate against codex terms",
+                        },
+                        focusAreas: {
+                            type: "array",
+                            items: {
+                                type: "string",
+                                enum: [
+                                    "error-handling",
+                                    "type-safety",
+                                    "performance",
+                                    "security",
+                                    "architecture",
+                                ],
                             },
-                            required: ["operation"],
+                            description: "Specific codex areas to focus validation on",
                         },
                     },
-                    {
-                        name: "context-analysis-validation",
-                        description: "Validate proper integration of contextual analysis components and architectural patterns",
-                        inputSchema: {
-                            type: "object",
-                            properties: {
-                                files: {
-                                    type: "array",
-                                    items: { type: "string" },
-                                    description: "Files to validate for context integration",
-                                },
-                                operation: {
-                                    type: "string",
-                                    description: "Operation context for validation",
-                                },
-                                checkPatterns: {
-                                    type: "array",
-                                    items: {
-                                        type: "string",
-                                        enum: [
-                                            "memory-optimization",
-                                            "error-handling",
-                                            "type-safety",
-                                            "dependency-injection",
-                                        ],
-                                    },
-                                    description: "Specific integration patterns to validate",
-                                },
+                    required: ["operation"],
+                },
+            },
+            {
+                name: "context-analysis-validation",
+                description: "Validate proper integration of contextual analysis components and architectural patterns",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        files: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "Files to validate for context integration",
+                        },
+                        operation: {
+                            type: "string",
+                            description: "Operation context for validation",
+                        },
+                        checkPatterns: {
+                            type: "array",
+                            items: {
+                                type: "string",
+                                enum: [
+                                    "memory-optimization",
+                                    "error-handling",
+                                    "type-safety",
+                                    "dependency-injection",
+                                ],
                             },
-                            required: ["files", "operation"],
+                            description: "Specific integration patterns to validate",
                         },
                     },
-                    {
-                        name: "quality-gate-check",
-                        description: "Perform comprehensive quality gate validation before commits with automated fixes and blocking decisions",
-                        inputSchema: {
+                    required: ["files", "operation"],
+                },
+            },
+            {
+                name: "quality-gate-check",
+                description: "Perform comprehensive quality gate validation before commits with automated fixes and blocking decisions",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        operation: {
+                            type: "string",
+                            description: "Operation to quality-gate check",
+                        },
+                        context: {
                             type: "object",
                             properties: {
-                                operation: {
-                                    type: "string",
-                                    description: "Operation to quality-gate check",
-                                },
-                                context: {
-                                    type: "object",
-                                    properties: {
-                                        files: { type: "array", items: { type: "string" } },
-                                        newCode: { type: "string" },
-                                        dependencies: { type: "array", items: { type: "string" } },
-                                        tests: { type: "array", items: { type: "string" } },
-                                    },
-                                    description: "Complete operation context for comprehensive validation",
-                                },
-                                strictMode: {
-                                    type: "boolean",
-                                    default: true,
-                                    description: "Enforce strict quality requirements",
-                                },
+                                files: { type: "array", items: { type: "string" } },
+                                newCode: { type: "string" },
+                                dependencies: { type: "array", items: { type: "string" } },
+                                tests: { type: "array", items: { type: "string" } },
                             },
-                            required: ["operation", "context"],
+                            description: "Complete operation context for comprehensive validation",
+                        },
+                        strictMode: {
+                            type: "boolean",
+                            default: true,
+                            description: "Enforce strict quality requirements",
                         },
                     },
-                    {
-                        name: "get-enforcement-status",
-                        description: "Get comprehensive enforcement statistics and rule compliance metrics",
-                        inputSchema: {
-                            type: "object",
-                            properties: {
-                                includeHistory: {
-                                    type: "boolean",
-                                    default: false,
-                                    description: "Include historical enforcement data",
-                                },
-                                focusAreas: {
-                                    type: "array",
-                                    items: {
-                                        type: "string",
-                                        enum: ["rules", "codex", "quality", "performance"],
-                                    },
-                                    description: "Specific areas to focus status reporting on",
-                                },
+                    required: ["operation", "context"],
+                },
+            },
+            {
+                name: "get-enforcement-status",
+                description: "Get comprehensive enforcement statistics and rule compliance metrics",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        includeHistory: {
+                            type: "boolean",
+                            default: false,
+                            description: "Include historical enforcement data",
+                        },
+                        focusAreas: {
+                            type: "array",
+                            items: {
+                                type: "string",
+                                enum: ["rules", "codex", "quality", "performance"],
                             },
+                            description: "Specific areas to focus status reporting on",
                         },
                     },
-                    {
-                        name: "run-pre-commit-validation",
-                        description: "Execute comprehensive pre-commit validation with all enforcer tools and automated remediation",
-                        inputSchema: {
-                            type: "object",
-                            properties: {
-                                files: {
-                                    type: "array",
-                                    items: { type: "string" },
-                                    description: "Files to validate before commit",
-                                },
-                                operation: {
-                                    type: "string",
-                                    default: "commit",
-                                    description: "Operation type for validation context",
-                                },
-                                autoFix: {
-                                    type: "boolean",
-                                    default: true,
-                                    description: "Automatically apply safe fixes",
-                                },
-                                strictBlocking: {
-                                    type: "boolean",
-                                    default: true,
-                                    description: "Block commit on any validation error",
-                                },
-                            },
-                            required: ["files"],
+                },
+            },
+            {
+                name: "run-pre-commit-validation",
+                description: "Execute comprehensive pre-commit validation with all enforcer tools and automated remediation",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        files: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "Files to validate before commit",
+                        },
+                        operation: {
+                            type: "string",
+                            default: "commit",
+                            description: "Operation type for validation context",
+                        },
+                        autoFix: {
+                            type: "boolean",
+                            default: true,
+                            description: "Automatically apply safe fixes",
+                        },
+                        strictBlocking: {
+                            type: "boolean",
+                            default: true,
+                            description: "Block commit on any validation error",
                         },
                     },
-                    {
-                        name: "security-scan",
-                        description: "Run security scan using npm audit, Trivy, ESLint security, and OWASP dependency check",
-                        inputSchema: {
-                            type: "object",
-                            properties: {
-                                includePromptValidation: {
-                                    type: "boolean",
-                                    default: false,
-                                    description: "Also validate prompts for injection attacks",
-                                },
-                                promptText: {
-                                    type: "string",
-                                    description: "Prompt text to validate for injection attacks",
-                                },
-                            },
+                    required: ["files"],
+                },
+            },
+            {
+                name: "security-scan",
+                description: "Run security scan using npm audit, Trivy, ESLint security, and OWASP dependency check",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        includePromptValidation: {
+                            type: "boolean",
+                            default: false,
+                            description: "Also validate prompts for injection attacks",
+                        },
+                        promptText: {
+                            type: "string",
+                            description: "Prompt text to validate for injection attacks",
                         },
                     },
-                ],
-            };
-        });
+                },
+            },
+        ];
+        this.handlers = {
+            "rule-validation": async (args) => this.ruleValidation(args),
+            "codex-enforcement": async (args) => this.codexEnforcement(args),
+            "context-analysis-validation": async (args) => this.contextAnalysisValidation(args),
+            "quality-gate-check": async (args) => this.qualityGateCheck(args),
+            "get-enforcement-status": async (args) => this.getEnforcementStatus(args),
+            "run-pre-commit-validation": async (args) => this.runPreCommitValidation(args),
+            "security-scan": async (args) => this.securityScan(args),
+        };
+        this.setupToolHandlers();
+    }
+    setupToolHandlers() {
+        this.setupListToolsHandler();
         this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
             const { name, arguments: args } = request.params;
+            const handler = this.handlers[name];
+            if (!handler)
+                throw new Error(`Unknown tool: ${name}`);
             try {
-                switch (name) {
-                    case "rule-validation":
-                        return await this.ruleValidation(args);
-                    case "codex-enforcement":
-                        return await this.codexEnforcement(args);
-                    case "context-analysis-validation":
-                        return await this.contextAnalysisValidation(args);
-                    case "quality-gate-check":
-                        return await this.qualityGateCheck(args);
-                    case "get-enforcement-status":
-                        return await this.getEnforcementStatus(args);
-                    case "run-pre-commit-validation":
-                        return await this.runPreCommitValidation(args);
-                    case "security-scan":
-                        return await this.securityScan(args);
-                    default:
-                        throw new Error(`Unknown tool: ${name}`);
-                }
+                return await handler(args);
             }
             catch (error) {
                 frameworkLogger.log("mcps/enforcer", "tool", "error", { tool: name, error: String(error) });
@@ -687,56 +671,6 @@ class XrayEnforcerToolsServer {
             filesProcessed: files.length,
         };
     }
-    async run() {
-        const transport = new StdioServerTransport();
-        await this.server.connect(transport);
-        // Server started - no startup logging to console
-        const cleanup = async (signal) => {
-            // Set a timeout to force exit if graceful shutdown fails
-            const timeout = setTimeout(() => {
-                frameworkLogger.log("mcps/enforcer", "shutdown", "error", { message: "Graceful shutdown timeout, forcing exit..." });
-                process.exit(1);
-            }, 5000); // 5 second timeout
-            try {
-                if (this.server && typeof this.server.close === "function") {
-                    await this.server.close();
-                }
-                clearTimeout(timeout);
-                process.exit(0);
-            }
-            catch (error) {
-                clearTimeout(timeout);
-                frameworkLogger.log("mcps/enforcer", "shutdown", "error", { message: `Error during server shutdown: ${String(error)}` });
-                process.exit(1);
-            }
-        };
-        // Handle multiple shutdown signals
-        process.on("SIGINT", () => cleanup("SIGINT"));
-        process.on("SIGTERM", () => cleanup("SIGTERM"));
-        process.on("SIGHUP", () => cleanup("SIGHUP"));
-        // Monitor parent process (opencode) and shutdown if it dies
-        const checkParent = () => {
-            try {
-                process.kill(process.ppid, 0); // Check if parent is alive
-                setTimeout(checkParent, 1000); // Check again in 1 second
-            }
-            catch (error) {
-                // Parent process died, shut down gracefully - no logging
-                cleanup("parent-process-death");
-            }
-        };
-        // Start monitoring parent process
-        setTimeout(checkParent, 2000); // Start checking after 2 seconds
-        // Handle uncaught exceptions and unhandled rejections
-        process.on("uncaughtException", (error) => {
-            frameworkLogger.log("mcps/enforcer", "uncaughtException", "error", { error: String(error) });
-            cleanup("uncaughtException");
-        });
-        process.on("unhandledRejection", (reason, promise) => {
-            frameworkLogger.log("mcps/enforcer", "unhandledRejection", "error", { error: String(reason) });
-            cleanup("unhandledRejection");
-        });
-    }
     async securityScan(args) {
         const { includePromptValidation, promptText } = args;
         const results = { tools: {}, summary: {} };