bin/completion.sh
Rama main · réplica de lectura del repositorio original.
#!/bin/bash
# User state and installed tools must never run with inherited root privileges.
# Individual maintenance operations request administrator access themselves.
if [[ "$EUID" -eq 0 ]]; then
printf '%s\n' 'Run Mole without sudo; it requests administrator access when needed.' >&2
exit 1
fi
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
ROOT_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
source "$ROOT_DIR/lib/core/common.sh"
source "$ROOT_DIR/lib/core/commands.sh"
command_names=()
for entry in "${MOLE_COMMANDS[@]}"; do
command_names+=("${entry%%:*}")
done
command_words="${command_names[*]}"
clean_option_words="--dry-run -n --external --whitelist --debug --help -h"
analyze_option_words="--json --help -h"
history_option_words="--json --limit --help -h"
purge_option_words="--paths --dry-run -n --yes --include-empty --debug --help -h"
emit_zsh_subcommands() {
for entry in "${MOLE_COMMANDS[@]}"; do
printf " '%s:%s'\n" "${entry%%:*}" "${entry#*:}"
done
}
emit_fish_completions() {
local cmd="$1"
for entry in "${MOLE_COMMANDS[@]}"; do
local name="${entry%%:*}"
local desc="${entry#*:}"
printf 'complete -f -c %s -n "__fish_mole_no_subcommand" -a %s -d "%s"\n' "$cmd" "$name" "$desc"
done
printf '\n'
printf 'complete -f -c %s -n "__fish_seen_subcommand_from clean" -l dry-run -s n -d "Preview cleanup without making changes"\n' "$cmd"
printf 'complete -c %s -n "__fish_seen_subcommand_from clean" -l external -r -a "(__fish_complete_directories)" -d "Clean OS metadata from an external volume"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from clean" -l whitelist -d "Manage protected paths"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from clean" -l debug -d "Show detailed logs"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from clean" -l help -s h -d "Show help"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from analyze analyse" -l json -d "Output analysis as JSON"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from analyze analyse" -l help -s h -d "Show help"\n' "$cmd"
printf 'complete -c %s -n "__fish_seen_subcommand_from analyze analyse; and not __fish_seen_argument -l json -l help -s h" -a "(__fish_complete_directories)" -d "Path to analyze"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from history" -l json -d "Output history as JSON"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from history" -l limit -r -d "Limit recent entries"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from history" -l help -s h -d "Show help"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from purge" -l paths -d "Edit custom scan directories"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from purge" -l dry-run -s n -d "Preview purge actions without making changes"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from purge" -l include-empty -d "Show zero-size project artifact directories"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from purge" -l yes -d "Confirm unattended cleanup of eligible artifacts"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from purge" -l debug -d "Show detailed logs"\n' "$cmd"
printf 'complete -f -c %s -n "__fish_seen_subcommand_from purge" -l help -s h -d "Show help"\n' "$cmd"
printf '\n'
printf 'complete -f -c %s -n "not __fish_mole_no_subcommand" -a bash -d "generate bash completion" -n "__fish_see_subcommand_path completion"\n' "$cmd"
printf 'complete -f -c %s -n "not __fish_mole_no_subcommand" -a zsh -d "generate zsh completion" -n "__fish_see_subcommand_path completion"\n' "$cmd"
printf 'complete -f -c %s -n "not __fish_mole_no_subcommand" -a fish -d "generate fish completion" -n "__fish_see_subcommand_path completion"\n' "$cmd"
}
# A dotfile chain is followed to its final file; a broken link or a cycle
# leaves the config untouched and says what to fix.
_completion_rc_target() {
mole_resolve_link_chain "$1" && return 0
log_error "Could not follow the shell config link $1, so it was left unchanged. Fix or remove that link, then rerun mo completion"
return 1
}
remove_stale_completion_entries() {
local config_file="$1"
local success_message="$2"
if [[ ! -f "$config_file" ]] || ! grep -Eq "$MOLE_COMPLETION_LINE_PATTERN" "$config_file" 2> /dev/null; then
return 1
fi
local original_mode="" temp_file target
target="$(_completion_rc_target "$config_file")" || return 1
original_mode="$("$STAT_BSD" -f '%Mp%Lp' "$target" 2> /dev/null || true)"
temp_file="$(mktemp)"
grep -Ev "$MOLE_COMPLETION_LINE_PATTERN" "$target" > "$temp_file" || true
mv "$temp_file" "$target"
[[ -n "$original_mode" ]] && chmod "$original_mode" "$target" 2> /dev/null || true
[[ -n "$success_message" ]] && echo -e "${GREEN}${ICON_SUCCESS}${NC} $success_message"
return 0
}
if [[ $# -gt 0 ]]; then
normalized_args=()
for arg in "$@"; do
case "$arg" in
"--dry-run" | "-n")
export MOLE_DRY_RUN=1
;;
*)
normalized_args+=("$arg")
;;
esac
done
if [[ ${#normalized_args[@]} -gt 0 ]]; then
set -- "${normalized_args[@]}"
else
set --
fi
fi
# Auto-install mode when run without arguments
if [[ $# -eq 0 ]]; then
if [[ "${MOLE_DRY_RUN:-0}" == "1" ]]; then
echo -e "${YELLOW}${ICON_DRY_RUN} DRY RUN MODE${NC}, shell config files will not be modified"
echo ""
fi
# Detect current shell
current_shell="${SHELL##*/}"
if [[ -z "$current_shell" ]]; then
current_shell="$(ps -p "$PPID" -o comm= 2> /dev/null | awk '{print $1}')"
fi
completion_name=""
if command -v mole > /dev/null 2>&1; then
completion_name="mole"
elif command -v mo > /dev/null 2>&1; then
completion_name="mo"
fi
# Fish uses a separate install path: write to ~/.config/fish/completions/ so
# both `mole` and `mo` load completions independently on terminal startup.
if [[ "$current_shell" == "fish" ]]; then
fish_dir="${HOME}/.config/fish/completions"
mole_file="${fish_dir}/mole.fish"
mo_file="${fish_dir}/mo.fish"
config_fish="${HOME}/.config/fish/config.fish"
if [[ -z "$completion_name" ]]; then
# Clean up any stale config.fish entries even when mole is not in PATH
if [[ "${MOLE_DRY_RUN:-0}" != "1" ]]; then
remove_stale_completion_entries "$config_fish" "Removed stale completion entries from config.fish" || true
fi
log_error "mole not found in PATH, install Mole before enabling completion"
exit 1
fi
if [[ "${MOLE_DRY_RUN:-0}" == "1" ]]; then
echo -e "${GRAY}${ICON_REVIEW} [DRY RUN] Would write Fish completions to:${NC}"
echo " $mole_file"
echo " $mo_file"
echo ""
echo -e "${GREEN}${ICON_SUCCESS}${NC} Dry run complete, no changes made"
exit 0
fi
# Remove stale config.fish source-based entries (previous install method)
if remove_stale_completion_entries "$config_fish" "Removed stale source-based entries from config.fish"; then
echo ""
fi
# Prompt only on first install; silently update if files exist
if [[ ! -f "$mole_file" ]]; then
echo ""
echo -e "${GRAY}Will write Fish completions to:${NC}"
echo " $mole_file"
echo " $mo_file"
echo ""
echo -ne "${PURPLE}${ICON_ARROW}${NC} Enable completion for ${GREEN}fish${NC}? ${GRAY}Enter confirm / Q cancel${NC}: "
if ! IFS= read -r -s -n1 key; then
echo ""
echo -e "${YELLOW}Cancelled${NC}"
exit 1
fi
drain_pending_input
echo ""
case "$key" in
$'\e' | [Qq] | [Nn])
echo -e "${YELLOW}Cancelled${NC}"
exit 0
;;
"" | $'\n' | $'\r' | [Yy]) ;;
*)
log_error "Invalid key"
exit 1
;;
esac
fi
mkdir -p "$fish_dir"
"$completion_name" completion fish > "$mole_file"
# mo.fish sources mole.fish so Fish loads mo completions on `mo<Tab>`
printf '# Mole completions for mo (alias) -- auto-generated, do not edit\n' > "$mo_file"
printf 'source %s\n' "$mole_file" >> "$mo_file"
if [[ -f "$mole_file" ]]; then
echo -e "${GREEN}${ICON_SUCCESS}${NC} Fish completions written to $fish_dir"
fi
echo ""
exit 0
fi
case "$current_shell" in
bash)
config_file="${HOME}/.bashrc"
[[ -f "${HOME}/.bash_profile" ]] && config_file="${HOME}/.bash_profile"
# shellcheck disable=SC2016
completion_line='if output="$('"$completion_name"' completion bash 2>/dev/null)"; then eval "$output"; fi'
;;
zsh)
config_file="${HOME}/.zshrc"
# shellcheck disable=SC2016
completion_line='if output="$('"$completion_name"' completion zsh 2>/dev/null)"; then eval "$output"; fi'
;;
*)
log_error "Unsupported shell: $current_shell"
echo " mole completion <bash|zsh|fish>"
exit 1
;;
esac
if [[ -z "$completion_name" ]]; then
if [[ -f "$config_file" ]] && grep -Eq "$MOLE_COMPLETION_LINE_PATTERN" "$config_file" 2> /dev/null; then
if [[ "${MOLE_DRY_RUN:-0}" == "1" ]]; then
echo -e "${GRAY}${ICON_REVIEW} [DRY RUN] Would remove stale completion entries from $config_file${NC}"
echo ""
else
original_mode=""
target="$(_completion_rc_target "$config_file")" || exit 1
original_mode="$("$STAT_BSD" -f '%Mp%Lp' "$target" 2> /dev/null || true)"
temp_file="$(mktemp)"
grep -Ev "$MOLE_COMPLETION_LINE_PATTERN" "$target" > "$temp_file" || true
mv "$temp_file" "$target"
if [[ -n "$original_mode" ]]; then
chmod "$original_mode" "$target" 2> /dev/null || true
fi
echo -e "${GREEN}${ICON_SUCCESS}${NC} Removed stale completion entries from $config_file"
echo ""
fi
fi
log_error "mole not found in PATH, install Mole before enabling completion"
exit 1
fi
# Check if already installed and normalize to latest line
if [[ -f "$config_file" ]] && grep -Eq "$MOLE_COMPLETION_LINE_PATTERN" "$config_file" 2> /dev/null; then
if [[ "${MOLE_DRY_RUN:-0}" == "1" ]]; then
echo -e "${GRAY}${ICON_REVIEW} [DRY RUN] Would normalize completion entry in $config_file${NC}"
echo ""
exit 0
fi
original_mode=""
target="$(_completion_rc_target "$config_file")" || exit 1
original_mode="$("$STAT_BSD" -f '%Mp%Lp' "$target" 2> /dev/null || true)"
temp_file="$(mktemp)"
grep -Ev "$MOLE_COMPLETION_LINE_PATTERN" "$target" > "$temp_file" || true
mv "$temp_file" "$target"
if [[ -n "$original_mode" ]]; then
chmod "$original_mode" "$target" 2> /dev/null || true
fi
{
echo ""
echo "# Mole shell completion"
echo "$completion_line"
} >> "$config_file"
echo ""
echo -e "${GREEN}${ICON_SUCCESS}${NC} Shell completion updated in $config_file"
echo ""
exit 0
fi
# Prompt user for installation
echo ""
echo -e "${GRAY}Will add to ${config_file}:${NC}"
echo " $completion_line"
echo ""
if [[ "${MOLE_DRY_RUN:-0}" == "1" ]]; then
echo -e "${GREEN}${ICON_SUCCESS}${NC} Dry run complete, no changes made"
exit 0
fi
echo -ne "${PURPLE}${ICON_ARROW}${NC} Enable completion for ${GREEN}${current_shell}${NC}? ${GRAY}Enter confirm / Q cancel${NC}: "
if ! IFS= read -r -s -n1 key; then
echo ""
echo -e "${YELLOW}Cancelled${NC}"
exit 1
fi
drain_pending_input
echo ""
case "$key" in
$'\e' | [Qq] | [Nn])
echo -e "${YELLOW}Cancelled${NC}"
exit 0
;;
"" | $'\n' | $'\r' | [Yy]) ;;
*)
log_error "Invalid key"
exit 1
;;
esac
# Create config file if it doesn't exist
if [[ ! -f "$config_file" ]]; then
mkdir -p "$(dirname "$config_file")"
touch "$config_file"
fi
# Remove previous Mole completion lines to avoid duplicates
if [[ -f "$config_file" ]]; then
original_mode=""
target="$(_completion_rc_target "$config_file")" || exit 1
original_mode="$("$STAT_BSD" -f '%Mp%Lp' "$target" 2> /dev/null || true)"
temp_file="$(mktemp)"
grep -Ev "$MOLE_COMPLETION_LINE_PATTERN" "$target" > "$temp_file" || true
mv "$temp_file" "$target"
if [[ -n "$original_mode" ]]; then
chmod "$original_mode" "$target" 2> /dev/null || true
fi
fi
# Add completion line
{
echo ""
echo "# Mole shell completion"
echo "$completion_line"
} >> "$config_file"
echo -e "${GREEN}${ICON_SUCCESS}${NC} Completion added to $config_file"
echo ""
echo ""
echo -e "${GRAY}To activate now:${NC}"
echo -e " ${GREEN}source $config_file${NC}"
exit 0
fi
case "$1" in
bash)
cat << EOF
_mole_completions()
{
local cur_word prev_word subcommand
cur_word="\${COMP_WORDS[\$COMP_CWORD]}"
prev_word="\${COMP_WORDS[\$COMP_CWORD-1]}"
subcommand="\${COMP_WORDS[1]}"
if [ "\$COMP_CWORD" -eq 1 ]; then
COMPREPLY=( \$(compgen -W "$command_words" -- "\$cur_word") )
else
case "\$subcommand" in
clean)
case "\$prev_word" in
--external)
COMPREPLY=( \$(compgen -d -- "\$cur_word") )
;;
*)
COMPREPLY=( \$(compgen -W "$clean_option_words" -- "\$cur_word") )
;;
esac
;;
analyze|analyse)
if [[ "\$cur_word" == -* ]]; then
COMPREPLY=( \$(compgen -W "$analyze_option_words" -- "\$cur_word") )
else
COMPREPLY=( \$(compgen -f -- "\$cur_word") )
fi
;;
history)
COMPREPLY=( \$(compgen -W "$history_option_words" -- "\$cur_word") )
;;
purge)
COMPREPLY=( \$(compgen -W "$purge_option_words" -- "\$cur_word") )
;;
completion)
COMPREPLY=( \$(compgen -W "bash zsh fish" -- "\$cur_word") )
;;
*)
COMPREPLY=()
;;
esac
fi
}
complete -F _mole_completions mole mo
EOF
;;
zsh)
printf '#compdef mole mo\n\n'
printf '_mole() {\n'
printf ' local -a subcommands\n'
printf ' subcommands=(\n'
emit_zsh_subcommands
printf ' )\n'
printf ' if (( CURRENT == 2 )); then\n'
printf " _describe 'subcommand' subcommands\n"
printf ' return\n'
printf ' fi\n'
printf " case \"\$words[2]\" in\n"
printf ' clean)\n'
printf ' _arguments \\\n'
printf " '--dry-run[Preview cleanup without making changes]' \\\\\n"
printf " '-n[Preview cleanup without making changes]' \\\\\n"
printf " '--external[Clean OS metadata from an external volume]:path:_files -/' \\\\\n"
printf " '--whitelist[Manage protected paths]' \\\\\n"
printf " '--debug[Show detailed logs]' \\\\\n"
printf " '(-h --help)'{-h,--help}'[Show help]'\n"
printf ' ;;\n'
printf ' analyze|analyse)\n'
printf ' _arguments \\\n'
printf " '--json[Output analysis as JSON]' \\\\\n"
printf " '(-h --help)'{-h,--help}'[Show help]' \\\\\n"
printf " '*:path:_files'\n"
printf ' ;;\n'
printf ' history)\n'
printf ' _arguments \\\n'
printf " '--json[Output history as JSON]' \\\\\n"
printf " '--limit[Limit recent entries]:limit:' \\\\\n"
printf " '(-h --help)'{-h,--help}'[Show help]'\n"
printf ' ;;\n'
printf ' purge)\n'
printf ' _arguments \\\n'
printf " '--paths[Edit custom scan directories]' \\\\\n"
printf " '--dry-run[Preview purge actions without making changes]' \\\\\n"
printf " '-n[Preview purge actions without making changes]' \\\\\n"
printf " '--include-empty[Show zero-size project artifact directories]' \\\\\n"
printf " '--yes[Confirm unattended cleanup of eligible artifacts]' \\\\\n"
printf " '--debug[Show detailed logs]' \\\\\n"
printf " '(-h --help)'{-h,--help}'[Show help]'\n"
printf ' ;;\n'
printf ' completion)\n'
printf " _arguments '1:shell:(bash zsh fish)'\n"
printf ' ;;\n'
printf ' *)\n'
printf " _describe 'subcommand' subcommands\n"
printf ' ;;\n'
printf ' esac\n'
printf '}\n\n'
printf 'compdef _mole mole mo\n'
;;
fish)
printf '# Completions for mole\n'
emit_fish_completions mole
printf '\n# Completions for mo (alias)\n'
emit_fish_completions mo
printf '\nfunction __fish_mole_no_subcommand\n'
printf ' for i in (commandline -opc)\n'
# shellcheck disable=SC2016
printf ' if contains -- $i %s\n' "$command_words"
printf ' return 1\n'
printf ' end\n'
printf ' end\n'
printf ' return 0\n'
printf 'end\n\n'
printf 'function __fish_see_subcommand_path\n'
printf ' string match -q -- "completion" (commandline -opc)[1]\n'
printf 'end\n'
;;
*)
cat << 'EOF'
Usage: mole completion [bash|zsh|fish]
Setup shell tab completion for mole and mo commands.
Auto-install:
mole completion # Auto-detect shell and install
mole completion --dry-run # Preview config changes without writing files
Manual install:
mole completion bash # Generate bash completion script
mole completion zsh # Generate zsh completion script
mole completion fish # Generate fish completion script
Examples:
# Auto-install (recommended)
mole completion
# Manual install - Bash
eval "$(mole completion bash)"
# Manual install - Zsh
eval "$(mole completion zsh)"
# Manual install - Fish
mole completion fish | source
EOF
[[ "$1" == "--help" || "$1" == "-h" ]] && exit 0
exit 1
;;
esac