feat(cli): enable cobra autocompletion

Author: mdaffad

internal/cli/root.go

@@ -60,7 +60,7 @@ func init() {
 		Hidden: true,
 	})
 
-	rootCmd.CompletionOptions.DisableDefaultCmd = true
+	rootCmd.CompletionOptions.DisableDefaultCmd = false
 
 	cobra.AddTemplateFunc("typeMap", func(cmds []*cobra.Command) map[string][]*cobra.Command {
 		m := make(map[string][]*cobra.Command)

scripts/setup.sh

@@ -100,4 +100,117 @@ fi
 
 echo "Done."
 echo ""
-echo "IMPORTANT: ensure you add /opt/pel/formae/bin to your PATH, and reload your shell configuration"
+echo "IMPORTANT: ensure you add /opt/pel/formae/bin to your PATH, and reload your shell configuration"
+echo ""
+
+CURRENT_SHELL=$(basename "$SHELL" 2>/dev/null || echo "bash")
+
+setup_bash_completions() {
+  echo "Setting up bash completions..."
+  
+  if [[ "$OS" == "darwin" ]]; then
+    # Check if bash-completion is installed on macOS
+    if ! brew list bash-completion &>/dev/null && ! brew list bash-completion@2 &>/dev/null; then
+      echo "Warning: bash-completion is not installed. Install it with:"
+      echo "  brew install bash-completion"
+      echo "Then add the following to your ~/.bash_profile or ~/.bashrc:"
+      echo '  [[ -r "/usr/local/etc/profile.d/bash_completion.sh" ]] && . "/usr/local/etc/profile.d/bash_completion.sh"'
+      echo ""
+      return
+    fi
+  fi
+  
+  # Generate completion file
+  ${INSTALLPREFIX}/formae/bin/formae completion bash > formae-cli-completion.bash 2>/dev/null || {
+    echo "Warning: Could not generate bash completions. Make sure formae is in your PATH."
+    return
+  }
+  
+  if [[ "$OS" == "darwin" ]]; then
+    # macOS with Homebrew bash-completion
+    completion_dir="/usr/local/etc/bash_completion.d"
+    if [[ -d "$completion_dir" ]]; then
+      if [[ $(id -u) = 0 ]]; then
+        cp formae-cli-completion.bash "$completion_dir/formae"
+      else
+        sudo cp formae-cli-completion.bash "$completion_dir/formae"
+      fi
+      echo "Bash completions installed to $completion_dir"
+    else
+      mkdir -p ~/.local/share/bash-completion/completions
+      cp formae-cli-completion.bash ~/.local/share/bash-completion/completions/formae
+      echo "Bash completions installed to ~/.local/share/bash-completion/completions"
+    fi
+  else
+    # Linux
+    mkdir -p ~/.local/share/bash-completion/completions
+    cp formae-cli-completion.bash ~/.local/share/bash-completion/completions/formae
+    echo "Bash completions installed to ~/.local/share/bash-completion/completions"
+  fi
+  
+  rm -f formae-cli-completion.bash
+  echo "Bash completions setup complete!"
+}
+
+setup_zsh_completions() {
+  echo "Setting up zsh completions..."
+  
+  # Create zsh completion directory if it doesn't exist
+  zsh_completion_dir="${HOME}/.zsh/completions"
+  mkdir -p "$zsh_completion_dir"
+  
+  ${INSTALLPREFIX}/formae/bin/formae completion zsh > "${zsh_completion_dir}/_formae" 2>/dev/null || {
+    echo "Warning: Could not generate zsh completions. Make sure formae is in your PATH."
+    return
+  }
+  
+  if ! echo "$FPATH" | grep -q "$zsh_completion_dir"; then
+    echo "Add the following to your ~/.zshrc:"
+    echo "  fpath=(${zsh_completion_dir} \$fpath)"
+    echo "  autoload -U compinit && compinit"
+  fi
+  
+  echo "Zsh completions installed to ${zsh_completion_dir}/_formae"
+  echo "Zsh completions setup complete!"
+}
+
+echo "Shell completions:"
+if [[ "$OS" == "darwin" ]]; then
+  echo "For zsh (default on macOS):"
+  echo "  Add completions with: formae completion zsh > ~/.zsh/completions/_formae"
+  echo "  Then add to ~/.zshrc: fpath=(~/.zsh/completions \$fpath) && autoload -U compinit && compinit"
+  echo ""
+  echo "For bash (requires bash-completion via Homebrew):"
+  echo "  First install: brew install bash-completion"
+  echo "  Then add completions with: formae completion bash > /usr/local/etc/bash_completion.d/formae"
+else
+  echo "For bash:"
+  echo "  formae completion bash > ~/.local/share/bash-completion/completions/formae"
+  echo "  mkdir -p ~/.local/share/bash-completion/completions"
+  echo ""
+  echo "For zsh:"
+  echo "  mkdir -p ~/.zsh/completions"
+  echo "  formae completion zsh > ~/.zsh/completions/_formae"
+  echo "  Add to ~/.zshrc: fpath=(~/.zsh/completions \$fpath) && autoload -U compinit && compinit"
+fi
+
+echo ""
+
+if ! "$skip_prompt"; then
+  echo "Would you like to install shell completions for $CURRENT_SHELL? (y/N)"
+  read -r install_completions
+  if [[ "$install_completions" =~ ^[Yy]$ ]]; then
+    case "$CURRENT_SHELL" in
+      bash)
+        setup_bash_completions
+        ;;
+      zsh)
+        setup_zsh_completions
+        ;;
+      *)
+        echo "Automatic completion setup not supported for $CURRENT_SHELL"
+        echo "Please follow the manual instructions above."
+        ;;
+    esac
+  fi
+fi