#!/usr/bin/env bash

clear

source "$(dirname "$0")/../colors.sh" > /dev/null 2>&1
source "$(dirname "$0")/../detect-distro.sh" > /dev/null 2>&1

print_message() {
    local color="$1"
    local message="$2"
    printf "%b:: %s%b\n" "$color" "$message" "$NC"
}

confirm() {
    while true; do
        read -p "$(printf "%b:: %s%b" "$CYAN" "$1 [y/N]: " "$NC")" answer
        case ${answer,,} in
            y | yes) return 0 ;;
            n | no | "") return 1 ;;
            *) print_message "$YELLOW" "Please answer with y/yes or n/no." ;;
        esac
    done
}

check_essential_dependencies() {
    local dependencies=("git" "wget" "curl")
    local missing=()

    for dep in "${dependencies[@]}"; do
        if ! command -v "$dep" &> /dev/null; then
            missing+=("$dep")
        fi
    done

    if [[ ${#missing[@]} -ne 0 ]]; then
        echo "Please wait, installing required dependencies..."

        case "$DISTRO" in
            "Arch") sudo pacman -S --noconfirm "${missing[@]}" > /dev/null 2>&1 ;;
            "Fedora") sudo dnf install -y "${missing[@]}" > /dev/null 2>&1 ;;
            "openSUSE") sudo zypper install -y "${missing[@]}" > /dev/null 2>&1 ;;
            *)
                exit 1
                ;;
        esac
    fi
}

check_aur_helper() {
    if command -v yay &> /dev/null; then
        AUR_HELPER="yay"
    elif command -v paru &> /dev/null; then
        AUR_HELPER="paru"
    else
        print_message "$CYAN" "No AUR helper found. Installing yay..."
        sudo pacman -S --needed --noconfirm git base-devel
        git clone https://aur.archlinux.org/yay.git
        cd yay || exit
        makepkg -si --noconfirm
        cd ..
        rm -rf yay
        AUR_HELPER="yay"
    fi
    print_message "$GREEN" "Using AUR helper: ${AUR_HELPER}"
}

check_default_shell() {
    local current_shell=$(basename "$SHELL")

    if [[ "$current_shell" != "zsh" ]]; then
        print_message "$YELLOW" "Current default shell: $current_shell"

        if confirm "Zsh is not your default shell. Do you want to change it to zsh?"; then
            print_message "$CYAN" "Changing default shell to zsh..."
            chsh -s /bin/zsh
            print_message "$GREEN" "Default shell changed to zsh. Please log out and log back in for the change to take effect."
        else
            print_message "${TEAL}" "Keeping current shell: $current_shell"
        fi
    else
        print_message "$GREEN" "Zsh is already your default shell."
    fi
}

install_zsh_dependencies() {
    print_message "$CYAN" "Installing Zsh dependencies..."
    case "$DISTRO" in
        "Arch")
            sudo pacman -S --noconfirm git zsh zsh-autosuggestions zsh-completions eza zsh-syntax-highlighting trash-cli
            ;;
        "Fedora")
            sudo dnf install -y git zsh zsh-autosuggestions zsh-syntax-highlighting unzip trash-cli

            # due to eza is no longer available on fedora 42 installing manually
            print_message "$CYAN" "Installing eza manually for Fedora..."

            if command -v eza &> /dev/null; then
                print_message "$GREEN" "eza is already installed."
            else
                local tmp_dir=$(mktemp -d)
                cd "$tmp_dir" || exit 1

                print_message "$CYAN" "Fetching latest eza release..."
                local latest_url=$(curl -s https://api.github.com/repos/eza-community/eza/releases/latest | grep -o "https://github.com/eza-community/eza/releases/download/.*/eza_x86_64-unknown-linux-gnu.zip" | head -1)

                if [ -z "$latest_url" ]; then
                    print_message "$YELLOW" "Could not determine latest version, using fallback version..."
                    latest_url="https://github.com/eza-community/eza/releases/download/v0.21.1/eza_x86_64-unknown-linux-gnu.zip"
                fi

                print_message "$CYAN" "Downloading eza from: $latest_url"
                if ! curl -L -o eza.zip "$latest_url"; then
                    print_message "$RED" "Failed to download eza. Exiting..."
                    cd "$HOME" || exit
                    rm -rf "$tmp_dir"
                    exit 1
                fi

                print_message "$CYAN" "Extracting eza..."
                unzip -q eza.zip

                print_message "$CYAN" "Installing eza to /usr/bin..."
                sudo cp eza /usr/bin/
                sudo chmod +x /usr/bin/eza

                cd "$HOME" || exit
                rm -rf "$tmp_dir"

                print_message "$GREEN" "eza installed successfully!"
            fi
            ;;
        "openSUSE")
            sudo zypper install -y zsh trash-cli eza
            ;;
    esac
}

