mirror of
https://github.com/openshift/openshift-mcp-server.git
synced 2025-10-17 14:27:48 +03:00
f138b06ba8
refactor(kubernetes): force usage of Derived kubernetes Prevents consumers of the kubernetes package the usage of public methods on a non-derived config instance. --- review(kubernetes): force usage of Derived kubernetes Addresses comment by ardaguclu
122 lines
3.0 KiB
Go
122 lines
3.0 KiB
Go
package mcp
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
"github.com/mark3labs/mcp-go/server"
|
|
"k8s.io/utils/ptr"
|
|
|
|
"github.com/manusa/kubernetes-mcp-server/pkg/kubernetes"
|
|
"github.com/manusa/kubernetes-mcp-server/pkg/output"
|
|
"github.com/manusa/kubernetes-mcp-server/pkg/version"
|
|
)
|
|
|
|
type Configuration struct {
|
|
Profile Profile
|
|
ListOutput output.Output
|
|
// When true, expose only tools annotated with readOnlyHint=true
|
|
ReadOnly bool
|
|
// When true, disable tools annotated with destructiveHint=true
|
|
DisableDestructive bool
|
|
Kubeconfig string
|
|
}
|
|
|
|
type Server struct {
|
|
configuration *Configuration
|
|
server *server.MCPServer
|
|
k *kubernetes.Manager
|
|
}
|
|
|
|
func NewSever(configuration Configuration) (*Server, error) {
|
|
s := &Server{
|
|
configuration: &configuration,
|
|
server: server.NewMCPServer(
|
|
version.BinaryName,
|
|
version.Version,
|
|
server.WithResourceCapabilities(true, true),
|
|
server.WithPromptCapabilities(true),
|
|
server.WithToolCapabilities(true),
|
|
server.WithLogging(),
|
|
),
|
|
}
|
|
if err := s.reloadKubernetesClient(); err != nil {
|
|
return nil, err
|
|
}
|
|
s.k.WatchKubeConfig(s.reloadKubernetesClient)
|
|
return s, nil
|
|
}
|
|
|
|
func (s *Server) reloadKubernetesClient() error {
|
|
k, err := kubernetes.NewManager(s.configuration.Kubeconfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.k = k
|
|
applicableTools := make([]server.ServerTool, 0)
|
|
for _, tool := range s.configuration.Profile.GetTools(s) {
|
|
if s.configuration.ReadOnly && !ptr.Deref(tool.Tool.Annotations.ReadOnlyHint, false) {
|
|
continue
|
|
}
|
|
if s.configuration.DisableDestructive && !ptr.Deref(tool.Tool.Annotations.ReadOnlyHint, false) && ptr.Deref(tool.Tool.Annotations.DestructiveHint, false) {
|
|
continue
|
|
}
|
|
applicableTools = append(applicableTools, tool)
|
|
}
|
|
s.server.SetTools(applicableTools...)
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) ServeStdio() error {
|
|
return server.ServeStdio(s.server)
|
|
}
|
|
|
|
func (s *Server) ServeSse(baseUrl string) *server.SSEServer {
|
|
options := make([]server.SSEOption, 0)
|
|
options = append(options, server.WithSSEContextFunc(contextFunc))
|
|
if baseUrl != "" {
|
|
options = append(options, server.WithBaseURL(baseUrl))
|
|
}
|
|
return server.NewSSEServer(s.server, options...)
|
|
}
|
|
|
|
func (s *Server) ServeHTTP() *server.StreamableHTTPServer {
|
|
options := []server.StreamableHTTPOption{
|
|
server.WithHTTPContextFunc(contextFunc),
|
|
}
|
|
return server.NewStreamableHTTPServer(s.server, options...)
|
|
}
|
|
|
|
func (s *Server) Close() {
|
|
if s.k != nil {
|
|
s.k.Close()
|
|
}
|
|
}
|
|
|
|
func NewTextResult(content string, err error) *mcp.CallToolResult {
|
|
if err != nil {
|
|
return &mcp.CallToolResult{
|
|
IsError: true,
|
|
Content: []mcp.Content{
|
|
mcp.TextContent{
|
|
Type: "text",
|
|
Text: err.Error(),
|
|
},
|
|
},
|
|
}
|
|
}
|
|
return &mcp.CallToolResult{
|
|
Content: []mcp.Content{
|
|
mcp.TextContent{
|
|
Type: "text",
|
|
Text: content,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func contextFunc(ctx context.Context, r *http.Request) context.Context {
|
|
return context.WithValue(ctx, kubernetes.AuthorizationHeader, r.Header.Get(kubernetes.AuthorizationHeader))
|
|
}
|