112 lines
3.5 KiB
TypeScript
112 lines
3.5 KiB
TypeScript
import * as vscode from 'vscode';
|
|
import * as fs from 'fs';
|
|
import * as fspath from 'path';
|
|
import { ISSEConnectionItem, IStdioConnectionItem, panels } from './global';
|
|
import * as OpenMCPService from '../resources/service';
|
|
|
|
|
|
export function getWebviewContent(context: vscode.ExtensionContext, panel: vscode.WebviewPanel): string | undefined {
|
|
const viewRoot = fspath.join(context.extensionPath, 'resources', 'renderer');
|
|
const htmlIndexPath = fspath.join(viewRoot, 'index.html');
|
|
const html = fs.readFileSync(htmlIndexPath, { encoding: 'utf-8' })?.replace(/(<link.+?href="|<script.+?src="|<img.+?src=")(.+?)"/g, (m, $1, $2) => {
|
|
const absLocalPath = fspath.resolve(viewRoot, $2);
|
|
const webviewUri = panel.webview.asWebviewUri(vscode.Uri.file(absLocalPath));
|
|
|
|
const replaceHref = $1 + webviewUri?.toString() + '"';
|
|
return replaceHref;
|
|
});
|
|
return html;
|
|
}
|
|
|
|
export function getLaunchCWD(context: vscode.ExtensionContext, uri: vscode.Uri) {
|
|
// TODO: 启动上下文？
|
|
// 获取当前打开的项目的路径
|
|
const workspaceFolder = vscode.workspace.getWorkspaceFolder(uri);
|
|
return workspaceFolder?.uri.fsPath || '';
|
|
}
|
|
|
|
|
|
export function revealOpenMcpWebviewPanel(
|
|
context: vscode.ExtensionContext,
|
|
panelKey: string,
|
|
option: (IStdioConnectionItem | ISSEConnectionItem) = {
|
|
type: 'stdio',
|
|
name: 'OpenMCP',
|
|
command: 'mcp',
|
|
args: ['run', 'main.py']
|
|
}
|
|
) {
|
|
if (panels.has(panelKey)) {
|
|
const panel = panels.get(panelKey);
|
|
panel?.reveal();
|
|
return panel;
|
|
}
|
|
|
|
const panel = vscode.window.createWebviewPanel(
|
|
'OpenMCP',
|
|
'OpenMCP',
|
|
vscode.ViewColumn.One,
|
|
{
|
|
enableScripts: true,
|
|
retainContextWhenHidden: true,
|
|
enableFindWidget: true
|
|
}
|
|
);
|
|
|
|
panels.set(panelKey, panel);
|
|
|
|
|
|
// 设置HTML内容
|
|
const html = getWebviewContent(context, panel);
|
|
panel.webview.html = html || '';
|
|
panel.iconPath = vscode.Uri.file(fspath.join(context.extensionPath, 'resources', 'renderer', 'images', 'openmcp.png'));
|
|
|
|
// 处理来自webview的消息
|
|
panel.webview.onDidReceiveMessage(message => {
|
|
const { command, data } = message;
|
|
console.log('receive message', message);
|
|
|
|
// 拦截消息，注入额外信息
|
|
switch (command) {
|
|
case 'vscode/launch-command':
|
|
const laucnResultMessage = option.type === 'stdio' ?
|
|
{
|
|
type: 'stdio',
|
|
commandString: option.command + ' ' + option.args.join(' '),
|
|
cwd: option.cwd
|
|
} :
|
|
{
|
|
type: 'sse',
|
|
url: option.url,
|
|
oauth: option.oauth
|
|
};
|
|
|
|
const launchResult = {
|
|
code: 200,
|
|
msg: laucnResultMessage
|
|
};
|
|
|
|
panel.webview.postMessage({
|
|
command: 'vscode/launch-command',
|
|
data: launchResult
|
|
});
|
|
|
|
break;
|
|
|
|
default:
|
|
OpenMCPService.messageController(command, data, panel.webview);
|
|
break;
|
|
}
|
|
|
|
});
|
|
|
|
panel.onDidDispose(async () => {
|
|
// 删除
|
|
panels.delete(panelKey);
|
|
|
|
// 退出
|
|
panel.dispose();
|
|
});
|
|
|
|
return panel;
|
|
} |