@@ -19,8 +19,10 @@ import (
1919
2020// MCPServer represents the core MCP server implementation
2121type MCPServer struct {
22- server * server.MCPServer
23- config * Config
22+ server * server.MCPServer
23+ config * Config
24+ wasmEngine * WASMEngine
25+ registeredTools []string
2426}
2527
2628// Config holds MCP server configuration
@@ -135,6 +137,7 @@ func NewMCPServer(config *Config) *MCPServer {
135137
136138 // Register WASM module tools from config
137139 log .Printf ("Registering %d WASM modules from config" , len (config .Modules ))
140+ registeredTools := []string {}
138141 for _ , module := range config .Modules {
139142 log .Printf ("Processing module: %s (WASM path: %s)" , module .Name , module .WASMPath )
140143 log .Printf ("Loading module with %d tools" , len (module .Tools ))
@@ -149,12 +152,17 @@ func NewMCPServer(config *Config) *MCPServer {
149152 log .Printf ("Failed to register tools from WASM module %s: %v" , module .WASMPath , err )
150153 } else {
151154 log .Printf ("Successfully registered %d tools for module: %s" , len (module .Tools ), module .Name )
155+ for _ , tool := range module .Tools {
156+ registeredTools = append (registeredTools , tool .Name )
157+ }
152158 }
153159 }
154160
155161 return & MCPServer {
156- server : mcpServer ,
157- config : config ,
162+ server : mcpServer ,
163+ config : config ,
164+ wasmEngine : wasmEngine ,
165+ registeredTools : registeredTools ,
158166 }
159167}
160168
@@ -186,19 +194,9 @@ func (s *MCPServer) Start() error {
186194 return
187195 }
188196
189- // Get all registered tools
190- toolNames := []string {}
191-
192- // Manually track tools from our config
193- for _ , module := range s .config .Modules {
194- for _ , tool := range module .Tools {
195- toolNames = append (toolNames , tool .Name )
196- }
197- }
198-
199197 w .Header ().Set ("Content-Type" , "application/json" )
200198 json .NewEncoder (w ).Encode (map [string ]interface {}{
201- "tools" : toolNames ,
199+ "tools" : s . registeredTools ,
202200 })
203201 })
204202
0 commit comments