|
9 | 9 | "github.com/charmbracelet/huh" |
10 | 10 | "github.com/spf13/cobra" |
11 | 11 |
|
| 12 | + "github.com/izzoa/polycode/internal/auth" |
12 | 13 | "github.com/izzoa/polycode/internal/config" |
13 | 14 | "github.com/izzoa/polycode/internal/mcp" |
14 | 15 | ) |
@@ -257,3 +258,184 @@ func runMCPTest(cmd *cobra.Command, args []string) error { |
257 | 258 | fmt.Printf(" ✓ Connected successfully (%d tools discovered)\n", toolCount) |
258 | 259 | return nil |
259 | 260 | } |
| 261 | + |
| 262 | +func runMCPSearch(cmd *cobra.Command, args []string) error { |
| 263 | + query := strings.Join(args, " ") |
| 264 | + |
| 265 | + rc := mcp.NewRegistryClient() |
| 266 | + ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) |
| 267 | + defer cancel() |
| 268 | + |
| 269 | + servers, _, err := rc.Search(ctx, query, 20) |
| 270 | + if err != nil { |
| 271 | + return fmt.Errorf("registry search failed: %w", err) |
| 272 | + } |
| 273 | + |
| 274 | + if len(servers) == 0 { |
| 275 | + fmt.Printf("No servers found for '%s'.\n", query) |
| 276 | + return nil |
| 277 | + } |
| 278 | + |
| 279 | + fmt.Printf("MCP Registry — %d results for '%s'\n\n", len(servers), query) |
| 280 | + fmt.Printf(" %-28s %-16s %-30s %s\n", "NAME", "TRANSPORT", "PACKAGE", "DESCRIPTION") |
| 281 | + fmt.Printf(" %-28s %-16s %-30s %s\n", "----", "---------", "-------", "-----------") |
| 282 | + for _, s := range servers { |
| 283 | + fmt.Printf(" %-28s %-16s %-30s %s\n", |
| 284 | + truncate(s.Name, 26), |
| 285 | + truncate(s.TransportLabel(), 14), |
| 286 | + truncate(s.PackageIdentifier(), 28), |
| 287 | + truncate(s.Description, 40), |
| 288 | + ) |
| 289 | + } |
| 290 | + return nil |
| 291 | +} |
| 292 | + |
| 293 | +func truncate(s string, maxLen int) string { |
| 294 | + if len(s) <= maxLen { |
| 295 | + return s |
| 296 | + } |
| 297 | + if maxLen <= 3 { |
| 298 | + return s[:maxLen] |
| 299 | + } |
| 300 | + return s[:maxLen-3] + "..." |
| 301 | +} |
| 302 | + |
| 303 | +func runMCPBrowse(cmd *cobra.Command, args []string) error { |
| 304 | + // Search query. |
| 305 | + var query string |
| 306 | + err := huh.NewInput(). |
| 307 | + Title("Search MCP Registry"). |
| 308 | + Placeholder("e.g., github, database, filesystem"). |
| 309 | + Value(&query). |
| 310 | + Run() |
| 311 | + if err != nil || query == "" { |
| 312 | + return nil // cancelled |
| 313 | + } |
| 314 | + |
| 315 | + rc := mcp.NewRegistryClient() |
| 316 | + ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) |
| 317 | + defer cancel() |
| 318 | + |
| 319 | + servers, _, err := rc.Search(ctx, query, 20) |
| 320 | + if err != nil { |
| 321 | + return fmt.Errorf("registry search failed: %w", err) |
| 322 | + } |
| 323 | + |
| 324 | + if len(servers) == 0 { |
| 325 | + fmt.Printf("No servers found for '%s'.\n", query) |
| 326 | + return nil |
| 327 | + } |
| 328 | + |
| 329 | + // Build selection list. |
| 330 | + opts := make([]huh.Option[int], len(servers)) |
| 331 | + for i, s := range servers { |
| 332 | + desc := s.Description |
| 333 | + if len(desc) > 60 { |
| 334 | + desc = desc[:57] + "..." |
| 335 | + } |
| 336 | + label := fmt.Sprintf("%-25s %s", s.Name, desc) |
| 337 | + opts[i] = huh.NewOption(label, i) |
| 338 | + } |
| 339 | + |
| 340 | + var selected int |
| 341 | + err = huh.NewSelect[int](). |
| 342 | + Title(fmt.Sprintf("Select server (%d results)", len(servers))). |
| 343 | + Options(opts...). |
| 344 | + Value(&selected). |
| 345 | + Run() |
| 346 | + if err != nil { |
| 347 | + return nil // cancelled |
| 348 | + } |
| 349 | + |
| 350 | + // Map to config. |
| 351 | + srv := servers[selected] |
| 352 | + cfg, envMeta := mcp.ToMCPServerConfig(srv) |
| 353 | + |
| 354 | + // Show what will be added. |
| 355 | + // Preflight: load config and check duplicates before prompting user. |
| 356 | + appCfg, err := config.Load() |
| 357 | + if err != nil { |
| 358 | + return fmt.Errorf("loading config: %w", err) |
| 359 | + } |
| 360 | + for _, s := range appCfg.MCP.Servers { |
| 361 | + if s.Name == cfg.Name { |
| 362 | + return fmt.Errorf("MCP server %q already exists — use a different name or remove the existing one first", cfg.Name) |
| 363 | + } |
| 364 | + } |
| 365 | + |
| 366 | + fmt.Printf("\nServer: %s\n", srv.Name) |
| 367 | + fmt.Printf(" Description: %s\n", srv.Description) |
| 368 | + if cfg.Command != "" { |
| 369 | + cmdStr := cfg.Command |
| 370 | + if len(cfg.Args) > 0 { |
| 371 | + cmdStr += " " + strings.Join(cfg.Args, " ") |
| 372 | + } |
| 373 | + fmt.Printf(" Command: %s\n", cmdStr) |
| 374 | + } |
| 375 | + if cfg.URL != "" { |
| 376 | + fmt.Printf(" URL: %s\n", cfg.URL) |
| 377 | + } |
| 378 | + if len(cfg.Env) > 0 { |
| 379 | + envKeys := make([]string, 0, len(cfg.Env)) |
| 380 | + for k := range cfg.Env { |
| 381 | + envKeys = append(envKeys, k) |
| 382 | + } |
| 383 | + fmt.Printf(" Env vars needed: %s\n", strings.Join(envKeys, ", ")) |
| 384 | + } |
| 385 | + |
| 386 | + // Confirm. |
| 387 | + var confirm bool |
| 388 | + err = huh.NewConfirm(). |
| 389 | + Title("Add this server?"). |
| 390 | + Value(&confirm). |
| 391 | + Run() |
| 392 | + if err != nil || !confirm { |
| 393 | + return nil |
| 394 | + } |
| 395 | + |
| 396 | + // Build secret lookup from metadata. |
| 397 | + secretVars := make(map[string]bool) |
| 398 | + for _, m := range envMeta { |
| 399 | + if m.IsSecret { |
| 400 | + secretVars[m.Name] = true |
| 401 | + } |
| 402 | + } |
| 403 | + |
| 404 | + // Prompt for required env var values. |
| 405 | + if len(cfg.Env) > 0 { |
| 406 | + store := auth.NewStore() |
| 407 | + for k := range cfg.Env { |
| 408 | + var val string |
| 409 | + input := huh.NewInput(). |
| 410 | + Title(fmt.Sprintf("Value for %s", k)). |
| 411 | + Value(&val) |
| 412 | + if secretVars[k] { |
| 413 | + input = input.EchoMode(huh.EchoModePassword) |
| 414 | + } |
| 415 | + err = input.Run() |
| 416 | + if err != nil { |
| 417 | + return nil |
| 418 | + } |
| 419 | + if secretVars[k] && val != "" { |
| 420 | + keyringKey := fmt.Sprintf("mcp_%s_%s", cfg.Name, k) |
| 421 | + if err := store.Set(keyringKey, val); err != nil { |
| 422 | + return fmt.Errorf("failed to store secret %s: %w", k, err) |
| 423 | + } |
| 424 | + cfg.Env[k] = "$KEYRING:" + keyringKey |
| 425 | + } else { |
| 426 | + cfg.Env[k] = val |
| 427 | + } |
| 428 | + } |
| 429 | + } |
| 430 | + |
| 431 | + appCfg.MCP.Servers = append(appCfg.MCP.Servers, cfg) |
| 432 | + if err := appCfg.Validate(); err != nil { |
| 433 | + return fmt.Errorf("invalid configuration: %w", err) |
| 434 | + } |
| 435 | + if err := appCfg.Save(); err != nil { |
| 436 | + return fmt.Errorf("saving config: %w", err) |
| 437 | + } |
| 438 | + |
| 439 | + fmt.Printf("\nMCP server '%s' added. %d server(s) configured.\n", cfg.Name, len(appCfg.MCP.Servers)) |
| 440 | + return nil |
| 441 | +} |
0 commit comments