Add the please command

Signed-off-by: Jean-Laurent de Morlhon <jeanlaurent@morlhon.net>

Author: jeanlaurent

cmd/root/alias.go

@@ -0,0 +1,200 @@
+package root
+
+import (
+	"fmt"
+	"path/filepath"
+	"sort"
+	"strings"
+
+	"github.com/spf13/cobra"
+
+	"github.com/docker/cagent/pkg/shortcuts"
+	"github.com/docker/cagent/pkg/telemetry"
+)
+
+// NewAliasCmd creates a new alias command for managing aliases
+func NewAliasCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "alias [<alias-name> <agent-path>]",
+		Short: "Manage aliases for agents",
+		Long: `Create and manage aliases for agent configurations or catalog references.
+
+When called without arguments, lists all registered aliases.`,
+		Example: `  # List all registered aliases (default)
+  cagent alias
+
+  # Create an alias for a catalog agent
+  cagent alias code agentcatalog/notion-expert
+
+  # Create an alias for a local agent file
+  cagent alias myagent ~/myagent.yaml
+
+  # List all registered aliases
+  cagent alias list
+
+  # Remove an alias
+  cagent alias remove code`,
+		Args: cobra.RangeArgs(0, 2),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			// Default to list when no arguments provided
+			if len(args) == 0 {
+				telemetry.TrackCommand("alias", []string{"list"})
+				return listAliases()
+			}
+
+			// Create new alias with 2 arguments
+			if len(args) == 2 {
+				telemetry.TrackCommand("alias", args)
+				return createAlias(args[0], args[1])
+			}
+
+			// Single argument is invalid in this context
+			return fmt.Errorf("invalid arguments: use 'alias list' or 'alias <name> <agent-path>'")
+		},
+	}
+
+	cmd.AddCommand(newAliasListCmd())
+	cmd.AddCommand(newAliasRemoveCmd())
+
+	return cmd
+}
+
+// newAliasListCmd creates the list subcommand
+func newAliasListCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "list",
+		Short: "List all registered aliases",
+		Args:  cobra.NoArgs,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			telemetry.TrackCommand("alias", []string{"list"})
+			return listAliases()
+		},
+	}
+}
+
+// newAliasRemoveCmd creates the remove subcommand
+func newAliasRemoveCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "remove <alias-name>",
+		Short: "Remove a registered alias",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			telemetry.TrackCommand("alias", append([]string{"remove"}, args...))
+			return removeAlias(args[0])
+		},
+	}
+}
+
+// createAlias creates a new alias
+func createAlias(name, agentPath string) error {
+	// Load existing aliases
+	s, err := shortcuts.Load()
+	if err != nil {
+		return fmt.Errorf("failed to load aliases: %w", err)
+	}
+
+	// Expand tilde in path if it's a local file path
+	if strings.HasPrefix(agentPath, "~/") {
+		if abs, err := expandTilde(agentPath); err == nil {
+			agentPath = abs
+		}
+	}
+
+	// Store the alias
+	s.Set(name, agentPath)
+
+	// Save to file
+	if err := s.Save(); err != nil {
+		return fmt.Errorf("failed to save aliases: %w", err)
+	}
+
+	fmt.Printf("Alias '%s' created successfully\n", name)
+	fmt.Printf("  Alias: %s\n", name)
+	fmt.Printf("  Agent: %s\n", agentPath)
+	fmt.Printf("\nYou can now run: cagent run %s\n", name)
+
+	return nil
+}
+
+// removeAlias removes an alias
+func removeAlias(name string) error {
+	s, err := shortcuts.Load()
+	if err != nil {
+		return fmt.Errorf("failed to load aliases: %w", err)
+	}
+
+	if !s.Delete(name) {
+		return fmt.Errorf("alias '%s' not found", name)
+	}
+
+	if err := s.Save(); err != nil {
+		return fmt.Errorf("failed to save aliases: %w", err)
+	}
+
+	fmt.Printf("Alias '%s' removed successfully\n", name)
+	return nil
+}
+
+func listAliases() error {
+	s, err := shortcuts.Load()
+	if err != nil {
+		return fmt.Errorf("failed to load aliases: %w", err)
+	}
+
+	allAliases := s.List()
+	if len(allAliases) == 0 {
+		fmt.Println("No aliases registered.")
+		fmt.Println("\nCreate an alias with: cagent alias <name> <agent-path>")
+		return nil
+	}
+
+	fmt.Printf("Registered aliases (%d):\n\n", len(allAliases))
+
+	// Sort aliases by name for consistent output
+	names := make([]string, 0, len(allAliases))
+	for name := range allAliases {
+		names = append(names, name)
+	}
+	sort.Strings(names)
+
+	// Find max name length for alignment
+	maxLen := 0
+	for _, name := range names {
+		if len(name) > maxLen {
+			maxLen = len(name)
+		}
+	}
+
+	for _, name := range names {
+		path := allAliases[name]
+		padding := strings.Repeat(" ", maxLen-len(name))
+		fmt.Printf("  %s%s → %s\n", name, padding, path)
+	}
+
+	fmt.Printf("\nRun an alias with: cagent run <alias>\n")
+
+	return nil
+}
+
+// expandTilde expands the tilde in a path to the user's home directory
+func expandTilde(path string) (string, error) {
+	if !strings.HasPrefix(path, "~/") {
+		return path, nil
+	}
+
+	homeDir, err := filepath.Abs(filepath.Join(filepath.Dir(path), ".."))
+	if err != nil {
+		return "", err
+	}
+
+	// Get actual home directory
+	homeDir, err = filepath.EvalSymlinks(homeDir)
+	if err != nil {
+		homeDir, err = filepath.Abs("~")
+		if err != nil {
+			return "", err
+		}
+	}
+
+	return filepath.Join(homeDir, strings.TrimPrefix(path, "~/")), nil
+}

