#!/usr/bin/env bash
# ============================================================================
# Dotfiles Health Check (Arch/CachyOS)
# ============================================================================

set -e

readonly DOTFILES_HOME="${DOTFILES_HOME:-.}"
readonly DOTFILES_VERSION="3.0.0"

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

# Track results
TOTAL_CHECKS=0
PASSED_CHECKS=0
FAILED_CHECKS=0
WARNING_CHECKS=0

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

print_header() {
    local user="${USER:-root}"
    local hostname="${HOSTNAME:-localhost}"
    local timestamp=$(date '+%a %b %d %H:%M')
    
    echo ""
    printf "${CYAN}+ ${NC}%-20s %30s %25s\n" "$user@$hostname" "dotfiles-doctor" "$timestamp"
    echo ""
}

# ============================================================================
# Health check functions
# ============================================================================

print_section() {
    echo -e "\n${BLUE}▶${NC} $1"
}

check_pass() {
    ((PASSED_CHECKS++))
    ((TOTAL_CHECKS++))
    echo -e "  ${GREEN}✓${NC} $1"
}

check_fail() {
    ((FAILED_CHECKS++))
    ((TOTAL_CHECKS++))
    echo -e "  ${RED}✗${NC} $1"
}

check_warn() {
    ((WARNING_CHECKS++))
    ((TOTAL_CHECKS++))
    echo -e "  ${YELLOW}⚠${NC} $1"
}

# ============================================================================
# Health checks
# ============================================================================

check_os() {
    print_section "Operating System"
    
    if [[ "$OSTYPE" == "linux-gnu" ]]; then
        if grep -qi "arch\|cachyos" /etc/os-release 2>/dev/null; then
            check_pass "Running on Arch/CachyOS"
        else
            check_fail "Not running on Arch/CachyOS"
        fi
    else
        check_fail "Not running on Linux"
    fi
}

check_shell() {
    print_section "Shell Configuration"
    
    if [[ -f "$HOME/.zshrc" ]]; then
        check_pass "Zsh configuration exists"
    else
        check_fail "Zsh configuration missing"
    fi
    
    if [[ "$SHELL" == *"zsh"* ]]; then
        check_pass "Zsh is default shell"
    else
        check_warn "Zsh is not default shell (current: $SHELL)"
    fi
}

check_symlinks() {
    print_section "Symlinks"
    
    local symlink_count=0
    local broken_count=0
    
    for symlink in ~/.zshrc ~/.gitconfig ~/.vimrc ~/.tmux.conf; do
        if [[ -L "$symlink" ]]; then
            ((symlink_count++))
            if [[ -e "$symlink" ]]; then
                check_pass "$(basename $symlink) → $(readlink $symlink)"
            else
                ((broken_count++))
                check_fail "$(basename $symlink) is broken"
            fi
        fi
    done
    
    if [[ $symlink_count -eq 0 ]]; then
        check_warn "No symlinks found (may not be installed yet)"
    fi
}

check_vim() {
    print_section "Editor Configuration"
    
    if command -v vim &> /dev/null; then
        local vim_version=$(vim --version | head -1)
        check_pass "Vim installed: $vim_version"
    else
        check_fail "Vim not installed"
    fi
    
    if command -v nvim &> /dev/null; then
        local nvim_version=$(nvim --version | head -1)
        check_pass "Neovim installed: $nvim_version"
    else
        check_warn "Neovim not installed (optional)"
    fi
}

check_git() {
    print_section "Git Configuration"
    
    if command -v git &> /dev/null; then
        check_pass "Git installed"
        
        if git config --global user.name &> /dev/null; then
            local git_user=$(git config --global user.name)
            check_pass "Git user configured: $git_user"
        else
            check_fail "Git user not configured"
        fi
        
        if git config --global user.email &> /dev/null; then
            local git_email=$(git config --global user.email)
            check_pass "Git email configured: $git_email"
        else
            check_fail "Git email not configured"
        fi
    else
        check_fail "Git not installed"
    fi
}