install_powerlevel10k() {
    print_message "$CYAN" "Installing Powerlevel10k..."
    case "$DISTRO" in
        "Arch")
            $AUR_HELPER -S --noconfirm zsh-theme-powerlevel10k-git
            echo 'source /usr/share/zsh-theme-powerlevel10k/powerlevel10k.zsh-theme' >> ~/.zshrc
            ;;
        "Fedora" | "openSUSE")
            sudo git clone --depth=1 https://github.com/romkatv/powerlevel10k.git /usr/share/zsh-theme-powerlevel10k
            echo 'source /usr/share/zsh-theme-powerlevel10k/powerlevel10k.zsh-theme' >> ~/.zshrc
            ;;
    esac
}

install_ohmyzsh() {
    if [[ ! -d /usr/share/oh-my-zsh ]]; then
        print_message "$CYAN" "Cloning Oh My Zsh..."
        sudo git clone https://github.com/ohmyzsh/ohmyzsh /usr/share/oh-my-zsh
    fi
}

install_ohmyzsh_plugins() {
    PLUGIN_DIR="/usr/share/oh-my-zsh/plugins"

    print_message "$CYAN" "Installing Zsh plugins..."
    cd "$PLUGIN_DIR" || exit 1
    [[ ! -d zsh-syntax-highlighting ]] && sudo git clone https://github.com/zsh-users/zsh-syntax-highlighting.git
    [[ ! -d zsh-256color ]] && sudo git clone https://github.com/chrissicool/zsh-256color.git
    [[ ! -d zsh-autosuggestions ]] && sudo git clone https://github.com/zsh-users/zsh-autosuggestions.git
}

config_zsh() {
    local backup_dir="$HOME/.config/carch/backups"
    mkdir -p "$backup_dir"

    P10K_CONFIG="$HOME/.p10k.zsh"
    if [[ -f "$P10K_CONFIG" ]]; then
        if confirm ".p10k.zsh found. Do you want to back it up?"; then
            local backup_path="$backup_dir/.p10k.zsh.bak.$RANDOM"
            mv "$P10K_CONFIG" "$backup_path"
            print_message "$GREEN" "Backup created: $backup_path"
        fi
    fi

    print_message "$CYAN" "Applying Powerlevel10k configuration..."
    curl -fsSL "https://raw.githubusercontent.com/harilvfs/dwm/refs/heads/main/config/.p10k.zsh" -o "$P10K_CONFIG"

    ZSHRC="$HOME/.zshrc"
    if [[ -f "$ZSHRC" ]]; then
        if confirm ".zshrc already exists. Use the recommended version?"; then
            local backup_path="$backup_dir/.zshrc.bak.$RANDOM"
            mv "$ZSHRC" "$backup_path"
            print_message "$GREEN" "Backup created: $backup_path"
            curl -fsSL "https://raw.githubusercontent.com/harilvfs/dwm/refs/heads/main/config/.zshrc" -o "$ZSHRC"
            print_message "$GREEN" "Applied recommended .zshrc."
        fi
    fi

    if ! grep -q '[ -f ~/.p10k.zsh ] && source ~/.p10k.zsh' "$ZSHRC"; then
        echo '[ -f ~/.p10k.zsh ] && source ~/.p10k.zsh' >> "$ZSHRC"
    fi
}

install_pokemon_colorscripts() {
    print_message "$CYAN" "Installing Pokémon Color Scripts..."
    case "$DISTRO" in
        "Arch")
            $AUR_HELPER -S --noconfirm pokemon-colorscripts-git
            ;;
        "Fedora" | "openSUSE")
            POKEMON_DIR="$HOME/pokemon-colorscripts"

            [[ -d "$POKEMON_DIR" ]] && rm -rf "$POKEMON_DIR"
            git clone --depth=1 https://gitlab.com/phoneybadger/pokemon-colorscripts.git "$POKEMON_DIR"

            if [[ -d "$POKEMON_DIR" ]]; then
                (cd "$POKEMON_DIR" && sudo ./install.sh)
                rm -rf "$POKEMON_DIR"
            else
                print_message "$RED" "Error: Pokémon Color Scripts failed to clone!"
                exit 1
            fi
            ;;
    esac
}

install_zoxide() {
    print_message "$CYAN" "Installing zoxide..."
    case "$DISTRO" in
        "Arch") sudo pacman -S --noconfirm zoxide ;;
        "Fedora") sudo dnf install -y zoxide ;;
        "openSUSE") sudo zypper install -y zoxide ;;
    esac
}

main() {
    check_essential_dependencies
    if [[ "$DISTRO" == "Arch" ]]; then
        check_aur_helper
    fi
    install_zsh_dependencies
    install_powerlevel10k
    install_ohmyzsh
    install_ohmyzsh_plugins
    config_zsh
    install_pokemon_colorscripts
    install_zoxide
    check_default_shell
    print_message "${TEAL}" "Zsh setup completed successfully!"
}

main