@rigstate/mcp 0.4.2 → 0.4.3

package/src/lib/schemas.ts

@@ -0,0 +1,191 @@
+import { z } from 'zod';
+
+// =============================================================================
+// ZOD SCHEMAS FOR TOOL INPUTS
+// =============================================================================
+
+export const QueryBrainInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    query: z.string().min(1, 'Query is required'),
+    limit: z.number().min(1).max(20).optional().default(8),
+    threshold: z.number().min(0).max(1).optional().default(0.5)
+});
+
+export const GetProjectContextInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID')
+});
+
+export const GetLatestDecisionsInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    limit: z.number().min(1).max(10).optional().default(5)
+});
+
+export const SaveDecisionInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    title: z.string().min(1, 'Title is required').max(200, 'Title too long'),
+    decision: z.string().min(1, 'Decision content is required'),
+    rationale: z.string().optional(),
+    category: z.enum(['decision', 'architecture', 'constraint', 'tech_stack', 'design_rule']).optional().default('decision'),
+    tags: z.array(z.string()).optional().default([])
+});
+
+export const SubmitIdeaInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    title: z.string().min(1, 'Title is required').max(200, 'Title too long'),
+    description: z.string().min(1, 'Description is required'),
+    category: z.enum(['feature', 'improvement', 'experiment', 'pivot']).optional().default('feature'),
+    tags: z.array(z.string()).optional().default([])
+});
+
+export const UpdateRoadmapInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    chunkId: z.string().uuid('Invalid chunk ID').optional(),
+    title: z.string().optional(),
+    status: z.enum(['LOCKED', 'ACTIVE', 'COMPLETED'])
+});
+
+export const RunArchitectureAuditInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    filePath: z.string().min(1, 'File path is required'),
+    content: z.string().min(1, 'Content is required')
+});
+
+export const ListRoadmapTasksInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID')
+});
+
+export const RefineLogicInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    originalReasoning: z.string().min(1, 'Original reasoning is required'),
+    userCorrection: z.string().min(1, 'User correction is required'),
+    scope: z.enum(['project', 'global']).optional().default('project')
+});
+
+export const GetLearnedInstructionsInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID').optional()
+});
+
+export const CheckAgentBridgeInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    action: z.enum(['check', 'update', 'submit_for_review']).optional().default('check'),
+    bridgeId: z.string().uuid('Invalid bridge ID').optional(),
+    status: z.enum(['PENDING', 'NEEDS_REVIEW', 'AWAITING_APPROVAL', 'APPROVED', 'REJECTED', 'EXECUTING', 'COMPLETED', 'FAILED']).optional(),
+    summary: z.string().optional(),
+    execution_summary: z.string().optional(),
+    proposal: z.string().optional()
+});
+
+export const CheckRulesSyncInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    currentRulesContent: z.string().optional()
+});
+
+export const GetPendingTasksInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID')
+});
+
+export const UpdateTaskStatusInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    taskId: z.string().uuid('Invalid task ID'),
+    status: z.enum(['EXECUTING', 'COMPLETED', 'FAILED']),
+    executionSummary: z.string().optional()
+});
+
+export const GetNextRoadmapStepInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    currentStepId: z.string().uuid('Invalid step ID').optional()
+});
+
+export const GenerateProfessionalPDFInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    reportType: z.enum(['SYSTEM_MANIFEST', 'INVESTOR_REPORT'])
+});
+
+export const ArchaeologicalScanInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    gitLog: z.string().describe('Git log output'),
+    fileTree: z.array(z.string()).describe('File paths')
+});
+
+export const ImportGhostFeaturesInputSchema = z.object({
+    projectId: z.string().uuid('Invalid project ID'),
+    features: z.array(z.any())
+});
+
+export const AnalyzeDependencyGraphInputSchema = z.object({
+    path: z.string().min(1).default('src')
+});
+
+export const AuditRlsStatusInputSchema = z.object({
+    projectId: z.string().uuid().optional()
+});
+
+export const AuditSecurityIntegrityInputSchema = z.object({
+    projectId: z.string().uuid(),
+    filePath: z.string().min(1),
+    content: z.string().min(1),
+    rules: z.array(z.string()).optional()
+});
+
+export const QueryProjectBrainInputSchema = QueryBrainInputSchema;
+
+export const FetchPackageHealthInputSchema = z.object({
+    packageName: z.string().min(1)
+});
+
+export const SaveToProjectBrainInputSchema = z.object({
+    projectId: z.string().uuid(),
+    title: z.string().min(1),
+    content: z.string().min(1),
+    category: z.enum(['DECISION', 'ARCHITECTURE', 'NOTE', 'LESSON_LEARNED']).default('NOTE'),
+    tags: z.array(z.string()).optional().default([])
+});
+
+export const UpdateRoadmapStatusInputSchema = z.object({
+    projectId: z.string().uuid(),
+    chunkId: z.string().uuid(),
+    status: z.enum(['TODO', 'IN_PROGRESS', 'COMPLETED'])
+});
+
+export const AddRoadmapChunkInputSchema = z.object({
+    projectId: z.string().uuid(),
+    title: z.string().min(1),
+    description: z.string().optional(),
+    featureId: z.string().optional(),
+    priority: z.enum(['LOW', 'MEDIUM', 'HIGH']).default('MEDIUM')
+});
+
+export const AnalyzeUiComponentInputSchema = z.object({
+    filePath: z.string()
+});
+
+export const ApplyDesignSystemInputSchema = z.object({
+    filePath: z.string()
+});
+
+export const FetchUiLibraryDocsInputSchema = z.object({
+    componentName: z.string(),
+    library: z.enum(['shadcn', 'lucide']).default('shadcn')
+});
+
+export const GenerateCursorRulesInputSchema = z.object({
+    projectId: z.string().uuid()
+});
+
+export const AnalyzeDatabasePerformanceInputSchema = z.object({
+    projectId: z.string().uuid(),
+    filePaths: z.array(z.string())
+});
+
+export const AuditIntegrityGateInputSchema = z.object({
+    projectId: z.string().uuid(),
+    filePaths: z.array(z.string()).optional().default([])
+});
+
+export const CompleteRoadmapTaskInputSchema = z.object({
+    projectId: z.string().uuid(),
+    summary: z.string().min(1),
+    taskId: z.string().uuid().optional(),
+    gitDiff: z.string().optional(),
+    integrityGate: z.any().optional()
+});

