ask: add fish UUID completions

2 files changed, 131 insertions, 9 deletions

diff --git a/internal/askcli/completion.go b/internal/askcli/completion.go
index f787849..28ad306 100644
--- a/internal/askcli/completion.go
+++ b/internal/askcli/completion.go
@@ -35,11 +35,61 @@ var askDepCompletionItems = []fishCompletionItem{
 	{name: "list", description: "List dependencies"},
 }
 
+var askUUIDCompletionItems = []fishCompletionItem{
+	{name: "info", description: "Show task details"},
+	{name: "annotate", description: "Add an annotation"},
+	{name: "start", description: "Start a task"},
+	{name: "stop", description: "Stop a task"},
+	{name: "done", description: "Mark a task complete"},
+	{name: "priority", description: "Set priority"},
+	{name: "tag", description: "Add or remove a tag"},
+	{name: "modify", description: "Modify task fields"},
+	{name: "denotate", description: "Remove an annotation"},
+	{name: "delete", description: "Delete a task"},
+}
+
+var askDepUUIDCompletionItems = []fishCompletionItem{
+	{name: "add", description: "Add a dependency"},
+	{name: "rm", description: "Remove a dependency"},
+	{name: "list", description: "List dependencies"},
+}
+
 func FishCompletion() string {
 	var b strings.Builder
+	writeFishPreamble(&b)
+	writeFishContextFunctions(&b)
+	writeFishTaskUUIDFunction(&b)
+	b.WriteString("complete -c ask -f\n")
+	b.WriteString("complete -c ask -s j -l json -d 'Emit JSON output'\n")
+	for _, item := range askRootCompletionItems {
+		writeFishCompletionLine(&b, "__ask_needs_root_completion", item)
+	}
+	for _, item := range askDepCompletionItems {
+		writeFishCompletionLine(&b, "__ask_in_dep_context", item)
+	}
+	for _, item := range askUUIDCompletionItems {
+		writeFishUUIDCompletionLine(&b, "__ask_in_uuid_context", item.description)
+	}
+	for _, item := range askDepUUIDCompletionItems {
+		writeFishUUIDCompletionLine(&b, "__ask_in_dep_uuid_context", item.description)
+	}
+	return b.String()
+}
+
+func writeFishPreamble(b *strings.Builder) {
 	b.WriteString("# Fish completion for ask.\n")
 	b.WriteString("# Install as ~/.config/fish/completions/ask.fish or")
 	b.WriteString(" $XDG_CONFIG_HOME/fish/completions/ask.fish.\n\n")
+}
+
+func writeFishContextFunctions(b *strings.Builder) {
+	writeFishNeedsRootCompletionFunction(b)
+	writeFishDepContextFunction(b)
+	writeFishUUIDContextFunction(b)
+	writeFishDepUUIDContextFunction(b)
+}
+
+func writeFishNeedsRootCompletionFunction(b *strings.Builder) {
 	b.WriteString("function __ask_needs_root_completion\n")
 	b.WriteString("    set -l tokens (commandline -opc)\n")
 	b.WriteString("    if test (count $tokens) -le 1\n")
@@ -52,6 +102,9 @@ func FishCompletion() string {
 	b.WriteString("    end\n")
 	b.WriteString("    return 0\n")
 	b.WriteString("end\n\n")
+}
+
+func writeFishDepContextFunction(b *strings.Builder) {
 	b.WriteString("function __ask_in_dep_context\n")
 	b.WriteString("    set -l tokens (commandline -opc)\n")
 	b.WriteString("    if test (count $tokens) -lt 2\n")
@@ -74,15 +127,70 @@ func FishCompletion() string {
 	b.WriteString("    end\n")
 	b.WriteString("    test $seen_dep -eq 1\n")
 	b.WriteString("end\n\n")
-	b.WriteString("complete -c ask -f\n")
-	b.WriteString("complete -c ask -s j -l json -d 'Emit JSON output'\n")
-	for _, item := range askRootCompletionItems {
-		writeFishCompletionLine(&b, "__ask_needs_root_completion", item)
-	}
-	for _, item := range askDepCompletionItems {
-		writeFishCompletionLine(&b, "__ask_in_dep_context", item)
-	}
-	return b.String()
+}
+
+func writeFishUUIDContextFunction(b *strings.Builder) {
+	b.WriteString("function __ask_in_uuid_context\n")
+	b.WriteString("    set -l tokens (commandline -opc)\n")
+	b.WriteString("    set -l positional\n")
+	b.WriteString("    for token in $tokens[2..-1]\n")
+	b.WriteString("        if string match -qr '^-' -- $token\n")
+	b.WriteString("            continue\n")
+	b.WriteString("        end\n")
+	b.WriteString("        set -a positional $token\n")
+	b.WriteString("    end\n")
+	b.WriteString("    if test (count $positional) -eq 0\n")
+	b.WriteString("        return 1\n")
+	b.WriteString("    end\n")
+	b.WriteString("    if test (count $positional) -gt 2\n")
+	b.WriteString("        return 1\n")
+	b.WriteString("    end\n")
+	b.WriteString("    switch $positional[1]\n")
+	b.WriteString("        case info annotate start stop done priority tag modify denotate delete\n")
+	b.WriteString("            return 0\n")
+	b.WriteString("        case '*'\n")
+	b.WriteString("            return 1\n")
+	b.WriteString("    end\n")
+	b.WriteString("    return 1\n")
+	b.WriteString("end\n\n")
+}
+
+func writeFishDepUUIDContextFunction(b *strings.Builder) {
+	b.WriteString("function __ask_in_dep_uuid_context\n")
+	b.WriteString("    set -l tokens (commandline -opc)\n")
+	b.WriteString("    set -l positional\n")
+	b.WriteString("    for token in $tokens[2..-1]\n")
+	b.WriteString("        if string match -qr '^-' -- $token\n")
+	b.WriteString("            continue\n")
+	b.WriteString("        end\n")
+	b.WriteString("        set -a positional $token\n")
+	b.WriteString("    end\n")
+	b.WriteString("    if test (count $positional) -lt 2\n")
+	b.WriteString("        return 1\n")
+	b.WriteString("    end\n")
+	b.WriteString("    if test $positional[1] != dep\n")
+	b.WriteString("        return 1\n")
+	b.WriteString("    end\n")
+	b.WriteString("    switch $positional[2]\n")
+	b.WriteString("        case add rm\n")
+	b.WriteString("            if test (count $positional) -le 4\n")
+	b.WriteString("                return 0\n")
+	b.WriteString("            end\n")
+	b.WriteString("        case list\n")
+	b.WriteString("            if test (count $positional) -le 3\n")
+	b.WriteString("                return 0\n")
+	b.WriteString("            end\n")
+	b.WriteString("        case '*'\n")
+	b.WriteString("            return 1\n")
+	b.WriteString("    end\n")
+	b.WriteString("    return 1\n")
+	b.WriteString("end\n\n")
+}
+
+func writeFishTaskUUIDFunction(b *strings.Builder) {
+	b.WriteString("function __ask_task_uuids\n")
+	b.WriteString("    command ask all --json 2>/dev/null | string match -r -a -g '\"uuid\":\"([^\"]+)\"'\n")
+	b.WriteString("end\n\n")
 }
 
 func writeFishCompletionLine(b *strings.Builder, condition string, item fishCompletionItem) {
@@ -94,3 +202,11 @@ func writeFishCompletionLine(b *strings.Builder, condition string, item fishComp
 	b.WriteString(strings.ReplaceAll(item.description, "'", "\\'"))
 	b.WriteString("'\n")
 }
+
+func writeFishUUIDCompletionLine(b *strings.Builder, condition, description string) {
+	b.WriteString("complete -c ask -n '")
+	b.WriteString(condition)
+	b.WriteString("' -a '(__ask_task_uuids)' -d '")
+	b.WriteString(strings.ReplaceAll(description, "'", "\\'"))
+	b.WriteString("'\n")
+}
diff --git a/internal/askcli/completion_test.go b/internal/askcli/completion_test.go
index e717575..439e483 100644
--- a/internal/askcli/completion_test.go
+++ b/internal/askcli/completion_test.go
@@ -35,6 +35,12 @@ func TestFishCompletion_IncludesCommandsAndExcludesExport(t *testing.T) {
 		"complete -c ask -n '__ask_in_dep_context' -a 'add' -d 'Add a dependency'",
 		"complete -c ask -n '__ask_in_dep_context' -a 'rm' -d 'Remove a dependency'",
 		"complete -c ask -n '__ask_in_dep_context' -a 'list' -d 'List dependencies'",
+		"function __ask_task_uuids",
+		"command ask all --json 2>/dev/null",
+		"complete -c ask -n '__ask_in_uuid_context' -a '(__ask_task_uuids)' -d 'Show task details'",
+		"complete -c ask -n '__ask_in_dep_uuid_context' -a '(__ask_task_uuids)' -d 'Add a dependency'",
+		"complete -c ask -n '__ask_in_dep_uuid_context' -a '(__ask_task_uuids)' -d 'Remove a dependency'",
+		"complete -c ask -n '__ask_in_dep_uuid_context' -a '(__ask_task_uuids)' -d 'List dependencies'",
 	} {
 		if !strings.Contains(script, line) {
 			t.Fatalf("script missing dep completion line %q", line)