import { VSCodeWebViewLike } from '../adapter';
import { connect, MCPClient, type MCPOptions } from './connect';
import { callTool, getPrompt, listPrompts, listResources, readResource } from './handler';


// TODO: 支持更多的 client
let client: MCPClient | undefined = undefined;

async function connectHandler(option: MCPOptions, webview: VSCodeWebViewLike) {
	try {
		client = await connect(option);
		const connectResult = {
			code: 200,
			msg: 'connect success\nHello from OpenMCP | virtual client version: 0.0.1'
		};
		webview.postMessage({ command: 'connect', data: connectResult });
	} catch (error) {

		// TODO: 这边获取到的 error 不够精致，如何才能获取到更加精准的错误
		// 比如	error: Failed to spawn: `server.py`
  		//		  Caused by: No such file or directory (os error 2)

		const connectResult = {
			code: 500,
			msg: (error as any).toString()
		};
		webview.postMessage({ command: 'connect', data: connectResult });
	}
}


export function messageController(command: string, data: any, webview: VSCodeWebViewLike) {
	switch (command) {
		case 'connect':
			connectHandler(data, webview);
			break;

		case 'prompts/list':
			listPrompts(client, webview);
			break;

		case 'prompts/get':
			getPrompt(client, data, webview);
			break;

		case 'resources/list':
			listResources(client, webview);
			break;

		case 'resources/read':
			readResource(client, data, webview);
			break;

		case 'tools/call':
			callTool(client, data, webview);
			break;

		default:
			break;
	}
}