Evandro Fonseca Junior
8834ef2b4f
- Added new Radix UI components including checkbox, label, popover, slider, switch, and tooltip to improve UI interactions. - Integrated react-syntax-highlighter for better code rendering in chat messages. - Updated package.json and pnpm-lock.yaml to include new dependencies and ensure compatibility. - Enhanced chat input to support web search toggling, improving user experience during message composition. - Implemented auto-saving of chat sessions and improved session management features. - Refactored chat message rendering to clean up metadata and improve display quality.
74 lines
2.0 KiB
TypeScript
74 lines
2.0 KiB
TypeScript
'use client';
|
|
|
|
import { useState, useEffect, useCallback } from 'react';
|
|
import { invoke } from '@tauri-apps/api/core';
|
|
|
|
export interface McpServerInfo {
|
|
name: string;
|
|
status: 'running' | 'stopped' | 'error';
|
|
pid?: number;
|
|
toolCount?: number;
|
|
}
|
|
|
|
export function useMcpServers() {
|
|
const [servers, setServers] = useState<McpServerInfo[]>([]);
|
|
const [isLoading, setIsLoading] = useState(false);
|
|
const [error, setError] = useState<string | null>(null);
|
|
|
|
const loadServers = useCallback(async () => {
|
|
try {
|
|
setIsLoading(true);
|
|
setError(null);
|
|
const serverStatuses = await invoke<McpServerInfo[]>('list_mcp_server_status');
|
|
|
|
// Get tool counts for each server
|
|
const allTools = await invoke<Array<{ server_name: string; tool: any }>>('get_all_mcp_tools');
|
|
|
|
// Count tools per server
|
|
const toolCounts = new Map<string, number>();
|
|
for (const toolInfo of allTools) {
|
|
toolCounts.set(toolInfo.server_name, (toolCounts.get(toolInfo.server_name) || 0) + 1);
|
|
}
|
|
|
|
// Enrich server info with tool counts
|
|
const enrichedServers = serverStatuses.map(server => ({
|
|
...server,
|
|
toolCount: toolCounts.get(server.name) || 0,
|
|
}));
|
|
|
|
setServers(enrichedServers);
|
|
} catch (err) {
|
|
console.error('Failed to load MCP servers:', err);
|
|
setError(err instanceof Error ? err.message : 'Failed to load MCP servers');
|
|
setServers([]);
|
|
} finally {
|
|
setIsLoading(false);
|
|
}
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
loadServers();
|
|
|
|
// Refresh servers every 10 seconds
|
|
const interval = setInterval(() => {
|
|
loadServers();
|
|
}, 10000);
|
|
|
|
return () => clearInterval(interval);
|
|
}, [loadServers]);
|
|
|
|
// Get only running servers
|
|
const getRunningServers = useCallback((): McpServerInfo[] => {
|
|
return servers.filter(server => server.status === 'running');
|
|
}, [servers]);
|
|
|
|
return {
|
|
servers,
|
|
runningServers: getRunningServers(),
|
|
isLoading,
|
|
error,
|
|
loadServers,
|
|
};
|
|
}
|
|
|