Fixes to load 'codex mcp-server' into wmcp

Author: radare

src/wmcp/lib/servers.go

@@ -105,14 +105,16 @@ func (s *MCPService) StartServerWithEnvAndTools(name, command string, env map[st
 			log.Printf("Warning: failed to load tools for server %s: %v", name, err)
 		}
 
-		if !s.NoPrompts {
+		if !s.NoPrompts && (!server.HasCapabilities || server.SupportsPrompts) {
 			if err := s.loadPrompts(server); err != nil {
 				log.Printf("Warning: failed to load prompts for server %s: %v", name, err)
 			}
 		}
 
-		if err := s.loadResources(server); err != nil {
-			log.Printf("Warning: failed to load resources for server %s: %v", name, err)
+		if !server.HasCapabilities || server.SupportsResources {
+			if err := s.loadResources(server); err != nil {
+				log.Printf("Warning: failed to load resources for server %s: %v", name, err)
+			}
 		}
 
 		if isSSE {
@@ -187,14 +189,16 @@ func (s *MCPService) StartServerWithEnvAndTools(name, command string, env map[st
 		log.Printf("Warning: failed to load tools for server %s: %v", name, err)
 	}
 
-	if !s.NoPrompts {
+	if !s.NoPrompts && (!server.HasCapabilities || server.SupportsPrompts) {
 		if err := s.loadPrompts(server); err != nil {
 			log.Printf("Warning: failed to load prompts for server %s: %v", name, err)
 		}
 	}
 
-	if err := s.loadResources(server); err != nil {
-		log.Printf("Warning: failed to load resources for server %s: %v", name, err)
+	if !server.HasCapabilities || server.SupportsResources {
+		if err := s.loadResources(server); err != nil {
+			log.Printf("Warning: failed to load resources for server %s: %v", name, err)
+		}
 	}
 
 	log.Printf("Started MCP server: %s", name)
@@ -230,6 +234,16 @@ func (s *MCPService) InitializeServer(server *MCPServer) error {
 		return fmt.Errorf("initialization failed: %v", response.Error)
 	}
 
+	if resultMap, ok := response.Result.(map[string]interface{}); ok {
+		if caps, ok := resultMap["capabilities"].(map[string]interface{}); ok {
+			server.Mutex.Lock()
+			server.HasCapabilities = true
+			_, server.SupportsPrompts = caps["prompts"]
+			_, server.SupportsResources = caps["resources"]
+			server.Mutex.Unlock()
+		}
+	}
+
 	initNotification := JSONRPCRequest{
 		JSONRPC: "2.0",
 		Method:  "notifications/initialized",

src/wmcp/lib/types.go

@@ -219,6 +219,9 @@ type MCPServer struct {
 	SessionID     string
 	SSEURL        string
 	SSEConnected  bool
+	HasCapabilities   bool
+	SupportsPrompts   bool
+	SupportsResources bool
 	Mutex         sync.RWMutex
 	stderrDone    chan struct{}
 	stderrActive  bool