install.sh

#!/usr/bin/env bash

# ============================================================================
# blxshell Installation Script
# ============================================================================
# Usage:
#   Local:  ./install.sh
#   Remote: curl -fsSL https://raw.githubusercontent.com/USER/REPO/main/install.sh | bash
#           bash <(curl -fsSL https://raw.githubusercontent.com/USER/REPO/main/install.sh)
# ============================================================================

set -eo pipefail

# Colors
readonly RED='\033[0;31m'
readonly GREEN='\033[0;32m'
readonly BLUE='\033[0;34m'
readonly YELLOW='\033[0;33m'
readonly CYAN='\033[0;36m'
readonly BOLD='\033[1m'
readonly NC='\033[0m'

# Configuration
readonly REPO_URL="https://github.com/binarylinuxx/dots.git"  
readonly REPO_BRANCH="main"
readonly BACKUP_DIR="$HOME/.cache/.config_backup"
readonly LOG_FILE="/tmp/blxshell_install_$(date +%Y%m%d_%H%M%S).log"
readonly BLXSHELL_HOME="$HOME/.local/blxshell"
readonly USER_BIN_DIR="$HOME/.local/bin"
readonly PACKAGES=("blxshell-quickshell-git" "blxshell-shell" "blxshell-audio" "blxshell-hyprland" "blxshell-font-googlesans" "blxshell-font-bitcount")
readonly OPENSUSE_PACKAGES=("blxshell-quickshell-git" "blxshell-shell" "blxshell-audio" "blxshell-hyprland" "blxshell-fonts" "gcc-c++" "python3-devel")
readonly OBS_REPO_URL="https://download.opensuse.org/repositories/home:binarylinuxx:blxshell/openSUSE_Tumbleweed/"
readonly OBS_REPO_ALIAS="home_binarylinuxx_blxshell"

# State (will be set by detect_install_mode)
SCRIPT_DIR=""
REMOTE_INSTALL=false
INSTALL_DOTFILES=false
BACKUP_DONE=false
CLEANUP_DIR=""

# ============================================================================
# Utility Functions
# ============================================================================

log() { echo "[$(date '+%H:%M:%S')] $*" >> "$LOG_FILE"; }

info()    { echo -e "${BLUE}::${NC} $1"; log "INFO: $1"; }
success() { echo -e "${GREEN}[+]${NC} $1"; log "SUCCESS: $1"; }
warning() { echo -e "${YELLOW}[!]${NC} $1"; log "WARNING: $1"; }
error()   { echo -e "${RED}[-]${NC} $1"; log "ERROR: $1"; }

die() { error "$1"; exit 1; }

confirm() {
    local prompt="$1" default="${2:-n}" response

    # If not interactive (piped), use default
    if [[ ! -t 0 ]]; then
        [[ "$default" =~ ^[Yy]$ ]]
        return $?
    fi

    [[ "$default" =~ ^[Yy] ]] && prompt="$prompt [Y/n]: " || prompt="$prompt [y/N]: "
    echo -en "${YELLOW}${prompt}${NC}"
    read -r response </dev/tty
    [[ "${response:-$default}" =~ ^[Yy]$ ]]
}

command_exists() { command -v "$1" &>/dev/null; }

# ============================================================================
# Installation Mode Detection
# ============================================================================

detect_install_mode() {
    # Method 1: Check if BASH_SOURCE points to a real file in a valid repo
    if [[ -n "${BASH_SOURCE[0]}" ]] && [[ -f "${BASH_SOURCE[0]}" ]]; then
        local script_dir
        script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd 2>/dev/null)"

        if [[ -n "$script_dir" ]] && [[ -d "$script_dir/arch-deps" ]]; then
            SCRIPT_DIR="$script_dir"
            REMOTE_INSTALL=false
            info "Detected: Running from local repository"
            return 0
        fi
    fi

    # Method 2: Check if we're in a git repo with the expected structure
    if git rev-parse --git-dir &>/dev/null; then
        local git_root
        git_root="$(git rev-parse --show-toplevel 2>/dev/null)"
        if [[ -n "$git_root" ]] && [[ -d "$git_root/arch-deps" ]]; then
            SCRIPT_DIR="$git_root"
            REMOTE_INSTALL=false
            info "Detected: Running from git repository"
            return 0
        fi
    fi

    # Method 3: Running from curl/pipe - need to clone
    REMOTE_INSTALL=true
    SCRIPT_DIR="/tmp/blxshell_install_$$"
    CLEANUP_DIR="$SCRIPT_DIR"
    info "Detected: Remote installation (will clone repository)"
}

