setup.sh

#!/usr/bin/env bash
# ==============================================================================
#  Fedora LinuxScripts Setup & Configuration Manager
# ==============================================================================
#  Interactive menu for installing/symlinking dotfiles and automation scripts.
# ==============================================================================

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
LIB_DIR="${SCRIPT_DIR}/scripts/lib"
BACKUP_DIR="${SCRIPT_DIR}/backups/$(date +%Y%m%d_%H%M%S)"

# Source shared modular libraries
if [[ -f "${LIB_DIR}/colors.sh" ]]; then
    # shellcheck source=scripts/lib/colors.sh
    . "${LIB_DIR}/colors.sh"
fi

if [[ -f "${LIB_DIR}/logger.sh" ]]; then
    # shellcheck source=scripts/lib/logger.sh
    . "${LIB_DIR}/logger.sh"
fi

if [[ -f "${LIB_DIR}/backup.sh" ]]; then
    # shellcheck source=scripts/lib/backup.sh
    . "${LIB_DIR}/backup.sh"
fi

print_header() {
    clear 2>/dev/null || true
    echo -e "${COLOR_CYAN}${COLOR_BOLD}"
    echo "============================================================"
    echo "       Fedora LinuxScripts & Dotfiles Setup Manager         "
    echo "============================================================"
    echo -e "${COLOR_RESET}"
}

install_config_item() {
    local name="$1"
    local src="$2"
    local dest="$3"

    mkdir -p "$(dirname "$dest")"
    backup_path "$dest" "${BACKUP_DIR}"
    rm -rf "$dest"
    ln -sf "$src" "$dest"
    echo -e "  ${COLOR_GREEN}✔ Installed ${name}:${COLOR_RESET} $dest -> $src"
}

install_script_item() {
    local name="$1"
    local src="$2"
    local dest_bin="${HOME}/.local/bin/${name}"

    mkdir -p "${HOME}/.local/bin"
    chmod +x "$src"
    rm -f "$dest_bin"
    ln -sf "$src" "$dest_bin"
    echo -e "  ${COLOR_GREEN}✔ Linked executable:${COLOR_RESET} ~/.local/bin/${name} -> $src"
}

print_statusline_activation_hint() {
    echo -e "\n${COLOR_YELLOW}ℹ Claude Statusline installed, but it is not active yet.${COLOR_RESET}"
    echo -e "  Add this to ~/.claude/settings.json to enable it:"
    echo -e "  ${COLOR_CYAN}\"statusLine\": { \"type\": \"command\", \"command\": \"node ~/.claude/statusline.js\", \"padding\": 0, \"refreshInterval\": 1 }${COLOR_RESET}"
}