package/src/lib/types.ts

@@ -1,10 +1,13 @@
 /**
  * Rigstate MCP Server - Types
- * 
  * Shared type definitions for the MCP server tools.
  */
 
 import { z } from 'zod';
+import * as schemas from './schemas.js';
+
+// Re-export schemas for convenience
+export * from './schemas.js';
 
 // =============================================================================
 // PROJECT CONTEXT TYPES
@@ -88,103 +91,42 @@ export interface DecisionsResponse {
 }
 
 // =============================================================================
-// ZOD SCHEMAS FOR TOOL INPUTS
-// =============================================================================
-
-export const QueryBrainInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    query: z.string().min(1, 'Query is required'),
-    limit: z.number().min(1).max(20).optional().default(8),
-    threshold: z.number().min(0).max(1).optional().default(0.5)
-});
-
-export const GetProjectContextInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID')
-});
-
-export const GetLatestDecisionsInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    limit: z.number().min(1).max(10).optional().default(5)
-});
-
-// =============================================================================
-// WRITE OPERATION SCHEMAS
+// TYPE ALIASES (Inferred from Schemas)
 // =============================================================================
 
-export const SaveDecisionInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    title: z.string().min(1, 'Title is required').max(200, 'Title too long'),
-    decision: z.string().min(1, 'Decision content is required'),
-    rationale: z.string().optional(),
-    category: z.enum(['decision', 'architecture', 'constraint', 'tech_stack', 'design_rule']).optional().default('decision'),
-    tags: z.array(z.string()).optional().default([])
-});
-
-export const SubmitIdeaInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    title: z.string().min(1, 'Title is required').max(200, 'Title too long'),
-    description: z.string().min(1, 'Description is required'),
-    category: z.enum(['feature', 'improvement', 'experiment', 'pivot']).optional().default('feature'),
-    tags: z.array(z.string()).optional().default([])
-});
-
-export const UpdateRoadmapInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    chunkId: z.string().uuid('Invalid chunk ID').optional(),
-    title: z.string().optional(),
-    status: z.enum(['LOCKED', 'ACTIVE', 'COMPLETED'])
-});
-
-export const RunArchitectureAuditInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    filePath: z.string().min(1, 'File path is required'),
-    content: z.string().min(1, 'Content is required')
-});
-
-export const ListRoadmapTasksInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID')
-});
-
-
+export type QueryBrainInput = z.infer<typeof schemas.QueryBrainInputSchema>;
+export type GetProjectContextInput = z.infer<typeof schemas.GetProjectContextInputSchema>;
+export type GetLatestDecisionsInput = z.infer<typeof schemas.GetLatestDecisionsInputSchema>;
+export type SaveDecisionInput = z.infer<typeof schemas.SaveDecisionInputSchema>;
+export type SubmitIdeaInput = z.infer<typeof schemas.SubmitIdeaInputSchema>;
+export type UpdateRoadmapInput = z.infer<typeof schemas.UpdateRoadmapInputSchema>;
+export type RunArchitectureAuditInput = z.infer<typeof schemas.RunArchitectureAuditInputSchema>;
+export type ListRoadmapTasksInput = z.infer<typeof schemas.ListRoadmapTasksInputSchema>;
+export type CheckAgentBridgeInput = z.infer<typeof schemas.CheckAgentBridgeInputSchema>;
+export type GetPendingTasksInput = z.infer<typeof schemas.GetPendingTasksInputSchema>;
+export type UpdateTaskStatusInput = z.infer<typeof schemas.UpdateTaskStatusInputSchema>;
+export type AnalyzeDependencyGraphInput = z.infer<typeof schemas.AnalyzeDependencyGraphInputSchema>;
+export type AuditRlsStatusInput = z.infer<typeof schemas.AuditRlsStatusInputSchema>;
+export type AuditSecurityIntegrityInput = z.infer<typeof schemas.AuditSecurityIntegrityInputSchema>;
+export type QueryProjectBrainInput = z.infer<typeof schemas.QueryProjectBrainInputSchema>;
+export type FetchPackageHealthInput = z.infer<typeof schemas.FetchPackageHealthInputSchema>;
+export type SaveToProjectBrainInput = z.infer<typeof schemas.SaveToProjectBrainInputSchema>;
+export type UpdateRoadmapStatusInput = z.infer<typeof schemas.UpdateRoadmapStatusInputSchema>;
+export type AddRoadmapChunkInput = z.infer<typeof schemas.AddRoadmapChunkInputSchema>;
+export type AnalyzeUiComponentInput = z.infer<typeof schemas.AnalyzeUiComponentInputSchema>;
+export type ApplyDesignSystemInput = z.infer<typeof schemas.ApplyDesignSystemInputSchema>;
+export type FetchUiLibraryDocsInput = z.infer<typeof schemas.FetchUiLibraryDocsInputSchema>;
+export type AnalyzeDatabasePerformanceInput = z.infer<typeof schemas.AnalyzeDatabasePerformanceInputSchema>;
+export type AuditIntegrityGateInput = z.infer<typeof schemas.AuditIntegrityGateInputSchema>;
+export type CompleteRoadmapTaskInput = z.infer<typeof schemas.CompleteRoadmapTaskInputSchema>;
 
 // =============================================================================
