85 lines
2.8 KiB
TypeScript
85 lines
2.8 KiB
TypeScript
import * as vscode from 'vscode';
|
|
import * as fs from 'fs';
|
|
import * as fspath from 'path';
|
|
|
|
import * as OpenMCPService from '../resources/service';
|
|
|
|
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;
|
|
}
|
|
|
|
function getLaunchCWD(context: vscode.ExtensionContext, uri: vscode.Uri) {
|
|
// TODO: 启动上下文？
|
|
// 获取当前打开的项目的路径
|
|
const workspaceFolder = vscode.workspace.getWorkspaceFolder(uri);
|
|
return workspaceFolder?.uri.fsPath || '';
|
|
}
|
|
|
|
export function activate(context: vscode.ExtensionContext) {
|
|
console.log('activate openmcp');
|
|
|
|
// 注册 showOpenMCP 命令
|
|
context.subscriptions.push(
|
|
vscode.commands.registerCommand('openmcp.showOpenMCP', async (uri: vscode.Uri) => {
|
|
|
|
const panel = vscode.window.createWebviewPanel(
|
|
'OpenMCP',
|
|
'OpenMCP',
|
|
vscode.ViewColumn.One,
|
|
{
|
|
enableScripts: true,
|
|
retainContextWhenHidden: true,
|
|
enableFindWidget: true
|
|
}
|
|
);
|
|
|
|
const cwd = getLaunchCWD(context, uri);
|
|
|
|
// 设置HTML内容
|
|
const html = getWebviewContent(context, panel);
|
|
panel.webview.html = html || '';
|
|
|
|
// 处理来自webview的消息
|
|
panel.webview.onDidReceiveMessage(message => {
|
|
const { command, data } = message;
|
|
console.log('receive message', message);
|
|
|
|
// 拦截消息，注入额外信息
|
|
switch (command) {
|
|
case 'connect':
|
|
data.cwd = cwd;
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
OpenMCPService.messageController(command, data, panel.webview as any);
|
|
});
|
|
|
|
panel.onDidDispose(() => {
|
|
panel.dispose();
|
|
});
|
|
})
|
|
);
|
|
|
|
// const provider = new WebviewViewProvider(context);
|
|
|
|
// context.subscriptions.push(
|
|
// vscode.window.registerWebviewViewProvider('webview-sidebar.view', provider)
|
|
// );
|
|
}
|
|
|
|
|
|
export function deactivate() {
|
|
|
|
} |