clone_repository() {
    info "Cloning repository to $SCRIPT_DIR..."

    if ! command_exists git; then
        info "Installing git..."
        if [[ "$IS_OPENSUSE" == true ]]; then
            sudo zypper install -y git || die "Failed to install git"
        else
            sudo pacman -S --needed --noconfirm git || die "Failed to install git"
        fi
    fi

    git clone --depth 1 --branch "$REPO_BRANCH" "$REPO_URL" "$SCRIPT_DIR" 2>&1 | tee -a "$LOG_FILE" || {
        die "Failed to clone repository from $REPO_URL"
    }

    success "Repository cloned successfully"
}

# ============================================================================
# Pre-flight Checks
# ============================================================================

preflight_checks() {
    info "Running pre-flight checks..."

    # Don't run as root
    [[ $EUID -eq 0 ]] && die "Do not run this script as root!"

    # Validate OS
    source "$SCRIPT_DIR/os_release_validator.sh"
    validate_os

    # Check internet
    local ping_host="archlinux.org"
    [[ "$IS_OPENSUSE" == true ]] && ping_host="opensuse.org"
    if ! ping -c 1 -W 3 "$ping_host" &>/dev/null; then
        warning "No internet connection detected"
        confirm "Continue anyway?" || exit 1
    fi

    # For remote install, we need internet
    if [[ "$REMOTE_INSTALL" == true ]]; then
        ping -c 1 -W 3 github.com &>/dev/null || die "Internet required for remote installation!"
    fi

    # Check disk space (minimum 1GB)
    local space
    space=$(df -BG "$HOME" | awk 'NR==2 {gsub(/G/,""); print $4}')
    if [[ "$space" -lt 1 ]]; then
        warning "Low disk space: ${space}GB available"
        confirm "Continue anyway?" || exit 1
    fi

    success "Pre-flight checks passed"
}

# ============================================================================
# Yay Installation
# ============================================================================

install_yay() {
    if command_exists yay; then
        success "yay already installed"
        return 0
    fi

    info "Installing yay..."
    sudo pacman -S --needed --noconfirm git base-devel || die "Failed to install base dependencies"

    local tmp_dir="/tmp/yay_install_$$"
    trap "rm -rf '$tmp_dir'" RETURN

    git clone https://aur.archlinux.org/yay-bin.git "$tmp_dir" || die "Failed to clone yay"
    (cd "$tmp_dir" && makepkg -si --noconfirm) || die "Failed to install yay"

    command_exists yay && success "yay installed" || die "yay installation failed"
}

# ============================================================================
# openSUSE Package Installation
# ============================================================================

add_obs_repo() {
    if zypper repos 2>/dev/null | grep -q "$OBS_REPO_ALIAS"; then
        success "OBS repo already added"
        sudo zypper --gpg-auto-import-keys refresh "$OBS_REPO_ALIAS" 2>&1 | tee -a "$LOG_FILE" || true
        return 0
    fi

    info "Adding blxshell OBS repository..."
    sudo zypper addrepo --refresh --name "blxshell (home:binarylinuxx:blxshell)" \
        "$OBS_REPO_URL" "$OBS_REPO_ALIAS" 2>&1 | tee -a "$LOG_FILE" || {
        die "Failed to add OBS repository"
    }

    info "Importing OBS repository GPG key..."
    sudo zypper --gpg-auto-import-keys refresh "$OBS_REPO_ALIAS" 2>&1 | tee -a "$LOG_FILE" || {
        die "Failed to refresh OBS repository"
    }

    success "OBS repository added"
}

