agent.go

package cmd

import (
	"fmt"
	"os"
	"path/filepath"
	"text/tabwriter"

	"github.com/GrayCodeAI/hawk/internal/multiagent/agents"
	"github.com/spf13/cobra"
	"golang.org/x/text/cases"
	"golang.org/x/text/language"
)

var agentCmd = &cobra.Command{
	Use:   "agent",
	Short: "Manage custom agent personas",
	Long:  "Create, list, and manage custom agent personas stored in Hawk user state.",
}

var agentListCmd = &cobra.Command{
	Use:   "list",
	Short: "List all available agents",
	RunE:  runAgentList,
}

var agentCreateCmd = &cobra.Command{
	Use:   "create <name>",
	Short: "Create a new agent persona",
	Args:  cobra.ExactArgs(1),
	RunE:  runAgentCreate,
}

var agentShowCmd = &cobra.Command{
	Use:   "show <name>",
	Short: "Show an agent's configuration",
	Args:  cobra.ExactArgs(1),
	RunE:  runAgentShow,
}

var agentRemoveCmd = &cobra.Command{
	Use:   "remove <name>",
	Short: "Remove an agent persona",
	Args:  cobra.ExactArgs(1),
	RunE:  runAgentRemove,
}

var (
	agentCreateDesc  string
	agentCreateModel string
)

func init() {
	agentCreateCmd.Flags().StringVarP(&agentCreateDesc, "description", "d", "", "Agent description")
	agentCreateCmd.Flags().StringVarP(&agentCreateModel, "model", "m", "", "Model to use (empty = inherit)")

	agentCmd.AddCommand(agentListCmd)
	agentCmd.AddCommand(agentCreateCmd)
	agentCmd.AddCommand(agentShowCmd)
	agentCmd.AddCommand(agentRemoveCmd)
}

func runAgentList(_ *cobra.Command, _ []string) error {
	all, err := agents.ListAll()
	if err != nil {
		return err
	}
	if len(all) == 0 {
		fmt.Printf("No agents found. Create one with: hawk agent create <name>\n")
		fmt.Printf("Agent directory: %s\n", agents.DefaultDir())
		return nil
	}

	w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
	_, _ = fmt.Fprintf(w, "NAME\tMODEL\tDESCRIPTION\n")
	for _, a := range all {
		model := a.Model
		if model == "" {
			model = "(inherit)"
		}
		desc := a.Description
		if len(desc) > 50 {
			desc = desc[:50] + "..."
		}
		_, _ = fmt.Fprintf(w, "%s\t%s\t%s\n", a.Name, model, desc)
	}
	return w.Flush()
}

func runAgentCreate(_ *cobra.Command, args []string) error {
	name := args[0]
	dir := agents.DefaultDir()
	if err := os.MkdirAll(dir, 0o755); err != nil {
		return err
	}

	path := filepath.Join(dir, name+".md")
	if _, err := os.Stat(path); err == nil {
		return fmt.Errorf("agent %q already exists at %s", name, path)
	}

	modelLine := "inherit"
	if agentCreateModel != "" {
		modelLine = agentCreateModel
	}
	desc := agentCreateDesc
	if desc == "" {
		desc = "Custom agent: " + name
	}

	content := fmt.Sprintf(`---
name: %s
description: %s
model: %s
---
# %s

You are a specialized agent. Complete tasks according to your expertise.

## Guidelines
- Be precise and focused
- Follow project conventions
- Report results clearly
	`, name, desc, modelLine, cases.Title(language.English).String(name))

	if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
		return err
	}

	fmt.Printf("Created agent %q at %s\n", name, path)
	fmt.Printf("Edit the file to customize the system prompt.\n")
	return nil
}

func runAgentShow(_ *cobra.Command, args []string) error {
	a, err := agents.Get(args[0])
	if err != nil {
		return err
	}

	fmt.Printf("Name:        %s\n", a.Name)
	fmt.Printf("Description: %s\n", a.Description)
	model := a.Model
	if model == "" {
		model = "(inherit from session)"
	}
	fmt.Printf("Model:       %s\n", model)
	fmt.Printf("File:        %s\n", a.FilePath)
	fmt.Printf("\n--- Prompt ---\n%s\n", a.Prompt)
	return nil
}

func runAgentRemove(_ *cobra.Command, args []string) error {
	a, err := agents.Get(args[0])
	if err != nil {
		return err
	}

	if err := os.Remove(a.FilePath); err != nil {
		return fmt.Errorf("remove %s: %w", a.FilePath, err)
	}
	fmt.Printf("Removed agent %q (%s)\n", a.Name, a.FilePath)
	return nil
}