-// TEACHER MODE SCHEMAS
+// RESPONSE INTERFACES
 // =============================================================================
 
-export const RefineLogicInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    originalReasoning: z.string().min(1, 'Original reasoning is required'),
-    userCorrection: z.string().min(1, 'User correction is required'),
-    scope: z.enum(['project', 'global']).optional().default('project')
-});
-
-export const GetLearnedInstructionsInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID').optional()
-});
-
-// =============================================================================
-// RESPONSE TYPES
-// =============================================================================
-
-export interface SaveDecisionResponse {
-    success: boolean;
-    memoryId: string;
-    message: string;
-}
-
-export interface SubmitIdeaResponse {
-    success: boolean;
-    ideaId: string;
-    message: string;
-}
-
-export interface UpdateRoadmapResponse {
-    success: boolean;
-    chunkId: string;
-    previousStatus: string;
-    newStatus: string;
-    message: string;
-}
+export interface SaveDecisionResponse { success: boolean; memoryId: string; message: string; }
+export interface SubmitIdeaResponse { success: boolean; ideaId: string; message: string; }
+export interface UpdateRoadmapResponse { success: boolean; chunkId: string; previousStatus: string; newStatus: string; message: string; }
 
 export interface AuditViolation {
     type: string;
@@ -202,54 +144,78 @@ export interface ArchitectureAuditResponse {
     summary: string;
 }
 
-export type QueryBrainInput = z.infer<typeof QueryBrainInputSchema>;
-export type GetProjectContextInput = z.infer<typeof GetProjectContextInputSchema>;
-export type GetLatestDecisionsInput = z.infer<typeof GetLatestDecisionsInputSchema>;
-export type SaveDecisionInput = z.infer<typeof SaveDecisionInputSchema>;
-export type SubmitIdeaInput = z.infer<typeof SubmitIdeaInputSchema>;
-export type UpdateRoadmapInput = z.infer<typeof UpdateRoadmapInputSchema>;
-export type RunArchitectureAuditInput = z.infer<typeof RunArchitectureAuditInputSchema>;
-export type ListRoadmapTasksInput = z.infer<typeof ListRoadmapTasksInputSchema>;
+export interface IntegrityCheckResult {
+    check: 'PERFORMANCE' | 'SECURITY' | 'FORTRESS_MATRIX';
+    status: 'PASS' | 'FAIL' | 'WARN';
+    message: string;
+    details?: any;
+}
 
+export interface IntegrityGateResponse {
+    passed: boolean;
+    mode: 'OPEN' | 'SOFT_LOCK' | 'HARD_LOCK';
+    checks: IntegrityCheckResult[];
+    summary: string;
+}
+
+export interface SecurityIntegrityResponse {
+    passed: boolean;
+    score: number;
+    violations: Array<{
+        id: string;
+        type: string;
+        severity: 'LOW' | 'MEDIUM' | 'HIGH' | 'FATAL';
+        title: string;
+        description: string;
+        recommendation: string;
+    }>;
+    summary: string;
+}
+
+// =============================================================================
+// ROADMAP & TASK TYPES
+// =============================================================================
+
+export interface RoadmapChunk {
+    id: string;
+    project_id: string;
+    feature_id: string | null;
+    title: string;
+    description: string | null;
+    status: string;
+    priority: string | null;
+    step_number: number;
+    prompt_content: string | null;
+    created_at: string;
+    completed_at: string | null;
+}
 
 export interface ListRoadmapTasksResponse {
     tasks: Array<{
         id: string;
         title: string;
-        priority: string;
+        priority: string | null;
         status: string;
-        step_number?: number;
+        step_number: number;
+        prompt_content: string | null;
     }>;
     formatted: string;
 }
 
-
-export const CheckAgentBridgeInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    action: z.enum(['check', 'update', 'submit_for_review']).optional().default('check'),
-    bridgeId: z.string().uuid('Invalid bridge ID').optional(),
-    status: z.enum(['PENDING', 'NEEDS_REVIEW', 'AWAITING_APPROVAL', 'APPROVED', 'REJECTED', 'EXECUTING', 'COMPLETED', 'FAILED']).optional(),
-    summary: z.string().optional(),
-    execution_summary: z.string().optional(),
-    proposal: z.string().optional()
-});
-
-export const CheckRulesSyncInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    currentRulesContent: z.string().optional() // Provide content to check, or tool tries to read file
-});
-
-export type CheckAgentBridgeInput = z.infer<typeof CheckAgentBridgeInputSchema>;
+// =============================================================================
+// AGENT BRIDGE & RULES TYPES
+// =============================================================================
 
 export interface AgentBridgeTask {
     id: string;
     project_id: string;
-    task_id: string;
+    task_id: string | null;
     status: string;
     summary: string | null;
     execution_summary: string | null;
     proposal: string | null;
     created_at: string;
+    updated_at?: string;
     completed_at: string | null;
     task_title?: string;
     task_description?: string;
@@ -258,8 +224,8 @@ export interface AgentBridgeTask {
 
 export interface CheckAgentBridgeResponse {
     success: boolean;
-    task?: AgentBridgeTask;
     message: string;
+    task?: AgentBridgeTask;
 }
 
 export interface CheckRulesSyncResponse {
@@ -270,146 +236,21 @@ export interface CheckRulesSyncResponse {
     offlineMode?: boolean;
 }
 
-export interface RoadmapChunk {
-    id: string;
-    project_id: string;
-    title: string;
-    description: string | null;
-    status: string;
-    priority: string;
-    step_number: number;
-    prompt_content?: string;
-}
-
 // =============================================================================
-// PENDING TASKS SCHEMAS (get_pending_tasks & update_task_status)
+// MISSION & RELEASE TYPES
 // =============================================================================
 
-export const GetPendingTasksInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID')
-});
-
-export const UpdateTaskStatusInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    taskId: z.string().uuid('Invalid task ID'),
-    status: z.enum(['EXECUTING', 'COMPLETED', 'FAILED']),
-    executionSummary: z.string().optional()
-});
-
-export type GetPendingTasksInput = z.infer<typeof GetPendingTasksInputSchema>;
-export type UpdateTaskStatusInput = z.infer<typeof UpdateTaskStatusInputSchema>;
-
-export const GetNextRoadmapStepInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    currentStepId: z.string().uuid('Invalid step ID').optional()
-});
-
-export const GenerateProfessionalPDFInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    reportType: z.enum(['SYSTEM_MANIFEST', 'INVESTOR_REPORT'])
-});
-
-export const ArchaeologicalScanInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    gitLog: z.string().describe('Git log output in format: hash:X\\ndate:X\\nauthor:X\\nmessage\\n---COMMIT---\\n...'),
-    fileTree: z.array(z.string()).describe('Array of file/directory paths in the repository')
-});
-
-export const ImportGhostFeaturesInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    features: z.array(z.object({
-        id: z.string(),
-        title: z.string(),
-        description: z.string(),
-        status: z.literal('COMPLETED'),
-        source: z.enum(['git', 'filesystem', 'combined']),
-        evidence: z.array(z.string()),
-        estimatedCompletionDate: z.string(),
-        priority: z.number()
-    }))
-});
-
-// =============================================================================
-// STRUCTURAL SHIELD (EINAR & SVEN)
-// =============================================================================
-
-export const AnalyzeDependencyGraphInputSchema = z.object({
-    path: z.string().min(1, 'Path is required').default('src')
-});
-
-export const AuditRlsStatusInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID').optional()
-});
-
-export type AnalyzeDependencyGraphInput = z.infer<typeof AnalyzeDependencyGraphInputSchema>;
-export type AuditRlsStatusInput = z.infer<typeof AuditRlsStatusInputSchema>;
-
-// =============================================================================
-// INTELLIGENCE CORE (MAJA & ASTRID)
-// =============================================================================
-
-export const QueryProjectBrainInputSchema = QueryBrainInputSchema; // Reuse existing structure
-
-export const FetchPackageHealthInputSchema = z.object({
-    packageName: z.string().min(1, 'Package name is required')
-});
-
-export type QueryProjectBrainInput = z.infer<typeof QueryProjectBrainInputSchema>;
-export type FetchPackageHealthInput = z.infer<typeof FetchPackageHealthInputSchema>;
-
-// =============================================================================
-// ACTIVE MEMORY & PLANNING (MAJA & KINE)
-// =============================================================================
-
-export const SaveToProjectBrainInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    title: z.string().min(1, 'Title is required'),
-    content: z.string().min(1, 'Content is required'),
-    category: z.enum(['DECISION', 'ARCHITECTURE', 'NOTE', 'LESSON_LEARNED']).default('NOTE'),
-    tags: z.array(z.string()).optional().default([])
-});
-
-export const UpdateRoadmapStatusInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    chunkId: z.string().uuid('Invalid chunk ID'),
-    status: z.enum(['TODO', 'IN_PROGRESS', 'COMPLETED'])
-});
-
-export const AddRoadmapChunkInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID'),
-    title: z.string().min(1, 'Title is required'),
-    description: z.string().optional(),
-    featureId: z.string().optional(),
-    priority: z.enum(['LOW', 'MEDIUM', 'HIGH']).default('MEDIUM')
-});
-
-export type SaveToProjectBrainInput = z.infer<typeof SaveToProjectBrainInputSchema>;
-export type UpdateRoadmapStatusInput = z.infer<typeof UpdateRoadmapStatusInputSchema>;
-export type AddRoadmapChunkInput = z.infer<typeof AddRoadmapChunkInputSchema>;
-
-// =============================================================================
-// UI/UX & RESEARCH (LINUS & ASTRID)
-// =============================================================================
-
-export const AnalyzeUiComponentInputSchema = z.object({
-    filePath: z.string().describe('Absolute path to the component file (.tsx, .css)')
-});
-
-export const ApplyDesignSystemInputSchema = z.object({
-    filePath: z.string().describe('Absolute path to the file to fix')
-});
-
-export const FetchUiLibraryDocsInputSchema = z.object({
-    componentName: z.string().describe('Name of the component (e.g. "button", "card")'),
-    library: z.enum(['shadcn', 'lucide']).default('shadcn')
-});
-
-export type AnalyzeUiComponentInput = z.infer<typeof AnalyzeUiComponentInputSchema>;
-export type ApplyDesignSystemInput = z.infer<typeof ApplyDesignSystemInputSchema>;
-export type FetchUiLibraryDocsInput = z.infer<typeof FetchUiLibraryDocsInputSchema>;
-
-export const GenerateCursorRulesInputSchema = z.object({
-    projectId: z.string().uuid('Invalid project ID')
-});
-
-export type GenerateCursorRulesInput = z.infer<typeof GenerateCursorRulesInputSchema>;
+export interface ReleaseManifest {
+    executive_summary: string;
+    security_certificate: {
+        status: 'PASSED' | 'FAILED' | 'WARNING';
+        message: string;
+        unsecured_tables: any[];
+    };
+    performance_report: {
+        status: 'PASSED' | 'FAILED' | 'WARNING';
+        message: string;
+        issues_found: number;
+    };
+    timestamp: string;
+}

