@vibescope/mcp-server 0.0.1 → 0.1.0

package/src/handlers/milestones.ts

@@ -7,10 +7,13 @@
  * - complete_milestone
  * - delete_milestone
  * - get_milestones
+ *
+ * MIGRATED: Uses Vibescope API client instead of direct Supabase
  */
 
 import type { Handler, HandlerRegistry } from './types.js';
 import { ValidationError, validateRequired, validateUUID } from '../validators.js';
+import { getApiClient } from '../api-client.js';
 
 export const addMilestone: Handler = async (args, ctx) => {
 	const { task_id, title, description, order_index } = args as {
@@ -24,52 +27,23 @@ export const addMilestone: Handler = async (args, ctx) => {
 	validateUUID(task_id, 'task_id');
 	validateRequired(title, 'title');
 
-	const { supabase, session } = ctx;
-
-	// Verify task exists
-	const { data: task, error: taskError } = await supabase
-		.from('tasks')
-		.select('id, project_id')
-		.eq('id', task_id)
-		.single();
+	const { session } = ctx;
+	const apiClient = getApiClient();
 
-	if (taskError || !task) {
-		throw new Error('Task not found');
-	}
+	const response = await apiClient.addMilestone(task_id, {
+		title,
+		description,
+		order_index
+	}, session.currentSessionId || undefined);
 
-	// Get the next order_index if not provided
-	let orderIdx = order_index;
-	if (orderIdx === undefined) {
-		const { data: maxOrder } = await supabase
-			.from('task_milestones')
-			.select('order_index')
-			.eq('task_id', task_id)
-			.order('order_index', { ascending: false })
-			.limit(1)
-			.single();
-
-		orderIdx = maxOrder ? maxOrder.order_index + 1 : 0;
+	if (!response.ok) {
+		throw new Error(`Failed to add milestone: ${response.error}`);
 	}
 
-	const { data: milestone, error } = await supabase
-		.from('task_milestones')
-		.insert({
-			task_id,
-			title,
-			description: description || null,
-			order_index: orderIdx,
-			created_by: 'agent',
-			created_by_session_id: session.currentSessionId,
-		})
-		.select()
-		.single();
-
-	if (error) throw error;
-
 	return {
 		result: {
 			success: true,
-			milestone,
+			milestone_id: response.data?.milestone_id,
 		},
 	};
 };
@@ -86,39 +60,35 @@ export const updateMilestone: Handler = async (args, ctx) => {
 	validateRequired(milestone_id, 'milestone_id');
 	validateUUID(milestone_id, 'milestone_id');
 
-	const updates: Record<string, unknown> = {};
-	if (title !== undefined) updates.title = title;
-	if (description !== undefined) updates.description = description;
-	if (order_index !== undefined) updates.order_index = order_index;
+	// Validate status if provided
 	if (status !== undefined) {
 		if (!['pending', 'in_progress', 'completed'].includes(status)) {
 			throw new ValidationError('status must be pending, in_progress, or completed');
 		}
-		updates.status = status;
-		if (status === 'completed') {
-			updates.completed_at = new Date().toISOString();
-		} else {
-			updates.completed_at = null;
-		}
 	}
 
-	if (Object.keys(updates).length === 0) {
+	// Check that at least one field is provided
+	if (title === undefined && description === undefined && status === undefined && order_index === undefined) {
 		throw new ValidationError('At least one field to update is required');
 	}
 
-	const { data: milestone, error } = await ctx.supabase
-		.from('task_milestones')
-		.update(updates)
-		.eq('id', milestone_id)
-		.select()
-		.single();
+	const apiClient = getApiClient();
+
+	const response = await apiClient.updateMilestone(milestone_id, {
+		title,
+		description,
+		status: status as 'pending' | 'in_progress' | 'completed' | undefined,
+		order_index
+	});
 
-	if (error) throw error;
+	if (!response.ok) {
+		throw new Error(`Failed to update milestone: ${response.error}`);
+	}
 
 	return {
 		result: {
 			success: true,
-			milestone,
+			milestone: response.data?.milestone,
 		},
 	};
 };
@@ -129,22 +99,18 @@ export const completeMilestone: Handler = async (args, ctx) => {
 	validateRequired(milestone_id, 'milestone_id');
 	validateUUID(milestone_id, 'milestone_id');
 
-	const { data: milestone, error } = await ctx.supabase
-		.from('task_milestones')
-		.update({
-			status: 'completed',
-			completed_at: new Date().toISOString(),
-		})
-		.eq('id', milestone_id)
-		.select()
-		.single();
+	const apiClient = getApiClient();
+
+	const response = await apiClient.completeMilestone(milestone_id);
 
-	if (error) throw error;
+	if (!response.ok) {
+		throw new Error(`Failed to complete milestone: ${response.error}`);
+	}
 
 	return {
 		result: {
 			success: true,
-			milestone,
+			milestone: response.data?.milestone,
 		},
 	};
 };
@@ -155,12 +121,13 @@ export const deleteMilestone: Handler = async (args, ctx) => {
 	validateRequired(milestone_id, 'milestone_id');
 	validateUUID(milestone_id, 'milestone_id');
 
-	const { error } = await ctx.supabase
-		.from('task_milestones')
-		.delete()
-		.eq('id', milestone_id);
+	const apiClient = getApiClient();
 
-	if (error) throw error;
+	const response = await apiClient.deleteMilestone(milestone_id);
+
+	if (!response.ok) {
+		throw new Error(`Failed to delete milestone: ${response.error}`);
+	}
 
 	return {
 		result: {
@@ -176,30 +143,24 @@ export const getMilestones: Handler = async (args, ctx) => {
 	validateRequired(task_id, 'task_id');
 	validateUUID(task_id, 'task_id');
 
-	const { data: milestones, error } = await ctx.supabase
-		.from('task_milestones')
-		.select('*')
-		.eq('task_id', task_id)
-		.order('order_index', { ascending: true });
+	const apiClient = getApiClient();
 
-	if (error) throw error;
+	const response = await apiClient.getMilestones(task_id);
 
-	// Calculate progress stats
-	const total = milestones?.length || 0;
-	const completed = milestones?.filter(m => m.status === 'completed').length || 0;
-	const in_progress = milestones?.filter(m => m.status === 'in_progress').length || 0;
-	const pending = total - completed - in_progress;
-	const progress_percentage = total > 0 ? Math.round((completed / total) * 100) : 0;
+	if (!response.ok) {
+		throw new Error(`Failed to get milestones: ${response.error}`);
+	}
 
+	// Stats are calculated server-side now
 	return {
 		result: {
-			milestones: milestones || [],
-			stats: {
-				total,
-				completed,
-				in_progress,
-				pending,
-				progress_percentage,
+			milestones: response.data?.milestones || [],
+			stats: response.data?.stats || {
+				total: 0,
+				completed: 0,
+				in_progress: 0,
+				pending: 0,
+				progress_percentage: 0,
 			},
 		},
 	};