#!/usr/bin/env bash
# Daily News - 系统服务一键安装
# 用法：./scripts/install.sh
# 不要用 sudo 直接调用本脚本；脚本会在需要时自行 sudo 提权。

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"

# ─── 用户检测 ────────────────────────────────────────────────
if [[ $EUID -eq 0 ]]; then
    if [[ -z "${SUDO_USER:-}" ]]; then
        echo "❌ 请勿直接用 root 运行本脚本，请用普通用户（脚本按需自动提权）"
        exit 1
    fi
    RUN_USER="$SUDO_USER"
else
    RUN_USER="$USER"
fi
RUN_GROUP="$(id -gn "$RUN_USER")"

# ─── uv 检测 ─────────────────────────────────────────────────
UV_BIN="$(command -v uv || true)"
if [[ -z "$UV_BIN" ]]; then
    echo "❌ 找不到 uv 命令，请先安装：https://docs.astral.sh/uv/"
    exit 1
fi

# ─── systemd 检测 ────────────────────────────────────────────
if ! command -v systemctl >/dev/null 2>&1; then
    echo "❌ 系统未检测到 systemctl（非 systemd 系统），本脚本仅支持 systemd Linux"
    exit 1
fi

cd "$PROJECT_DIR"

echo "🚀 Daily News - 系统服务安装"
echo "────────────────────────────────────────"
echo "  项目目录: $PROJECT_DIR"
echo "  运行身份: $RUN_USER:$RUN_GROUP"
echo "  uv 路径:  $UV_BIN"
echo "────────────────────────────────────────"
echo ""

# 提前提权一次，后续 sudo 命令直接复用凭据缓存
echo "🔐 需要 sudo 权限来写入 /etc/systemd/system/"
sudo -v
echo ""

# ─── [1/6] 同步依赖 ──────────────────────────────────────────
echo "📦 [1/6] 同步依赖 (uv sync)..."
uv sync
echo "✓ 依赖已就绪"
echo ""

# ─── [2/6] 验证 .env ────────────────────────────────────────
echo "🔐 [2/6] 验证 .env..."
if [[ ! -f "$PROJECT_DIR/.env" ]]; then
    echo "❌ .env 不存在，请先复制 .env.example 并填入 API Key"
    exit 1
fi
echo "✓ .env 存在"
echo ""

# ─── [3/6] LLM 健康检查 ─────────────────────────────────────
echo "🔍 [3/6] 校验 LLM 接口（仅在此处校验，运行时不再校验）..."
if ! uv run python -m src.main check; then
    echo ""
    echo "❌ LLM 校验失败，已中止安装"
    echo "请检查 config.json 中的 llm.baseUrl/model 以及 .env 中的 API Key"
    exit 1
fi
echo ""

# ─── [4/6] 生成单元文件 ─────────────────────────────────────
echo "📝 [4/6] 渲染 systemd 单元模板..."
STAGE_DIR="$(mktemp -d)"
trap 'rm -rf "$STAGE_DIR"' EXIT

uv run python "$PROJECT_DIR/scripts/_gen_units.py" \
    --project-dir "$PROJECT_DIR" \
    --user "$RUN_USER" \
    --group "$RUN_GROUP" \
    --uv-bin "$UV_BIN" \
    --output-dir "$STAGE_DIR"

# 同时渲染 daily-news 包装脚本（/usr/local/bin/daily-news）
sed "s|{{PROJECT_DIR}}|$PROJECT_DIR|g" \
    "$PROJECT_DIR/scripts/daily-news.tmpl" > "$STAGE_DIR/daily-news"
echo "  ✓ daily-news"
echo ""

# ─── [5/6] 安装到 systemd ───────────────────────────────────
echo "📥 [5/6] 安装到系统..."
sudo install -m 644 "$STAGE_DIR/dnews-fetch.service" /etc/systemd/system/
sudo install -m 644 "$STAGE_DIR/dnews-fetch.timer"   /etc/systemd/system/
sudo install -m 644 "$STAGE_DIR/dnews-push.service"  /etc/systemd/system/
sudo install -m 644 "$STAGE_DIR/dnews-push.timer"    /etc/systemd/system/

# 日志保留策略（journald 命名空间 drop-in）
sudo mkdir -p /etc/systemd/journald@dnews.conf.d
sudo install -m 644 "$STAGE_DIR/journald-dnews.conf" \
    /etc/systemd/journald@dnews.conf.d/retention.conf

# daily-news 包装脚本到系统 PATH
sudo install -m 755 "$STAGE_DIR/daily-news" /usr/local/bin/daily-news

sudo systemctl daemon-reload

# 命名空间 journald 实例首次会在 fetch/push 首次产出日志时自动拉起；
# 若已存在（重装场景），重启以加载新保留策略。
sudo systemctl restart systemd-journald@dnews.service 2>/dev/null || true

echo "✓ 单元文件 + daily-news 包装脚本已安装"
echo ""

# ─── [6/6] 启用并启动 timer ─────────────────────────────────
echo "🚦 [6/6] 启用并（重）启动 timer..."
# 先 enable 注册开机自启
sudo systemctl enable dnews-fetch.timer dnews-push.timer
# 再 restart 强制按新单元文件重置内存中的 timer 状态
# （enable --now 对已运行的 timer 是 no-op，无法应用新 OnCalendar/OnActiveSec 等改动）
sudo systemctl restart dnews-fetch.timer dnews-push.timer
echo ""

# ─── 完成 ───────────────────────────────────────────────────
echo "✅ 安装完成！"
echo ""
echo "下次触发时间："
systemctl list-timers 'dnews-*' --no-pager || true
echo ""
echo "常用命令（系统 PATH 中可直接调用）："
echo "  daily-news status      查看状态 + 最近日志"
echo "  daily-news logs        实时跟随日志"
echo "  daily-news stop        停止"
echo "  daily-news start       启动"
echo "  daily-news restart     重启 timer"
echo ""
echo "手动立即触发一次任务（不影响下次调度）："
echo "  sudo systemctl start dnews-fetch.service"
echo "  sudo systemctl start dnews-push.service"
echo ""
echo "卸载：$PROJECT_DIR/scripts/uninstall.sh"