install_opensuse_packages() {
    echo -e "\n${BOLD}${BLUE}══════ Installing Packages (openSUSE) ══════${NC}\n"

    add_obs_repo

    info "Packages to install:"
    for pkg in "${OPENSUSE_PACKAGES[@]}"; do
        echo -e "  ${GREEN}✓${NC} $pkg"
    done

    info "Installing blxshell packages via zypper..."
    # shellcheck disable=SC2086
    if ! sudo zypper install -y --allow-vendor-change "${OPENSUSE_PACKAGES[@]}" 2>&1 | tee -a "$LOG_FILE"; then
        error "Failed to install packages"
        return 1
    fi

    success "All packages installed!"
}

# ============================================================================
# Package Installation
# ============================================================================

get_depends() {
    local pkgbuild="$1"
    bash -c "source '$pkgbuild' 2>/dev/null && echo \"\${depends[*]}\""
}

install_package() {
    local pkg_name="$1"
    local pkg_dir="$SCRIPT_DIR/arch-deps/$pkg_name"

    echo -e "\n${CYAN}━━━ $pkg_name ━━━${NC}"

    [[ -f "$pkg_dir/PKGBUILD" ]] || { error "PKGBUILD not found for $pkg_name"; return 1; }

    local depends
    depends=$(get_depends "$pkg_dir/PKGBUILD")

    if [[ -n "$depends" ]]; then
        info "Installing dependencies: $depends"
        # shellcheck disable=SC2086
        if ! yay -S --sudoloop --needed --noconfirm --asdeps --overwrite '*' $depends 2>&1 | tee -a "$LOG_FILE"; then
            error "Failed to install dependencies"
            return 1
        fi
    fi

    info "Building $pkg_name..."
    (cd "$pkg_dir" && makepkg -Afc --noconfirm) 2>&1 | tee -a "$LOG_FILE" || {
        error "Failed to build $pkg_name"
        return 1
    }

    local pkg_file
    pkg_file=$(find "$pkg_dir" -maxdepth 1 -name "*.pkg.tar.zst" -newer "$pkg_dir/PKGBUILD" | head -1)

    [[ -n "$pkg_file" ]] || { error "No package file found"; return 1; }

    info "Installing $pkg_name..."
    sudo pacman -U --noconfirm --overwrite '*' "$pkg_file" 2>&1 | tee -a "$LOG_FILE" || {
        error "Failed to install $pkg_name"
        return 1
    }

    success "$pkg_name installed"
}