package/src/server/core.ts

@@ -0,0 +1,40 @@
+import { CallToolRequestSchema, McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { AuthContext } from './types.js';
+import { registry } from '../lib/tool-registry.js';
+import {
+    GetProjectContextInputSchema,
+    AuditSecurityIntegrityInputSchema
+} from '../lib/types.js';
+import { getProjectContext } from '../tools/get-project-context.js';
+import { auditSecurityIntegrity } from '../tools/security-tools.js';
+import { watcherState } from './telemetry.js';
+
+export function setupToolHandlers(server: Server, authContext: AuthContext) {
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name } = request.params;
+        const args = request.params.arguments as any;
+
+        // 1. Check Registry (Modern Tools)
+        try {
+            return await registry.callTool(name, args, {
+                supabase: authContext.supabase,
+                userId: authContext.userId
+            });
+        } catch (e: any) {
+            if (!e.message?.includes(`Tool '${name}' not found`)) {
+                throw new McpError(ErrorCode.InvalidParams, e.message);
+            }
+        }
+
+        // 2. Legacy Switch (Modularized)
+        switch (name) {
+            case 'get_agent_status': {
+                return { content: [{ type: 'text', text: JSON.stringify(watcherState, null, 2) }] };
+            }
+
+            default:
+                throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`);
+        }
+    });
+}