Helios-9 MCP Server

// Local type definitions interface MCPTool { name: string description: string inputSchema: any } import { supabaseService } from '../lib/api-client.js' import { requireAuth } from '../lib/auth.js' import { logger } from '../lib/logger.js' import { z } from 'zod' // Input schemas for initiative tools const ListInitiativesSchema = z.object({ project_id: z.string().uuid().optional(), status: z.enum(['planning', 'active', 'on_hold', 'completed', 'cancelled']).optional(), priority: z.enum(['critical', 'high', 'medium', 'low']).optional(), search: z.string().optional(), limit: z.number().int().positive().max(100).default(20) }) const GetInitiativeSchema = z.object({ initiative_id: z.string().uuid() }) const CreateInitiativeSchema = z.object({ name: z.string().min(1).max(255), objective: z.string().min(1), description: z.string().optional(), status: z.enum(['planning', 'active', 'on_hold', 'completed', 'cancelled']).default('planning'), priority: z.enum(['critical', 'high', 'medium', 'low']).default('medium'), owner_id: z.string().uuid(), project_ids: z.array(z.string().uuid()).min(1), // Required: at least one project start_date: z.string().datetime().optional(), target_date: z.string().datetime().optional(), metadata: z.object({}).optional(), tags: z.array(z.string()).optional() }) const UpdateInitiativeSchema = z.object({ initiative_id: z.string().uuid(), name: z.string().min(1).max(255).optional(), objective: z.string().min(1).optional(), description: z.string().optional(), status: z.enum(['planning', 'active', 'on_hold', 'completed', 'cancelled']).optional(), priority: z.enum(['critical', 'high', 'medium', 'low']).optional(), owner_id: z.string().uuid().optional(), start_date: z.string().datetime().optional(), target_date: z.string().datetime().optional(), metadata: z.object({}).optional(), tags: z.array(z.string()).optional() }) /** * List all initiatives */ export const listInitiativesTool: MCPTool = { name: 'list_initiatives', description: 'List all initiatives with optional filtering', inputSchema: { type: 'object', properties: { project_id: { type: 'string', format: 'uuid', description: 'Filter by project' }, status: { type: 'string', enum: ['planning', 'active', 'on_hold', 'completed', 'cancelled'], description: 'Filter by status' }, priority: { type: 'string', enum: ['critical', 'high', 'medium', 'low'], description: 'Filter by priority' }, search: { type: 'string', description: 'Search initiatives by name or objective' }, limit: { type: 'number', minimum: 1, maximum: 100, default: 20, description: 'Maximum number of initiatives to return' } } } } export const listInitiatives = requireAuth(async (args: any) => { const { project_id, status, priority, search, limit } = ListInitiativesSchema.parse(args) logger.info('Listing initiatives', { project_id, status, priority, search, limit }) const initiatives = await supabaseService.getInitiatives( { project_id, status, priority, search }, { limit }, { field: 'updated_at', order: 'desc' } ) // The API already returns enriched initiatives with counts return { initiatives: initiatives, total: initiatives.length, filters_applied: { project_id, status, priority, search } } }) /** * Get initiative details */ export const getInitiativeTool: MCPTool = { name: 'get_initiative', description: 'Get detailed information about a specific initiative', inputSchema: { type: 'object', properties: { initiative_id: { type: 'string', format: 'uuid', description: 'The unique identifier of the initiative' } }, required: ['initiative_id'] } } export const getInitiative = requireAuth(async (args: any) => { const { initiative_id } = GetInitiativeSchema.parse(args) logger.info('Getting initiative details', { initiative_id }) const initiative = await supabaseService.getInitiative(initiative_id) // If the API returns enriched data with tasks, milestones, and documents, calculate from that const statistics = { total_tasks: initiative.tasks?.length || initiative.task_count || 0, completed_tasks: initiative.tasks?.filter((t: any) => t.status === 'done').length || 0, total_milestones: initiative.milestones?.length || initiative.milestone_count || 0, completed_milestones: initiative.milestones?.filter((m: any) => m.status === 'completed').length || 0, total_documents: initiative.documents?.length || initiative.document_count || 0 } const completion_percentage = statistics.total_tasks > 0 ? Math.round((statistics.completed_tasks / statistics.total_tasks) * 100) : 0 return { initiative: { ...initiative, completion_percentage, statistics } } }) /** * Create new initiative */ export const createInitiativeTool: MCPTool = { name: 'create_initiative', description: 'Create a new initiative', inputSchema: { type: 'object', properties: { name: { type: 'string', minLength: 1, maxLength: 255, description: 'The name of the initiative' }, objective: { type: 'string', minLength: 1, description: 'The strategic objective of the initiative' }, description: { type: 'string', description: 'Optional detailed description' }, status: { type: 'string', enum: ['planning', 'active', 'on_hold', 'completed', 'cancelled'], default: 'planning', description: 'Initial status of the initiative' }, priority: { type: 'string', enum: ['critical', 'high', 'medium', 'low'], default: 'medium', description: 'Priority level' }, owner_id: { type: 'string', format: 'uuid', description: 'ID of the initiative owner' }, project_ids: { type: 'array', items: { type: 'string', format: 'uuid' }, minItems: 1, description: 'IDs of projects this initiative belongs to (at least one required)' }, start_date: { type: 'string', format: 'date-time', description: 'Optional start date' }, target_date: { type: 'string', format: 'date-time', description: 'Optional target completion date' }, metadata: { type: 'object', description: 'Optional metadata' }, tags: { type: 'array', items: { type: 'string' }, description: 'Optional tags' } }, required: ['name', 'objective', 'owner_id', 'project_ids'] } } export const createInitiative = requireAuth(async (args: any) => { const initiativeData = CreateInitiativeSchema.parse(args) logger.info('Creating new initiative', { name: initiativeData.name, project_ids: initiativeData.project_ids }) // Create the initiative with project_ids included const initiative = await supabaseService.createInitiative({ name: initiativeData.name, objective: initiativeData.objective, description: initiativeData.description || null, status: initiativeData.status, priority: initiativeData.priority, owner_id: initiativeData.owner_id, start_date: initiativeData.start_date || null, target_date: initiativeData.target_date || null, metadata: initiativeData.metadata || {}, tags: initiativeData.tags || [], project_ids: initiativeData.project_ids // Pass project_ids to API } as any) logger.info('Initiative created successfully', { initiative_id: initiative.id, name: initiative.name }) return { initiative, message: `Initiative "${initiative.name}" created successfully` } }) /** * Update existing initiative */ export const updateInitiativeTool: MCPTool = { name: 'update_initiative', description: 'Update initiative details', inputSchema: { type: 'object', properties: { initiative_id: { type: 'string', format: 'uuid', description: 'The unique identifier of the initiative to update' }, name: { type: 'string', minLength: 1, maxLength: 255, description: 'New name for the initiative' }, objective: { type: 'string', minLength: 1, description: 'New objective' }, description: { type: 'string', description: 'New description' }, status: { type: 'string', enum: ['planning', 'active', 'on_hold', 'completed', 'cancelled'], description: 'New status' }, priority: { type: 'string', enum: ['critical', 'high', 'medium', 'low'], description: 'New priority' }, owner_id: { type: 'string', format: 'uuid', description: 'New owner ID' }, start_date: { type: 'string', format: 'date-time', description: 'New start date' }, target_date: { type: 'string', format: 'date-time', description: 'New target date' }, metadata: { type: 'object', description: 'New metadata' }, tags: { type: 'array', items: { type: 'string' }, description: 'New tags' } }, required: ['initiative_id'] } } export const updateInitiative = requireAuth(async (args: any) => { const { initiative_id, ...updates } = UpdateInitiativeSchema.parse(args) logger.info('Updating initiative', { initiative_id, updates }) const initiative = await supabaseService.updateInitiative(initiative_id, updates) logger.info('Initiative updated successfully', { initiative_id: initiative.id }) return { initiative, message: `Initiative "${initiative.name}" updated successfully` } }) /** * Get initiative context for AI agents */ export const getInitiativeContextTool: MCPTool = { name: 'get_initiative_context', description: 'Get rich context about an initiative for AI understanding', inputSchema: { type: 'object', properties: { initiative_id: { type: 'string', format: 'uuid', description: 'The unique identifier of the initiative' } }, required: ['initiative_id'] } } export const getInitiativeContext = requireAuth(async (args: any) => { const { initiative_id } = GetInitiativeSchema.parse(args) logger.info('Getting initiative context for AI', { initiative_id }) const context = await supabaseService.getInitiativeContext(initiative_id) return { context } }) /** * Get initiative insights */ export const getInitiativeInsightsTool: MCPTool = { name: 'get_initiative_insights', description: 'Get AI-powered insights and recommendations for an initiative', inputSchema: { type: 'object', properties: { initiative_id: { type: 'string', format: 'uuid', description: 'The unique identifier of the initiative' } }, required: ['initiative_id'] } } export const getInitiativeInsights = requireAuth(async (args: any) => { const { initiative_id } = GetInitiativeSchema.parse(args) logger.info('Getting initiative insights', { initiative_id }) const insights = await supabaseService.getInitiativeInsights(initiative_id) return { insights } }) /** * Search workspace */ export const searchWorkspaceTool: MCPTool = { name: 'search_workspace', description: 'Search across all entity types with initiative awareness', inputSchema: { type: 'object', properties: { query: { type: 'string', minLength: 2, description: 'The search query to use' }, filters: { type: 'object', properties: { type: { type: 'string', enum: ['project', 'initiative', 'task', 'document', 'milestone'] }, project_id: { type: 'string', format: 'uuid' }, initiative_id: { type: 'string', format: 'uuid' }, status: { type: 'string' } } }, limit: { type: 'number', minimum: 1, maximum: 100, default: 20 } }, required: ['query'] } } const SearchWorkspaceSchema = z.object({ query: z.string().min(2), filters: z.object({ type: z.enum(['project', 'initiative', 'task', 'document', 'milestone']).optional(), project_id: z.string().uuid().optional(), initiative_id: z.string().uuid().optional(), status: z.string().optional() }).optional(), limit: z.number().int().positive().max(100).default(20) }) export const searchWorkspace = requireAuth(async (args: any) => { const { query, filters, limit } = SearchWorkspaceSchema.parse(args) logger.info('Searching workspace', { query, filters, limit }) const results = await supabaseService.searchWorkspace(query, filters, limit) return results }) /** * Get enhanced project context (includes initiatives) */ export const getEnhancedProjectContextTool: MCPTool = { name: 'get_enhanced_project_context', description: 'Get project context including all initiatives', inputSchema: { type: 'object', properties: { project_id: { type: 'string', format: 'uuid', description: 'The unique identifier of the project' } }, required: ['project_id'] } } export const getEnhancedProjectContext = requireAuth(async (args: any) => { const { project_id } = z.object({ project_id: z.string().uuid() }).parse(args) logger.info('Getting enhanced project context', { project_id }) const context = await supabaseService.getEnhancedProjectContext(project_id) return { context } }) /** * Get workspace context */ export const getWorkspaceContextTool: MCPTool = { name: 'get_workspace_context', description: 'Get complete workspace hierarchy and insights', inputSchema: { type: 'object', properties: {} } } export const getWorkspaceContext = requireAuth(async (args: any) => { logger.info('Getting workspace context') const context = await supabaseService.getWorkspaceContext() return { context } }) /** * Associate document with initiative */ export const associateDocumentWithInitiativeTool: MCPTool = { name: 'associate_document_with_initiative', description: 'Link an existing document to an initiative', inputSchema: { type: 'object', properties: { initiative_id: { type: 'string', format: 'uuid', description: 'The initiative ID to associate the document with' }, document_id: { type: 'string', format: 'uuid', description: 'The document ID to associate with the initiative' } }, required: ['initiative_id', 'document_id'] } } export const associateDocumentWithInitiative = requireAuth(async (args: any) => { const { initiative_id, document_id } = z.object({ initiative_id: z.string().uuid(), document_id: z.string().uuid() }).parse(args) logger.info('Associating document with initiative', { initiative_id, document_id }) // Call the API to create the association const response = await supabaseService.associateDocumentWithInitiative(initiative_id, document_id) logger.info('Document associated with initiative successfully') return { success: true, message: `Document successfully associated with initiative`, initiative_id, document_id } }) /** * Disassociate document from initiative */ export const disassociateDocumentFromInitiativeTool: MCPTool = { name: 'disassociate_document_from_initiative', description: 'Remove the link between a document and an initiative', inputSchema: { type: 'object', properties: { initiative_id: { type: 'string', format: 'uuid', description: 'The initiative ID to disassociate the document from' }, document_id: { type: 'string', format: 'uuid', description: 'The document ID to disassociate from the initiative' } }, required: ['initiative_id', 'document_id'] } } export const disassociateDocumentFromInitiative = requireAuth(async (args: any) => { const { initiative_id, document_id } = z.object({ initiative_id: z.string().uuid(), document_id: z.string().uuid() }).parse(args) logger.info('Disassociating document from initiative', { initiative_id, document_id }) // Call the API to remove the association const response = await supabaseService.disassociateDocumentFromInitiative(initiative_id, document_id) logger.info('Document disassociated from initiative successfully') return { success: true, message: `Document successfully disassociated from initiative`, initiative_id, document_id } }) // Export all initiative tools export const initiativeTools = { listInitiativesTool, getInitiativeTool, createInitiativeTool, updateInitiativeTool, getInitiativeContextTool, getInitiativeInsightsTool, searchWorkspaceTool, getEnhancedProjectContextTool, getWorkspaceContextTool, associateDocumentWithInitiativeTool, disassociateDocumentFromInitiativeTool } export const initiativeHandlers = { list_initiatives: listInitiatives, get_initiative: getInitiative, create_initiative: createInitiative, update_initiative: updateInitiative, get_initiative_context: getInitiativeContext, get_initiative_insights: getInitiativeInsights, search_workspace: searchWorkspace, get_enhanced_project_context: getEnhancedProjectContext, get_workspace_context: getWorkspaceContext, associate_document_with_initiative: associateDocumentWithInitiative, disassociate_document_from_initiative: disassociateDocumentFromInitiative }