Supports manual shutdown of MCP servers

Author: aiqlcom

src/main/IPCs.ts

@@ -40,7 +40,7 @@ interface ManifestResponse {
  * */
 export default class IPCs {
   static clients: ClientObj[] = []
-  static currentFeatures: any[] = []
+  static currentFeatures: FeatureObj[] = []
 
   static initialize(): void {
     // Get application version
@@ -56,16 +56,24 @@ export default class IPCs {
       return pathToFileURL(normalize(resolve(Constants.ASSETS_PATH.dxt))).toString()
     })
 
+    ipcMain.handle('msgMcpServersStop', async () => {
+      IPCs.stopAllServers()
+
+      const configs = await loadConfig()
+
+      const features = configs.map((params) => {
+        return registerIpcHandlers(params)
+      })
+
+      this.currentFeatures = features
+
+      return true
+    })
+
     ipcMain.handle(
       'msgMcpServersInit',
       async (event: IpcMainEvent, metadata: ConfigMcpMetadata) => {
-        this.clients.forEach((client: ClientObj) => {
-          if (client.connection?.transport) {
-            disconnect(client.connection.transport)
-          }
-        })
-
-        IPCs.removeAllHandlers()
+        IPCs.stopAllServers()
 
         const progressCallback: McpProgressCallback = (name, message, status) => {
           mcpServersCallback({ name, message, status })
@@ -249,6 +257,17 @@ export default class IPCs {
     this.currentFeatures = newFeatures
   }
 
+  static stopAllServers() {
+    this.clients.forEach((client: ClientObj) => {
+      if (client.connection?.transport) {
+        disconnect(client.connection.transport)
+        delete client.connection.transport
+      }
+    })
+
+    IPCs.removeAllHandlers()
+  }
+
   static removeAllHandlers() {
     for (const [eventName] of handlerRegistry) {
       ipcMain.removeHandler(eventName)

src/preload/index.ts

@@ -25,6 +25,7 @@ const mainAvailChannels: string[] = [
   'msgCommandSelectionNotify',
   'msgGetApiToken',
   'msgMcpServersInit',
+  'msgMcpServersStop',
   'msgWindowReload'
 ]
 

src/renderer/components/pages/McpProcessPage.vue

@@ -43,13 +43,16 @@ const allSuccess = computed(() => {
   return Object.values(progressMap.value).every((item) => item.status === 'success')
 })
 
-watch(isLoading, (newVal) => {
+watch(isLoading, (newVal, oldVal) => {
   if (newVal) {
     progressMap.value = {}
     mcpDialog.value = true
   } else {
     setTimeout(() => {
-      if (allSuccess.value) {
+      console.log(progressMap.value)
+      if (allSuccess.value || oldVal === 'stop') {
+        // If all server are started successful, or all servers are forced to stop
+        // no need to stick the dialog for process visualization
         mcpDialog.value = false
       }
     }, 500)

src/renderer/locales/en.json

@@ -44,7 +44,8 @@
     "config": "Server Config",
     "file": "Archive File",
     "new": "New MCP Server",
-    "init": "Init MCP Servers",
+    "init": "Start MCP Servers",
+    "stop": "Stop MCP Servers",
     "read": "Read",
     "minutes": "minutes",
     "open": "Reveal in File Explorer",

src/renderer/locales/zh.json

@@ -44,7 +44,8 @@
     "config": "服务端配置",
     "file": "文件压缩包",
     "new": "新的 MCP 服务端",
-    "init": "初始化 MCP 服务端",
+    "init": "启动 MCP 服务端",
+    "stop": "停止 MCP 服务端",
     "read": "阅读",
     "minutes": "分钟",
     "open": "在文件管理器中打开",

src/renderer/screens/mcp/McpSideDock.vue

@@ -19,8 +19,19 @@ const addDialog = ref(false)
 
 const isDragActive = ref(false)
 
+async function stopAllMcpServers() {
+  layoutStore.mcpLoading = 'stop'
+  try {
+    await McpEvent.stop()
+    await mcpStore.updateServers()
+  } finally {
+    layoutStore.mcpLoading = false
+    mcpStore.version++
+  }
+}
+
 async function activeAllMcpServers() {
-  layoutStore.mcpLoading = true
+  layoutStore.mcpLoading = 'start'
   try {
     const configs = getServers()
     const result = await McpEvent.init(configs)
@@ -76,6 +87,14 @@ const items = [
         @click="activeAllMcpServers()"
       >
       </v-btn>
+      <v-btn
+        v-tooltip:top="$t('mcp.stop')"
+        icon="mdi-power-off"
+        color="error"
+        :loading="layoutStore.mcpLoading"
+        @click="stopAllMcpServers()"
+      >
+      </v-btn>
       <v-btn
         v-tooltip:top="$t('mcp.open')"
         color="primary"

src/renderer/store/layout.ts

@@ -12,6 +12,8 @@ const PATH_TO_SCREEN = {
 type ScreenKey = keyof typeof PATH_TO_SCREEN
 type ScreenValue = (typeof PATH_TO_SCREEN)[ScreenKey]
 
+type LoadingStatus = 'start' | 'stop' | false
+
 export const getScreenFromPath = (path: string): ScreenValue => {
   return PATH_TO_SCREEN[path as ScreenKey] ?? 0
 }
@@ -20,7 +22,7 @@ export const useLayoutStore = defineStore('layoutStore', {
   state: () => ({
     sidebar: true,
     apiKeyShow: false,
-    mcpLoading: false,
+    mcpLoading: false as LoadingStatus,
     screen: 0 // The selected screen is a list: 0,1,2,...
   })
 })

src/renderer/utils/index.ts

@@ -184,12 +184,17 @@ class Mcp {
     return window.mainApi.invoke('msgMcpServersInit', filteredConfigs)
   }
 
+  static async msgMcpServersStop(): Promise<any> {
+    return window.mainApi.invoke('msgMcpServersStop')
+  }
+
   static async msgMcpServersWatch(callback: any): Promise<any> {
     return window.mainApi.on('msgMcpServersWatch', callback)
   }
 }
 
 export const McpEvent = {
   init: Mcp.msgMcpServersInit,
+  stop: Mcp.msgMcpServersStop,
   watch: Mcp.msgMcpServersWatch
 }