#!/bin/sh
# POSIX-only, /bin/sh compatible, ASCII-only
# Dry-run by default; use --write to apply changes
# Logs: $LOG_DIR, Plan: $PLAN_OUT
set -u

# Load env
if [ -f "/etc/synomap/synomap.env" ]; then
  . /etc/synomap/synomap.env
fi

QB_URL="${QB_URL:-}"; QB_USER="${QB_USER:-}"; QB_PASS="${QB_PASS:-}"
MAP_FILE="${MAP_FILE:-}"; DEST_ROOT="${DEST_ROOT:-}"
QB_CATEGORIES="${QB_CATEGORIES:-sonarr,radarr}"
QB_EXCLUDE_CATEGORIES="${QB_EXCLUDE_CATEGORIES:-hum}"
TAG_DONE="${TAG_DONE:-SYNO}"; PLAN_OUT="${PLAN_OUT:-/root/synomap_plan.txt}"
LOG_DIR="${LOG_DIR:-/var/log/synomap}"; LOCK_DIR="${LOCK_DIR:-/var/lock/synomap}"
UMASK_VAL="${UMASK:-002}"
QB_TIMEOUT="${QB_TIMEOUT:-10}"
QB_FETCH_LIMIT="${QB_FETCH_LIMIT:-500}"
QB_FETCH_FACTOR="${QB_FETCH_FACTOR:-20}"

umask "$UMASK_VAL"; mkdir -p "$LOG_DIR" "$LOCK_DIR" 2>/dev/null || true

DRY=1; WRITE=0; PLAN=""; LIMIT=""
log(){ printf "%s %s\n" "$(date +%Y-%m-%dT%H:%M:%S)" "$*" | tee -a "$LOG_DIR/synomap.log"; }
nok(){ log "[NOK] $*"; }; ok(){ log "[OK]  $*"; }; note(){ log "[NOTE] $*"; }

qb_cookie=""
qb_login(){
  qb_cookie="$(mktemp)"
  curl -m "$QB_TIMEOUT" -sS -c "$qb_cookie" \
    --data-urlencode "username=$QB_USER" \
    --data-urlencode "password=$QB_PASS" \
    "$QB_URL/api/v2/auth/login" >/dev/null 2>&1 || return 1
  return 0
}
qb_api(){
  m="$1"; shift; p="$1"; shift
  if [ "$m" = "GET" ]; then
    curl -m "$QB_TIMEOUT" -sS -b "$qb_cookie" "$QB_URL$p" 2>/dev/null
  else
    curl -m "$QB_TIMEOUT" -sS -b "$qb_cookie" -X POST "$QB_URL$p" "$@" 2>/dev/null
  fi
}
qb_info_many(){
  lim="$QB_FETCH_LIMIT"
  if [ -n "$LIMIT" ]; then
    case "$LIMIT" in *[!0-9]*|"") : ;; *) lim=$((LIMIT * QB_FETCH_FACTOR));; esac
  fi
  qb_api GET "/api/v2/torrents/info?limit=$lim"
}
qb_set_location(){ qb_api POST "/api/v2/torrents/setLocation" --data "hashes=$1" --data-urlencode "location=$2"; }
qb_add_tags(){ qb_api POST "/api/v2/torrents/addTags" --data "hashes=$1" --data-urlencode "tags=$2"; }
qb_recheck(){ qb_api POST "/api/v2/torrents/recheck" --data "hashes=$1"; }
poll_state(){
  h="$1"; tm="${2:-120}"; i=0
  while [ "$i" -lt "$tm" ]; do
    st="$(qb_api GET "/api/v2/torrents/info?hashes=$h" | tr -d '\r\n')"
    state="$(printf "%s" "$st" | sed -n 's/.*\"state\":\"\([^\"]*\)\".*/\1/p')"
    [ -n "$state" ] && log "[STATE] $h = $state"
    case "$state" in
      checking*|queuedForChecking) : ;;
      *) break ;;
    esac
    sleep 2; i=$((i+2))
  done
}

