Update too

Author: andylouisqin

src/commands/uninstall.ts

@@ -2,8 +2,13 @@ import {Args, Command, Flags} from '@oclif/core'
 import * as fs from 'node:fs/promises'
 import * as os from 'node:os'
 import * as path from 'node:path'
+import { execFile } from 'node:child_process'
+import { promisify } from 'node:util'
 
 import { servers } from '../data/servers/index.js'
+
+const execFileAsync = promisify(execFile)
+
 interface MCPServerTransport {
   transport: {
     args: string[];
@@ -70,6 +75,10 @@ export default class Uninstall extends Command {
         // Ignore errors reading Continue config
       }
 
+      // Try to get installed servers from VSCode and VSCode Insiders
+      await getInstalledServersFromVSCode('code', installedServers)
+      await getInstalledServersFromVSCode('code-insiders', installedServers)
+
       const matches = [...installedServers].filter(id =>
         id.toLowerCase().startsWith(input.toLowerCase())
       )
@@ -85,14 +94,16 @@ export default class Uninstall extends Command {
   static override examples = [
     '<%= config.bin %> <%= command.id %> server-name',
     '<%= config.bin %> <%= command.id %> server-name1 server-name2 --client claude',
+    '<%= config.bin %> <%= command.id %> server-name --client vscode',
+    '<%= config.bin %> <%= command.id %> server-name --client vscode-insiders',
   ]
 
   static override flags = {
     client: Flags.string({
       char: 'c',
       default: 'claude',
       description: 'Uninstall the MCP servers from this client',
-      options: ['claude', 'continue'],
+      options: ['claude', 'continue', 'vscode', 'vscode-insiders'],
     }),
   }
 
@@ -251,10 +262,71 @@ export default class Uninstall extends Command {
           this.error('An unknown error occurred')
         }
       }
+    } else if (client === 'vscode' || client === 'vscode-insiders') {
+      const command = client === 'vscode' ? 'code' : 'code-insiders'
+
+      try {
+        // Use VSCode's CLI to remove the server
+        const json = JSON.stringify({ name: serverName })
+        await execFileAsync(command, ['--remove-mcp', json])
+
+        this.log(`🗑️  Successfully uninstalled ${serverName} from ${client}`)
+      } catch (error: unknown) {
+        if (error instanceof Error) {
+          this.error(`Error uninstalling server from ${client}: ${error.message}`)
+        } else {
+          this.error(`An unknown error occurred while uninstalling from ${client}`)
+        }
+      }
+    }
+  }
+
+  private async uninstallOnWindows(serverName: string, client: string): Promise<void> {
+    if (client === 'vscode' || client === 'vscode-insiders') {
+      const command = client === 'vscode' ? 'code.cmd' : 'code-insiders.cmd'
+
+      try {
+        // Use VSCode's CLI to remove the server
+        const json = JSON.stringify({ name: serverName })
+        await execFileAsync(command, ['--remove-mcp', json])
+
+        this.log(`🗑️  Successfully uninstalled ${serverName} from ${client}`)
+      } catch (error: unknown) {
+        if (error instanceof Error) {
+          this.error(`Error uninstalling server from ${client}: ${error.message}`)
+        } else {
+          this.error(`An unknown error occurred while uninstalling from ${client}`)
+        }
+      }
+    } else {
+      throw new Error('Windows uninstallation not implemented yet for this client')
     }
   }
+}
 
-  private async uninstallOnWindows(_serverName: string, _client: string): Promise<void> {
-    throw new Error('Windows uninstallation not implemented yet')
+// Helper function to get installed servers from VSCode or VSCode Insiders
+async function getInstalledServersFromVSCode(command: string, installedServers: Set<string>): Promise<void> {
+  try {
+    const results = await execFileAsync(command, ['--list-mcp'])
+    if (results.stdout) {
+      for (const line of results.stdout.split('\n')) {
+        if (line.trim()) {
+          try {
+            const serverInfo = JSON.parse(line.trim())
+            const server = servers.find(srv =>
+              serverInfo.command === srv.config.command &&
+              JSON.stringify(serverInfo.args.slice(0, srv.config.args.length)) === JSON.stringify(srv.config.args)
+            )
+            if (server) {
+              installedServers.add(server.id)
+            }
+          } catch {
+            // Ignore parsing errors
+          }
+        }
+      }
+    }
+  } catch {
+    // Ignore errors reading VSCode servers
   }
 }