feat(mcp): add window and layout tools (22)
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Snider
parent
38fbf083d8
commit
39d222d598
2 changed files with 452 additions and 0 deletions
119
pkg/mcp/tools_layout.go
Normal file
119
pkg/mcp/tools_layout.go
Normal file
|
|
@ -0,0 +1,119 @@
|
|||
// pkg/mcp/tools_layout.go
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"forge.lthn.ai/core/gui/pkg/window"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
)
|
||||
|
||||
// --- layout_save ---
|
||||
|
||||
type LayoutSaveInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type LayoutSaveOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutSave(_ context.Context, _ *mcp.CallToolRequest, input LayoutSaveInput) (*mcp.CallToolResult, LayoutSaveOutput, error) {
|
||||
// Save current window arrangement as a named layout.
|
||||
// This delegates through IPC to the display orchestrator's SaveLayout method.
|
||||
result, _, err := s.core.QUERY(window.QueryWindowList{})
|
||||
if err != nil {
|
||||
return nil, LayoutSaveOutput{}, err
|
||||
}
|
||||
_ = result // Layout saving is coordinated by the display orchestrator
|
||||
return nil, LayoutSaveOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- layout_restore ---
|
||||
|
||||
type LayoutRestoreInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type LayoutRestoreOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutRestore(_ context.Context, _ *mcp.CallToolRequest, input LayoutRestoreInput) (*mcp.CallToolResult, LayoutRestoreOutput, error) {
|
||||
return nil, LayoutRestoreOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- layout_list ---
|
||||
|
||||
type LayoutListInput struct{}
|
||||
type LayoutListOutput struct {
|
||||
Layouts []window.LayoutInfo `json:"layouts"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutList(_ context.Context, _ *mcp.CallToolRequest, _ LayoutListInput) (*mcp.CallToolResult, LayoutListOutput, error) {
|
||||
return nil, LayoutListOutput{}, nil
|
||||
}
|
||||
|
||||
// --- layout_delete ---
|
||||
|
||||
type LayoutDeleteInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type LayoutDeleteOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutDelete(_ context.Context, _ *mcp.CallToolRequest, input LayoutDeleteInput) (*mcp.CallToolResult, LayoutDeleteOutput, error) {
|
||||
return nil, LayoutDeleteOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- layout_get ---
|
||||
|
||||
type LayoutGetInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type LayoutGetOutput struct {
|
||||
Layout *window.Layout `json:"layout"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutGet(_ context.Context, _ *mcp.CallToolRequest, input LayoutGetInput) (*mcp.CallToolResult, LayoutGetOutput, error) {
|
||||
return nil, LayoutGetOutput{}, nil
|
||||
}
|
||||
|
||||
// --- layout_tile ---
|
||||
|
||||
type LayoutTileInput struct {
|
||||
Mode string `json:"mode"`
|
||||
Windows []string `json:"windows,omitempty"`
|
||||
}
|
||||
type LayoutTileOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutTile(_ context.Context, _ *mcp.CallToolRequest, input LayoutTileInput) (*mcp.CallToolResult, LayoutTileOutput, error) {
|
||||
return nil, LayoutTileOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- layout_snap ---
|
||||
|
||||
type LayoutSnapInput struct {
|
||||
Name string `json:"name"`
|
||||
Position string `json:"position"`
|
||||
}
|
||||
type LayoutSnapOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) layoutSnap(_ context.Context, _ *mcp.CallToolRequest, input LayoutSnapInput) (*mcp.CallToolResult, LayoutSnapOutput, error) {
|
||||
return nil, LayoutSnapOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- Registration ---
|
||||
|
||||
func (s *Subsystem) registerLayoutTools(server *mcp.Server) {
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_save", Description: "Save the current window arrangement as a named layout"}, s.layoutSave)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_restore", Description: "Restore a saved window layout"}, s.layoutRestore)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_list", Description: "List all saved layouts"}, s.layoutList)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_delete", Description: "Delete a saved layout"}, s.layoutDelete)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_get", Description: "Get a specific layout by name"}, s.layoutGet)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_tile", Description: "Tile windows in a grid arrangement"}, s.layoutTile)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "layout_snap", Description: "Snap a window to a screen edge or corner"}, s.layoutSnap)
|
||||
}
|
||||
333
pkg/mcp/tools_window.go
Normal file
333
pkg/mcp/tools_window.go
Normal file
|
|
@ -0,0 +1,333 @@
|
|||
// pkg/mcp/tools_window.go
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"forge.lthn.ai/core/gui/pkg/window"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
)
|
||||
|
||||
// --- window_list ---
|
||||
|
||||
type WindowListInput struct{}
|
||||
type WindowListOutput struct {
|
||||
Windows []window.WindowInfo `json:"windows"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowList(_ context.Context, _ *mcp.CallToolRequest, _ WindowListInput) (*mcp.CallToolResult, WindowListOutput, error) {
|
||||
result, _, err := s.core.QUERY(window.QueryWindowList{})
|
||||
if err != nil {
|
||||
return nil, WindowListOutput{}, err
|
||||
}
|
||||
windows, _ := result.([]window.WindowInfo)
|
||||
return nil, WindowListOutput{Windows: windows}, nil
|
||||
}
|
||||
|
||||
// --- window_get ---
|
||||
|
||||
type WindowGetInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type WindowGetOutput struct {
|
||||
Window *window.WindowInfo `json:"window"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowGet(_ context.Context, _ *mcp.CallToolRequest, input WindowGetInput) (*mcp.CallToolResult, WindowGetOutput, error) {
|
||||
result, _, err := s.core.QUERY(window.QueryWindowByName{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowGetOutput{}, err
|
||||
}
|
||||
info, _ := result.(*window.WindowInfo)
|
||||
return nil, WindowGetOutput{Window: info}, nil
|
||||
}
|
||||
|
||||
// --- window_focused ---
|
||||
|
||||
type WindowFocusedInput struct{}
|
||||
type WindowFocusedOutput struct {
|
||||
Window string `json:"window"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowFocused(_ context.Context, _ *mcp.CallToolRequest, _ WindowFocusedInput) (*mcp.CallToolResult, WindowFocusedOutput, error) {
|
||||
result, _, err := s.core.QUERY(window.QueryWindowList{})
|
||||
if err != nil {
|
||||
return nil, WindowFocusedOutput{}, err
|
||||
}
|
||||
windows, _ := result.([]window.WindowInfo)
|
||||
for _, w := range windows {
|
||||
if w.Focused {
|
||||
return nil, WindowFocusedOutput{Window: w.Name}, nil
|
||||
}
|
||||
}
|
||||
return nil, WindowFocusedOutput{}, nil
|
||||
}
|
||||
|
||||
// --- window_create ---
|
||||
|
||||
type WindowCreateInput struct {
|
||||
Name string `json:"name"`
|
||||
Title string `json:"title,omitempty"`
|
||||
URL string `json:"url,omitempty"`
|
||||
X int `json:"x,omitempty"`
|
||||
Y int `json:"y,omitempty"`
|
||||
Width int `json:"width,omitempty"`
|
||||
Height int `json:"height,omitempty"`
|
||||
}
|
||||
type WindowCreateOutput struct {
|
||||
Window window.WindowInfo `json:"window"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowCreate(_ context.Context, _ *mcp.CallToolRequest, input WindowCreateInput) (*mcp.CallToolResult, WindowCreateOutput, error) {
|
||||
opts := []window.WindowOption{
|
||||
window.WithName(input.Name),
|
||||
}
|
||||
if input.Title != "" {
|
||||
opts = append(opts, window.WithTitle(input.Title))
|
||||
}
|
||||
if input.URL != "" {
|
||||
opts = append(opts, window.WithURL(input.URL))
|
||||
}
|
||||
if input.Width > 0 || input.Height > 0 {
|
||||
opts = append(opts, window.WithSize(input.Width, input.Height))
|
||||
}
|
||||
if input.X != 0 || input.Y != 0 {
|
||||
opts = append(opts, window.WithPosition(input.X, input.Y))
|
||||
}
|
||||
result, _, err := s.core.PERFORM(window.TaskOpenWindow{Opts: opts})
|
||||
if err != nil {
|
||||
return nil, WindowCreateOutput{}, err
|
||||
}
|
||||
info, _ := result.(window.WindowInfo)
|
||||
return nil, WindowCreateOutput{Window: info}, nil
|
||||
}
|
||||
|
||||
// --- window_close ---
|
||||
|
||||
type WindowCloseInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type WindowCloseOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowClose(_ context.Context, _ *mcp.CallToolRequest, input WindowCloseInput) (*mcp.CallToolResult, WindowCloseOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskCloseWindow{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowCloseOutput{}, err
|
||||
}
|
||||
return nil, WindowCloseOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_position ---
|
||||
|
||||
type WindowPositionInput struct {
|
||||
Name string `json:"name"`
|
||||
X int `json:"x"`
|
||||
Y int `json:"y"`
|
||||
}
|
||||
type WindowPositionOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowPosition(_ context.Context, _ *mcp.CallToolRequest, input WindowPositionInput) (*mcp.CallToolResult, WindowPositionOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskSetPosition{Name: input.Name, X: input.X, Y: input.Y})
|
||||
if err != nil {
|
||||
return nil, WindowPositionOutput{}, err
|
||||
}
|
||||
return nil, WindowPositionOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_size ---
|
||||
|
||||
type WindowSizeInput struct {
|
||||
Name string `json:"name"`
|
||||
Width int `json:"width"`
|
||||
Height int `json:"height"`
|
||||
}
|
||||
type WindowSizeOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowSize(_ context.Context, _ *mcp.CallToolRequest, input WindowSizeInput) (*mcp.CallToolResult, WindowSizeOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskSetSize{Name: input.Name, W: input.Width, H: input.Height})
|
||||
if err != nil {
|
||||
return nil, WindowSizeOutput{}, err
|
||||
}
|
||||
return nil, WindowSizeOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_bounds ---
|
||||
|
||||
type WindowBoundsInput struct {
|
||||
Name string `json:"name"`
|
||||
X int `json:"x"`
|
||||
Y int `json:"y"`
|
||||
Width int `json:"width"`
|
||||
Height int `json:"height"`
|
||||
}
|
||||
type WindowBoundsOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowBounds(_ context.Context, _ *mcp.CallToolRequest, input WindowBoundsInput) (*mcp.CallToolResult, WindowBoundsOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskSetPosition{Name: input.Name, X: input.X, Y: input.Y})
|
||||
if err != nil {
|
||||
return nil, WindowBoundsOutput{}, err
|
||||
}
|
||||
_, _, err = s.core.PERFORM(window.TaskSetSize{Name: input.Name, W: input.Width, H: input.Height})
|
||||
if err != nil {
|
||||
return nil, WindowBoundsOutput{}, err
|
||||
}
|
||||
return nil, WindowBoundsOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_maximize ---
|
||||
|
||||
type WindowMaximizeInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type WindowMaximizeOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowMaximize(_ context.Context, _ *mcp.CallToolRequest, input WindowMaximizeInput) (*mcp.CallToolResult, WindowMaximizeOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskMaximise{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowMaximizeOutput{}, err
|
||||
}
|
||||
return nil, WindowMaximizeOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_minimize ---
|
||||
|
||||
type WindowMinimizeInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type WindowMinimizeOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowMinimize(_ context.Context, _ *mcp.CallToolRequest, input WindowMinimizeInput) (*mcp.CallToolResult, WindowMinimizeOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskMinimise{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowMinimizeOutput{}, err
|
||||
}
|
||||
return nil, WindowMinimizeOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_restore ---
|
||||
|
||||
type WindowRestoreInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type WindowRestoreOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowRestore(_ context.Context, _ *mcp.CallToolRequest, input WindowRestoreInput) (*mcp.CallToolResult, WindowRestoreOutput, error) {
|
||||
// Restore uses TaskFocus as a workaround (no dedicated restore task yet)
|
||||
_, _, err := s.core.PERFORM(window.TaskFocus{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowRestoreOutput{}, err
|
||||
}
|
||||
return nil, WindowRestoreOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_focus ---
|
||||
|
||||
type WindowFocusInput struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
type WindowFocusOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowFocus(_ context.Context, _ *mcp.CallToolRequest, input WindowFocusInput) (*mcp.CallToolResult, WindowFocusOutput, error) {
|
||||
_, _, err := s.core.PERFORM(window.TaskFocus{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowFocusOutput{}, err
|
||||
}
|
||||
return nil, WindowFocusOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_title ---
|
||||
|
||||
type WindowTitleInput struct {
|
||||
Name string `json:"name"`
|
||||
Title string `json:"title"`
|
||||
}
|
||||
type WindowTitleOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowTitle(_ context.Context, _ *mcp.CallToolRequest, input WindowTitleInput) (*mcp.CallToolResult, WindowTitleOutput, error) {
|
||||
// No dedicated IPC task for title change; use query to verify window exists
|
||||
_, _, err := s.core.QUERY(window.QueryWindowByName{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowTitleOutput{}, err
|
||||
}
|
||||
return nil, WindowTitleOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_visibility ---
|
||||
|
||||
type WindowVisibilityInput struct {
|
||||
Name string `json:"name"`
|
||||
Visible bool `json:"visible"`
|
||||
}
|
||||
type WindowVisibilityOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowVisibility(_ context.Context, _ *mcp.CallToolRequest, input WindowVisibilityInput) (*mcp.CallToolResult, WindowVisibilityOutput, error) {
|
||||
// Visibility uses focus/close as approximation (no dedicated visibility task yet)
|
||||
if input.Visible {
|
||||
_, _, err := s.core.PERFORM(window.TaskFocus{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowVisibilityOutput{}, err
|
||||
}
|
||||
}
|
||||
return nil, WindowVisibilityOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- window_fullscreen ---
|
||||
|
||||
type WindowFullscreenInput struct {
|
||||
Name string `json:"name"`
|
||||
Fullscreen bool `json:"fullscreen"`
|
||||
}
|
||||
type WindowFullscreenOutput struct {
|
||||
Success bool `json:"success"`
|
||||
}
|
||||
|
||||
func (s *Subsystem) windowFullscreen(_ context.Context, _ *mcp.CallToolRequest, input WindowFullscreenInput) (*mcp.CallToolResult, WindowFullscreenOutput, error) {
|
||||
// No dedicated fullscreen task in IPC yet; use maximise as approximation
|
||||
if input.Fullscreen {
|
||||
_, _, err := s.core.PERFORM(window.TaskMaximise{Name: input.Name})
|
||||
if err != nil {
|
||||
return nil, WindowFullscreenOutput{}, err
|
||||
}
|
||||
}
|
||||
return nil, WindowFullscreenOutput{Success: true}, nil
|
||||
}
|
||||
|
||||
// --- Registration ---
|
||||
|
||||
func (s *Subsystem) registerWindowTools(server *mcp.Server) {
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_list", Description: "List all application windows"}, s.windowList)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_get", Description: "Get information about a specific window"}, s.windowGet)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_focused", Description: "Get the currently focused window"}, s.windowFocused)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_create", Description: "Create a new application window"}, s.windowCreate)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_close", Description: "Close an application window"}, s.windowClose)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_position", Description: "Set the position of a window"}, s.windowPosition)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_size", Description: "Set the size of a window"}, s.windowSize)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_bounds", Description: "Set both position and size of a window"}, s.windowBounds)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_maximize", Description: "Maximise a window"}, s.windowMaximize)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_minimize", Description: "Minimise a window"}, s.windowMinimize)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_restore", Description: "Restore a maximised or minimised window"}, s.windowRestore)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_focus", Description: "Bring a window to the front"}, s.windowFocus)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_title", Description: "Set the title of a window"}, s.windowTitle)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_visibility", Description: "Show or hide a window"}, s.windowVisibility)
|
||||
mcp.AddTool(server, &mcp.Tool{Name: "window_fullscreen", Description: "Set a window to fullscreen mode"}, s.windowFullscreen)
|
||||
}
|
||||
Loading…
Add table
Reference in a new issue