@vibescope/mcp-server 0.0.1 → 0.1.0

package/src/handlers/discovery.test.ts

@@ -1,35 +1,6 @@
-import { describe, it, expect, vi } from 'vitest';
-import type { SupabaseClient } from '@supabase/supabase-js';
-import type { HandlerContext, TokenUsage } from './types.js';
+import { describe, it, expect } from 'vitest';
 import { discoverTools, getToolInfo } from './discovery.js';
-
-// ============================================================================
-// Test Utilities
-// ============================================================================
-
-function createMockSupabase() {
-	return {} as unknown as SupabaseClient;
-}
-
-function createMockContext(supabase: SupabaseClient): HandlerContext {
-	const defaultTokenUsage: TokenUsage = {
-		callCount: 5,
-		totalTokens: 2500,
-		byTool: {},
-	};
-
-	return {
-		supabase,
-		auth: { userId: 'user-123', apiKeyId: 'api-key-123' },
-		session: {
-			instanceId: 'instance-abc',
-			currentSessionId: 'session-123',
-			currentPersona: 'Wave',
-			tokenUsage: defaultTokenUsage,
-		},
-		updateSession: vi.fn(),
-	};
-}
+import { createMockContext } from './__test-utils__.js';
 
 // ============================================================================
 // discoverTools Tests
@@ -37,8 +8,7 @@ function createMockContext(supabase: SupabaseClient): HandlerContext {
 
 describe('discoverTools', () => {
 	it('should return all categories when no category specified', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: unknown[]; total_tools: number };
@@ -50,8 +20,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return category info with tool counts', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string; description: string; tool_count: number }[] };
@@ -63,8 +32,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return tools for specific category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({ category: 'session' }, ctx);
 		const res = result.result as { category: string; description: string; tools: unknown[] };
@@ -76,8 +44,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return tools with name and brief for specific category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({ category: 'tasks' }, ctx);
 		const res = result.result as { tools: { name: string; brief: string }[] };
@@ -89,8 +56,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should return error for unknown category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({ category: 'nonexistent' }, ctx);
 		const res = result.result as { error: string; available: string[] };
@@ -101,8 +67,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include session category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string }[] };
@@ -112,8 +77,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include tasks category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string }[] };
@@ -123,8 +87,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include deployment category', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { categories: { name: string }[] };
@@ -134,8 +97,7 @@ describe('discoverTools', () => {
 	});
 
 	it('should include usage hint in response', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await discoverTools({}, ctx);
 		const res = result.result as { usage: string };
@@ -152,8 +114,7 @@ describe('discoverTools', () => {
 
 describe('getToolInfo', () => {
 	it('should return error for missing tool_name', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({}, ctx);
 		const res = result.result as { error: string };
@@ -162,8 +123,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return error for unknown tool', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'nonexistent_tool' }, ctx);
 		const res = result.result as { error: string; hint: string };
@@ -173,8 +133,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for start_work_session', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'start_work_session' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -185,8 +144,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for add_task', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'add_task' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -197,8 +155,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for complete_task', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'complete_task' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -208,8 +165,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for request_deployment', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'request_deployment' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -220,8 +176,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info with parameters section', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'add_task' }, ctx);
 		const res = result.result as { info: string };
@@ -230,8 +185,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for get_help', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'get_help' }, ctx);
 		const res = result.result as { tool: string; info: string };
@@ -242,8 +196,7 @@ describe('getToolInfo', () => {
 	});
 
 	it('should return info for discover_tools', async () => {
-		const supabase = createMockSupabase();
-		const ctx = createMockContext(supabase);
+		const ctx = createMockContext();
 
 		const result = await getToolInfo({ tool_name: 'discover_tools' }, ctx);
 		const res = result.result as { tool: string; info: string };

package/src/handlers/discovery.ts

@@ -186,6 +186,12 @@ const TOOL_CATEGORIES: Record<string, { description: string; tools: Array<{ name
 			{ name: 'get_task_costs', brief: 'Cost per task' },
 		],
 	},
+	knowledge: {
+		description: 'Queryable knowledge base from project data',
+		tools: [
+			{ name: 'query_knowledge_base', brief: 'Aggregated project knowledge in one call' },
+		],
+	},
 };
 
 // Detailed tool info (fetched on-demand)
@@ -899,6 +905,32 @@ Get cost breakdown by task for a project.
 - limit: Max tasks to return (default: 20)
 
 **Returns:** Tasks sorted by estimated cost with model breakdown`,
+
+	// Knowledge base tools
+	query_knowledge_base: `# query_knowledge_base
+Query aggregated project knowledge in a single call. Reduces token usage by combining multiple data sources.
+
+**Parameters:**
+- project_id (required): Project UUID
+- scope: 'summary' (default) or 'detailed' (includes rationales, descriptions)
+- categories: Array of categories to include (default: all)
+  - findings: Audit findings (security, performance, code quality)
+  - qa: Questions and answers
+  - decisions: Architectural decisions
+  - completed_tasks: Tasks with completion summaries
+  - blockers: Resolved blockers with resolution notes
+  - progress: Recent progress logs
+- limit: Max items per category (default: 5, max: 20)
+- search_query: Optional text search across all knowledge
+
+**Returns:**
+- project: name, goal, tech_stack
+- stats: counts for each category, findings by severity
+- Category data based on selection
+
+**Token savings:** Replaces up to 6 separate tool calls (get_findings, get_decisions, get_tasks, get_blockers, etc.) with one call.
+
+**Example:** query_knowledge_base(project_id, categories: ["findings", "decisions"], limit: 10)`,
 };
 
 export const discoverTools: Handler = async (args) => {