update install script

Author: jjchen01

install.sh

@@ -141,6 +141,38 @@ install() {
     fi
     log_info "Version $VERSION written to config"
 
+    # Install shell completions to config directory (not modifying user shell files)
+    COMPLETION_DIR="$CONFIG_DIR/completions"
+    mkdir -p "$COMPLETION_DIR"
+
+    # Detect current shell
+    SHELL_NAME="$(basename "$SHELL" 2>/dev/null || echo "bash")"
+
+    case "$SHELL_NAME" in
+        zsh)
+            "$BINARY_NAME" completion zsh > "$COMPLETION_DIR/zsh"
+            log_info "Zsh completion installed to $COMPLETION_DIR/zsh"
+            log_info "Add to your .zshrc: source $COMPLETION_DIR/zsh"
+            ;;
+        fish)
+            "$BINARY_NAME" completion fish > "$COMPLETION_DIR/fish"
+            log_info "Fish completion installed to $COMPLETION_DIR/fish"
+            log_info "Add to your config.fish: source $COMPLETION_DIR/fish"
+            ;;
+        bash)
+            "$BINARY_NAME" completion bash > "$COMPLETION_DIR/bash"
+            log_info "Bash completion installed to $COMPLETION_DIR/bash"
+            log_info "Add to your .bashrc: source $COMPLETION_DIR/bash"
+            ;;
+        *)
+            # Install all completions
+            "$BINARY_NAME" completion bash > "$COMPLETION_DIR/bash" 2>/dev/null || true
+            "$BINARY_NAME" completion zsh > "$COMPLETION_DIR/zsh" 2>/dev/null || true
+            "$BINARY_NAME" completion fish > "$COMPLETION_DIR/fish" 2>/dev/null || true
+            log_info "Completions installed to $COMPLETION_DIR"
+            ;;
+    esac
+
     # Cleanup
     cd /
     rm -rf "$TEMP_DIR"

internal/commands/completion.go

@@ -0,0 +1,73 @@
+package commands
+
+import (
+	"os"
+
+	"github.com/spf13/cobra"
+)
+
+var completionCmd = &cobra.Command{
+	Use:    "completion [shell]",
+	Short:  "Generate shell completion scripts",
+	Hidden: true,
+	Long: `Generate shell completion scripts for menlo.
+
+To load completions:
+
+Bash:
+
+  $ source <(menlo completion bash)
+
+  # To load completions for each session, execute once:
+  # Linux:
+  $ menlo completion bash > /etc/bash_completion.d/menlo
+  # macOS:
+  $ menlo completion bash > /usr/local/etc/bash_completion.d/menlo
+
+Zsh:
+
+  # If shell completion is not already enabled in your environment,
+  # you will need to enable it.  You can execute the following once:
+
+  $ echo "autoload -U compinit; compinit" >> ~/.zshrc
+
+  # To load completions for each session, execute once:
+  $ menlo completion zsh > "${fpath[1]}/_menlo"
+
+  # You will need to start a new shell for this setup to take effect.
+
+Fish:
+
+  $ menlo completion fish | source
+
+  # To load completions for each session, execute once:
+  $ menlo completion fish > ~/.config/fish/completions/menlo.fish
+
+PowerShell:
+
+  PS> menlo completion powershell | Out-String | Invoke-Expression
+
+  # To load completions for every session, add the output of the previous
+  # command to your powershell profile.
+`,
+	DisableFlagsInUseLine: true,
+	ValidArgs:             []string{"bash", "zsh", "fish", "powershell"},
+	Args:                  cobra.MatchAll(cobra.ExactArgs(1), cobra.OnlyValidArgs),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		switch args[0] {
+		case "bash":
+			return rootCmd.GenBashCompletion(os.Stdout)
+		case "zsh":
+			return rootCmd.GenZshCompletion(os.Stdout)
+		case "fish":
+			return rootCmd.GenFishCompletion(os.Stdout, true)
+		case "powershell":
+			return rootCmd.GenPowerShellCompletionWithDesc(os.Stdout)
+		}
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(completionCmd)
+}