@vibescope/mcp-server 0.5.0 → 0.5.1

package/src/handlers/cloud-agents.ts

@@ -1,156 +1,156 @@
-/**
- * Cloud Agents Handlers
- *
- * Handles cloud agent management:
- * - cleanup_stale_cloud_agents
- * - list_cloud_agents
- */
-
-import type { Handler, HandlerRegistry } from './types.js';
-import { success, error } from './types.js';
-import { parseArgs, uuidValidator, createEnumValidator } from '../validators.js';
-import { getApiClient } from '../api-client.js';
-
-// Argument schemas
-const cleanupStaleAgentsSchema = {
-	project_id: { type: 'string' as const, required: true as const, validate: uuidValidator },
-	stale_minutes: { type: 'number' as const, default: 5 },
-	include_running: { type: 'boolean' as const, default: false },
-	dry_run: { type: 'boolean' as const, default: false },
-};
-
-const listCloudAgentsSchema = {
-	project_id: { type: 'string' as const, required: true as const, validate: uuidValidator },
-	status: { 
-		type: 'string' as const, 
-		default: 'all',
-		validate: createEnumValidator(['starting', 'running', 'stopped', 'failed', 'all'])
-	},
-};
-
-interface CleanupResponse {
-	cleaned: number;
-	failed?: number;
-	wouldClean?: number;
-	dryRun?: boolean;
-	agents: Array<{
-		id: string;
-		name?: string;
-		success?: boolean;
-		error?: string;
-		status?: string;
-		createdAt?: string;
-	}>;
-	message?: string;
-}
-
-interface ListAgentsResponse {
-	agents: Array<{
-		id: string;
-		name?: string;
-		status: string;
-		created_at: string;
-		last_heartbeat?: string;
-		public_ip?: string;
-		ecs_task_id?: string;
-	}>;
-}
-
-/**
- * Clean up stale cloud agents that failed to start or lost connection.
- * Only operates on agents in the specified project (security scoped).
- */
-export const cleanupStaleCloudAgents: Handler = async (args, _ctx) => {
-	const { project_id, stale_minutes, include_running, dry_run } = parseArgs(args, cleanupStaleAgentsSchema);
-
-	const apiClient = getApiClient();
-
-	// Call the cleanup endpoint via fetch (since it's a new endpoint not in the client)
-	const response = await apiClient.proxy<CleanupResponse>('cleanup_stale_cloud_agents', {
-		project_id,
-		staleMinutes: stale_minutes,
-		includeRunning: include_running,
-		dryRun: dry_run,
-	});
-
-	if (!response.ok || !response.data) {
-		return error(response.error || 'Failed to cleanup stale agents');
-	}
-
-	const data = response.data!;
-
-	if (dry_run) {
-		return success({
-			dryRun: true,
-			wouldClean: data.wouldClean || 0,
-			agents: data.agents,
-			message: `Found ${data.wouldClean || 0} stale agents that would be cleaned up`
-		});
-	}
-
-	return success({
-		cleaned: data.cleaned,
-		failed: data.failed || 0,
-		agents: data.agents,
-		message: `Cleaned up ${data.cleaned} stale agents`
-	});
-};
-
-/**
- * List cloud agents for a project with optional status filter.
- */
-export const listCloudAgents: Handler = async (args, _ctx) => {
-	const { project_id, status } = parseArgs(args, listCloudAgentsSchema);
-
-	const apiClient = getApiClient();
-
-	const response = await apiClient.proxy<ListAgentsResponse>('list_cloud_agents', {
-		project_id,
-		status: status === 'all' ? undefined : status,
-	});
-
-	if (!response.ok || !response.data) {
-		return error(response.error || 'Failed to list cloud agents');
-	}
-
-	return success({
-		agents: response.data!.agents,
-		count: response.data!.agents.length
-	});
-};
-
-/**
- * Update agent status message (for dashboard display).
- */
-export const updateAgentStatus: Handler = async (args, ctx) => {
-	const statusMessage = args.status_message as string;
-	const projectId = (args.project_id || ctx.session.currentProjectId) as string;
-	const agentName = args.agent_name as string | undefined;
-
-	if (!statusMessage) {
-		return error('status_message is required');
-	}
-
-	const apiClient = getApiClient();
-	
-	const response = await apiClient.proxy<{ success: boolean }>('update_agent_status', {
-		project_id: projectId,
-		agent_name: agentName,
-		status_message: statusMessage,
-	});
-
-	if (!response.ok) {
-		return error(response.error || 'Failed to update status');
-	}
-
-	return success({ updated: true, status_message: statusMessage });
-};
-
-/**
- * Cloud agents handlers registry
- */
-export const cloudAgentHandlers: HandlerRegistry = {
-	cleanup_stale_cloud_agents: cleanupStaleCloudAgents,
-	list_cloud_agents: listCloudAgents,
-	update_agent_status: updateAgentStatus,
-};
+/**
+ * Cloud Agents Handlers
+ *
+ * Handles cloud agent management:
+ * - cleanup_stale_cloud_agents
+ * - list_cloud_agents
+ */
+
+import type { Handler, HandlerRegistry } from './types.js';
+import { success, error } from './types.js';
+import { parseArgs, uuidValidator, createEnumValidator } from '../validators.js';
+import { getApiClient } from '../api-client.js';
+
+// Argument schemas
+const cleanupStaleAgentsSchema = {
+	project_id: { type: 'string' as const, required: true as const, validate: uuidValidator },
+	stale_minutes: { type: 'number' as const, default: 5 },
+	include_running: { type: 'boolean' as const, default: false },
+	dry_run: { type: 'boolean' as const, default: false },
+};
+
+const listCloudAgentsSchema = {
+	project_id: { type: 'string' as const, required: true as const, validate: uuidValidator },
+	status: { 
+		type: 'string' as const, 
+		default: 'all',
+		validate: createEnumValidator(['starting', 'running', 'stopped', 'failed', 'all'])
+	},
+};
+
+interface CleanupResponse {
+	cleaned: number;
+	failed?: number;
+	wouldClean?: number;
+	dryRun?: boolean;
+	agents: Array<{
+		id: string;
+		name?: string;
+		success?: boolean;
+		error?: string;
+		status?: string;
+		createdAt?: string;
+	}>;
+	message?: string;
+}
+
+interface ListAgentsResponse {
+	agents: Array<{
+		id: string;
+		name?: string;
+		status: string;
+		created_at: string;
+		last_heartbeat?: string;
+		public_ip?: string;
+		ecs_task_id?: string;
+	}>;
+}
+
+/**
+ * Clean up stale cloud agents that failed to start or lost connection.
+ * Only operates on agents in the specified project (security scoped).
+ */
+export const cleanupStaleCloudAgents: Handler = async (args, _ctx) => {
+	const { project_id, stale_minutes, include_running, dry_run } = parseArgs(args, cleanupStaleAgentsSchema);
+
+	const apiClient = getApiClient();
+
+	// Call the cleanup endpoint via fetch (since it's a new endpoint not in the client)
+	const response = await apiClient.proxy<CleanupResponse>('cleanup_stale_cloud_agents', {
+		project_id,
+		staleMinutes: stale_minutes,
+		includeRunning: include_running,
+		dryRun: dry_run,
+	});
+
+	if (!response.ok || !response.data) {
+		return error(response.error || 'Failed to cleanup stale agents');
+	}
+
+	const data = response.data!;
+
+	if (dry_run) {
+		return success({
+			dryRun: true,
+			wouldClean: data.wouldClean || 0,
+			agents: data.agents,
+			message: `Found ${data.wouldClean || 0} stale agents that would be cleaned up`
+		});
+	}
+
+	return success({
+		cleaned: data.cleaned,
+		failed: data.failed || 0,
+		agents: data.agents,
+		message: `Cleaned up ${data.cleaned} stale agents`
+	});
+};
+
+/**
+ * List cloud agents for a project with optional status filter.
+ */
+export const listCloudAgents: Handler = async (args, _ctx) => {
+	const { project_id, status } = parseArgs(args, listCloudAgentsSchema);
+
+	const apiClient = getApiClient();
+
+	const response = await apiClient.proxy<ListAgentsResponse>('list_cloud_agents', {
+		project_id,
+		status: status === 'all' ? undefined : status,
+	});
+
+	if (!response.ok || !response.data) {
+		return error(response.error || 'Failed to list cloud agents');
+	}
+
+	return success({
+		agents: response.data!.agents,
+		count: response.data!.agents.length
+	});
+};
+
+/**
+ * Update agent status message (for dashboard display).
+ */
+export const updateAgentStatus: Handler = async (args, ctx) => {
+	const statusMessage = args.status_message as string;
+	const projectId = (args.project_id || ctx.session.currentProjectId) as string;
+	const agentName = args.agent_name as string | undefined;
+
+	if (!statusMessage) {
+		return error('status_message is required');
+	}
+
+	const apiClient = getApiClient();
+	
+	const response = await apiClient.proxy<{ success: boolean }>('update_agent_status', {
+		project_id: projectId,
+		agent_name: agentName,
+		status_message: statusMessage,
+	});
+
+	if (!response.ok) {
+		return error(response.error || 'Failed to update status');
+	}
+
+	return success({ updated: true, status_message: statusMessage });
+};
+
+/**
+ * Cloud agents handlers registry
+ */
+export const cloudAgentHandlers: HandlerRegistry = {
+	cleanup_stale_cloud_agents: cleanupStaleCloudAgents,
+	list_cloud_agents: listCloudAgents,
+	update_agent_status: updateAgentStatus,
+};