107 lines
2.5 KiB
TypeScript
107 lines
2.5 KiB
TypeScript
|
|
import { PostMessageble } from '../adapter';
|
|
import { connect, MCPClient, type MCPOptions } from './connect';
|
|
import { callTool, getPrompt, getServerVersion, listPrompts, listResources, listResourceTemplates, listTools, readResource } from './handler';
|
|
import { chatCompletionHandler } from './llm';
|
|
import { panelLoadHandler, panelSaveHandler } from './panel';
|
|
import { settingLoadHandler, settingSaveHandler } from './setting';
|
|
import { ping } from './util';
|
|
|
|
|
|
// TODO: 支持更多的 client
|
|
let client: MCPClient | undefined = undefined;
|
|
|
|
async function connectHandler(option: MCPOptions, webview: PostMessageble) {
|
|
try {
|
|
console.log('ready to connect', option);
|
|
|
|
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)
|
|
|
|
console.log('error', error);
|
|
|
|
const connectResult = {
|
|
code: 500,
|
|
msg: (error as any).toString()
|
|
};
|
|
webview.postMessage({ command: 'connect', data: connectResult });
|
|
}
|
|
}
|
|
|
|
|
|
export function messageController(command: string, data: any, webview: PostMessageble) {
|
|
switch (command) {
|
|
case 'connect':
|
|
connectHandler(data, webview);
|
|
break;
|
|
|
|
case 'server/version':
|
|
getServerVersion(client, 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/templates/list':
|
|
listResourceTemplates(client, webview);
|
|
break;
|
|
|
|
case 'resources/read':
|
|
readResource(client, data, webview);
|
|
break;
|
|
|
|
case 'tools/list':
|
|
listTools(client, webview);
|
|
break;
|
|
|
|
case 'tools/call':
|
|
callTool(client, data, webview);
|
|
break;
|
|
|
|
case 'ping':
|
|
ping(client, webview);
|
|
break;
|
|
|
|
case 'setting/save':
|
|
settingSaveHandler(client, data, webview);
|
|
break;
|
|
|
|
case 'setting/load':
|
|
settingLoadHandler(client, webview);
|
|
break;
|
|
|
|
case 'panel/save':
|
|
panelSaveHandler(client, data, webview);
|
|
break;
|
|
|
|
case 'panel/load':
|
|
panelLoadHandler(client, webview);
|
|
break;
|
|
|
|
case 'llm/chat/completions':
|
|
chatCompletionHandler(client, data, webview);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
} |