@rigstate/mcp 0.7.6 → 0.7.7

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@rigstate/mcp",
-    "version": "0.7.6",
+    "version": "0.7.7",
     "description": "Rigstate MCP Server - Model Context Protocol for AI Editors",
     "type": "module",
     "main": "./dist/index.js",

package/src/lib/project-context-utils.ts

@@ -1,4 +1,4 @@
-import { ProjectContextResponse, TechStackInfo } from '../lib/types.js';
+import { TechStackInfo } from '../lib/types.js';
 
 export async function buildProjectSummary(
     project: any,
@@ -8,14 +8,15 @@ export async function buildProjectSummary(
     agentTasks: any[],
     roadmapItems: any[],
     stackDef: any,
-    supabase: any
+    supabase: any,
+    userId: string
 ): Promise<string> {
     const summaryParts: string[] = [];
 
     summaryParts.push(`Project Type: ${project.project_type?.toUpperCase() || 'UNKNOWN'}`);
 
     // Active Mission Parameters
-    if (stackDef) {
+    if (stackDef || activeTask || nextTask) {
         summaryParts.push('\n=== ACTIVE MISSION PARAMETERS ===');
         if (activeTask) {
             summaryParts.push(`⚠️  CURRENT OBJECTIVE: T-${activeTask.step_number}: ${activeTask.title}`);
@@ -44,17 +45,23 @@ export async function buildProjectSummary(
                 summaryParts.push(`   Tags: ${activeTask.tags.join(', ')}`);
             }
 
-            // Enhanced Feature Context
+            // Enhanced Feature Context (Secure)
             if (activeTask.feature_id) {
-                const { data: feature } = await supabase
-                    .from('project_features')
-                    .select('name, description')
-                    .eq('id', activeTask.feature_id)
-                    .single();
-
-                if (feature) {
-                    summaryParts.push(`\n   Parent Feature: ${feature.name}`);
-                    summaryParts.push(`   Feature Vision: ${feature.description}`);
+                try {
+                    const { data: dbFeatures } = await supabase
+                        .rpc('get_project_features_secure', {
+                            p_project_id: project.id,
+                            p_user_id: userId
+                        });
+
+                    const feature = dbFeatures?.find((f: any) => f.id === activeTask.feature_id);
+
+                    if (feature) {
+                        summaryParts.push(`\n   Parent Feature: ${feature.name}`);
+                        summaryParts.push(`   Feature Vision: ${feature.description}`);
+                    }
+                } catch (e) {
+                    console.warn('Feature context fetch failed', e);
                 }
             }
 
@@ -80,13 +87,15 @@ export async function buildProjectSummary(
         }
 
         summaryParts.push('\n=== CURRENT STACK ===');
-        if (stackDef.frontend) summaryParts.push(`Frontend: ${stackDef.frontend.framework} (${stackDef.frontend.language})`);
-        if (stackDef.backend) summaryParts.push(`Backend: ${stackDef.backend.service} (${stackDef.backend.database})`);
-        if (stackDef.styling) summaryParts.push(`Styling: ${stackDef.styling.framework} ${stackDef.styling.library || ''}`);
-        if (stackDef.hosting) summaryParts.push(`Infrastructure: ${stackDef.hosting.provider}`);
-    } else {
-        if (techStack.framework) summaryParts.push(`Framework: ${techStack.framework}`);
-        if (techStack.orm) summaryParts.push(`ORM: ${techStack.orm}`);
+        if (stackDef) {
+            if (stackDef.frontend) summaryParts.push(`Frontend: ${stackDef.frontend.framework} (${stackDef.frontend.language})`);
+            if (stackDef.backend) summaryParts.push(`Backend: ${stackDef.backend.service} (${stackDef.backend.database})`);
+            if (stackDef.styling) summaryParts.push(`Styling: ${stackDef.styling.framework} ${stackDef.styling.library || ''}`);
+            if (stackDef.hosting) summaryParts.push(`Infrastructure: ${stackDef.hosting.provider}`);
+        } else {
+            if (techStack.framework) summaryParts.push(`Framework: ${techStack.framework}`);
+            if (techStack.orm) summaryParts.push(`ORM: ${techStack.orm}`);
+        }
     }
 
     if (project.description) {
@@ -103,8 +112,8 @@ export async function buildProjectSummary(
 
         if (spec.featureList && Array.isArray(spec.featureList)) {
             summaryParts.push('\nKey Features & Nuances:');
-            spec.featureList.filter((f: any) => f.priority === 'MVP').forEach((f: any) => {
-                summaryParts.push(`- ${f.name}: ${f.description}`);
+            spec.featureList.filter((f: any) => f.priority === 'MVP' || f.priority === 'HIGH').slice(0, 10).forEach((f: any) => {
+                summaryParts.push(`- ${f.name}: ${f.description?.substring(0, 200)}`);
             });
         }
     }
@@ -124,7 +133,7 @@ export async function buildProjectSummary(
         summaryParts.push('\nLatest AI Executions:');
         agentTasks.forEach((t: any) => {
             const time = t.completed_at ? new Date(t.completed_at).toLocaleString() : 'Recently';
-            summaryParts.push(`- [${time}] ${t.roadmap_title || 'Task'}: ${t.execution_summary || 'Completed'}`);
+            summaryParts.push(`- [${time}] ${t.roadmap_title || 'Task'}: ${t.execution_summary?.substring(0, 100) || 'Completed'}`);
         });
     }
 

package/src/tools/get-project-context.ts

@@ -165,7 +165,6 @@ export async function getProjectContext(
         )].slice(0, 10) as string[];
     }
 
-    // Build summary via utility
     const summary = await buildProjectSummary(
         project,
         techStack,
@@ -174,7 +173,8 @@ export async function getProjectContext(
         agentTasks || [],
         roadmapItems || [],
         stackDef,
-        supabase
+        supabase,
+        userId
     );
 
     const response: ProjectContextResponse = {

package/src/tools/list-features.ts

@@ -17,53 +17,41 @@ export const listFeaturesTool: ToolDefinition<typeof InputSchema> = {
 Useful for understanding the strategic context and major milestones.`,
     schema: InputSchema,
     handler: async ({ projectId }, { supabase, userId }) => {
-        // 1. Verify project access
-        const { data: hasAccess, error: accessError } = await supabase
-            .rpc('check_project_access_secure', {
+        // 1. Verify project access and fetch context
+        const { data: projectRow, error: projectError } = await supabase
+            .rpc('get_project_context_secure', {
                 p_project_id: projectId,
                 p_user_id: userId
-            });
+            })
+            .single() as any;
 
-        if (accessError || !hasAccess) {
-            throw new Error('Project not found or access denied');
+        if (projectError || !projectRow) {
+            throw new Error('Project details not found or access denied');
         }
 
-        // 2. Fetch project details
-        const { data: project, error: projectError } = await supabase
-            .from('projects')
-            .select('id, functional_spec')
-            .eq('id', projectId)
-            .single();
-
-        if (projectError || !project) {
-            throw new Error('Project details not found');
-        }
-
-        // 2. Primary Strategy: Fetch from 'project_features'
+        // 2. Primary Strategy: Fetch from 'project_features' via secure RPC
         const { data: dbFeatures, error: dbError } = await supabase
-            .from('project_features')
-            .select('id, name, description, status')
-            .eq('project_id', projectId)
-            .neq('status', 'shadow') // Exclude shadow features by default unless asked
-            .order('created_at', { ascending: false });
+            .rpc('get_project_features_secure', {
+                p_project_id: projectId,
+                p_user_id: userId
+            });
 
         let featuresList: any[] = [];
         let source = 'DB';
 
         if (!dbError && dbFeatures && dbFeatures.length > 0) {
-            featuresList = dbFeatures.map(f => ({
+            featuresList = dbFeatures.map((f: any) => ({
                 ...f,
-                title: f.name // Map back to title specifically for uniform handling below
+                title: f.name
             }));
         } else {
             // 3. Fallback Strategy: Extract from 'functional_spec'
             source = 'FALLBACK_SPEC';
-            // Log warning (In a real system, use a structured logger. Here we console.error to stderr)
-            console.error(`[WARN] Project ${projectId}: 'project_features' empty or missing. Falling back to 'functional_spec'.`);
+            const spec = projectRow.functional_spec as any;
+            const features = spec?.featureList || spec?.features;
 
-            const spec = project.functional_spec as any;
-            if (spec && typeof spec === 'object' && Array.isArray(spec.features)) {
-                featuresList = spec.features.map((f: any) => ({
+            if (Array.isArray(features)) {
+                featuresList = features.map((f: any) => ({
                     id: 'legacy',
                     title: f.name || f.title,
                     description: f.description,
@@ -89,5 +77,3 @@ Useful for understanding the strategic context and major milestones.`,
 };
 
 registry.register(listFeaturesTool);
-
-

package/src/tools/planning-tools.ts

@@ -60,25 +60,21 @@ export async function saveToProjectBrain(
 
     const fullContent = `# ${title}\n\n${content}`;
 
-    const { data, error } = await supabase
-        .from('project_memories')
-        .insert({
-            project_id: projectId,
-            content: fullContent,
-            category: category.toLowerCase(),
-            tags: tags,
-            importance: (category === 'DECISION' || category === 'ARCHITECTURE') ? 9 : 5,
-            is_active: true,
-            source_type: 'chat_manual'
-        })
-        .select('id')
-        .single();
+    const { data: memoryId, error } = await supabase
+        .rpc('save_project_memory_secure', {
+            p_project_id: projectId,
+            p_user_id: userId,
+            p_content: fullContent,
+            p_category: category.toLowerCase(),
+            p_tags: tags || [],
+            p_importance: (category === 'DECISION' || category === 'ARCHITECTURE') ? 9 : 5
+        });
 
     if (error) throw new Error(`Failed to save memory: ${error.message}`);
 
     return {
         success: true,
-        memoryId: data.id,
+        memoryId: memoryId,
         message: `✅ Saved [${category}] "${title}" to Project Brain.`
     };
 }

package/src/tools/query-brain.ts

@@ -87,42 +87,22 @@ export async function queryBrain(
     }
 
     // Try semantic search first using the match_memories RPC
-    // This requires the embedding to be generated, so we'll try
-    // Generate embedding if possible for semantic search
     const embedding = await generateQueryEmbedding(query);
     let memories: MemoryRecord[] = [];
 
-    // Use the hybrid search RPC
+    // Use the secure search RPC
     const { data: searchResults, error: searchError } = await supabase
-        .rpc('hybrid_search_memories', {
+        .rpc('query_project_brain_secure', {
             p_project_id: projectId,
+            p_user_id: userId,
             p_query: query,
-            p_embedding: embedding || null,
-            p_limit: limit,
-            p_similarity_threshold: threshold || 0.1
+            p_limit: limit
         });
 
-    if (searchError) {
-        // Fallback to basic recent fetch if RPC fails
-        const { data: recentMemories } = await supabase
-            .from('project_memories')
-            .select('id, content, category, tags, importance, created_at')
-            .eq('project_id', projectId)
-            .eq('is_active', true)
-            .order('created_at', { ascending: false })
-            .limit(limit);
-
-        if (recentMemories) {
-            memories = recentMemories.map(m => ({
-                id: m.id,
-                content: m.content,
-                category: m.category || 'general',
-                tags: m.tags || [],
-                netVotes: m.importance || 0,
-                createdAt: m.created_at
-            }));
-        }
-    } else if (searchResults) {
+    if (searchError || !searchResults) {
+        console.error('Brain query failed:', searchError);
+        memories = [];
+    } else {
         memories = searchResults.map((m: any) => ({
             id: m.id,
             content: m.content,
@@ -133,31 +113,39 @@ export async function queryBrain(
         }));
     }
 
-    // --- NEW: FETCH RELEVANT FEATURES ---
-    // Simple fuzzy search until we vector embedding for features.
+    // --- FETCH RELEVANT FEATURES FROM SPEC ---
     let relevantFeatures: any[] = [];
     try {
-        const { data: features } = await supabase
-            .from('project_features')
-            .select('name, status, description')
-            .eq('project_id', projectId)
-            .or(`name.ilike.%${query}%,description.ilike.%${query}%`)
-            .limit(3);
-
-        if (features) relevantFeatures = features;
+        const { data: projectRow } = await supabase
+            .rpc('get_project_context_secure', {
+                p_project_id: projectId,
+                p_user_id: userId
+            })
+            .single() as any;
+
+        if (projectRow?.functional_spec) {
+            const spec = typeof projectRow.functional_spec === 'string'
+                ? JSON.parse(projectRow.functional_spec)
+                : projectRow.functional_spec;
+
+            const features = spec.featureList || spec.features || [];
+            relevantFeatures = features
+                .filter((f: any) => (f.name || f.title)?.toLowerCase().includes(query.toLowerCase()))
+                .slice(0, 3);
+        }
     } catch (e) {
         console.warn('Feature fetch failed in brain query', e);
     }
 
     // Format memories into a readable context block
-    const contextLines = memories.map((m) => {
+    const contextLines = memories.map((m: any) => {
         const voteIndicator = m.netVotes && m.netVotes < 0 ? ` [⚠️ POORLY RATED: ${m.netVotes}]` : '';
         const tagStr = m.tags && m.tags.length > 0 ? ` [${m.tags.join(', ')}]` : '';
         const category = m.category ? m.category.toUpperCase() : 'GENERAL';
         return `- [${category}]${tagStr}${voteIndicator}: ${m.content}`;
     });
 
-    const searchType = embedding ? 'TRIPLE-HYBRID (Vector + FTS + Fuzzy)' : 'HYBRID (FTS + Fuzzy)';
+    const searchType = embedding ? 'TRIPLE-HYBRID (Vector + FTS + Fuzzy)' : 'SECURE-GATEWAY (Fuzzy)';
 
     let formatted = `=== PROJECT BRAIN: RELEVANT MEMORIES ===
 Search Mode: ${searchType}
@@ -165,7 +153,7 @@ Query: "${query}"`;
 
     if (relevantFeatures.length > 0) {
         formatted += `\n\n=== RELATED FEATURES ===\n` +
-            relevantFeatures.map((f: any) => `- ${f.name} [${f.status}]`).join('\n');
+            relevantFeatures.map((f: any) => `- ${f.name || f.title} [${f.status || 'Active'}]`).join('\n');
     }
 
     formatted += `\n\nFound ${memories.length} relevant memories:\n\n${contextLines.join('\n')}\n\n==========================================`;