mirror of
https://github.com/tw93/Mole.git
synced 2026-02-04 19:09:43 +00:00
334 lines
10 KiB
Bash
Executable File
334 lines
10 KiB
Bash
Executable File
#!/bin/bash
|
|
# Mole - Main Entry Point
|
|
# A comprehensive macOS maintenance tool
|
|
#
|
|
# 🧹 Clean - Remove junk files and optimize system
|
|
# 🗑️ Uninstall - Remove applications completely
|
|
# 📊 Analyze - Interactive disk space explorer
|
|
#
|
|
# Usage:
|
|
# ./mole # Interactive main menu
|
|
# ./mole clean # Direct clean mode
|
|
# ./mole uninstall # Direct uninstall mode
|
|
# ./mole analyze # Disk space explorer
|
|
# ./mole --help # Show help
|
|
|
|
set -euo pipefail
|
|
|
|
# Get script directory
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
# Source common functions
|
|
source "$SCRIPT_DIR/lib/common.sh"
|
|
|
|
# Version info
|
|
VERSION="1.4.0"
|
|
MOLE_TAGLINE="can dig deep to clean your Mac."
|
|
|
|
# Check for updates (non-blocking, cached)
|
|
check_for_updates() {
|
|
local cache="$HOME/.cache/mole/version_check"
|
|
local msg_cache="$HOME/.cache/mole/update_message"
|
|
mkdir -p "$(dirname "$cache")" 2>/dev/null
|
|
|
|
# Skip if checked within 24 hours
|
|
if [[ -f "$cache" ]]; then
|
|
local age=$(($(date +%s) - $(stat -f%m "$cache" 2>/dev/null || echo 0)))
|
|
[[ $age -lt 86400 ]] && return
|
|
fi
|
|
|
|
# Background version check (save to file, don't output)
|
|
(
|
|
local latest=$(curl -fsSL --connect-timeout 2 --max-time 3 -H "Cache-Control: no-cache" \
|
|
"https://raw.githubusercontent.com/tw93/mole/main/mole" 2>/dev/null | \
|
|
grep '^VERSION=' | head -1 | sed 's/VERSION="\(.*\)"/\1/')
|
|
|
|
if [[ -n "$latest" && "$VERSION" != "$latest" && "$(printf '%s\n' "$VERSION" "$latest" | sort -V | head -1)" == "$VERSION" ]]; then
|
|
echo -e "${YELLOW}📢 New version ${GREEN}${latest}${YELLOW} available (current: ${VERSION})\n Run ${GREEN}mole update${YELLOW} to upgrade${NC}" > "$msg_cache"
|
|
else
|
|
echo -n > "$msg_cache"
|
|
fi
|
|
touch "$cache" 2>/dev/null
|
|
) &
|
|
disown 2>/dev/null || true
|
|
}
|
|
|
|
# Show update notification if available
|
|
show_update_notification() {
|
|
local msg_cache="$HOME/.cache/mole/update_message"
|
|
if [[ -f "$msg_cache" && -s "$msg_cache" ]]; then
|
|
cat "$msg_cache"
|
|
echo
|
|
fi
|
|
}
|
|
|
|
show_brand_banner() {
|
|
cat << EOF
|
|
${GREEN} __ __ _ ${NC}
|
|
${GREEN}| \/ | ___ | | ___ ${NC}
|
|
${GREEN}| |\/| |/ _ \| |/ _ \\${NC}
|
|
${GREEN}| | | | (_) | | __/${NC} ${BLUE}https://github.com/tw93/mole${NC}
|
|
${GREEN}|_| |_|\___/|_|\___|${NC} ${GREEN}${MOLE_TAGLINE}${NC}
|
|
|
|
EOF
|
|
}
|
|
|
|
animate_mole_intro() {
|
|
# Skip animation if stdout isn't a TTY (non-interactive)
|
|
if [[ ! -t 1 ]]; then
|
|
return
|
|
fi
|
|
|
|
clear_screen
|
|
printf '\n'
|
|
hide_cursor
|
|
|
|
local -a mole_lines=()
|
|
while IFS= read -r line; do
|
|
mole_lines+=("$line")
|
|
done <<'EOF'
|
|
/\_/\
|
|
____/ o o \
|
|
/~____ =o= /
|
|
(______)__m_m)
|
|
/ \
|
|
__/ /\ \__
|
|
/__/ \__\_
|
|
EOF
|
|
|
|
local idx
|
|
local body_cutoff=4
|
|
local body_color="${PURPLE}"
|
|
local ground_color="${GREEN}"
|
|
for idx in "${!mole_lines[@]}"; do
|
|
if (( idx < body_cutoff )); then
|
|
printf "%s\n" "${body_color}${mole_lines[$idx]}${NC}"
|
|
else
|
|
printf "%s\n" "${ground_color}${mole_lines[$idx]}${NC}"
|
|
fi
|
|
sleep 0.1
|
|
done
|
|
|
|
printf '\n'
|
|
sleep 0.5
|
|
|
|
printf '\033[2J\033[H'
|
|
show_cursor
|
|
}
|
|
|
|
show_version() {
|
|
printf 'Mole version %s\n' "$VERSION"
|
|
}
|
|
|
|
show_help() {
|
|
show_brand_banner
|
|
echo
|
|
printf "%s%s%s\n" "$BLUE" "USAGE" "$NC"
|
|
printf " %s%s%s [command]\n\n" "$GREEN" "mole" "$NC"
|
|
|
|
printf "%s%s%s\n" "$BLUE" "COMMANDS" "$NC"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole" "$NC" "Interactive main menu"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole clean" "$NC" "Deeper system cleanup"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole clean --dry-run" "$NC" "Preview cleanup (no deletions)"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole clean --whitelist" "$NC" "Manage protected caches"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole uninstall" "$NC" "Remove applications completely"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole analyze" "$NC" "Interactive disk space explorer"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole update" "$NC" "Update Mole to the latest version"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole --version" "$NC" "Show installed version"
|
|
printf " %s%-28s%s %s\n" "$GREEN" "mole --help" "$NC" "Show this help message"
|
|
|
|
printf "\n%s%s%s\n" "$BLUE" "MORE" "$NC"
|
|
printf " https://github.com/tw93/mole\n\n"
|
|
}
|
|
|
|
# Simple update function
|
|
update_mole() {
|
|
# Check if installed via Homebrew
|
|
if command -v brew >/dev/null 2>&1 && brew list mole >/dev/null 2>&1; then
|
|
echo -e "${GREEN}Homebrew detected, use: ${BLUE}brew upgrade mole${NC}"
|
|
exit 0
|
|
fi
|
|
|
|
log_info "Updating Mole..."
|
|
|
|
local installer_url="https://raw.githubusercontent.com/tw93/mole/main/install.sh"
|
|
local tmp_installer
|
|
tmp_installer="$(mktemp)" || { log_error "Failed to create temp file"; exit 1; }
|
|
|
|
# Download installer with timeout
|
|
if command -v curl >/dev/null 2>&1; then
|
|
if ! curl -fsSL --connect-timeout 10 --max-time 60 "$installer_url" -o "$tmp_installer"; then
|
|
rm -f "$tmp_installer"
|
|
log_error "Failed to download installer"
|
|
exit 1
|
|
fi
|
|
elif command -v wget >/dev/null 2>&1; then
|
|
if ! wget --timeout=10 --tries=3 -qO "$tmp_installer" "$installer_url"; then
|
|
rm -f "$tmp_installer"
|
|
log_error "Failed to download installer"
|
|
exit 1
|
|
fi
|
|
else
|
|
rm -f "$tmp_installer"
|
|
log_error "curl or wget required"
|
|
exit 1
|
|
fi
|
|
|
|
chmod +x "$tmp_installer"
|
|
|
|
# Determine install directory
|
|
local mole_path
|
|
mole_path="$(command -v mole 2>/dev/null || echo "$0")"
|
|
local install_dir
|
|
install_dir="$(cd "$(dirname "$mole_path")" && pwd)"
|
|
|
|
# Run installer quietly
|
|
if "$tmp_installer" --prefix "$install_dir" --config "$HOME/.config/mole" --update 2>/dev/null; then
|
|
log_success "Updated successfully"
|
|
else
|
|
if "$tmp_installer" --prefix "$install_dir" --config "$HOME/.config/mole" 2>/dev/null; then
|
|
log_success "Updated successfully"
|
|
else
|
|
rm -f "$tmp_installer"
|
|
log_error "Update failed"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
rm -f "$tmp_installer"
|
|
|
|
# Clear version check cache after successful update
|
|
rm -f "$HOME/.cache/mole/version_check" "$HOME/.cache/mole/update_message"
|
|
}
|
|
|
|
# Display main menu options
|
|
show_main_menu() {
|
|
local selected="${1:-1}"
|
|
local full_draw="${2:-true}"
|
|
|
|
if [[ "$full_draw" == "true" ]]; then
|
|
clear_screen
|
|
echo ""
|
|
show_brand_banner
|
|
show_update_notification
|
|
echo ""
|
|
printf '\033[s'
|
|
else
|
|
printf '\033[u\033[0J'
|
|
fi
|
|
|
|
show_menu_option 1 "Clean Mac - Remove junk files and optimize" "$([[ $selected -eq 1 ]] && echo true || echo false)"
|
|
show_menu_option 2 "Uninstall Apps - Remove applications completely" "$([[ $selected -eq 2 ]] && echo true || echo false)"
|
|
show_menu_option 3 "Analyze Disk - Interactive space explorer" "$([[ $selected -eq 3 ]] && echo true || echo false)"
|
|
show_menu_option 4 "Help & Information - Usage guide and tips" "$([[ $selected -eq 4 ]] && echo true || echo false)"
|
|
show_menu_option 5 "Exit - Close Mole" "$([[ $selected -eq 5 ]] && echo true || echo false)"
|
|
|
|
if [[ -t 0 ]]; then
|
|
echo ""
|
|
echo -e " ${GRAY}↑/↓${NC} Navigate ${GRAY}|${NC} ${GRAY}Enter${NC} Select ${GRAY}|${NC} ${GRAY}Q/ESC${NC} Quit"
|
|
fi
|
|
}
|
|
|
|
# Interactive main menu loop
|
|
interactive_main_menu() {
|
|
# Show intro animation only once per terminal tab
|
|
if [[ -t 1 ]]; then
|
|
local tty_name=$(tty 2>/dev/null || echo "")
|
|
if [[ -n "$tty_name" ]]; then
|
|
local flag_file="/tmp/mole_intro_$(echo "$tty_name" | tr -c '[:alnum:]_' '_')"
|
|
if [[ ! -f "$flag_file" ]]; then
|
|
animate_mole_intro
|
|
touch "$flag_file" 2>/dev/null || true
|
|
fi
|
|
fi
|
|
fi
|
|
local current_option=1
|
|
local first_draw=true
|
|
cleanup_and_exit() {
|
|
show_cursor
|
|
echo ""
|
|
echo "Thank you for using Mole!"
|
|
exit 0
|
|
}
|
|
|
|
trap cleanup_and_exit INT
|
|
hide_cursor
|
|
|
|
while true; do
|
|
show_main_menu $current_option "$first_draw"
|
|
if [[ "$first_draw" == "true" ]]; then
|
|
first_draw=false
|
|
fi
|
|
|
|
local key=$(read_key)
|
|
[[ $? -ne 0 ]] && continue
|
|
|
|
case "$key" in
|
|
"UP") ((current_option > 1)) && ((current_option--)) ;;
|
|
"DOWN") ((current_option < 5)) && ((current_option++)) ;;
|
|
"ENTER"|"$current_option")
|
|
show_cursor
|
|
case $current_option in
|
|
1)
|
|
exec "$SCRIPT_DIR/bin/clean.sh"
|
|
;;
|
|
2) exec "$SCRIPT_DIR/bin/uninstall.sh" ;;
|
|
3) exec "$SCRIPT_DIR/bin/analyze.sh" ;;
|
|
4) clear; show_help; exit 0 ;;
|
|
5) cleanup_and_exit ;;
|
|
esac
|
|
;;
|
|
"QUIT") cleanup_and_exit ;;
|
|
[1-5])
|
|
show_cursor
|
|
case $key in
|
|
1)
|
|
exec "$SCRIPT_DIR/bin/clean.sh"
|
|
;;
|
|
2) exec "$SCRIPT_DIR/bin/uninstall.sh" ;;
|
|
3) exec "$SCRIPT_DIR/bin/analyze.sh" ;;
|
|
4) clear; show_help; exit 0 ;;
|
|
5) cleanup_and_exit ;;
|
|
esac
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
main() {
|
|
case "${1:-""}" in
|
|
"clean")
|
|
exec "$SCRIPT_DIR/bin/clean.sh" "${@:2}"
|
|
;;
|
|
"uninstall")
|
|
exec "$SCRIPT_DIR/bin/uninstall.sh"
|
|
;;
|
|
"analyze")
|
|
exec "$SCRIPT_DIR/bin/analyze.sh" "${@:2}"
|
|
;;
|
|
"update")
|
|
update_mole
|
|
exit 0
|
|
;;
|
|
"help"|"--help"|"-h")
|
|
show_help
|
|
exit 0
|
|
;;
|
|
"version"|"--version"|"-V")
|
|
show_version
|
|
exit 0
|
|
;;
|
|
"")
|
|
check_for_updates
|
|
interactive_main_menu
|
|
;;
|
|
*)
|
|
echo "Unknown command: $1"
|
|
echo "Use 'mole --help' for usage information."
|
|
exit 1
|
|
;;
|
|
esac
|
|
}
|
|
|
|
main "$@"
|