feat: implement dynamic shell completion for margin keys and IDs (#22)

Author: codevalve

cmd/delete.go

@@ -27,6 +27,23 @@ var deleteCmd = &cobra.Command{
 
 		fmt.Printf("%s Entry %s deleted.\n", successStyle.Render("✓"), id)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		s, err := storage.NewStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		ids, err := s.GetRecentIDs()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		return ids, cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

cmd/done.go

@@ -46,6 +46,23 @@ var doneCmd = &cobra.Command{
 
 		fmt.Printf("%s Task %s marked as done.\n", successStyle.Render("✓"), id)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		s, err := storage.NewStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		ids, err := s.GetRecentIDs()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		return ids, cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

cmd/edit.go

@@ -30,6 +30,23 @@ var editCmd = &cobra.Command{
 
 		fmt.Printf("%s Entry %s updated.\n", successStyle.Render("✓"), id)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		s, err := storage.NewStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		ids, err := s.GetRecentIDs()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		return ids, cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

cmd/list.go

@@ -124,6 +124,26 @@ var listCmd = &cobra.Command{
 			renderEntry(e, maxMargin)
 		}
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		s, err := storage.NewStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		keys, err := s.GetUniqueMarginKeys()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		// Add static time filters
+		keys = append(keys, "today", "3d", "7d", "30d")
+
+		return keys, cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

cmd/migrate.go

@@ -67,6 +67,23 @@ var migrateCmd = &cobra.Command{
 
 		fmt.Printf("%s Task %s migrated to today (New ID: %s).\n", successStyle.Render("✓"), id, newID)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		s, err := storage.NewStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		ids, err := s.GetRecentIDs()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		return ids, cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

cmd/pin.go

@@ -32,6 +32,23 @@ var pinCmd = &cobra.Command{
 		}
 		fmt.Printf("Entry %s %s.\n", id, status)
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		if len(args) != 0 {
+			return nil, cobra.ShellCompDirectiveNoFileComp
+		}
+
+		s, err := storage.NewStorage()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		ids, err := s.GetRecentIDs()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		return ids, cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

internal/storage/storage.go

@@ -8,6 +8,7 @@ import (
 	"os"
 	"path/filepath"
 	"rapide/internal/model"
+	"sort"
 	"time"
 )
 
@@ -303,6 +304,56 @@ func (s *Storage) TrimBefore(cutoff time.Time) (int, error) {
 	return count, s.saveAll(toKeep)
 }
 
+func (s *Storage) GetUniqueMarginKeys() ([]string, error) {
+	entries, err := s.List()
+	if err != nil {
+		return nil, err
+	}
+
+	keysMap := make(map[string]struct{})
+	for _, e := range entries {
+		if e.MarginKey != "" {
+			keysMap[e.MarginKey] = struct{}{}
+		}
+	}
+
+	keys := make([]string, 0, len(keysMap))
+	for k := range keysMap {
+		keys = append(keys, k)
+	}
+	return keys, nil
+}
+
+func (s *Storage) GetRecentIDs() ([]string, error) {
+	entries, err := s.List()
+	if err != nil {
+		return nil, err
+	}
+
+	// Only return IDs for relevant items (maybe last 50?)
+	// Sort by timestamp desc
+	sort.Slice(entries, func(i, j int) bool {
+		return entries[i].Timestamp.After(entries[j].Timestamp)
+	})
+
+	limit := 50
+	if len(entries) < limit {
+		limit = len(entries)
+	}
+
+	ids := make([]string, 0, limit)
+	for i := 0; i < limit; i++ {
+		// Include content snippet in completion description if possible
+		// Cobra supports "id\tDescription" format
+		snippet := entries[i].Content
+		if len(snippet) > 30 {
+			snippet = snippet[:27] + "..."
+		}
+		ids = append(ids, fmt.Sprintf("%s\t%s %s", entries[i].ID, entries[i].Bullet, snippet))
+	}
+	return ids, nil
+}
+
 func generateID(e model.Entry) string {
 	h := sha1.New()
 	h.Write([]byte(fmt.Sprintf("%d%s%s", e.Timestamp.UnixNano(), e.Bullet, e.Content)))