@vibescope/mcp-server 0.2.3 → 0.2.5

package/src/handlers/roles.ts

@@ -1,208 +1,226 @@
-/**
- * Role Management Handlers
- *
- * Handles agent role configuration and assignment:
- * - get_role_settings: Get role configuration for a project
- * - update_role_settings: Update role settings for a project
- * - set_session_role: Set the role for the current session
- * - get_agents_by_role: Get active agents grouped by their assigned roles
- */
-
-import type { Handler, HandlerRegistry, AgentRole } from './types.js';
-import { getApiClient } from '../api-client.js';
-
-export const getRoleSettings: Handler = async (args, _ctx) => {
-	const { project_id } = args as { project_id: string };
-
-	if (!project_id) {
-		return {
-			result: { error: 'project_id is required' },
-		};
-	}
-
-	const apiClient = getApiClient();
-	const response = await apiClient.proxy<{
-		roles: Array<{
-			role: AgentRole;
-			enabled: boolean;
-			display_name: string | null;
-			description: string | null;
-			priority_filter: number[] | null;
-			fallback_activities: string[] | null;
-			auto_assign_validation: boolean;
-			auto_assign_deployment: boolean;
-		}>;
-	}>('get_role_settings', { project_id });
-
-	if (!response.ok) {
-		return {
-			result: { error: response.error || 'Failed to get role settings' },
-		};
-	}
-
-	return { result: response.data };
-};
-
-export const updateRoleSettings: Handler = async (args, _ctx) => {
-	const {
-		project_id,
-		role,
-		enabled,
-		display_name,
-		description,
-		priority_filter,
-		fallback_activities,
-		auto_assign_validation,
-		auto_assign_deployment,
-	} = args as {
-		project_id: string;
-		role: AgentRole;
-		enabled?: boolean;
-		display_name?: string;
-		description?: string;
-		priority_filter?: number[];
-		fallback_activities?: string[];
-		auto_assign_validation?: boolean;
-		auto_assign_deployment?: boolean;
-	};
-
-	if (!project_id) {
-		return {
-			result: { error: 'project_id is required' },
-		};
-	}
-
-	if (!role) {
-		return {
-			result: { error: 'role is required' },
-		};
-	}
-
-	const apiClient = getApiClient();
-	const response = await apiClient.proxy<{
-		success: boolean;
-		role: AgentRole;
-	}>('update_role_settings', {
-		project_id,
-		role,
-		enabled,
-		display_name,
-		description,
-		priority_filter,
-		fallback_activities,
-		auto_assign_validation,
-		auto_assign_deployment,
-	});
-
-	if (!response.ok) {
-		return {
-			result: { error: response.error || 'Failed to update role settings' },
-		};
-	}
-
-	return { result: response.data };
-};
-
-export const setSessionRole: Handler = async (args, ctx) => {
-	const { role, role_config } = args as {
-		role: AgentRole;
-		role_config?: Record<string, unknown>;
-	};
-	const { session, updateSession } = ctx;
-
-	if (!role) {
-		return {
-			result: { error: 'role is required' },
-		};
-	}
-
-	const validRoles: AgentRole[] = ['developer', 'validator', 'deployer', 'reviewer', 'maintainer'];
-	if (!validRoles.includes(role)) {
-		return {
-			result: {
-				error: `Invalid role: ${role}. Must be one of: ${validRoles.join(', ')}`,
-			},
-		};
-	}
-
-	// Update local session state
-	updateSession({ currentRole: role });
-
-	// If there's an active session, update it on the server too
-	if (session.currentSessionId) {
-		const apiClient = getApiClient();
-		const response = await apiClient.proxy<{
-			success: boolean;
-			session_id: string;
-			role: AgentRole;
-		}>('set_session_role', {
-			session_id: session.currentSessionId,
-			role,
-			role_config,
-		});
-
-		if (!response.ok) {
-			return {
-				result: { error: response.error || 'Failed to update session role' },
-			};
-		}
-
-		return {
-			result: {
-				success: true,
-				session_id: session.currentSessionId,
-				role,
-				message: `Session role updated to ${role}. Task filtering and fallback suggestions will now be optimized for this role.`,
-			},
-		};
-	}
-
-	return {
-		result: {
-			success: true,
-			role,
-			message: `Local role set to ${role}. Start a session to persist this role.`,
-		},
-	};
-};
-
-export const getAgentsByRole: Handler = async (args, _ctx) => {
-	const { project_id } = args as { project_id: string };
-
-	if (!project_id) {
-		return {
-			result: { error: 'project_id is required' },
-		};
-	}
-
-	const apiClient = getApiClient();
-	const response = await apiClient.proxy<{
-		agents_by_role: Record<AgentRole, Array<{
-			session_id: string;
-			agent_name: string;
-			status: string;
-			current_task_id: string | null;
-			current_task_title: string | null;
-			last_synced_at: string;
-		}>>;
-		total_active: number;
-	}>('get_agents_by_role', { project_id });
-
-	if (!response.ok) {
-		return {
-			result: { error: response.error || 'Failed to get agents by role' },
-		};
-	}
-
-	return { result: response.data };
-};
-
-/**
- * Role handlers registry
- */
-export const roleHandlers: HandlerRegistry = {
-	get_role_settings: getRoleSettings,
-	update_role_settings: updateRoleSettings,
-	set_session_role: setSessionRole,
-	get_agents_by_role: getAgentsByRole,
-};
+/**
+ * Role Management Handlers
+ *
+ * Handles agent role configuration and assignment:
+ * - get_role_settings: Get role configuration for a project
+ * - update_role_settings: Update role settings for a project
+ * - set_session_role: Set the role for the current session
+ * - get_agents_by_role: Get active agents grouped by their assigned roles
+ */
+
+import type { Handler, HandlerRegistry, AgentRole } from './types.js';
+import { getApiClient } from '../api-client.js';
+
+export const getRoleSettings: Handler = async (args, _ctx) => {
+	const { project_id } = args as { project_id: string };
+
+	if (!project_id) {
+		return {
+			result: { error: 'project_id is required' },
+		};
+	}
+
+	const apiClient = getApiClient();
+	const response = await apiClient.proxy<{
+		roles: Array<{
+			role: AgentRole;
+			enabled: boolean;
+			display_name: string | null;
+			description: string | null;
+			priority_filter: number[] | null;
+			fallback_activities: string[] | null;
+			auto_assign_validation: boolean;
+			auto_assign_deployment: boolean;
+		}>;
+	}>('get_role_settings', { project_id });
+
+	if (!response.ok) {
+		return {
+			result: { error: response.error || 'Failed to get role settings' },
+		};
+	}
+
+	return { result: response.data };
+};
+
+export const updateRoleSettings: Handler = async (args, _ctx) => {
+	const {
+		project_id,
+		role,
+		enabled,
+		display_name,
+		description,
+		priority_filter,
+		fallback_activities,
+		auto_assign_validation,
+		auto_assign_deployment,
+	} = args as {
+		project_id: string;
+		role: AgentRole;
+		enabled?: boolean;
+		display_name?: string;
+		description?: string;
+		priority_filter?: number[];
+		fallback_activities?: string[];
+		auto_assign_validation?: boolean;
+		auto_assign_deployment?: boolean;
+	};
+
+	if (!project_id) {
+		return {
+			result: { error: 'project_id is required' },
+		};
+	}
+
+	if (!role) {
+		return {
+			result: { error: 'role is required' },
+		};
+	}
+
+	const apiClient = getApiClient();
+	const response = await apiClient.proxy<{
+		success: boolean;
+		role: AgentRole;
+	}>('update_role_settings', {
+		project_id,
+		role,
+		enabled,
+		display_name,
+		description,
+		priority_filter,
+		fallback_activities,
+		auto_assign_validation,
+		auto_assign_deployment,
+	});
+
+	if (!response.ok) {
+		return {
+			result: { error: response.error || 'Failed to update role settings' },
+		};
+	}
+
+	return { result: response.data };
+};
+
+export const setSessionRole: Handler = async (args, ctx) => {
+	const { role, role_config } = args as {
+		role: AgentRole;
+		role_config?: Record<string, unknown>;
+	};
+	const { session, updateSession } = ctx;
+
+	if (!role) {
+		return {
+			result: { error: 'role is required' },
+		};
+	}
+
+	const validRoles: AgentRole[] = ['developer', 'validator', 'deployer', 'reviewer', 'maintainer'];
+	if (!validRoles.includes(role)) {
+		return {
+			result: {
+				error: `Invalid role: ${role}. Must be one of: ${validRoles.join(', ')}`,
+			},
+		};
+	}
+
+	// Update local session state
+	updateSession({ currentRole: role });
+
+	// If there's an active session, update it on the server too
+	if (session.currentSessionId) {
+		const apiClient = getApiClient();
+		const response = await apiClient.proxy<{
+			success: boolean;
+			session_id: string;
+			role: AgentRole;
+		}>('set_session_role', {
+			session_id: session.currentSessionId,
+			role,
+			role_config,
+		});
+
+		if (!response.ok) {
+			return {
+				result: { error: response.error || 'Failed to update session role' },
+			};
+		}
+
+		return {
+			result: {
+				success: true,
+				session_id: session.currentSessionId,
+				role,
+				message: `Session role updated to ${role}. Task filtering and fallback suggestions will now be optimized for this role.`,
+			},
+		};
+	}
+
+	return {
+		result: {
+			success: true,
+			role,
+			message: `Local role set to ${role}. Start a session to persist this role.`,
+		},
+	};
+};
+
+export const getAgentsByRole: Handler = async (args, _ctx) => {
+	const { project_id, counts_only = true } = args as { project_id: string; counts_only?: boolean };
+
+	if (!project_id) {
+		return {
+			result: { error: 'project_id is required' },
+		};
+	}
+
+	const apiClient = getApiClient();
+
+	// Type varies based on counts_only
+	if (counts_only) {
+		const response = await apiClient.proxy<{
+			agents_by_role: Record<AgentRole, number>;
+			total_active: number;
+		}>('get_agents_by_role', { project_id, counts_only: true });
+
+		if (!response.ok) {
+			return {
+				result: { error: response.error || 'Failed to get agents by role' },
+			};
+		}
+
+		return { result: response.data };
+	}
+
+	// Full details mode
+	const response = await apiClient.proxy<{
+		agents_by_role: Record<AgentRole, Array<{
+			session_id: string;
+			agent_name: string;
+			status: string;
+			current_task_id: string | null;
+			current_task_title: string | null;
+			last_synced_at: string;
+		}>>;
+		total_active: number;
+	}>('get_agents_by_role', { project_id, counts_only: false });
+
+	if (!response.ok) {
+		return {
+			result: { error: response.error || 'Failed to get agents by role' },
+		};
+	}
+
+	return { result: response.data };
+};
+
+/**
+ * Role handlers registry
+ */
+export const roleHandlers: HandlerRegistry = {
+	get_role_settings: getRoleSettings,
+	update_role_settings: updateRoleSettings,
+	set_session_role: setSessionRole,
+	get_agents_by_role: getAgentsByRole,
+};