// Copyright 2025 eat-pray-ai & OpenWaygate
// SPDX-License-Identifier: Apache-2.0
package cmd
import (
"fmt"
"log/slog"
"net/http"
"os"
"time"
"github.com/modelcontextprotocol/go-sdk/mcp"
"github.com/spf13/cobra"
)
const (
mcpShort = "Start MCP server"
mcpLong = "Start MCP server to handle requests from clients"
modeUsage = "stdio|http"
portUsage = "Port to listen on for HTTP mode"
)
var (
mode string
port int
)
var Server = mcp.NewServer(
&mcp.Implementation{Name: "yutu", Version: Version},
&mcp.ServerOptions{
Instructions: "Automate YouTube operations",
PageSize: 99,
KeepAlive: 13 * time.Second,
Capabilities: &mcp.ServerCapabilities{
Experimental: nil,
Completions: nil,
Logging: &mcp.LoggingCapabilities{},
Prompts: nil,
Resources: &mcp.ResourceCapabilities{
ListChanged: true,
Subscribe: true,
},
Tools: &mcp.ToolCapabilities{
ListChanged: true,
},
},
},
)
var mcpCmd = &cobra.Command{
Use: "mcp",
Short: mcpShort,
Long: mcpLong,
Run: func(cmd *cobra.Command, args []string) {
var err error
ctx := cmd.Context()
addr := fmt.Sprintf(":%d", port)
slog.InfoContext(
ctx, "starting MCP server",
"mode", mode,
"version", Version,
)
switch mode {
case "stdio":
t := &mcp.LoggingTransport{
Transport: &mcp.StdioTransport{},
Writer: os.Stderr,
}
err = Server.Run(ctx, t)
case "http":
handler := mcp.NewStreamableHTTPHandler(
func(*http.Request) *mcp.Server {
return Server
}, nil,
)
slog.InfoContext(
ctx, "http server configuration",
"url", fmt.Sprintf("http://localhost:%d/mcp", port),
)
err = http.ListenAndServe(addr, handler)
default:
slog.ErrorContext(
ctx, "invalid mode", "mode", mode, "valid_modes", "stdio, http",
)
os.Exit(1)
}
if err != nil {
slog.ErrorContext(
ctx, "starting server failed", "error", err, "mode", mode,
)
os.Exit(1)
}
},
}
func init() {
RootCmd.AddCommand(mcpCmd)
mcpCmd.Flags().StringVarP(&mode, "mode", "m", "stdio", modeUsage)
mcpCmd.Flags().IntVarP(&port, "port", "p", 8216, portUsage)
}
