add: more commands

internal/pkg/tools/give/give.go

@@ -0,0 +1,101 @@
+package give
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/ollama/ollama/api"
+	"tipsy.codes/charles/mc-god/v2/internal/pkg/rcon"
+)
+
+type Give struct{}
+
+func Get() *Give {
+	return &Give{}
+}
+
+func (g *Give) Do(ctx context.Context, toolCall api.ToolCall, client *rcon.Client) error {
+	// Extract the arguments from the tool call
+	args := toolCall.Function.Arguments
+
+	player, found := args.Get("player")
+	if !found {
+		return fmt.Errorf("missing player argument")
+	}
+	playerString, ok := player.(string)
+	if !ok {
+		return fmt.Errorf("incorrect data type %v; want string", player)
+	}
+
+	item, found := args.Get("item")
+	if !found {
+		return fmt.Errorf("missing item argument")
+	}
+	itemString, ok := item.(string)
+	if !ok {
+		return fmt.Errorf("incorrect data type %T; want int", item)
+	}
+
+	// Handle count (optional, default to 1)
+	count := 1
+	countVal, found := args.Get("count")
+	if found {
+		if countFloat, ok := countVal.(float64); ok {
+			count = int(countFloat)
+		} else {
+			return fmt.Errorf("incorrect data type %T; want number", countVal)
+		}
+	}
+
+	// Validate that we have valid player and item names (basic validation)
+	if strings.TrimSpace(playerString) == "" {
+		return fmt.Errorf("player and item names cannot be empty")
+	}
+
+	// Send the give command to the Minecraft server
+	command := "/give " + playerString + " " + itemString + " " + fmt.Sprintf("%d", count)
+	_, err := client.Execute(command)
+	if err != nil {
+		return fmt.Errorf("failed to execute give command: %w", err)
+	}
+
+	return nil
+}
+
+func (g *Give) Desc() api.Tool {
+	toolPropertiesMap := api.NewToolPropertiesMap()
+	toolPropertiesMap.Set("player", api.ToolProperty{
+		Type:        api.PropertyType{"string"},
+		Description: "The player to give the item to",
+	})
+	toolPropertiesMap.Set("item", api.ToolProperty{
+		Type: api.PropertyType{"string"},
+		Description: `The item to give. Items can include:
+		- dirt
+		- carrot
+		- reeds
+		`,
+	})
+	toolPropertiesMap.Set("count", api.ToolProperty{
+		Type:        api.PropertyType{"integer"},
+		Description: "The number of items to give (default is 1)",
+	})
+
+	return api.Tool{
+		Type: "function",
+		Function: api.ToolFunction{
+			Name:        g.Name(),
+			Description: "Give items to a player in Minecraft",
+			Parameters: api.ToolFunctionParameters{
+				Type:       "object",
+				Properties: toolPropertiesMap,
+				Required:   []string{"player", "item"},
+			},
+		},
+	}
+}
+
+func (g *Give) Name() string {
+	return "give"
+}