Add completions (#8)

.gitignore

@@ -3,3 +3,4 @@ bin
 
 dist/
 .github_token
+completions

.goreleaser.yaml

@@ -10,10 +10,8 @@ version: 1
 
 before:
   hooks:
-    # You may remove this if you don't use go modules.
     - go mod tidy
-    # you may remove this if you don't need go generate
-    - go generate ./...
+    - ./scripts/completions.zsh
 
 builds:
   - env:
@@ -23,6 +21,10 @@ builds:
 
 archives:
   - format: tar.gz
+    files:
+      - README.md
+      - LICENSE
+      - completions/*
     # this name template makes the OS and Arch compatible with the results of `uname`.
     name_template: >-
       {{ .ProjectName }}_
@@ -43,6 +45,10 @@ env_files:
   github_token: .github_token
 
 brews:
-  - repository:
+  - extra_install: |-
+      zsh_completion.install "completions/{{ .ProjectName }}.zsh" => "{{ .ProjectName }}"
+    homepage: "https://github.com/Bhacaz/gostacking"
+    description: "Git stacking using merge"
+    repository:
       owner: Bhacaz
       name: homebrew-tap

README.md

@@ -89,8 +89,8 @@ goreleaser release
 
 ## Release
 
-1. Update `.version`
-2. `zsh release.zsh`
+1. Update the version in file `.version`
+2. `zsh scripts/release.zsh`
 
 ## TODOs
 

cmd/root.go

@@ -40,4 +40,5 @@ func init() {
 	// Cobra also supports local flags, which will only run
 	// when this action is called directly.
 	// 	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+	rootCmd.CompletionOptions.HiddenDefaultCmd = true
 }

cmd/switch.go

@@ -21,6 +21,9 @@ var switchCmd = &cobra.Command{
 			stack.Manager().SwitchByName(args[0])
 		}
 	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		return stack.Manager().ListStacksForCompletion(toComplete), cobra.ShellCompDirectiveNoFileComp
+	},
 }
 
 func init() {

internal/stack/stack_manager.go

@@ -6,6 +6,7 @@ import (
 	"github.com/Bhacaz/gostacking/internal/git"
 	"log"
 	"slices"
+	"strings"
 )
 
 type StacksManager struct {
@@ -95,6 +96,17 @@ func (sm StacksManager) List() {
 	}
 }
 
+func (sm StacksManager) ListStacksForCompletion(toComplete string) []string {
+	data := sm.load()
+	var stacks []string
+	for _, stack := range data.Stacks {
+		if toComplete == "" || strings.HasPrefix(stack.Name, toComplete) {
+			stacks = append(stacks, stack.Name)
+		}
+	}
+	return stacks
+}
+
 func (sm StacksManager) SwitchByName(stackName string) {
 	data := sm.load()
 	data.CurrentStack = stackName

scripts/completions.zsh

@@ -0,0 +1,7 @@
+#!/bin/zsh
+
+set -e
+rm -rf completions
+mkdir completions
+
+go run . completion zsh > "completions/gostacking.zsh"