#!/bin/sh
# =============================================================================
#  追光AI CRON 容器启动入口
#  - 用 supercronic 调度 /app/crontab.txt 里的任务
#  - 跟 host cron 行为一致，但完全在容器内
#  - 增强：信号处理、错误处理、日志轮转集成
# =============================================================================
set -e

cd /app

# =============================================================================
# 信号处理 - 优雅关闭
# =============================================================================
SUPERCRONIC_PID=""

cleanup() {
  echo "[entrypoint-cron] Received shutdown signal, stopping gracefully..."
  if [ -n "$SUPERCRONIC_PID" ]; then
    kill -TERM "$SUPERCRONIC_PID" 2>/dev/null || true
    # 等待进程结束，最多 30 秒
    for i in $(seq 1 30); do
      if ! kill -0 "$SUPERCRONIC_PID" 2>/dev/null; then
        echo "[entrypoint-cron] Process exited cleanly"
        break
      fi
      sleep 1
    done
    # 如果还没退出，强制终止
    if kill -0 "$SUPERCRONIC_PID" 2>/dev/null; then
      echo "[entrypoint-cron] Force killing process..."
      kill -9 "$SUPERCRONIC_PID" 2>/dev/null || true
    fi
  fi
  exit 0
}

trap cleanup TERM INT QUIT

# =============================================================================
# 日志轮转配置（通过环境变量控制）
# =============================================================================
# LOG_MAX_SIZE: 单个日志文件最大大小（默认 100M）
# LOG_MAX_FILES: 保留的历史日志文件数（默认 7）
# LOG_COMPRESS: 是否压缩历史日志（默认 true）
LOG_MAX_SIZE="${LOG_MAX_SIZE:-100M}"
LOG_MAX_FILES="${LOG_MAX_FILES:-7}"
LOG_COMPRESS="${LOG_COMPRESS:-true}"

# 如果启用了日志文件输出，配置 logrotate
if [ -n "$LOG_FILE" ]; then
  echo "[entrypoint-cron] Log file configured: $LOG_FILE"
  echo "[entrypoint-cron] Log rotation: max_size=$LOG_MAX_SIZE, max_files=$LOG_MAX_FILES, compress=$LOG_COMPRESS"
  
  # 创建 logrotate 配置
  cat > /etc/logrotate.d/zhuiguang-cron <<EOF
$LOG_FILE {
    size $LOG_MAX_SIZE
    rotate $LOG_MAX_FILES
    compress
    delaycompress
    missingok
    notifempty
    create 0644 root root
    postrotate
      # 通知 supercronic 重新打开日志文件（如果支持）
      true
    endscript
}
EOF
fi

# =============================================================================
# Seed data volume if empty
# =============================================================================
if [ ! -f /app/data/bot-characters.json ] && [ -d /app/data-seed ]; then
  echo "[entrypoint] Seeding data volume..."
  cp -r /app/data-seed/* /app/data/
fi

# =============================================================================
# .env 校验
# =============================================================================
if [ ! -f /app/.env ]; then
  echo "[entrypoint-cron] FATAL: /app/.env not found. Mount .env file into container."
  exit 1
fi

# =============================================================================
# crontab 文件校验
# =============================================================================
if [ ! -f /app/crontab.txt ]; then
  echo "[entrypoint-cron] FATAL: /app/crontab.txt not found. Mount crontab into container."
  exit 1
fi

# =============================================================================
# 健康检查端点（可选）
# =============================================================================
if [ "$HEALTH_CHECK_ENABLED" = "true" ]; then
  HEALTH_PORT="${HEALTH_CHECK_PORT:-8311}"
  echo "[entrypoint-cron] Health check enabled on port $HEALTH_PORT"
  
  # 启动简单的健康检查 HTTP 服务器
  (
    while true; do
      echo -e "HTTP/1.1 200 OK\r\nContent-Length: 2\r\n\r\nOK" | nc -l -p "$HEALTH_PORT" -q 1 2>/dev/null || true
    done
  ) &
  HEALTH_PID=$!
  echo "[entrypoint-cron] Health check server started (PID: $HEALTH_PID)"
fi

# =============================================================================
# supercronic 启动
# =============================================================================
echo "[entrypoint-cron] Starting supercronic..."
echo "[entrypoint-cron] crontab:"
sed 's/^/  /' /app/crontab.txt

# 构建启动参数
SUPERCRONIC_ARGS=""

# Prometheus 监控（可选）
if [ "$PROMETHEUS_ENABLED" != "false" ]; then
  PROMETHEUS_PORT="${PROMETHEUS_PORT:-8310}"
  SUPERCRONIC_ARGS="$SUPERCRONIC_ARGS -prometheus-listen-address 0.0.0.0:$PROMETHEUS_PORT"
  echo "[entrypoint-cron] Prometheus metrics enabled on port $PROMETHEUS_PORT"
fi

# 日志分离（可选）
if [ "$SPLIT_LOGS" = "true" ]; then
  SUPERCRONIC_ARGS="$SUPERCRONIC_ARGS -split-logs"
  echo "[entrypoint-cron] Split logs enabled (stdout/stderr separated)"
fi

# 调试模式
if [ "$DEBUG" = "true" ]; then
  SUPERCRONIC_ARGS="$SUPERCRONIC_ARGS -debug"
  echo "[entrypoint-cron] Debug mode enabled"
fi

# 启动 supercronic（后台运行以捕获 PID）
/usr/local/bin/supercronic-linux-amd64 $SUPERCRONIC_ARGS /app/crontab.txt &
SUPERCRONIC_PID=$!

echo "[entrypoint-cron] Supercronic started with PID: $SUPERCRONIC_PID"
echo "[entrypoint-cron] Container ready"

# 等待 supercronic 进程（同时保持信号处理能力）
wait "$SUPERCRONIC_PID"
EXIT_CODE=$?

echo "[entrypoint-cron] Supercronic exited with code: $EXIT_CODE"

# 清理健康检查服务器
if [ -n "$HEALTH_PID" ]; then
  kill "$HEALTH_PID" 2>/dev/null || true
fi

exit $EXIT_CODE