index.ts•5.54 kB
/**
* MCP Tools services.logger.info('Enhanced ARC MCP tools registered successfully');
}
// Keep the original tools as reference/backup but don't register them
function registerOriginalArcToolsBackup(server: any, services: ServiceContext): void {ation
*
* Registers all ARC management tools with the MCP server.
*/
import type { ServiceContext } from '../types/arc.js';
import { registerEnhancedArcTools } from './enhanced-tools.js';
/**
* Register all ARC management tools with enhanced real-time capabilities
*/
export function registerArcTools(server: any, services: ServiceContext): void {
// Register enhanced tools with real-time progress updates
registerEnhancedArcTools(server, services);
// Original tools are replaced by enhanced versions - no need to register both
services.logger.info('Enhanced ARC MCP tools registered successfully');
}
/**
* Register original ARC management tools (fallback)
*/
function registerOriginalArcTools(server: any, services: ServiceContext): void {
// Natural language processing tool
server.registerTool(
'arc_process_natural_language',
{
title: 'Process Natural Language ARC Command',
description: 'Process natural language commands for ARC operations like "Install ARC", "Scale runners", "Check status"'
},
async (params: { query: string }) => {
const result = await processNaturalLanguageQuery(params.query, services);
return {
content: [
{
type: 'text',
text: JSON.stringify(result, null, 2)
}
],
structuredContent: result
};
}
);
// Installation tools
server.registerTool(
'arc_install_controller',
{
title: 'Install ARC Controller',
description: 'Install GitHub Actions Runner Controller in Kubernetes cluster'
},
async (params: any) => {
const result = await services.installer.installController(params);
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
structuredContent: result
};
}
);
// Status and monitoring tools
server.registerTool(
'arc_get_status',
{
title: 'Get ARC Status',
description: 'Get comprehensive status of ARC installation and runners'
},
async (params: any) => {
const result = await services.installer.getStatus();
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
structuredContent: result
};
}
);
// Scaling operations
server.registerTool(
'arc_scale_runners',
{
title: 'Scale ARC Runners',
description: 'Scale GitHub Actions runners up or down'
},
async (params: any) => {
services.logger.info('Scaling ARC runners', params);
const result = {
success: true,
currentReplicas: 2,
targetReplicas: params.replicas,
message: `Scaled ${params.scaleSetName} to ${params.replicas} replicas`
};
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
structuredContent: result
};
}
);
services.logger.info('All ARC MCP tools registered successfully (enhanced + original)');
}
/**
* Process natural language queries for ARC operations
*/
async function processNaturalLanguageQuery(query: string, services: ServiceContext): Promise<any> {
if (!query || typeof query !== 'string') {
return {
action: 'unknown',
parameters: {},
confidence: 0.0,
interpretation: 'Invalid or empty query provided',
suggestion: 'Try commands like "Install ARC", "Scale runners to 5", or "Check ARC status"'
};
}
const lowercaseQuery = query.toLowerCase();
// Simple pattern matching for demo - would use proper NLP in production
if (lowercaseQuery.includes('install') && lowercaseQuery.includes('arc')) {
return {
action: 'install_controller',
parameters: { namespace: 'arc-systems', version: '0.27.6' },
confidence: 0.9,
interpretation: 'User wants to install ARC controller'
};
}
if (lowercaseQuery.includes('scale') && lowercaseQuery.includes('runner')) {
const numbers = query.match(/\d+/);
const replicas = numbers ? parseInt(numbers[0]) : 2;
return {
action: 'scale_runners',
parameters: { replicas, scaleSetName: 'default-runners' },
confidence: 0.8,
interpretation: `User wants to scale runners to ${replicas} replicas`
};
}
if (lowercaseQuery.includes('status') || lowercaseQuery.includes('check')) {
return {
action: 'get_status',
parameters: {},
confidence: 0.95,
interpretation: 'User wants to check ARC status'
};
}
return {
action: 'unknown',
parameters: {},
confidence: 0.1,
interpretation: 'Query not recognized',
suggestion: 'Try commands like "Install ARC", "Scale runners to 5", or "Check ARC status"'
};
}