Merge pull request #11 from bpradipt/completion

Add command completion support

Author: bpradipt

README.md

@@ -151,6 +151,82 @@ make build
 sudo make install
 ```
 
+## Shell Completion
+
+kubectl-coco supports command autocompletion for bash and zsh shells.
+
+### Bash
+
+**Prerequisites:**
+
+Install bash-completion.
+
+For MacOS:
+
+```bash
+brew install bash-completion@2
+
+# Add to your ~/.bash_profile:
+[[ -r "/opt/homebrew/etc/profile.d/bash_completion.sh" ]] && . "/opt/homebrew/etc/profile.d/bash_completion.sh"
+
+# Reload profile:
+source ~/.bash_profile
+```
+
+For Linux:
+
+```bash
+# Ubuntu/Debian:
+apt-get install bash-completion
+
+# CentOS/RHEL:
+yum install bash-completion
+```
+
+**Installation:**
+
+For current session:
+```bash
+source <(kubectl-coco completion bash)
+```
+
+For all sessions (permanent):
+```bash
+# MacOS:
+kubectl-coco completion bash > $(brew --prefix)/etc/bash_completion.d/kubectl-coco
+
+# Linux:
+kubectl-coco completion bash > /etc/bash_completion.d/kubectl-coco
+
+# Then restart your shell
+```
+
+**For kubectl plugin (`kubectl coco`):**
+
+Install kubectl completion first:
+
+```bash
+# MacOS:
+kubectl completion bash > $(brew --prefix)/etc/bash_completion.d/kubectl
+
+# Linux:
+kubectl completion bash > /etc/bash_completion.d/kubectl
+```
+
+### Zsh
+
+Enable completion if not already enabled:
+```bash
+echo "autoload -U compinit; compinit" >> ~/.zshrc
+```
+
+Install kubectl-coco completion:
+```bash
+kubectl-coco completion zsh > "${fpath[1]}/_kubectl-coco"
+```
+
+Start a new shell for completion to take effect.
+
 ## Usage
 
 ### Initialize Configuration

claude.md

@@ -45,6 +45,7 @@ make clean
 # Build and test locally
 ./kubectl-coco init --help
 ./kubectl-coco apply --help
+./kubectl-coco completion bash  # Generate shell completion
 
 # Test transformation without applying
 ./kubectl-coco apply -f examples/pod.yaml --skip-apply

cmd/completion.go

@@ -0,0 +1,142 @@
+package cmd
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"os"
+	"strings"
+
+	"github.com/spf13/cobra"
+)
+
+var completionCmd = &cobra.Command{
+	Use:   "completion [bash|zsh]",
+	Short: "Generate shell completion script",
+	Long: `Generate shell completion script for kubectl-coco.
+
+PREREQUISITES:
+
+  Bash requires bash-completion to be installed:
+    macOS: brew install bash-completion@2
+    Linux: Usually pre-installed (apt-get/yum install bash-completion)
+
+  Add to your ~/.bash_profile (macOS):
+    [[ -r "/opt/homebrew/etc/profile.d/bash_completion.sh" ]] && . "/opt/homebrew/etc/profile.d/bash_completion.sh"
+
+BASH:
+
+  Current session:
+    $ source <(kubectl-coco completion bash)
+
+  Permanent:
+    $ kubectl-coco completion bash > $(brew --prefix)/etc/bash_completion.d/kubectl-coco
+    $ source ~/.bash_profile
+
+ZSH:
+
+  Install completion:
+    $ kubectl-coco completion zsh > "${fpath[1]}/_kubectl-coco"
+    $ exec zsh
+
+KUBECTL PLUGIN:
+
+  For 'kubectl coco' completion, install kubectl completion:
+    $ kubectl completion bash > $(brew --prefix)/etc/bash_completion.d/kubectl
+`,
+	DisableFlagsInUseLine: true,
+	ValidArgs:             []string{"bash", "zsh"},
+	Args:                  cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs),
+	RunE: func(_ *cobra.Command, args []string) error {
+		switch args[0] {
+		case "bash":
+			return genBashCompletionWithPlugin(os.Stdout)
+		case "zsh":
+			return genZshCompletionWithPlugin(os.Stdout)
+		}
+		return nil
+	},
+}
+
+func genBashCompletionWithPlugin(out io.Writer) error {
+	// Generate standard completion
+	buf := new(bytes.Buffer)
+	if err := rootCmd.GenBashCompletion(buf); err != nil {
+		return err
+	}
+
+	completion := buf.String()
+
+	// Find the last "if [[ $(type -t compopt)" block and remove it
+	// We'll replace it with our own that includes kubectl_coco
+	lastIfIdx := strings.LastIndex(completion, "if [[ $(type -t compopt)")
+	if lastIfIdx > 0 {
+		// Find the start of this line
+		lineStart := strings.LastIndex(completion[:lastIfIdx], "\n")
+		if lineStart < 0 {
+			lineStart = 0
+		} else {
+			lineStart++ // Move past the newline
+		}
+		completion = completion[:lineStart]
+	}
+
+	// Write the modified completion
+	if _, err := fmt.Fprint(out, completion); err != nil {
+		return err
+	}
+
+	// Add kubectl plugin completion (kubectl coco and kubectl-coco)
+	pluginCompletion := `
+# kubectl plugin completion
+if [[ $(type -t compopt) = "builtin" ]]; then
+    complete -o default -F __start_kubectl-coco kubectl-coco kubectl_coco
+else
+    complete -o default -o nospace -F __start_kubectl-coco kubectl-coco kubectl_coco
+fi
+
+# ex: ts=4 sw=4 et filetype=sh
+`
+	_, err := fmt.Fprint(out, pluginCompletion)
+	return err
+}
+
+func genZshCompletionWithPlugin(out io.Writer) error {
+	// Generate standard completion
+	buf := new(bytes.Buffer)
+	if err := rootCmd.GenZshCompletion(buf); err != nil {
+		return err
+	}
+
+	completion := buf.String()
+
+	// Find and remove the last compdef line
+	lastCompdefIdx := strings.LastIndex(completion, "compdef _kubectl-coco kubectl-coco")
+	if lastCompdefIdx > 0 {
+		// Find the start of this line
+		lineStart := strings.LastIndex(completion[:lastCompdefIdx], "\n")
+		if lineStart < 0 {
+			lineStart = 0
+		} else {
+			lineStart++ // Move past the newline
+		}
+		completion = completion[:lineStart]
+	}
+
+	// Write the modified completion
+	if _, err := fmt.Fprint(out, completion); err != nil {
+		return err
+	}
+
+	// Add kubectl plugin completion (kubectl coco and kubectl-coco)
+	pluginCompletion := `
+# kubectl plugin completion
+compdef _kubectl-coco kubectl-coco kubectl_coco
+`
+	_, err := fmt.Fprint(out, pluginCompletion)
+	return err
+}
+
+func init() {
+	rootCmd.AddCommand(completionCmd)
+}