#!/usr/bin/env bash
# ============================================================
# 把「每日 17:00 增量抓数据」注册为 macOS launchd 用户级定时任务
#
# 用法：
#   ./deploy/install-sync-schedule.sh install     安装（幂等，可重复执行）
#   ./deploy/install-sync-schedule.sh uninstall   停止并移除
#   ./deploy/install-sync-schedule.sh status      查看状态、下次触发时间与最近日志
#   ./deploy/install-sync-schedule.sh reinstall   重新渲染 plist 并重载
#   ./deploy/install-sync-schedule.sh run         立刻按计划表跑一次（排障用）
#   ./deploy/install-sync-schedule.sh dry-run     立刻跑一次「只看清单不抓数」
#
# 为什么用 launchd 而不是 cron：
#   - 本机已经用 launchd 托管 web 服务（install-service.sh），同一套机制更好排查；
#   - macOS 上 cron 需要额外的 Full Disk Access 授权，且睡眠错过的任务**不会补跑**，
#     而 launchd 的 StartCalendarInterval 会在唤醒后补跑 —— 对「每天补缺口」的任务
#     来说，补跑是刚需（漏一天就等于多一天缺口）。
#
# 若你不用 launchd，也可以用 crontab 一行搞定：
#   0 17 * * * cd <项目根> && PYTHONPATH=src .venv/bin/python -m hdiv sync daily \
#     >> logs/daily-sync.log 2>&1
# ============================================================
set -euo pipefail

PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
TEMPLATE="${PROJECT_ROOT}/deploy/com.hddiv.sync.plist.example"
TARGET="${HOME}/Library/LaunchAgents/com.hddiv.sync.plist"
LABEL="com.hddiv.sync"
RUNNER="${PROJECT_ROOT}/deploy/daily-sync.sh"
LOG="${PROJECT_ROOT}/logs/daily-sync.log"

die() { echo "错误: $*" >&2; exit 1; }

is_loaded() { launchctl list 2>/dev/null | grep -q "[[:space:]]${LABEL}$"; }

render() {
  [[ -f "${TEMPLATE}" ]] || die "缺少模板：${TEMPLATE}"
  [[ -x "${RUNNER}" ]] || die "缺少可执行脚本：${RUNNER}"
  mkdir -p "${PROJECT_ROOT}/logs" "${HOME}/Library/LaunchAgents"
  # 路径含中文，用 | 作 sed 分隔符避免转义问题
  sed -e "s|__PROJECT_ROOT__|${PROJECT_ROOT}|g" "${TEMPLATE}" > "${TARGET}"
  if grep -q "__" "${TARGET}"; then
    die "plist 中仍有未替换的占位符：$(grep -o '__[A-Z_]*__' "${TARGET}" | sort -u | tr '\n' ' ')"
  fi
  plutil -lint "${TARGET}" >/dev/null || die "plist 格式非法：${TARGET}"
  echo "已渲染 ${TARGET}"
}

cmd_install() {
  render
  if is_loaded; then
    echo "任务已加载，先卸载再加载以应用新配置…"
    launchctl unload -w "${TARGET}" 2>/dev/null || true
  fi
  launchctl load -w "${TARGET}"
  cmd_status
}

cmd_uninstall() {
  if [[ -f "${TARGET}" ]]; then
    launchctl unload -w "${TARGET}" 2>/dev/null || true
    rm -f "${TARGET}"
    echo "已停止并移除 ${TARGET}"
  else
    echo "${TARGET} 不存在，跳过"
  fi
}

cmd_status() {
  echo "=== launchd 定时任务 ==="
  if is_loaded; then
    launchctl list | grep "${LABEL}" | awk '{printf "  PID=%s  上次退出码=%s  %s\n", $1, $2, $3}'
    echo "  （PID 为 - 表示当前没有正在运行的实例；退出码 0 = 上次同步成功）"
  else
    echo "  未加载。执行 ./deploy/install-sync-schedule.sh install 安装"
  fi
  echo "=== 计划 ==="
  echo "  每天 17:00 执行 ${RUNNER}"
  echo "  睡眠/关机错过会在唤醒后补跑一次"
  echo "=== 最近日志（${LOG}）==="
  if [[ -f "${LOG}" ]]; then
    tail -12 "${LOG}"
  else
    echo "  尚无日志（任务还没跑过）"
  fi
}

cmd_run() {
  echo "立即执行一次（等价于 17:00 触发）…"
  "${RUNNER}" "$@"
  echo "完成，退出码 $?。日志：${LOG}"
}

case "${1:-install}" in
  install)   cmd_install ;;
  reinstall) render; launchctl unload -w "${TARGET}" 2>/dev/null || true
             launchctl load -w "${TARGET}"; cmd_status ;;
  uninstall) cmd_uninstall ;;
  status)    cmd_status ;;
  run)       shift || true; cmd_run "$@" ;;
  dry-run)   shift || true; cmd_run --dry-run "$@" ;;
  *) die "未知动作：$1（可用：install|reinstall|uninstall|status|run|dry-run）" ;;
esac