# ------------------------------------------------------------------------------
# Configuration Installation Menu
# ------------------------------------------------------------------------------
menu_install_configs() {
    print_header
    echo -e "${COLOR_BOLD}Select configurations to install (Symlink with auto-backup):${COLOR_RESET}\n"
    echo "  1) All Configurations (Kitty, Starship, Bash, Claude, AGY)"
    echo "  2) Kitty Terminal (~/.config/kitty/)"
    echo "  3) Starship Prompt (~/.config/starship.toml)"
    echo "  4) Bash Shell (~/.bashrc)"
    echo "  5) Claude Code Statusline & Gemini Worker Skill (~/.claude/)"
    echo "  6) Antigravity CLI Statusline (~/.gemini/antigravity-cli/statusline.js)"
    echo "  7) Back to Main Menu"
    echo ""
    read -rp "Enter choice [1-7]: " config_choice

    case "$config_choice" in
        1)
            echo -e "\n${COLOR_BOLD}Installing all configurations...${COLOR_RESET}"
            install_config_item "Kitty (Full Package)" "${SCRIPT_DIR}/config/kitty" "${HOME}/.config/kitty"
            install_config_item "Starship" "${SCRIPT_DIR}/config/starship/starship.toml" "${HOME}/.config/starship.toml"
            install_config_item "Bash" "${SCRIPT_DIR}/config/bash/.bashrc" "${HOME}/.bashrc"
            install_config_item "Claude Statusline" "${SCRIPT_DIR}/config/claude/statusline.js" "${HOME}/.claude/statusline.js"
            install_config_item "Claude Statusline (Test)" "${SCRIPT_DIR}/config/claude/statusline.test.js" "${HOME}/.claude/statusline.test.js"
            install_config_item "Claude Gemini Worker Skill" "${SCRIPT_DIR}/config/claude/skills/gemini-worker" "${HOME}/.claude/skills/gemini-worker"
            install_config_item "Antigravity Statusline" "${SCRIPT_DIR}/config/agy/statusline.js" "${HOME}/.gemini/antigravity-cli/statusline.js"
            install_config_item "Antigravity Statusline (Test)" "${SCRIPT_DIR}/config/agy/statusline.test.js" "${HOME}/.gemini/antigravity-cli/statusline.test.js"
            print_statusline_activation_hint
            ;;
        2)
            echo -e "\n${COLOR_BOLD}Installing Kitty configuration...${COLOR_RESET}"
            install_config_item "Kitty (Full Package)" "${SCRIPT_DIR}/config/kitty" "${HOME}/.config/kitty"
            ;;
        3)
            echo -e "\n${COLOR_BOLD}Installing Starship configuration...${COLOR_RESET}"
            install_config_item "Starship" "${SCRIPT_DIR}/config/starship/starship.toml" "${HOME}/.config/starship.toml"
            ;;
        4)
            echo -e "\n${COLOR_BOLD}Installing Bash configuration...${COLOR_RESET}"
            install_config_item "Bash" "${SCRIPT_DIR}/config/bash/.bashrc" "${HOME}/.bashrc"
            ;;
        5)
            echo -e "\n${COLOR_BOLD}Installing Claude Code Statusline & Gemini Worker Skill...${COLOR_RESET}"
            install_config_item "Claude Statusline" "${SCRIPT_DIR}/config/claude/statusline.js" "${HOME}/.claude/statusline.js"
            install_config_item "Claude Statusline (Test)" "${SCRIPT_DIR}/config/claude/statusline.test.js" "${HOME}/.claude/statusline.test.js"
            install_config_item "Claude Gemini Worker Skill" "${SCRIPT_DIR}/config/claude/skills/gemini-worker" "${HOME}/.claude/skills/gemini-worker"
            print_statusline_activation_hint
            ;;
        6)
            echo -e "\n${COLOR_BOLD}Installing Antigravity CLI Statusline...${COLOR_RESET}"
            install_config_item "Antigravity Statusline" "${SCRIPT_DIR}/config/agy/statusline.js" "${HOME}/.gemini/antigravity-cli/statusline.js"
            install_config_item "Antigravity Statusline (Test)" "${SCRIPT_DIR}/config/agy/statusline.test.js" "${HOME}/.gemini/antigravity-cli/statusline.test.js"
            ;;
        7)
            return 0
            ;;
        *)
            echo -e "\n${COLOR_RED}Invalid option selected.${COLOR_RESET}"
            ;;
    esac

    echo ""
    read -rp "Press [Enter] to continue..."
}

# ------------------------------------------------------------------------------
# Script Installation Menu
# ------------------------------------------------------------------------------
menu_install_scripts() {
    print_header
    echo -e "${COLOR_BOLD}Select automation scripts to link to ~/.local/bin:${COLOR_RESET}\n"
    echo "  1) All Scripts (update, check-updates, ezin)"
    echo "  2) System Updater (update.sh -> ~/.local/bin/update)"
    echo "  3) Update Checker (check-updates.sh -> ~/.local/bin/check-updates)"
    echo "  4) Archive Installer (ezin.sh -> ~/.local/bin/ezin)"
    echo "  5) Back to Main Menu"
    echo ""
    read -rp "Enter choice [1-5]: " script_choice

    # Ensure scripts have execute permissions
    chmod +x "${SCRIPT_DIR}/scripts/"*.sh

    case "$script_choice" in
        1)
            echo -e "\n${COLOR_BOLD}Linking all scripts to ~/.local/bin...${COLOR_RESET}"
            install_script_item "update" "${SCRIPT_DIR}/scripts/update.sh"
            install_script_item "check-updates" "${SCRIPT_DIR}/scripts/check-updates.sh"
            install_script_item "ezin" "${SCRIPT_DIR}/scripts/ezin.sh"
            ;;
        2)
            echo -e "\n${COLOR_BOLD}Linking update script...${COLOR_RESET}"
            install_script_item "update" "${SCRIPT_DIR}/scripts/update.sh"
            ;;
        3)
            echo -e "\n${COLOR_BOLD}Linking check-updates script...${COLOR_RESET}"
            install_script_item "check-updates" "${SCRIPT_DIR}/scripts/check-updates.sh"
            ;;
        4)
            echo -e "\n${COLOR_BOLD}Linking ezin installer...${COLOR_RESET}"
            install_script_item "ezin" "${SCRIPT_DIR}/scripts/ezin.sh"
            ;;
        5)
            return 0
            ;;
        *)
            echo -e "\n${COLOR_RED}Invalid option selected.${COLOR_RESET}"
            ;;
    esac

    echo ""
    read -rp "Press [Enter] to continue..."
}

