#!/usr/bin/env bash
set -euo pipefail

script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck source=/dev/null
source "${script_dir}/openclaw-heartbeat-cron-common.sh"

apply=0
enable_after_create=0
replace_existing=0
announce=0

skill_root="$(cd "${script_dir}/.." && pwd)"
interval="$(aquaclaw_heartbeat_default_interval)"
job_name="$(aquaclaw_heartbeat_default_job_name)"
session_target="$(aquaclaw_heartbeat_default_session)"
thinking_level="$(aquaclaw_heartbeat_default_thinking)"
timeout_seconds="$(aquaclaw_heartbeat_default_timeout_seconds)"
delivery_channel=""
delivery_session_key=""
target_to=""
account_id=""

while [[ $# -gt 0 ]]; do
  case "$1" in
    --apply)
      apply=1
      shift
      ;;
    --enable)
      enable_after_create=1
      shift
      ;;
    --replace)
      replace_existing=1
      shift
      ;;
    --announce)
      announce=1
      shift
      ;;
    --skill-root)
      skill_root="$2"
      shift 2
      ;;
    --every)
      interval="$2"
      shift 2
      ;;
    --name)
      job_name="$2"
      shift 2
      ;;
    --session)
      session_target="$2"
      shift 2
      ;;
    --thinking)
      thinking_level="$2"
      shift 2
      ;;
    --timeout-seconds)
      timeout_seconds="$2"
      shift 2
      ;;
    --channel)
      delivery_channel="$2"
      shift 2
      ;;
    --to)
      target_to="$2"
      shift 2
      ;;
    --account)
      account_id="$2"
      shift 2
      ;;
    --session-key)
      delivery_session_key="$2"
      shift 2
      ;;
    -h|--help)
      cat <<'EOF'
Usage: install-openclaw-heartbeat-cron.sh [options]

Options:
  --apply                 Actually create or update the cron job
  --enable                Leave the job enabled after install/update
  --replace               Update an existing job with the same name instead of failing
  --announce              Deliver cron summaries to the resolved chat target
  --skill-root <path>     AquaClaw skill repo path
  --every <duration>      Cron interval, for example 15m
  --name <name>           Cron job name
  --session <target>      OpenClaw cron session target
  --thinking <level>      OpenClaw cron thinking level
  --timeout-seconds <n>   OpenClaw cron timeout
  --channel <name>        Delivery channel override
  --to <dest>             Delivery target override
  --account <id>          Delivery account id override
  --session-key <key>     Delivery session key override
EOF
      exit 0
      ;;
    *)
      echo "unknown option: $1" >&2
      exit 1
      ;;
  esac
done

message="$(aquaclaw_heartbeat_build_message "$skill_root")"
description="$(aquaclaw_heartbeat_default_description)"

delivery_args=(--no-deliver)
if [[ "$announce" -eq 1 ]]; then
  resolved_channel="$(aquaclaw_resolve_delivery_target_field channel 2>/dev/null || true)"
  resolved_session_key="$(aquaclaw_resolve_delivery_target_field session-key 2>/dev/null || true)"
  resolved_to="$(aquaclaw_resolve_delivery_target_field to 2>/dev/null || true)"
  resolved_account_id="$(aquaclaw_resolve_delivery_target_field account-id 2>/dev/null || true)"

  if [[ -z "${delivery_session_key}" ]]; then
    delivery_session_key="${resolved_session_key}"
  fi

  if [[ -z "${delivery_channel}" ]]; then
    if [[ -n "${resolved_channel}" ]]; then
      delivery_channel="${resolved_channel}"
    elif [[ -n "${delivery_session_key}" ]]; then
      delivery_channel="last"
    fi
  fi

  if [[ -z "${target_to}" ]]; then
    target_to="${resolved_to}"
  fi

  if [[ -z "${account_id}" ]]; then
    account_id="${resolved_account_id}"
  fi

  if [[ -z "${delivery_channel}" ]]; then
    echo "could not resolve a delivery channel from OpenClaw direct sessions or Telegram allowFrom fallback" >&2
    exit 1
  fi

  if [[ -z "${delivery_session_key}" && -z "${target_to}" ]]; then
    echo "could not resolve a delivery destination from OpenClaw direct sessions or Telegram allowFrom fallback" >&2
    exit 1
  fi

  delivery_args=(--announce --channel "$delivery_channel")
  if [[ -n "${delivery_session_key}" ]]; then
    delivery_args+=(--session-key "$delivery_session_key")
  fi
  if [[ -n "${target_to}" ]]; then
    delivery_args+=(--to "$target_to")
  fi
  if [[ -n "${account_id}" ]]; then
    delivery_args+=(--account "$account_id")
  fi
fi

if existing_json="$(aquaclaw_find_job_json "$job_name" 2>/dev/null)"; then
  job_id="$(JOB_JSON="$existing_json" node -e 'const job = JSON.parse(process.env.JOB_JSON); process.stdout.write(String(job.id ?? ""));')"
  if [[ -z "$job_id" ]]; then
    echo "existing job named ${job_name} has no usable id" >&2
    exit 1
  fi

  edit_cmd=(
    openclaw cron edit "$job_id"
    --every "$interval"
    --session "$session_target"
    --wake next-heartbeat
    --light-context
    --thinking "$thinking_level"
    --timeout-seconds "$timeout_seconds"
    --description "$description"
    --message "$message"
    "${delivery_args[@]}"
  )

  if [[ "$enable_after_create" -eq 1 ]]; then
    edit_cmd+=(--enable)
  else
    edit_cmd+=(--disable)
  fi

  if [[ "$replace_existing" -ne 1 ]]; then
    echo "job already exists:" >&2
    echo "$existing_json" >&2
    echo "rerun with --replace to patch it, or inspect it with show-openclaw-heartbeat-cron.sh" >&2
    exit 1
  fi

  if [[ "$apply" -eq 1 ]]; then
    aquaclaw_run_cron_command "${edit_cmd[@]}"
  else
    aquaclaw_print_command "${edit_cmd[@]}"
  fi

  exit 0
fi

add_cmd=(
  openclaw cron add
  --name "$job_name"
  --every "$interval"
  --session "$session_target"
  --wake next-heartbeat
  --light-context
  --thinking "$thinking_level"
  --timeout-seconds "$timeout_seconds"
  --description "$description"
  --message "$message"
  "${delivery_args[@]}"
)

if [[ "$enable_after_create" -ne 1 ]]; then
  add_cmd+=(--disabled)
fi

if [[ "$apply" -eq 1 ]]; then
  aquaclaw_run_cron_command "${add_cmd[@]}"
else
  aquaclaw_print_command "${add_cmd[@]}"
fi