while [ $# -gt 0 ]; do
  case "$1" in
    --write) DRY=0; WRITE=1; shift;;
    --dry-run) DRY=1; WRITE=0; shift;;
    --plan) PLAN="$2"; shift 2;;
    --limit) LIMIT="$2"; shift 2;;
    *) note "arg ignored: $1"; shift;;
  esac
done

# synomap_miniplan_pick.sh
# Output: hash|src_save_path|dst_path|category

[ -n "$PLAN" ] && PLAN_OUT="$PLAN"

if [ ! -f "$MAP_FILE" ]; then
  nok "MAP_FILE missing: $MAP_FILE"; exit 0
fi
ok "Loading mapping: $MAP_FILE"
M="$(mktemp)"; grep -v '^[[:space:]]*$' "$MAP_FILE" | grep '|' > "$M" || true

ok "Login to qBittorrent"
if ! qb_login; then nok "qB login failed"; rm -f "$M"; exit 0; fi

J="$(qb_info_many)"
if [ -z "$J" ]; then nok "qB info empty (timeout or zero results)"; rm -f "$M" "$qb_cookie"; exit 0; fi

tmp="$(mktemp)"; : > "$tmp"

if command -v jq >/dev/null 2>&1; then
  # JQ path: filter categories, ignore already tagged SYNO
  printf "%s" "$J" | jq -cr '
    .[]
    | {h:.hash, sp:.save_path, c:.category, tags:(.tags // "")}
    | select((.c=="sonarr" or .c=="radarr") and (.tags | contains("SYNO") | not))
    | [.h, .sp, .c] | @tsv
  ' | while IFS=$(printf "\t") read -r h sp catg; do
      dst=""
      while IFS='|' read -r mcat msrc mdst; do
        [ "$catg" = "$mcat" ] || continue
        case "$sp" in
          "$msrc"*) rel="${sp#$msrc}"; dst="$mdst$rel"; break;;
          *) : ;;
        esac
      done < "$M"
      [ -n "$dst" ] && printf "%s|%s|%s|%s\n" "$h" "$sp" "$dst" "$catg" >> "$tmp"
    done
else
  # Fallback sed (less robust)
  echo "[NOTE] jq absent, fallback sed JSON parsing" >&2
  printf "%s" "$J" | tr -d '\n\r' | sed 's/},{/}\n{/g' | while IFS= read -r obj; do
    h="$(printf "%s" "$obj" | sed -n 's/.*\"hash\":\"\([^\"]*\)\".*/\1/p' | head -n1)"
    sp="$(printf "%s" "$obj" | sed -n 's/.*\"save_path\":\"\([^\"]*\)\".*/\1/p' | head -n1)"
    catg="$(printf "%s" "$obj" | sed -n 's/.*\"category\":\"\([^\"]*\)\".*/\1/p' | head -n1)"
    tags="$(printf "%s" "$obj" | sed -n 's/.*\"tags\":\"\([^\"]*\)\".*/\1/p' | head -n1)"
    [ -z "$h" ] && continue
    [ -z "$sp" ] && continue
    [ -z "$catg" ] && catg=""
    case ",$tags," in *",SYNO,"*) continue;; esac
    dst=""
    while IFS='|' read -r mcat msrc mdst; do
      [ "$catg" = "$mcat" ] || continue
      case "$sp" in
        "$msrc"*) rel="${sp#$msrc}"; dst="$mdst$rel"; break;;
        *) : ;;
      esac
    done < "$M"
    [ -n "$dst" ] && printf "%s|%s|%s|%s\n" "$h" "$sp" "$dst" "$catg" >> "$tmp"
  done
fi

count="$(wc -l < "$tmp" | tr -d ' ')"
[ -n "$LIMIT" ] && [ "$LIMIT" -gt 0 ] && head -n "$LIMIT" "$tmp" > "$tmp.head" && mv "$tmp.head" "$tmp"

[ "$DRY" -eq 1 ] && ok "Plan preview ($count items):" && sed -n '1,50p' "$tmp"
mv "$tmp" "$PLAN_OUT"; ok "Plan written to $PLAN_OUT"
rm -f "$M" "$qb_cookie"
exit 0