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)
tools.AddNetbirdNetworkTools(s)
tools.AddNetbirdPostureCheckTools(s)
tools.AddNetbirdPortAllocationTools(s)
tools.AddNetbirdNameserverTools(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:8001", "The host and port to start the sse server on")
flag.Parse()
if err := run(transport, *addr); err != nil {
panic(err)
}
}