#!/bin/bash
set -euo pipefail

# ============================================
# NyaGallery 一键部署脚本（清华源版 + npm淘宝源）
# 适用系统：Ubuntu 20.04+ / Debian 11+
# 功能：自动换清华源、安装依赖、部署前后端
# ============================================

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

WORK_DIR="${HOME}/NyaGallery"
PYTHON_CMD="python3"
NODE_VERSION_REQUIRED=18

info() { echo -e "${GREEN}[INFO]${NC} $1"; }
warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
error() { echo -e "${RED}[ERROR]${NC} $1"; exit 1; }

detect_os() {
    if [ -f /etc/os-release ]; then
        . /etc/os-release
        OS=$ID
        VER=$VERSION_ID
    else
        error "无法检测操作系统"
    fi
    info "检测到系统: $OS $VER"
    if [[ "$OS" != "ubuntu" && "$OS" != "debian" ]]; then
        error "本脚本仅支持 Ubuntu/Debian"
    fi
}

change_apt_source() {
    info "备份当前 APT 源并更换为清华源..."
    BACKUP_DIR="/etc/apt/backup_$(date +%Y%m%d_%H%M%S)"
    sudo mkdir -p "$BACKUP_DIR"
    sudo cp /etc/apt/sources.list "$BACKUP_DIR/sources.list.bak"
    info "已备份到 $BACKUP_DIR"
    if [[ "$OS" == "ubuntu" ]]; then
        CODENAME=$(lsb_release -sc)
        sudo tee /etc/apt/sources.list > /dev/null << EOF
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ $CODENAME main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ $CODENAME-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ $CODENAME-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ $CODENAME-security main restricted universe multiverse
EOF
    elif [[ "$OS" == "debian" ]]; then
        CODENAME=$(lsb_release -sc)
        sudo tee /etc/apt/sources.list > /dev/null << EOF
deb https://mirrors.tuna.tsinghua.edu.cn/debian $CODENAME main contrib non-free
deb https://mirrors.tuna.tsinghua.edu.cn/debian $CODENAME-updates main contrib non-free
deb https://mirrors.tuna.tsinghua.edu.cn/debian-security $CODENAME-security main contrib non-free
EOF
    fi
    sudo apt-get update -qq
}

change_pip_source() {
    info "设置 pip 源为清华源..."
    mkdir -p "$HOME/.pip"
    cat > "$HOME/.pip/pip.conf" << EOF
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = pypi.tuna.tsinghua.edu.cn
EOF
}

install_base_deps() {
    sudo apt-get install -y -qq git curl wget build-essential software-properties-common
}

install_python() {
    if command -v python3.11 &>/dev/null; then
        PYTHON_CMD="python3.11"; return
    fi
    if command -v python3.12 &>/dev/null; then
        PYTHON_CMD="python3.12"; return
    fi
    sudo add-apt-repository -y ppa:deadsnakes/ppa
    sudo apt-get update -qq
    sudo apt-get install -y python3.11 python3.11-venv python3.11-dev
    PYTHON_CMD="python3.11"
}

purge_nodejs() {
    sudo apt-get remove -y nodejs npm 2>/dev/null || true
    sudo apt-get autoremove -y 2>/dev/null || true
    sudo rm -f /etc/apt/sources.list.d/nodesource.list* 2>/dev/null || true
}

install_nodejs() {
    if command -v node &>/dev/null; then
        NODE_VER=$(node -v | cut -d'v' -f2 | cut -d. -f1)
        [ "$NODE_VER" -ge 18 ] && return
    fi
    purge_nodejs
    curl -fsSL https://deb.nodesource.com/setup_18.x | sudo -E bash -
    sudo apt-get install -y nodejs
}

clone_repo() {
    if [ ! -d "$WORK_DIR/.git" ]; then
        git clone https://github.com/NayaCcR/NyaGallery.git "$WORK_DIR"
    fi
    cd "$WORK_DIR"
}

install_backend() {
    $PYTHON_CMD -m venv venv
    source venv/bin/activate
    pip install --upgrade pip
    pip install -e ".[media,pixiv,pixiv-login,postgres,redis]"
}

install_frontend() {
    cd frontend
    npm config set registry https://registry.npmmirror.com
    npm install
    cd ..
}

init_database() {
    mkdir -p storage/original storage/metadata storage/cache
    [ ! -f "storage/nyagallery.db" ] && nyagallery --storage storage setup
}

create_admin() {
    read -p "管理员用户名 (默认: admin): " admin_user
    admin_user=${admin_user:-admin}
    read -s -p "密码: " admin_pass; echo
    read -s -p "确认密码: " admin_pass2; echo
    [ "$admin_pass" != "$admin_pass2" ] && { error "密码不一致"; }
    nyagallery --storage storage setup --username "$admin_user" --role admin --password "$admin_pass"
}

main() {
    [ "$EUID" -eq 0 ] && { warn "不建议用 root，建议普通用户 + sudo"; read -p "继续？(y/N): " c; [[ ! "$c" =~ ^[Yy]$ ]] && exit 0; }
    detect_os
    change_apt_source; change_pip_source; install_base_deps
    install_python; install_nodejs; clone_repo
    install_backend; install_frontend; init_database; create_admin
    echo "部署完成！运行 cd $WORK_DIR && source venv/bin/activate && nyagallery --storage storage serve"
}

main