@vibescope/mcp-server 0.0.1 → 0.2.0

package/src/handlers/cost.ts

@@ -7,9 +7,11 @@
  * - add_cost_alert
  * - update_cost_alert
  * - delete_cost_alert
+ * - get_task_costs
  */
 
 import type { Handler, HandlerRegistry } from './types.js';
+import { getApiClient } from '../api-client.js';
 
 /**
  * Get cost summary for a project (daily, weekly, or monthly)
@@ -21,8 +23,6 @@ export const getCostSummary: Handler = async (args, ctx) => {
 		limit?: number;
 	};
 
-	const { supabase } = ctx;
-
 	if (!project_id) {
 		return {
 			result: { error: 'project_id is required' },
@@ -30,45 +30,17 @@ export const getCostSummary: Handler = async (args, ctx) => {
 		};
 	}
 
-	// Select the appropriate view based on period
-	const viewName = `${period}_cost_summary`;
-
-	const { data, error } = await supabase
-		.from(viewName)
-		.select('*')
-		.eq('project_id', project_id)
-		.order(period === 'daily' ? 'date' : period === 'weekly' ? 'week_start' : 'month_start', { ascending: false })
-		.limit(limit);
+	const apiClient = getApiClient();
+	const response = await apiClient.getCostSummary(project_id, { period, limit });
 
-	if (error) {
+	if (!response.ok) {
 		return {
-			result: { error: `Failed to get cost summary: ${error.message}` },
+			result: { error: response.error || 'Failed to get cost summary' },
 			isError: true,
 		};
 	}
 
-	// Calculate totals
-	const totals = (data || []).reduce(
-		(acc, row) => ({
-			sessions: acc.sessions + (row.session_count || 0),
-			tokens: acc.tokens + (row.total_tokens || 0),
-			calls: acc.calls + (row.total_calls || 0),
-			cost: acc.cost + parseFloat(row.estimated_cost_usd || '0'),
-		}),
-		{ sessions: 0, tokens: 0, calls: 0, cost: 0 }
-	);
-
-	return {
-		result: {
-			period,
-			project_id,
-			summary: data || [],
-			totals: {
-				...totals,
-				cost: Math.round(totals.cost * 100) / 100,
-			},
-		},
-	};
+	return { result: response.data };
 };
 
 /**
@@ -76,33 +48,18 @@ export const getCostSummary: Handler = async (args, ctx) => {
  */
 export const getCostAlerts: Handler = async (args, ctx) => {
 	const { project_id } = args as { project_id?: string };
-	const { supabase, auth } = ctx;
-
-	let query = supabase
-		.from('cost_alerts')
-		.select('*')
-		.eq('user_id', auth.userId)
-		.order('threshold_amount', { ascending: true });
-
-	if (project_id) {
-		query = query.eq('project_id', project_id);
-	}
 
-	const { data, error } = await query;
+	const apiClient = getApiClient();
+	const response = await apiClient.getCostAlerts();
 
-	if (error) {
+	if (!response.ok) {
 		return {
-			result: { error: `Failed to get cost alerts: ${error.message}` },
+			result: { error: response.error || 'Failed to get cost alerts' },
 			isError: true,
 		};
 	}
 
-	return {
-		result: {
-			alerts: data || [],
-			count: data?.length || 0,
-		},
-	};
+	return { result: response.data };
 };
 
 /**
@@ -121,8 +78,6 @@ export const addCostAlert: Handler = async (args, ctx) => {
 		alert_type?: 'warning' | 'critical';
 	};
 
-	const { supabase, auth } = ctx;
-
 	if (!threshold_amount || threshold_amount <= 0) {
 		return {
 			result: { error: 'threshold_amount must be a positive number' },
@@ -137,32 +92,22 @@ export const addCostAlert: Handler = async (args, ctx) => {
 		};
 	}
 
-	const { data, error } = await supabase
-		.from('cost_alerts')
-		.insert({
-			user_id: auth.userId,
-			project_id: project_id || null,
-			threshold_amount,
-			threshold_period,
-			alert_type,
-		})
-		.select()
-		.single();
-
-	if (error) {
+	const apiClient = getApiClient();
+	const response = await apiClient.addCostAlert({
+		project_id,
+		threshold_amount,
+		threshold_period,
+		alert_type
+	});
+
+	if (!response.ok) {
 		return {
-			result: { error: `Failed to create cost alert: ${error.message}` },
+			result: { error: response.error || 'Failed to create cost alert' },
 			isError: true,
 		};
 	}
 
-	return {
-		result: {
-			success: true,
-			alert: data,
-			message: `Alert created: ${alert_type} when ${threshold_period} cost exceeds $${threshold_amount}`,
-		},
-	};
+	return { result: response.data };
 };
 
 /**
@@ -183,8 +128,6 @@ export const updateCostAlert: Handler = async (args, ctx) => {
 		enabled?: boolean;
 	};
 
-	const { supabase, auth } = ctx;
-
 	if (!alert_id) {
 		return {
 			result: { error: 'alert_id is required' },
@@ -192,7 +135,12 @@ export const updateCostAlert: Handler = async (args, ctx) => {
 		};
 	}
 
-	const updates: Record<string, unknown> = {};
+	const updates: {
+		threshold_amount?: number;
+		threshold_period?: 'daily' | 'weekly' | 'monthly';
+		alert_type?: 'warning' | 'critical';
+		enabled?: boolean;
+	} = {};
 	if (threshold_amount !== undefined) updates.threshold_amount = threshold_amount;
 	if (threshold_period !== undefined) updates.threshold_period = threshold_period;
 	if (alert_type !== undefined) updates.alert_type = alert_type;
@@ -205,27 +153,17 @@ export const updateCostAlert: Handler = async (args, ctx) => {
 		};
 	}
 
-	const { data, error } = await supabase
-		.from('cost_alerts')
-		.update(updates)
-		.eq('id', alert_id)
-		.eq('user_id', auth.userId)
-		.select()
-		.single();
+	const apiClient = getApiClient();
+	const response = await apiClient.updateCostAlert(alert_id, updates);
 
-	if (error) {
+	if (!response.ok) {
 		return {
-			result: { error: `Failed to update cost alert: ${error.message}` },
+			result: { error: response.error || 'Failed to update cost alert' },
 			isError: true,
 		};
 	}
 
-	return {
-		result: {
-			success: true,
-			alert: data,
-		},
-	};
+	return { result: response.data };
 };
 
 /**
@@ -233,7 +171,6 @@ export const updateCostAlert: Handler = async (args, ctx) => {
  */
 export const deleteCostAlert: Handler = async (args, ctx) => {
 	const { alert_id } = args as { alert_id: string };
-	const { supabase, auth } = ctx;
 
 	if (!alert_id) {
 		return {
@@ -242,25 +179,17 @@ export const deleteCostAlert: Handler = async (args, ctx) => {
 		};
 	}
 
-	const { error } = await supabase
-		.from('cost_alerts')
-		.delete()
-		.eq('id', alert_id)
-		.eq('user_id', auth.userId);
+	const apiClient = getApiClient();
+	const response = await apiClient.deleteCostAlert(alert_id);
 
-	if (error) {
+	if (!response.ok) {
 		return {
-			result: { error: `Failed to delete cost alert: ${error.message}` },
+			result: { error: response.error || 'Failed to delete cost alert' },
 			isError: true,
 		};
 	}
 
-	return {
-		result: {
-			success: true,
-			deleted_alert_id: alert_id,
-		},
-	};
+	return { result: response.data };
 };
 
 /**
@@ -272,8 +201,6 @@ export const getTaskCosts: Handler = async (args, ctx) => {
 		limit?: number;
 	};
 
-	const { supabase } = ctx;
-
 	if (!project_id) {
 		return {
 			result: { error: 'project_id is required' },
@@ -281,32 +208,17 @@ export const getTaskCosts: Handler = async (args, ctx) => {
 		};
 	}
 
-	const { data, error } = await supabase
-		.from('task_costs')
-		.select('*')
-		.eq('project_id', project_id)
-		.order('estimated_cost_usd', { ascending: false })
-		.limit(limit);
+	const apiClient = getApiClient();
+	const response = await apiClient.getTaskCosts(project_id, limit);
 
-	if (error) {
+	if (!response.ok) {
 		return {
-			result: { error: `Failed to get task costs: ${error.message}` },
+			result: { error: response.error || 'Failed to get task costs' },
 			isError: true,
 		};
 	}
 
-	const totalCost = (data || []).reduce(
-		(sum, task) => sum + parseFloat(task.estimated_cost_usd || '0'),
-		0
-	);
-
-	return {
-		result: {
-			project_id,
-			tasks: data || [],
-			total_cost_usd: Math.round(totalCost * 100) / 100,
-		},
-	};
+	return { result: response.data };
 };
 
 /**