#!/usr/bin/env bash
# ============================================================================
# Update Dotfiles Script
# ============================================================================
# Updates dotfiles from the git repository and relinks files
#
# Usage:
#   dotfiles-update.sh              # Pull and re-run install
#   dotfiles-update.sh --skip-deps  # Pull and re-run install without deps
#   dotfiles-update.sh --pull-only  # Only git pull, don't re-run install
#
# Aliases: dfu, dfupdate
# ============================================================================

set -e

# ============================================================================
# Options
# ============================================================================

SKIP_DEPS=true   # Default to skipping deps on updates
PULL_ONLY=false

for arg in "$@"; do
    case "$arg" in
        --skip-deps)
            SKIP_DEPS=true
            ;;
        --with-deps)
            SKIP_DEPS=false
            ;;
        --pull-only)
            PULL_ONLY=true
            ;;
        --help|-h)
            echo "Usage: dotfiles-update.sh [OPTIONS]"
            echo
            echo "Options:"
            echo "  --skip-deps    Skip dependency check (default for updates)"
            echo "  --with-deps    Run full dependency check"
            echo "  --pull-only    Only git pull, don't re-run install script"
            echo "  --help         Show this help message"
            echo
            echo "Aliases:"
            echo "  dfu, dfupdate  Update dotfiles"
            echo
            exit 0
            ;;
    esac
done

# ============================================================================
# Load Configuration
# ============================================================================

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
DOTFILES_CONF="${SCRIPT_DIR}/../dotfiles.conf"
[[ -f "$DOTFILES_CONF" ]] || DOTFILES_CONF="$HOME/.dotfiles/dotfiles.conf"

if [[ -f "$DOTFILES_CONF" ]]; then
    source "$DOTFILES_CONF"
else
    # Fallback defaults
    DOTFILES_DIR="$HOME/.dotfiles"
    DOTFILES_BRANCH="main"
    DOTFILES_GITHUB_USER="adlee-was-taken"
    DOTFILES_REPO_NAME="dotfiles"
    DOTFILES_RAW_URL="https://raw.githubusercontent.com/${DOTFILES_GITHUB_USER}/${DOTFILES_REPO_NAME}/${DOTFILES_BRANCH}"
fi

# ============================================================================
# Colors
# ============================================================================

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

# ============================================================================
# MOTD-style header
# ============================================================================

_M_WIDTH=66

print_header() {
    local user="${USER:-root}"
    local hostname="${HOSTNAME:-$(hostname -s 2>/dev/null)}"
    local script_name="dotfiles-update"
    local datetime=$(date '+%a %b %d %H:%M')
    
    # Colors
    local _M_RESET=$'\033[0m'
    local _M_BOLD=$'\033[1m'
    local _M_DIM=$'\033[2m'
    local _M_BLUE=$'\033[38;5;39m'
    local _M_GREY=$'\033[38;5;242m'
    
    # Build horizontal line
    local hline=""
    for ((i=0; i<_M_WIDTH; i++)); do hline+="═"; done
    local inner=$((_M_WIDTH - 2))
    
    # Header content
    local h_left="✦ ${user}@${hostname}"
    local h_center="${script_name}"
    local h_right="${datetime}"
    local h_pad=$(((inner - ${#h_left} - ${#h_center} - ${#h_right}) / 2))
    local h_spaces=""
    for ((i=0; i<h_pad; i++)); do h_spaces+=" "; done
    
    echo ""
    echo -e "${_M_GREY}╒${hline}╕${_M_RESET}"
    echo -e "${_M_GREY}│${_M_RESET} ${_M_BOLD}${_M_BLUE}${h_left}${_M_RESET}${h_spaces}${_M_DIM}${h_center}${h_spaces}${h_right}${_M_RESET} ${_M_GREY}│${_M_RESET}"
    echo -e "${_M_GREY}╘${hline}╛${_M_RESET}"
    echo ""
}

print_success() {
    echo -e "${GREEN}✓${NC} $1"
}

print_warning() {
    echo -e "${YELLOW}⚠${NC} $1"
}

print_error() {
    echo -e "${RED}✗${NC} $1"
}

print_step() {
    echo -e "${GREEN}==>${NC} $1"
}

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

print_header

if [ ! -d "$DOTFILES_DIR" ]; then
    print_error "Dotfiles directory not found: $DOTFILES_DIR"
    echo "Run the installation script first:"
    echo "  curl -fsSL ${DOTFILES_RAW_URL}/install.sh | bash"
    exit 1
fi

cd "$DOTFILES_DIR"

print_step "Updating dotfiles from repository..."
git pull origin "$DOTFILES_BRANCH"

if [ $? -eq 0 ]; then
    print_success "Dotfiles updated successfully"

    if [[ "$PULL_ONLY" == true ]]; then
        echo
        print_success "Pull complete (--pull-only mode)"
        echo "Run ./install.sh manually to re-link files"
        exit 0
    fi

    if [ -f "$DOTFILES_DIR/install.sh" ]; then
        echo
        read -p "Run install script to update links? [Y/n]: " response
        response=${response:-y}

        if [[ "$response" =~ ^[Yy]$ ]]; then
            if [[ "$SKIP_DEPS" == true ]]; then
                "$DOTFILES_DIR/install.sh" --skip-deps
            else
                "$DOTFILES_DIR/install.sh"
            fi
        fi
    fi

    echo
    print_success "Update complete!"
    echo -e "Reload your shell: ${CYAN}reload${NC} or ${CYAN}source ~/.zshrc${NC}"
else
    print_error "Failed to update dotfiles"
    exit 1
fi