-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathmcp-hub.el
282 lines (252 loc) · 11.6 KB
/
mcp-hub.el
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
;;; mcp-hub.el --- manager mcp server -*- lexical-binding: t; -*-
;; Copyright (C) 2025 lizqwer scott
;; Author: lizqwer scott <[email protected]>
;; Keywords: ai, mcp
;; This program is free software; you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation, either version 3 of the License, or
;; (at your option) any later version.
;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
;; GNU General Public License for more details.
;; You should have received a copy of the GNU General Public License
;; along with this program. If not, see <https://www.gnu.org/licenses/>.
;;; Commentary:
;;
;;; Code:
(require 'mcp)
(defcustom mcp-hub-servers nil
"Configuration for MCP servers.
Each server configuration is a list of the form (NAME . (:command COMMAND :args ARGS)) or (NAME . (:url URL)), where:
- NAME is a string identifying the server.
- COMMAND is the command to start the server.
- ARGS is a list of arguments passed to the command.
- URL is a string arguments to connect sse mcp server.
Example:
(setq mcp-hub-servers
'(;; (\"filesystem\" . (:command \"npx\" :args (\"-y\" \"@modelcontextprotocol/server-filesystem\" \"/home/lizqwer/MyProject/\")))
(\"fetch\" . (:command \"uvx\" :args (\"mcp-server-fetch\")))
(\"qdrant\" . (:url \"http://localhost:8000/sse\"))))
"
:group 'mcp-hub
:type '(list (cons string (list symbol string))))
(defun mcp-hub--start-server (server)
(apply #'mcp-connect-server
(append (list (car server))
(cdr server)
(list :initial-callback
#'(lambda (connection)
(mcp-hub-update))
:tools-callback
#'(lambda (connection tools)
(mcp-hub-update))
:prompts-callback
#'(lambda (connection prompts)
(mcp-hub-update))
:resources-callback
#'(lambda (connection resources)
(mcp-hub-update))
:error-callback
#'(lambda (code message)
(mcp-hub-update))))))
;;;###autoload
(cl-defun mcp-hub-get-all-tool (&key asyncp categoryp)
"Retrieve all available tools from connected MCP servers.
This function collects all tools from currently connected MCP servers,
filtering out any invalid entries. Each tool is created as a text tool
that can be used for interaction.
When ASYNCP is non-nil, the tools will be created asynchronously.
Returns a list of text tools created from all valid tools across all
connected servers. The list excludes any tools that couldn't be created
due to missing or invalid names.
Example:
(mcp-hub-get-all-tool) ; Get all tools synchronously
(mcp-hub-get-all-tool t) ; Get all tools asynchronously"
(let ((res ))
(maphash #'(lambda (name server)
(when (and server
(equal (mcp--status server)
'connected))
(when-let* ((tools (mcp--tools server))
(tool-names (mapcar #'(lambda (tool) (plist-get tool :name)) tools)))
(dolist (tool-name tool-names)
(push (let ((tool (mcp-make-text-tool name tool-name asyncp)))
(if categoryp
(plist-put
tool
:category
(format "mcp-%s"
name))
tool))
res)))))
mcp-server-connections)
(nreverse res)))
;;;###autoload
(defun mcp-hub-start-all-server ()
"Start all configured MCP servers.
This function will attempt to start each server listed in `mcp-hub-servers'
if it's not already running."
(interactive)
(mapcar #'(lambda (server)
(unless (gethash (car server)
mcp-server-connections)
(condition-case err
(mcp-hub--start-server server)
(error
(message "start %s server error: %s" err)))))
mcp-hub-servers))
;;;###autoload
(defun mcp-hub-close-all-server ()
"Stop all running MCP servers.
This function will attempt to stop each server listed in `mcp-hub-servers'
that is currently running."
(interactive)
(mapcar #'(lambda (server)
(when (gethash (car server)
mcp-server-connections)
(mcp-stop-server (car server))))
mcp-hub-servers)
(mcp-hub-update))
;;;###autoload
(defun mcp-hub-restart-all-server ()
"Restart all configured MCP servers.
This function first stops all running servers, then starts them again.
It's useful for applying configuration changes or recovering from errors."
(interactive)
(mcp-hub-close-all-server)
(mcp-hub-start-all-server))
(defun mcp-hub-get-servers ()
"Retrieve status information for all configured servers.
Returns a list of server statuses, where each status is a plist containing:
- :name - The server's name
- :status - Either 'connected or 'stop
- :tools - Available tools (if connected)
- :resources - Available resources (if connected)
- :prompts - Available prompts (if connected)"
(mapcar #'(lambda (server)
(let ((name (car server)))
(if-let* ((connection (gethash name mcp-server-connections)))
(list :name name
:type (mcp--connection-type connection)
:status (mcp--status connection)
:tools (mcp--tools connection)
:resources (mcp--resources connection)
:prompts (mcp--prompts connection))
(list :name name :status 'stop))))
mcp-hub-servers))
(defun mcp-hub-update (&optional arg silent)
"Update the MCP Hub display with current server status.
If called interactively, ARG is the prefix argument.
When SILENT is non-nil, suppress any status messages.
This function refreshes the *Mcp-Hub* buffer with the latest server information,
including connection status, available tools, resources, and prompts."
(interactive "P")
(when-let* ((server-list (mcp-hub-get-servers))
(server-show (mapcar #'(lambda (server)
(let* ((name (plist-get server :name))
(status (plist-get server :status)))
(append (list name
(symbol-name (plist-get server :type))
(pcase status
('connected
(propertize (symbol-name status)
'face 'success))
('error
(propertize (symbol-name status)
'face 'error))
(_
(symbol-name status))))
(if (equal status 'connected)
(mapcar #'(lambda (x)
(format "%d"
(length x)))
(list (plist-get server :tools)
(plist-get server :resources)
(plist-get server :prompts)))
(list "nil" "nil" "nil")))))
server-list)))
(with-current-buffer (get-buffer-create "*Mcp-Hub*")
(setq tabulated-list-entries
(cl-mapcar #'(lambda (statu index)
(list (format "%d" index)
(vconcat statu)))
server-show
(number-sequence 1 (length server-list))))
(tabulated-list-print t))))
;;;###autoload
(defun mcp-hub ()
"View mcp hub server"
(interactive)
;; start all server
(when (and mcp-hub-servers
(= (hash-table-count mcp-server-connections)
0))
(mcp-hub-start-all-server))
;; show buffer
(pop-to-buffer "*Mcp-Hub*" nil)
(mcp-hub-mode))
;;;###autoload
(defun mcp-hub-start-server ()
"Start the currently selected MCP server.
This function starts the server that is currently highlighted in the *Mcp-Hub*
buffer. It sets up callbacks for connection status, tools, prompts, and resources
updates, and refreshes the hub view after starting the server."
(interactive)
(when-let* ((server (tabulated-list-get-entry))
(name (elt server 0))
(server-arg (cl-find name mcp-hub-servers :key #'car :test #'equal)))
(mcp-hub--start-server server-arg)
(mcp-hub-update)))
;;;###autoload
(defun mcp-hub-close-server ()
"Stop the currently selected MCP server.
This function stops the server that is currently highlighted in the *Mcp-Hub*
buffer and updates the hub view to reflect the change in status."
(interactive)
(when-let* ((server (tabulated-list-get-entry))
(name (elt server 0)))
(mcp-stop-server name)
(mcp-hub-update)))
;;;###autoload
(defun mcp-hub-restart-server ()
"Restart the currently selected MCP server.
This function stops and then starts the server that is currently highlighted
in the *Mcp-Hub* buffer. It's useful for applying configuration changes or
recovering from errors."
(interactive)
(mcp-hub-close-server)
(mcp-hub-start-server))
;;;###autoload
(defun mcp-hub-view-log ()
"View the event log for the currently selected MCP server.
This function opens a buffer showing the event log for the server that is
currently highlighted in the *Mcp-Hub* buffer."
(interactive)
(when-let* ((server (tabulated-list-get-entry))
(name (elt server 0)))
(switch-to-buffer (format "*%s events*"
name))))
(define-derived-mode mcp-hub-mode tabulated-list-mode "Mcp Hub"
"A major mode for viewing a list of mcp server."
(setq-local revert-buffer-function #'mcp-hub-update)
(setq tabulated-list-format
[("Name" 18 t)
("Type" 10 t)
("Status" 15 t)
("Tools" 10 t)
("Resources" 10 t)
("Prompts" 10 t)])
(setq tabulated-list-padding 2)
(setq tabulated-list-sort-key '("Name" . nil))
(tabulated-list-init-header)
(keymap-set mcp-hub-mode-map "l" #'mcp-hub-view-log)
(keymap-set mcp-hub-mode-map "s" #'mcp-hub-start-server)
(keymap-set mcp-hub-mode-map "k" #'mcp-hub-close-server)
(keymap-set mcp-hub-mode-map "r" #'mcp-hub-restart-server)
(keymap-set mcp-hub-mode-map "S" #'mcp-hub-start-all-server)
(keymap-set mcp-hub-mode-map "R" #'mcp-hub-restart-all-server)
(keymap-set mcp-hub-mode-map "K" #'mcp-hub-close-all-server)
(mcp-hub-update))
(provide 'mcp-hub)
;;; mcp-hub.el ends here