check_optional_tools() {
    print_section "Optional Tools"
    
    if command -v fzf &> /dev/null; then
        check_pass "fzf installed (fuzzy finder)"
    else
        check_warn "fzf not installed (command palette requires this)"
    fi
    
    if command -v lastpass-cli &> /dev/null || command -v lpass &> /dev/null; then
        check_pass "LastPass CLI installed"
    else
        check_warn "LastPass CLI not installed (password manager)"
    fi
    
    if command -v tmux &> /dev/null; then
        check_pass "Tmux installed"
    else
        check_warn "Tmux not installed (workspaces require this)"
    fi
    
    if command -v age &> /dev/null || command -v gpg &> /dev/null; then
        check_pass "Encryption tool available (age or gpg)"
    else
        check_warn "No encryption tool (vault requires age or gpg)"
    fi
    
    if command -v bat &> /dev/null; then
        check_pass "bat installed (syntax highlighting)"
    else
        check_warn "bat not installed (optional enhancement)"
    fi
    
    if command -v eza &> /dev/null; then
        check_pass "eza installed (ls replacement)"
    else
        check_warn "eza not installed (optional enhancement)"
    fi
}

check_pacman() {
    print_section "Package Manager"
    
    if command -v pacman &> /dev/null; then
        check_pass "Pacman available"
    else
        check_fail "Pacman not found (this is Arch/CachyOS only)"
    fi
}

check_permissions() {
    print_section "File Permissions"
    
    if [[ -f "$DOTFILES_HOME/install.sh" ]]; then
        if [[ -x "$DOTFILES_HOME/install.sh" ]]; then
            check_pass "install.sh is executable"
        else
            check_fail "install.sh is not executable"
        fi
    fi
    
    if [[ -d "$DOTFILES_HOME/bin" ]]; then
        local non_exec=$(find "$DOTFILES_HOME/bin" -type f ! -perm /u+x 2>/dev/null | wc -l)
        if [[ $non_exec -eq 0 ]]; then
            check_pass "All scripts in bin/ are executable"
        else
            check_fail "$non_exec scripts in bin/ are not executable"
        fi
    fi
}

check_zsh_plugins() {
    print_section "Zsh Plugins"
    
    if [[ -d "$HOME/.oh-my-zsh" ]]; then
        check_pass "Oh My Zsh installed"
        
        if [[ -d "$HOME/.oh-my-zsh/custom/plugins/zsh-autosuggestions" ]]; then
            check_pass "zsh-autosuggestions installed"
        else
            check_warn "zsh-autosuggestions not installed"
        fi
        
        if [[ -d "$HOME/.oh-my-zsh/custom/plugins/zsh-syntax-highlighting" ]]; then
            check_pass "zsh-syntax-highlighting installed"
        else
            check_warn "zsh-syntax-highlighting not installed"
        fi
        
        if [[ -f "$HOME/.oh-my-zsh/themes/adlee.zsh-theme" ]]; then
            check_pass "adlee theme installed"
        else
            check_warn "adlee theme not installed"
        fi
    else
        check_warn "Oh My Zsh not installed"
    fi
}

check_dotfiles_dir() {
    print_section "Dotfiles Directory"
    
    if [[ -d "$DOTFILES_HOME" ]]; then
        check_pass "Dotfiles directory found: $DOTFILES_HOME"
    else
        check_fail "Dotfiles directory not found: $DOTFILES_HOME"
        return
    fi
    
    if [[ -f "$DOTFILES_HOME/dotfiles.conf" ]]; then
        check_pass "Configuration file exists"
    else
        check_warn "Configuration file missing"
    fi
    
    if [[ -d "$DOTFILES_HOME/.git" ]]; then
        check_pass "Git repository initialized"
    else
        check_warn "Not a git repository"
    fi
}

# ============================================================================
# Print summary
# ============================================================================

print_summary() {
    echo ""
    printf "${CYAN}─%.0s${NC}" {1..70}; echo ""
    
    if [[ $FAILED_CHECKS -eq 0 ]]; then
        echo -e "${GREEN}✓${NC} All checks passed ($PASSED_CHECKS/$TOTAL_CHECKS)"
    else
        echo -e "${RED}✗${NC} Some checks failed"
        echo -e "  ${GREEN}Passed:${NC} $PASSED_CHECKS"
        echo -e "  ${RED}Failed:${NC} $FAILED_CHECKS"
        if [[ $WARNING_CHECKS -gt 0 ]]; then
            echo -e "  ${YELLOW}Warnings:${NC} $WARNING_CHECKS"
        fi
    fi
    
    echo ""
    
    if [[ $FAILED_CHECKS -gt 0 ]]; then
        echo -e "${YELLOW}💡 Tip:${NC} Run 'dotfiles-doctor.sh --fix' to attempt automatic fixes"
        echo ""
        return 1
    fi
}

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

main() {
    print_header
    
    check_os
    check_pacman
    check_shell
    check_vim
    check_git
    check_dotfiles_dir
    check_symlinks
    check_zsh_plugins
    check_optional_tools
    check_permissions
    
    print_summary
}

main "$@"