install_metapackages() {
    echo -e "\n${BOLD}${BLUE}══════ Installing Metapackages ══════${NC}\n"

    install_yay
    sudo pacman -S --needed --noconfirm base-devel &>/dev/null

    info "Packages to install:"
    for pkg in "${PACKAGES[@]}"; do
        if [[ -d "$SCRIPT_DIR/arch-deps/$pkg" ]]; then
            echo -e "  ${GREEN}✓${NC} $pkg"
        else
            echo -e "  ${RED}✗${NC} $pkg ${YELLOW}(missing)${NC}"
        fi
    done

    local failed=()
    for pkg in "${PACKAGES[@]}"; do
        [[ -d "$SCRIPT_DIR/arch-deps/$pkg" ]] || { warning "Skipping $pkg"; continue; }
        install_package "$pkg" || failed+=("$pkg")
    done

    if [[ ${#failed[@]} -gt 0 ]]; then
        error "Failed packages: ${failed[*]}"
        return 1
    fi

    success "All metapackages installed!"
}

# ============================================================================
# Backup & Dotfiles
# ============================================================================

backup_config() {
    local timestamp backup_path
    timestamp=$(date +%Y%m%d_%H%M%S)
    backup_path="$BACKUP_DIR/config_$timestamp"

    info "Creating backup at $backup_path..."
    mkdir -p "$BACKUP_DIR"

    if cp -r "$HOME/.config" "$backup_path"; then
        ln -sfn "$backup_path" "$BACKUP_DIR/latest"
        success "Backup completed"
        BACKUP_DONE=true
    else
        error "Backup failed!"
        return 1
    fi
}

install_dotfiles() {
    [[ -d "$SCRIPT_DIR/.config" ]] || { error "No .config in $SCRIPT_DIR"; return 1; }

    info "Installing dotfiles..."

    if [[ "$BACKUP_DONE" == false && -d "$HOME/.config" ]]; then
        warning "No backup exists - creating one now..."
        backup_config || return 1
    fi

    rm -rf "$HOME/.config"
    cp -r "$SCRIPT_DIR/.config" "$HOME/"    
    
    success "Dotfiles installed to $HOME/.config"
}

install_blxshell_runtime() {
    local source_dir="$SCRIPT_DIR/.config/quickshell"
    local backup_path timestamp

    [[ -d "$source_dir" ]] || { error "No quickshell runtime in $source_dir"; return 1; }

    info "Installing blxshell runtime to $BLXSHELL_HOME..."
    mkdir -p "$BACKUP_DIR" "$HOME/.local"

    if [[ -e "$BLXSHELL_HOME" ]]; then
        timestamp=$(date +%Y%m%d_%H%M%S)
        backup_path="$BACKUP_DIR/blxshell_$timestamp"
        info "Backing up existing blxshell runtime to $backup_path..."
        mv "$BLXSHELL_HOME" "$backup_path" || { error "Failed to backup existing blxshell runtime"; return 1; }
    fi

    cp -r "$source_dir" "$BLXSHELL_HOME" || { error "Failed to install blxshell runtime"; return 1; }

    # These are local/generated development artifacts, not part of the runtime.
    rm -rf "$BLXSHELL_HOME/.ruff_cache" "$BLXSHELL_HOME/.obsidian" "$BLXSHELL_HOME/col_gen/.venv"
    rm -f "$BLXSHELL_HOME"/*_grim.png "$BLXSHELL_HOME"/shot-*.png

    success "blxshell runtime installed"
}

install_blxshell_cli() {
    [[ -f "$SCRIPT_DIR/blxshell" ]] || { error "CLI not found at $SCRIPT_DIR/blxshell"; return 1; }

    info "Installing blxshell CLI to $USER_BIN_DIR/blxshell..."
    mkdir -p "$USER_BIN_DIR"
    install -m 755 "$SCRIPT_DIR/blxshell" "$USER_BIN_DIR/blxshell" || { error "Failed to install blxshell CLI"; return 1; }

    if [[ ":$PATH:" != *":$USER_BIN_DIR:"* ]]; then
        warning "$USER_BIN_DIR is not in PATH; add it to your shell profile to run 'blxshell' directly"
    fi

    success "blxshell CLI installed"
}

# ============================================================================
# Color Generator Setup
# ============================================================================

setup_col_gen() {
    local col_gen_dir="$BLXSHELL_HOME/col_gen"

    [[ -d "$col_gen_dir" ]] || { warning "col_gen directory not found, skipping"; return 0; }

    info "Setting up color generator (col_gen)..."

    # Install uv if not present
    if ! command_exists uv; then
        info "Installing uv..."
        curl -LsSf https://astral.sh/uv/install.sh | sh || die "Failed to install uv"
        export PATH="$HOME/.local/bin:$PATH"
        command_exists uv || die "uv not found after installation"
    fi

    # Create venv and install dependencies
    info "Creating Python virtual environment..."
    (cd "$col_gen_dir" && uv sync) 2>&1 | tee -a "$LOG_FILE" || die "Failed to setup col_gen venv"

    success "Color generator ready"
}

# ============================================================================
# Greeter Setup
# ============================================================================

readonly GREETER_WRAPPER_DIR="$SCRIPT_DIR/.config/quickshell/greeter-wrapper"
readonly GREETER_BIN_DEST="/usr/local/bin/greeter-launcher"
readonly GREETER_CACHE_DIR="/var/cache/greeter"
readonly GREETD_CONFIG="/etc/greetd/config.toml"

install_greeter() {
    echo -e "\n${BOLD}${BLUE}══════ Setting Up Greeter ══════${NC}\n"

    info "Building greeter-launcher..."
    (cd "$GREETER_WRAPPER_DIR" && make clean && make) 2>&1 | tee -a "$LOG_FILE" \
        || { error "Failed to build greeter-launcher"; return 1; }

    info "Installing greeter-launcher to $GREETER_BIN_DEST..."
    sudo install -m 755 "$GREETER_WRAPPER_DIR/build/greeter-launcher" "$GREETER_BIN_DEST" \
        || { error "Failed to install greeter-launcher"; return 1; }

    # Ensure greeter group/user exist
    if ! getent group greeter &>/dev/null; then
        info "Creating greeter group..."
        sudo groupadd -r greeter || { error "Failed to create greeter group"; return 1; }
    fi
    if ! getent passwd greeter &>/dev/null; then
        info "Creating greeter user..."
        sudo useradd -r -g greeter -d "$GREETER_CACHE_DIR" -s /usr/sbin/nologin \
            -c "Greeter User" greeter || { error "Failed to create greeter user"; return 1; }
    fi

    # Set up cache directory
    info "Setting up $GREETER_CACHE_DIR..."
    sudo install -d -m 2770 -o root -g greeter "$GREETER_CACHE_DIR"
    for d in .config .config/quickshell .cache .local .local/state .local/share; do
        sudo install -d -m 0750 -o root -g greeter "$GREETER_CACHE_DIR/$d"
    done

    # ACLs so greeter user can traverse home and read Colors.json + wallpapers
    if command_exists setfacl; then
        info "Setting ACLs for greeter access..."
        sudo setfacl -m "u:greeter:--x,m:--x"   "$HOME"
        sudo setfacl -m "u:greeter:r-x,m:r-x"   "$HOME/.config"
        sudo setfacl -m "u:greeter:r-x,m:r-x"   "$HOME/.config/quickshell"
        sudo setfacl -m "u:greeter:r--,m:r--"   "$HOME/.config/quickshell/Colors.json" 2>/dev/null || true
        for d in .local .local/share .local/share/wallpapers; do
            [[ -d "$HOME/$d" ]] && sudo setfacl -m "u:greeter:r-x,m:r-x" "$HOME/$d"
        done
        sudo setfacl -R -m "u:greeter:r-X,m:r-X" "$HOME/.local/share/wallpapers" 2>/dev/null || true
    else
        warning "setfacl not found — greeter may not read Colors.json (install 'acl')"
    fi

    # Write greetd config
    info "Writing $GREETD_CONFIG..."
    sudo tee "$GREETD_CONFIG" > /dev/null << EOF
[terminal]
vt = 1

[default_session]
user = "greeter"
command = "$GREETER_BIN_DEST"
EOF

    # Enable greetd + graphical target
    if confirm "Enable greetd to start at boot?" "y"; then
        sudo systemctl set-default graphical.target  2>&1 | tee -a "$LOG_FILE" || true
        sudo systemctl enable greetd                 2>&1 | tee -a "$LOG_FILE" \
            || warning "Failed to enable greetd (enable manually: systemctl enable greetd)"
        success "greetd enabled — will start on next boot"
    else
        info "greetd not enabled — run: sudo systemctl enable greetd"
    fi

    success "Greeter setup complete"
}

# ============================================================================
# Main
# ============================================================================

show_banner() {
    echo -e "${CYAN}"
    cat << 'EOF'
  _     _          _          _ _
 | |__ | |_  _____| |__   ___| | |
 | '_ \| \ \/ / __| '_ \ / _ \ | |
 | |_) | |>  <\__ \ | | |  __/ | |
 |_.__/|_/_/\_\___/_| |_|\___|_|_|

EOF
    echo -e "${NC}${BOLD}Installation Script${NC} | Log: ${BLUE}$LOG_FILE${NC}\n"
}

greet_user() {
    echo -e "Hello ${BOLD}$USER${NC}!\n"

    warning "blxshell changed: the shell now runs through the 'blxshell' CLI."
    info "Installer will place the runtime in $BLXSHELL_HOME and the CLI in $USER_BIN_DIR/blxshell."

    if [[ "$REMOTE_INSTALL" == true ]]; then
        info "Remote install mode - dotfiles will be installed by default"
        INSTALL_DOTFILES=true
        if [[ -d "$HOME/.config" ]]; then
            backup_config || { confirm "Continue without backup?" || exit 1; }
        fi
        return
    fi

    if confirm "Install dotfiles (will replace ~/.config)?" "n"; then
        INSTALL_DOTFILES=true
        if [[ -d "$HOME/.config" ]]; then
            if confirm "Backup existing ~/.config first?" "y"; then
                backup_config || { confirm "Continue without backup?" || exit 1; }
            else
                warning "Config will be overwritten without backup!"
                confirm "Are you absolutely sure?" || exit 1
            fi
        fi
    fi
}

cleanup() {
    local exit_code=$?

    # Clean up cloned repo if remote install
    if [[ -n "$CLEANUP_DIR" ]] && [[ -d "$CLEANUP_DIR" ]]; then
        info "Cleaning up temporary files..."
        rm -rf "$CLEANUP_DIR"
    fi

    if [[ $exit_code -ne 0 ]]; then
        echo -e "\n${RED}━━━ Installation Failed ━━━${NC}"
        error "Check log: $LOG_FILE"
        [[ "$BACKUP_DONE" == true ]] && info "Backup at: $BACKUP_DIR/latest"
    fi
}

trap cleanup EXIT

main() {
    show_banner

    # Detect if running locally or from curl
    detect_install_mode

    # Clone if remote install
    if [[ "$REMOTE_INSTALL" == true ]]; then
        clone_repository
    fi

    cd "$SCRIPT_DIR"

    # Verify directory structure
    [[ -d "$SCRIPT_DIR/arch-deps" ]] || die "'arch-deps' directory not found!"

    preflight_checks
    greet_user

    echo -e "\n${BOLD}${YELLOW}══════ Starting Installation ══════${NC}"

    if [[ "$SKIP_PACKAGES" == true ]]; then
        warning "Skipping package installation (unsupported system)"
        INSTALL_DOTFILES=true
    elif [[ "$IS_OPENSUSE" == true ]]; then
        install_opensuse_packages || exit 1
    else
        install_metapackages || exit 1
    fi

    echo -e "\n${BOLD}${BLUE}══════ Installing blxshell CLI Runtime ══════${NC}\n"
    install_blxshell_runtime || exit 1
    install_blxshell_cli || exit 1

    if [[ "$INSTALL_DOTFILES" == true ]]; then
        echo -e "\n${BOLD}${BLUE}══════ Installing Dotfiles ══════${NC}\n"
        install_dotfiles || exit 1
    fi

    echo -e "\n${BOLD}${BLUE}══════ Setting Up Color Generator ══════${NC}\n"
    setup_col_gen || exit 1

    echo -e "\n${BOLD}${BLUE}══════ Setting Default Shell ══════${NC}\n"
    info "Setting fish as default shell..."
    chsh -s /usr/bin/fish || warning "Failed to set fish as default shell (run: chsh -s /usr/bin/fish)"

    echo -e "\n${BOLD}${BLUE}══════ Configuring Groups ══════${NC}\n"
    info "Adding $USER to NetworkManager group..."
    # Only Arch uses a 'network' group — openSUSE uses PolicyKit, no group needed
    if getent group network &>/dev/null; then
        sudo /usr/sbin/usermod -aG network "$USER" && success "Added $USER to network" \
            || warning "Failed to add $USER to network group"
    else
        success "NetworkManager managed via PolicyKit — no group needed"
    fi

    if confirm "Set up greeter (greetd login screen)?" "y"; then
        install_greeter || warning "Greeter setup failed — run install_greeter manually"
    fi

    echo -e "\n${BOLD}${GREEN}══════ Installation Complete! ══════${NC}\n"

    echo -e "\n${GREEN}Run '${BOLD}blxshell start${NC}${GREEN}' to launch the shell, then reboot and login into Hyprland (NOT UWSM).${NC}\n"
}

main "$@"