Merge pull request 'feat: auto-configure DevStar MCP Server for Github Copilot and Trae' (#6) from feature/auto-configure-mcp into main

Reviewed-on: #6

.gitea/workflows/devstar-vscode-release.yaml

@@ -7,17 +7,59 @@ on:
     branches:
       - main
 
+permissions:
+  contents: write
+
 jobs:
   build:
     runs-on: ubuntu-latest
     container:
       image: node:20-alpine
     steps:
+      - name: 安装 Git
+        run: |
+          apk add --no-cache git
+
       - name: 拉取代码
-        uses: https://devstar.cn/actions/checkout@v4
+        uses: actions/checkout@v4
         with:
+          token: ${{ secrets.GITHUB_TOKEN }}
           fetch-depth: 0
 
+      - name: 配置 Git
+        run: |
+          # 添加工作目录为安全目录
+          git config --global --add safe.directory "$GITHUB_WORKSPACE"
+          git config --global --add safe.directory /github/workspace
+          
+          # 配置用户信息
+          git config --global user.name "github-actions[bot]"
+          git config --global user.email "github-actions[bot]@users.noreply.github.com"
+
+      - name: 自动递增版本号
+        if: github.event_name == 'push' && github.ref == 'refs/heads/main'
+        run: |
+          apk add --no-cache jq
+          CURRENT_VERSION=$(jq -r '.version' package.json)
+          echo "当前版本: $CURRENT_VERSION"
+          
+          # 分解版本号
+          IFS='.' read -r MAJOR MINOR PATCH <<< "$CURRENT_VERSION"
+          
+          # 递增补丁版本号
+          NEW_PATCH=$((PATCH + 1))
+          NEW_VERSION="$MAJOR.$MINOR.$NEW_PATCH"
+          echo "新版本: $NEW_VERSION"
+          
+          # 更新 package.json
+          jq --arg version "$NEW_VERSION" '.version = $version' package.json > package.json.tmp
+          mv package.json.tmp package.json
+          
+          # 提交版本变更
+          git add package.json
+          git commit -m "chore: bump version to $NEW_VERSION [skip ci]"
+          git push
+
       - name: 安装依赖
         run: |
           npm install
@@ -27,7 +69,7 @@ jobs:
           npm run package
       
       - name: 发布插件
-        if: gitea.event_name == 'push' && gitea.ref == 'refs/heads/main'
+        if: github.event_name == 'push' && github.ref == 'refs/heads/main'
         run: |
           npm run publish
         env:

package.json

@@ -2,7 +2,7 @@
   "name": "devstar",
   "displayName": "%displayName%",
   "description": "%description%",
-  "version": "0.4.1",
+  "version": "0.4.3",
   "keywords": [],
   "publisher": "mengning",
   "engines": {

src/home.ts

@@ -79,7 +79,33 @@ export default class DSHome {
       async (message) => {
         const data = message.data;
         const need_return = message.need_return;
-        if (!need_return) {
+
+        if (need_return) {
+          // 处理需要返回结果的消息
+          switch (message.command) {
+            case 'getUserToken':
+              panel.webview.postMessage({
+                command: 'getUserToken',
+                data: { userToken: this.user.getUserTokenFromLocal() }
+              });
+              break;
+
+            case 'getUsername':
+              panel.webview.postMessage({
+                command: 'getUsername',
+                data: { username: this.user.getUsernameFromLocal() }
+              });
+              break;
+
+            case 'getDevstarDomain':
+              panel.webview.postMessage({
+                command: 'getDevstarDomain',
+                data: { devstarDomain: this.devstarDomain }
+              });
+              break;
+          }
+        } else {
+          // 处理不需要返回结果的消息
           switch (message.command) {
             case 'openExternalUrl':
               const url = message.url || (data && data.url);
@@ -96,6 +122,18 @@ export default class DSHome {
                 vscode.window.showErrorMessage('打开链接失败: 链接地址无效');
               }
               break;
+
+            case 'showInformationNotification':
+              if (data && data.message) {
+                vscode.window.showInformationMessage(data.message);
+              }
+              break;
+
+            case 'showErrorNotification':
+              if (data && data.message) {
+                vscode.window.showErrorMessage(data.message);
+              }
+              break;
           }
         }
       },

src/main.ts

@@ -66,31 +66,17 @@ export class DevStarExtension {
           const path = params.get('path');
           const accessToken = params.get('access_token');
           const devstarUsername = params.get('devstar_username');
-          const rawDevstarDomain = params.get('devstar_domain');
+          const devstarDomain = params.get('devstar_domain');
-          let devstarDomain = rawDevstarDomain;
+          const forwardPortsParam = params.get('forwardPorts');
-          if (rawDevstarDomain) {
-            try {
-              const url = new URL(rawDevstarDomain);
-              devstarDomain = `${url.protocol}//${url.hostname}`;
 
-              // 从 rawDevstarDomain 的查询参数中提取 forwardPorts
+          // 处理 forwardPorts 参数
-              const forwardPortsParam = url.searchParams.get('forwardPorts');
           if (forwardPortsParam) {
             const ports = forwardPortsParam.split(',').map(port => parseInt(port, 10)).filter(port => !isNaN(port));
-                console.log('解析到的 forwardPorts 参数:', ports);
             context.globalState.update('forwardPorts', ports);
           } else {
             // 如果没有 forwardPorts 参数，清除 globalState 中的旧值
-                console.log('未找到 forwardPorts 参数，清除旧的 forwardPorts 配置');
             context.globalState.update('forwardPorts', undefined);
           }
-            } catch (error) {
-              console.error('Invalid devstar_domain URL:', error);
-            }
-          }
-          console.log('sanitized_devstar_domain:', devstarDomain);
-
-          // 使用修正后的 devstar_domain
           if (devstarDomain) {
             this.user.setDevstarDomain(devstarDomain);
             this.remoteContainer.setUser(this.user);

src/remote-container.ts

@@ -263,24 +263,32 @@ export default class RemoteContainer {
         `root@${hostname}`
       ];
 
-      const sshProcess = spawn('ssh', sshArgs);
+      // 使用 detached 选项让 SSH 进程独立运行，不随父进程退出
+      const sshProcess = spawn('ssh', sshArgs, {
+        detached: true,  // 让进程在后台独立运行
+        stdio: 'ignore'  // 忽略输入输出，避免进程挂起
+      });
+
+      // 解除父进程对子进程的引用，使其真正独立
+      sshProcess.unref();
 
       sshProcess.on('error', (error: Error) => {
         reject(error);
       });
 
-      sshProcess.stdout.on('data', (_data: Buffer) => {
+      // 由于使用了 stdio: 'ignore'，这些事件监听器不再需要
-      });
+      // sshProcess.stdout.on('data', (data: Buffer) => {
+      //   console.log(`[SSH stdout] ${data.toString()}`);
+      // });
 
-      sshProcess.stderr.on('data', (_data: Buffer) => {
+      // sshProcess.stderr.on('data', (data: Buffer) => {
-      });
+      //   console.error(`[SSH stderr] ${data.toString()}`);
+      // });
 
-      if (!this.sshProcesses) {
+      // 注意：由于进程已 detached 和 unref，不再需要保存到 sshProcesses Map
-        this.sshProcesses = new Map();
+      // 因为我们无法也不需要控制这些独立进程的生命周期
-      }
-      const key = `${hostname}:${sshPort}:${containerPort}`;
-      this.sshProcesses.set(key, sshProcess);
 
+      // 等待 SSH 连接建立
       setTimeout(() => {
         resolve();
       }, 2000);
@@ -350,11 +358,12 @@ export default class RemoteContainer {
   }
 
   /**
-   * local env
+   * local env - 使用 Extension API 打开远程文件夹
    */
   async openRemoteFolder(host: string, port: number, _username: string, path: string, context: vscode.ExtensionContext): Promise<void> {
     try {
       const sshConfig = await this.getSSHConfig(host);
+
       if (sshConfig) {
         try {
           // 调用 setupPortForwardingFromGlobalState 方法
@@ -363,14 +372,36 @@ export default class RemoteContainer {
         } catch (portError) {
           vscode.window.showWarningMessage('端口映射设置失败，但容器连接已建立');
         }
+
+        // 自动配置本地 VSCode 的 MCP Server
+        try {
+          await this.configureMCPServerLocally();
+        } catch (mcpError) {
+          console.error('[MCP] 本地 MCP 配置失败:', mcpError);
+          // 不阻塞主流程，仅记录错误
         }
 
-      let terminal = vscode.window.activeTerminal || vscode.window.createTerminal(`Ext Terminal`);
+        // 自动配置容器内 AI IDE 的 MCP Server
-      terminal.show(true);
+        try {
+          await this.configureMCPServerInContainer(sshConfig.hostname, port);
+        } catch (mcpError) {
+          console.error('[MCP] 容器内 MCP 配置失败:', mcpError);
+          // 不阻塞主流程，仅记录错误
+        }
+      }
 
-      const command = `code --remote ssh-remote+root@${host}:${port} ${path} --reuse-window`;
+      // 使用 VSCode Extension API 打开远程连接
+      // 使用 SSH config 中的 Host 别名，让 SSH config 的用户配置生效
+      // 格式: vscode-remote://ssh-remote+host/path
+      // 注意：不在 URI 中指定用户名，让 SSH config 中的 User root 配置生效
+      const remoteUri = vscode.Uri.parse(
+        `vscode-remote://ssh-remote+${host}${path}`
+      );
 
-      terminal.sendText(command);
+      // 尝试使用 vscode.openFolder 命令
+      await vscode.commands.executeCommand('vscode.openFolder', remoteUri, {
+        forceNewWindow: false
+      });
 
     } catch (error) {
       const errorMessage = error instanceof Error ? error.message : '未知错误';
@@ -456,18 +487,251 @@ export default class RemoteContainer {
       console.log('未找到 forwardPorts 参数，跳过端口映射设置。');
     }
   }
+
+  /**
+   * 在容器内配置 AI IDE 的 MCP Server
+   * 支持 Trae IDE 等 AI IDE 的 MCP 自动配置
+   */
+  private async configureMCPServerInContainer(hostname: string, sshPort: number): Promise<void> {
+    // 获取用户 token
+    const userToken = this.user.getUserTokenFromLocal();
+    if (!userToken) {
+      console.log('[MCP] 用户未登录，跳过 MCP Server 配置');
+      return;
+    }
+
+    // 获取 DevStar 域名
+    const devstarDomain = this.user.getDevstarDomain();
+    const mcpUrl = `${devstarDomain}/api/mcp`;
+
+    console.log(`[MCP] 开始配置 MCP Server: ${mcpUrl}`);
+
+    // Trae IDE 的 MCP 配置路径
+    const traeMcpPath = '/root/.trae-server/data/Machine/mcp.json';
+
+    // MCP 配置内容（注意：使用 "mcpServers" 格式）
+    const mcpConfig = {
+      mcpServers: {
+        devstar: {
+          type: 'http',
+          url: mcpUrl,
+          headers: {
+            Authorization: `Bearer ${userToken}`
+          }
+        }
+      }
+    };
+
+    const configJson = JSON.stringify(mcpConfig, null, 2);
+
+    // 使用 SSH 连接到容器并配置
+    const ssh = new NodeSSH();
+
+    try {
+      await ssh.connect({
+        host: hostname,
+        username: 'root',
+        port: sshPort,
+        privateKeyPath: this.user.getUserPrivateKeyPath(),
+        readyTimeout: 10000,
+      });
+
+      console.log('[MCP] SSH 连接成功');
+
+      // 检查现有配置
+      const checkScript = `
+        if [ -f "${traeMcpPath}" ]; then
+          cat "${traeMcpPath}"
+        else
+          echo "FILE_NOT_EXISTS"
+        fi
+      `;
+
+      const checkResult = await ssh.execCommand(checkScript);
+
+      let needUpdate = true;
+
+      if (checkResult.stdout !== 'FILE_NOT_EXISTS') {
+        try {
+          const existingConfig = JSON.parse(checkResult.stdout);
+          const existingDevstar = existingConfig.mcpServers?.devstar;
+
+          if (existingDevstar) {
+            // 检查 URL 和 token 是否匹配
+            const urlMatch = existingDevstar.url === mcpUrl;
+            const tokenMatch = existingDevstar.headers?.Authorization === `Bearer ${userToken}`;
+
+            if (urlMatch && tokenMatch) {
+              console.log('[MCP] DevStar MCP 配置已存在且正确，无需更新');
+              needUpdate = false;
+            } else {
+              console.log(`[MCP] DevStar MCP 配置需要更新 (URL匹配: ${urlMatch}, Token匹配: ${tokenMatch})`);
+            }
+          }
+        } catch (parseError) {
+          console.log('[MCP] 解析现有配置失败，将创建新配置');
+        }
+      } else {
+        console.log('[MCP] 容器内无配置文件，需要创建');
+      }
+
+      // 创建或更新配置文件
+      if (needUpdate) {
+        const setupScript = `
+          mkdir -p /root/.trae-server/data/Machine && \
+          cat > ${traeMcpPath} << 'EOF'
+${configJson}
+EOF
+          echo "MCP 配置已更新"
+        `;
+
+        const result = await ssh.execCommand(setupScript);
+
+        if (result.code === 0) {
+          console.log('[MCP] DevStar MCP 配置成功');
+        } else {
+          console.error(`[MCP] 配置失败: ${result.stderr}`);
+        }
+      }
+
+    } catch (error) {
+      console.error('[MCP] SSH 连接或配置失败:', error);
+      throw error;
+    } finally {
+      try {
+        await ssh.dispose();
+      } catch (e) {
+        // ignore dispose error
+      }
+    }
+  }
+
+  /**
+   * 在本地 VSCode 中配置 MCP Server
+   * 支持 GitHub Copilot 等使用本地 MCP 配置
+   */
+  private async configureMCPServerLocally(): Promise<void> {
+    // 获取用户 token
+    const userToken = this.user.getUserTokenFromLocal();
+    if (!userToken) {
+      console.log('[MCP] 用户未登录，跳过本地 MCP Server 配置');
+      return;
+    }
+
+    // 获取 DevStar 域名
+    const devstarDomain = this.user.getDevstarDomain();
+    const mcpUrl = `${devstarDomain}/api/mcp`;
+
+    console.log(`[MCP] 开始配置本地 MCP Server: ${mcpUrl}`);
+
+    // 根据操作系统确定配置文件路径
+    let mcpConfigPath: string;
+    const platform = os.platform();
+    const homedir = os.homedir();
+
+    if (platform === 'darwin') {
+      // macOS
+      mcpConfigPath = path.join(homedir, 'Library/Application Support/Code/User/mcp.json');
+    } else if (platform === 'win32') {
+      // Windows
+      mcpConfigPath = path.join(homedir, 'AppData/Roaming/Code/User/mcp.json');
+    } else {
+      // Linux
+      mcpConfigPath = path.join(homedir, '.config/Code/User/mcp.json');
+    }
+
+    console.log(`[MCP] 配置文件路径: ${mcpConfigPath}`);
+
+    // MCP 配置内容（本地 VSCode 使用 "servers" 格式）
+    const mcpConfig = {
+      servers: {
+        devstar: {
+          type: 'http',
+          url: mcpUrl,
+          headers: {
+            Authorization: `Bearer ${userToken}`
+          }
+        }
+      }
+    };
+
+    const configJson = JSON.stringify(mcpConfig, null, 2);
+
+    try {
+      // 检查现有配置
+      let existingConfig: any = {};
+      let needUpdate = true;
+
+      if (fs.existsSync(mcpConfigPath)) {
+        try {
+          const content = fs.readFileSync(mcpConfigPath, 'utf8');
+          existingConfig = JSON.parse(content);
+          const existingDevstar = existingConfig.servers?.devstar;
+
+          if (existingDevstar) {
+            // 检查 URL 和 token 是否匹配
+            const urlMatch = existingDevstar.url === mcpUrl;
+            const tokenMatch = existingDevstar.headers?.Authorization === `Bearer ${userToken}`;
+
+            if (urlMatch && tokenMatch) {
+              console.log('[MCP] 本地 DevStar MCP 配置已存在且正确，无需更新');
+              needUpdate = false;
+            } else {
+              console.log(`[MCP] 本地 DevStar MCP 配置需要更新 (URL匹配: ${urlMatch}, Token匹配: ${tokenMatch})`);
+            }
+          }
+        } catch (parseError) {
+          console.log('[MCP] 解析现有配置失败，将创建新配置');
+        }
+      } else {
+        console.log('[MCP] 本地无配置文件，需要创建');
+      }
+
+      // 创建或更新配置文件
+      if (needUpdate) {
+        // 确保目录存在
+        const configDir = path.dirname(mcpConfigPath);
+        if (!fs.existsSync(configDir)) {
+          fs.mkdirSync(configDir, { recursive: true });
+        }
+
+        // 合并现有配置（保留其他 MCP servers）
+        const newConfig = {
+          ...existingConfig,
+          servers: {
+            ...existingConfig.servers,
+            ...mcpConfig.servers
+          }
+        };
+
+        fs.writeFileSync(mcpConfigPath, JSON.stringify(newConfig, null, 2), 'utf8');
+        console.log('[MCP] 本地 DevStar MCP 配置成功');
+      }
+
+    } catch (error) {
+      console.error('[MCP] 本地 MCP 配置失败:', error);
+      throw error;
+    }
+  }
 }
 
 /**
- * 打开项目（无须插件登录）
+ * 打开项目（无须插件登录）- 使用 Extension API
  */
-export async function openProjectWithoutLogging(hostname: string, port: number, username: string, path: string): Promise<void> {
+export async function openProjectWithoutLogging(host: string, _port: number, _username: string, path: string): Promise<void> {
-  const command = `code --remote ssh-remote+${username}@${hostname}:${port} ${path} --reuse-window`;
-
   try {
-    let terminal = vscode.window.activeTerminal || vscode.window.createTerminal(`Ext Terminal`);
+    // 使用 VSCode Extension API 打开远程连接
-    terminal.show(true);
+    // 使用 SSH config 中的 Host 别名，让 SSH config 的用户配置生效
-    terminal.sendText(command);
+    // 格式: vscode-remote://ssh-remote+host/path
+    const remoteUri = vscode.Uri.parse(
+      `vscode-remote://ssh-remote+${host}${path}`
+    );
+
+    // 使用 vscode.openFolder 命令
+    await vscode.commands.executeCommand('vscode.openFolder', remoteUri, {
+      forceNewWindow: false
+    });
+
   } catch (error) {
     const errorMessage = error instanceof Error ? error.message : '未知错误';
     vscode.window.showErrorMessage(`无登录打开项目失败: ${errorMessage}`);