#!/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"

# Build category normalization map and (optional) mapping overrides
# Accepts:
#  - CATEGORY|SRC_ROOT|DST_ROOT
#  - CATEGORY|anyPathUnderSRC_ROOT   (infer SRC_ROOT=/data/torrents/completed/<cat>, DST_ROOT=$DEST_ROOT/<cat>)
# Lines beginning with # or empty are ignored.
if [ -f "$MAP_FILE" ]; then
  ok "Loading mapping: $MAP_FILE"
  while IFS= read -r line; do
    case "$line" in ""|"#"* ) continue;; esac
    cat="$(printf "%s" "$line" | cut -d'|' -f1 | tr '[:upper:]' '[:lower:]')"
    nf=$(printf "%s" "$line" | awk -F'|' '{print NF}')
    if [ "$nf" -ge 3 ]; then
      src="$(printf "%s" "$line" | cut -d'|' -f2)"
      dst="$(printf "%s" "$line" | cut -d'|' -f3-)"
      eval "SRC_$cat=\"$src\""
      eval "DST_$cat=\"$dst\""
    elif [ "$nf" -ge 2 ]; then
      eval "SRC_$cat=\"/data/torrents/completed/$cat\""
      eval "DST_$cat=\"$DEST_ROOT/$cat\""
    fi
  done < "$MAP_FILE"
else
  nok "MAP_FILE missing: $MAP_FILE"; exit 0
fi

ok "Login to qBittorrent"
if ! qb_login; then nok "qB login failed"; exit 0; fi

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

incl="$(printf "%s" "$QB_CATEGORIES" | tr '[:upper:]' '[:lower:]' | tr ',' ' ')"
excl="$(printf "%s" "$QB_EXCLUDE_CATEGORIES" | tr '[:upper:]' '[:lower:]' | tr ',' ' ')"

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

emit_line(){
  h="$1"; sp="$2"; catg="$3"
  cl="$(printf "%s" "$catg" | tr '[:upper:]' '[:lower:]')"
  eval "sr=\${SRC_$cl:-/data/torrents/completed/$cl}"
  eval "ds=\${DST_$cl:-$DEST_ROOT/$cl}"
  case "$sp" in
    "$sr"*) rel="${sp#$sr}"; out="$ds$rel" ;;
    *) rel="" ; out="" ;;
  esac
  [ -n "$out" ] && printf "%s|%s|%s|%s\n" "$h" "$sp" "$out" "$cl" >> "$tmp"
}

if command -v jq >/dev/null 2>&1; then
  printf "%s" "$J" | jq -cr '.[] | {h:.hash, sp:.save_path, c:(.category//""), tags:(.tags//"")} | @base64' \
  | while read -r row; do
      obj="$(printf "%s" "$row" | base64 -d)"
      h="$(printf "%s" "$obj" | jq -r '.h')"
      sp="$(printf "%s" "$obj" | jq -r '.sp')"
      c0="$(printf "%s" "$obj" | jq -r '.c')"
      t="$(printf "%s" "$obj" | jq -r '.tags')"
      [ -z "$h" ] || [ -z "$sp" ] && continue
      cl="$(printf "%s" "$c0" | tr '[:upper:]' '[:lower:]')"
      case ",$t," in *",$TAG_DONE,"*) continue;; esac
      skip=0; for x in $excl; do [ "$cl" = "$x" ] && skip=1 && break; done; [ "$skip" -eq 1 ] && continue
      want=0; for c in $incl; do [ "$cl" = "$c" ] && want=1 && break; done; [ "$want" -eq 0 ] && continue
      emit_line "$h" "$sp" "$cl"
    done
else
  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)"
    c0="$(printf "%s" "$obj" | sed -n 's/.*\"category\":\"\([^\"]*\)\".*/\1/p' | head -n1)"
    t="$(printf "%s" "$obj" | sed -n 's/.*\"tags\":\"\([^\"]*\)\".*/\1/p' | head -n1)"
    [ -z "$h" ] && continue
    [ -z "$sp" ] && continue
    cl="$(printf "%s" "$c0" | tr '[:upper:]' '[:lower:]')"
    case ",$t," in *",$TAG_DONE,"*) continue;; esac
    skip=0; for x in $excl; do [ "$cl" = "$x" ] && skip=1 && break; done; [ "$skip" -eq 1 ] && continue
    want=0; for c in $incl; do [ "$cl" = "$c" ] && want=1 && break; done; [ "$want" -eq 0 ] && continue
    emit_line "$h" "$sp" "$cl"
  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 "$qb_cookie"
exit 0