# ------------------------------------------------------------------------------
# Full Setup (Both Configs & Scripts)
# ------------------------------------------------------------------------------
install_both() {
    print_header
    echo -e "${COLOR_BOLD}Performing Full Setup (Configurations + Scripts)...${COLOR_RESET}\n"

    echo -e "${COLOR_BOLD}1. Installing configurations...${COLOR_RESET}"
    install_config_item "Kitty (Full Package)" "${SCRIPT_DIR}/config/kitty" "${HOME}/.config/kitty"
    install_config_item "Starship" "${SCRIPT_DIR}/config/starship/starship.toml" "${HOME}/.config/starship.toml"
    install_config_item "Bash" "${SCRIPT_DIR}/config/bash/.bashrc" "${HOME}/.bashrc"
    install_config_item "Claude Statusline" "${SCRIPT_DIR}/config/claude/statusline.js" "${HOME}/.claude/statusline.js"
    install_config_item "Claude Statusline (Test)" "${SCRIPT_DIR}/config/claude/statusline.test.js" "${HOME}/.claude/statusline.test.js"
    install_config_item "Claude Gemini Worker Skill" "${SCRIPT_DIR}/config/claude/skills/gemini-worker" "${HOME}/.claude/skills/gemini-worker"
    install_config_item "Antigravity Statusline" "${SCRIPT_DIR}/config/agy/statusline.js" "${HOME}/.gemini/antigravity-cli/statusline.js"
    install_config_item "Antigravity Statusline (Test)" "${SCRIPT_DIR}/config/agy/statusline.test.js" "${HOME}/.gemini/antigravity-cli/statusline.test.js"
    print_statusline_activation_hint

    echo -e "\n${COLOR_BOLD}2. Linking scripts to ~/.local/bin...${COLOR_RESET}"
    chmod +x "${SCRIPT_DIR}/scripts/"*.sh
    install_script_item "update" "${SCRIPT_DIR}/scripts/update.sh"
    install_script_item "check-updates" "${SCRIPT_DIR}/scripts/check-updates.sh"
    install_script_item "ezin" "${SCRIPT_DIR}/scripts/ezin.sh"

    echo -e "\n${COLOR_GREEN}${COLOR_BOLD}✔ Full setup completed successfully!${COLOR_RESET}"
    if [[ -d "${BACKUP_DIR:-}" ]]; then
        echo -e "  Backups saved to: ${COLOR_CYAN}${BACKUP_DIR}${COLOR_RESET}"
    fi

    echo ""
    read -rp "Press [Enter] to continue..."
}

# ------------------------------------------------------------------------------
# Main Interactive Menu Loop
# ------------------------------------------------------------------------------
main_menu() {
    while true; do
        print_header
        echo -e "${COLOR_BOLD}What would you like to install?${COLOR_RESET}\n"
        echo "  1) Configurations (Kitty, Starship, Bash, Claude, AGY)"
        echo "  2) Automation Scripts (Link to ~/.local/bin)"
        echo "  3) Both (Full Installation)"
        echo "  4) Exit"
        echo ""
        read -rp "Enter choice [1-4]: " main_choice

        case "$main_choice" in
            1) menu_install_configs ;;
            2) menu_install_scripts ;;
            3) install_both ;;
            4)
                echo -e "\n${COLOR_GREEN}Exiting setup. Goodbye!${COLOR_RESET}\n"
                exit 0
                ;;
            *)
                echo -e "\n${COLOR_RED}Invalid choice. Please choose 1, 2, 3, or 4.${COLOR_RESET}"
                sleep 1
                ;;
        esac
    done
}

main_menu