81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
// pkg/mcp/tools_environment.go
|
|
package mcp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"forge.lthn.ai/core/gui/pkg/environment"
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
)
|
|
|
|
// --- theme_get ---
|
|
|
|
type ThemeGetInput struct{}
|
|
type ThemeGetOutput struct {
|
|
Theme environment.ThemeInfo `json:"theme"`
|
|
}
|
|
|
|
func (s *Subsystem) themeGet(_ context.Context, _ *mcp.CallToolRequest, _ ThemeGetInput) (*mcp.CallToolResult, ThemeGetOutput, error) {
|
|
result, _, err := s.core.QUERY(environment.QueryTheme{})
|
|
if err != nil {
|
|
return nil, ThemeGetOutput{}, err
|
|
}
|
|
theme, ok := result.(environment.ThemeInfo)
|
|
if !ok {
|
|
return nil, ThemeGetOutput{}, fmt.Errorf("unexpected result type from theme query")
|
|
}
|
|
return nil, ThemeGetOutput{Theme: theme}, nil
|
|
}
|
|
|
|
// --- theme_system ---
|
|
|
|
type ThemeSystemInput struct{}
|
|
type ThemeSystemOutput struct {
|
|
Info environment.EnvironmentInfo `json:"info"`
|
|
}
|
|
|
|
func (s *Subsystem) themeSystem(_ context.Context, _ *mcp.CallToolRequest, _ ThemeSystemInput) (*mcp.CallToolResult, ThemeSystemOutput, error) {
|
|
result, _, err := s.core.QUERY(environment.QueryInfo{})
|
|
if err != nil {
|
|
return nil, ThemeSystemOutput{}, err
|
|
}
|
|
info, ok := result.(environment.EnvironmentInfo)
|
|
if !ok {
|
|
return nil, ThemeSystemOutput{}, fmt.Errorf("unexpected result type from environment info query")
|
|
}
|
|
return nil, ThemeSystemOutput{Info: info}, nil
|
|
}
|
|
|
|
// --- theme_set ---
|
|
|
|
type ThemeSetInput struct {
|
|
Theme string `json:"theme"`
|
|
}
|
|
type ThemeSetOutput struct {
|
|
Theme environment.ThemeInfo `json:"theme"`
|
|
}
|
|
|
|
func (s *Subsystem) themeSet(_ context.Context, _ *mcp.CallToolRequest, input ThemeSetInput) (*mcp.CallToolResult, ThemeSetOutput, error) {
|
|
_, _, err := s.core.PERFORM(environment.TaskSetTheme{Theme: input.Theme})
|
|
if err != nil {
|
|
return nil, ThemeSetOutput{}, err
|
|
}
|
|
result, _, err := s.core.QUERY(environment.QueryTheme{})
|
|
if err != nil {
|
|
return nil, ThemeSetOutput{}, err
|
|
}
|
|
theme, ok := result.(environment.ThemeInfo)
|
|
if !ok {
|
|
return nil, ThemeSetOutput{}, fmt.Errorf("unexpected result type from theme query")
|
|
}
|
|
return nil, ThemeSetOutput{Theme: theme}, nil
|
|
}
|
|
|
|
// --- Registration ---
|
|
|
|
func (s *Subsystem) registerEnvironmentTools(server *mcp.Server) {
|
|
mcp.AddTool(server, &mcp.Tool{Name: "theme_get", Description: "Get the current application theme"}, s.themeGet)
|
|
mcp.AddTool(server, &mcp.Tool{Name: "theme_system", Description: "Get system environment and theme information"}, s.themeSystem)
|
|
mcp.AddTool(server, &mcp.Tool{Name: "theme_set", Description: "Set the application theme override"}, s.themeSet)
|
|
}
|