mcp-netbird

package main import ( "context" "flag" "fmt" "log" "os" "github.com/mark3labs/mcp-go/server" mcpnetbird "github.com/aantti/mcp-netbird" "github.com/aantti/mcp-netbird/tools" ) func newServer() *server.MCPServer { s := server.NewMCPServer( "mcp-netbird", "0.1.0", ) tools.AddNetbirdPeerTools(s) tools.AddNetbirdGroupTools(s) tools.AddNetbirdPolicyTools(s) return s } func run(transport, addr string) error { s := newServer() switch transport { case "stdio": srv := server.NewStdioServer(s) srv.SetContextFunc(mcpnetbird.ComposedStdioContextFunc) return srv.Listen(context.Background(), os.Stdin, os.Stdout) case "sse": srv := server.NewSSEServer(s, server.WithSSEContextFunc(mcpnetbird.ComposedSSEContextFunc), ) log.Printf("SSE server listening on %s", addr) if err := srv.Start(addr); err != nil { return fmt.Errorf("server error: %v", err) } default: return fmt.Errorf( "invalid transport type: %s. must be 'stdio' or 'sse'", transport, ) } return nil } func main() { var transport string flag.StringVar(&transport, "t", "stdio", "Transport type (stdio or sse)") flag.StringVar( &transport, "transport", "stdio", "Transport type (stdio or sse)", ) addr := flag.String("sse-address", "localhost:8000", "The host and port to start the sse server on") flag.Parse() if err := run(transport, *addr); err != nil { panic(err) } }