cmd/root/root.go

@@ -123,6 +123,7 @@ func NewRootCmd() *cobra.Command {
 	cmd.AddCommand(NewCatalogCmd())
 	cmd.AddCommand(NewBuildCmd())
 	cmd.AddCommand(NewPrintCmd())
+	cmd.AddCommand(NewAliasCmd())
 
 	return cmd
 }

cmd/root/run.go

@@ -28,6 +28,7 @@ import (
 	"github.com/docker/cagent/pkg/remote"
 	"github.com/docker/cagent/pkg/runtime"
 	"github.com/docker/cagent/pkg/session"
+	"github.com/docker/cagent/pkg/shortcuts"
 	"github.com/docker/cagent/pkg/team"
 	"github.com/docker/cagent/pkg/teamloader"
 	"github.com/docker/cagent/pkg/telemetry"
@@ -94,6 +95,15 @@ func doRunCommand(ctx context.Context, args []string, exec bool) error {
 	slog.Debug("Starting agent", "agent", agentName, "debug_mode", debugMode)
 
 	agentFilename := args[0]
+
+	// Try to resolve as an alias first
+	if aliases, aliasErr := shortcuts.Load(); aliasErr == nil {
+		if resolvedPath, ok := aliases.Get(agentFilename); ok {
+			slog.Debug("Resolved alias", "alias", agentFilename, "path", resolvedPath)
+			agentFilename = resolvedPath
+		}
+	}
+
 	if !strings.Contains(agentFilename, "\n") && (strings.Contains(agentFilename, ".yaml") || strings.Contains(agentFilename, ".yml")) {
 		if abs, err := filepath.Abs(agentFilename); err == nil {
 			agentFilename = abs

pkg/shortcuts/shortcuts.go

@@ -0,0 +1,100 @@
+package shortcuts
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/goccy/go-yaml"
+
+	"github.com/docker/cagent/pkg/paths"
+)
+
+// Shortcuts represents the shortcuts configuration
+type Shortcuts struct {
+	Shortcuts map[string]string `yaml:"shortcuts"`
+}
+
+// GetShortcutsFilePath returns the path to the shortcuts file
+func GetShortcutsFilePath() string {
+	return filepath.Join(paths.GetConfigDir(), "shortcuts.yaml")
+}
+
+// Load loads shortcuts from the configuration file
+func Load() (*Shortcuts, error) {
+	return LoadFrom(GetShortcutsFilePath())
+}
+
+// LoadFrom loads shortcuts from a specific file path
+func LoadFrom(path string) (*Shortcuts, error) {
+	// If file doesn't exist, return empty shortcuts
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return &Shortcuts{Shortcuts: make(map[string]string)}, nil
+	}
+
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read shortcuts file: %w", err)
+	}
+
+	var s Shortcuts
+	if err := yaml.Unmarshal(data, &s); err != nil {
+		return nil, fmt.Errorf("failed to parse shortcuts file: %w", err)
+	}
+
+	if s.Shortcuts == nil {
+		s.Shortcuts = make(map[string]string)
+	}
+
+	return &s, nil
+}
+
+// Save saves shortcuts to the configuration file
+func (s *Shortcuts) Save() error {
+	return s.SaveTo(GetShortcutsFilePath())
+}
+
+// SaveTo saves shortcuts to a specific file path
+func (s *Shortcuts) SaveTo(path string) error {
+	// Ensure directory exists
+	dir := filepath.Dir(path)
+	if err := os.MkdirAll(dir, 0o755); err != nil {
+		return fmt.Errorf("failed to create directory: %w", err)
+	}
+
+	data, err := yaml.Marshal(s)
+	if err != nil {
+		return fmt.Errorf("failed to marshal shortcuts: %w", err)
+	}
+
+	if err := os.WriteFile(path, data, 0o644); err != nil {
+		return fmt.Errorf("failed to write shortcuts file: %w", err)
+	}
+
+	return nil
+}
+
+// Get retrieves the agent path for a shortcut
+func (s *Shortcuts) Get(name string) (string, bool) {
+	path, ok := s.Shortcuts[name]
+	return path, ok
+}
+
+// Set creates or updates a shortcut
+func (s *Shortcuts) Set(name, agentPath string) {
+	s.Shortcuts[name] = agentPath
+}
+
+// Delete removes a shortcut
+func (s *Shortcuts) Delete(name string) bool {
+	if _, exists := s.Shortcuts[name]; exists {
+		delete(s.Shortcuts, name)
+		return true
+	}
+	return false
+}
+
+// List returns all shortcuts
+func (s *Shortcuts) List() map[string]string {
+	return s.Shortcuts
+}