#!/usr/bin/env bash
# The whole job, in one place. Inputs (environment):
#   DRY_RUN      true|false   (true: text stages only, voice reachability, push-permission test, commits nothing to the branch)
#   PDF_INPUT    optional path relative to the repo root (manual runs). Empty = every PDF waiting in _incoming/
#   BRANCH       branch to commit back to
#   AUDIOBOOK_*  setting overrides (see _pipeline/settings.py)
set -uo pipefail
cd "${GITHUB_WORKSPACE:-$(git rev-parse --show-toplevel)}"
export PATH="$PWD/.venv/bin:$PATH"
DRY_RUN="${DRY_RUN:-false}"
BRANCH="${BRANCH:-main}"
git config user.name "audiobook-bot"
git config user.email "audiobook-bot@noreply.invalid"

# The checkout is sparse (code and _incoming only). Books are pulled in only when needed.
sparse_add() { printf '%s\n' "$@" | git sparse-checkout add --stdin; }
esc() { printf '%s' "$1" | sed -e 's/[][*?\\#!]/\\&/g'; }

pdfs=()
if [ -n "${PDF_INPUT:-}" ]; then
  sparse_add "/$(esc "$PDF_INPUT")"
  [ -f "$PDF_INPUT" ] || { echo "FAILED: no such file in the repo: $PDF_INPUT"; exit 1; }
  pdfs=("$PDF_INPUT")
else
  while IFS= read -r f; do pdfs+=("$f"); done < <(find _incoming -maxdepth 1 -type f -iname '*.pdf' | sort)
fi
if [ "${#pdfs[@]}" -eq 0 ]; then
  echo "Nothing to do: no PDF in _incoming/."
  exit 0
fi
echo "PDFs to process: ${#pdfs[@]}"; printf '  %s\n' "${pdfs[@]}"

push_with_retry() {   # push HEAD to the branch; if the branch moved, rebase and try again
  local i
  for i in 1 2 3 4 5; do
    if git push origin "HEAD:refs/heads/$BRANCH"; then return 0; fi
    echo "push attempt $i failed; fetching and rebasing"
    git fetch -q --depth=100 origin "$BRANCH" && git rebase FETCH_HEAD || { git rebase --abort 2>/dev/null; return 1; }
  done
  return 1
}

failed=0
for pdf in "${pdfs[@]}"; do
  echo; echo "################ $pdf"
  result="$(mktemp)"
  rel="$(python3 _pipeline/cli.py bookpath "$pdf")" || { echo "FAILED: $rel"; failed=1; continue; }
  echo "book folder: $rel"
  e="$(esc "$rel")"
  if git cat-file -e "HEAD:$rel" 2>/dev/null; then
    if [ "$DRY_RUN" = "true" ]; then   # dry run compares text only: skip audio
      sparse_add "/$e/book.json" "/$e/warnings.log" "/$e/LICENSE_STATUS.md" "/$e/text/" "/$e/script/" "/$e/figures/"
    else                               # a real run needs the existing audio so it can skip it
      sparse_add "/$e/"
    fi
    echo "this book already exists in the repo; existing files fetched"
  fi
  if [ "$DRY_RUN" = "true" ]; then
    export AUDIOBOOK_BOOKS_DIR="/tmp/audiobook-dry/books"; rm -rf /tmp/audiobook-dry
    mkdir -p /tmp/audiobook-dry
    args=(--dry-run --compare "$PWD/books")
  else
    unset AUDIOBOOK_BOOKS_DIR
    args=()
  fi
  python3 _pipeline/cli.py process "$pdf" "${args[@]}" --result-file "$result"
  rc=$?
  if [ $rc -ne 0 ]; then
    echo "FAILED: process exited $rc for $pdf. Nothing was committed. The PDF stays where it is."
    git checkout -q -- . 2>/dev/null; git clean -qfd -- books 2>/dev/null
    failed=1; continue
  fi
  [ "$DRY_RUN" = "true" ] && continue

  bookdir="$(python3 -c "import json,sys; print(json.load(open(sys.argv[1]))['book_dir'])" "$result")"
  case "$pdf" in
    _incoming/*)
      git rm -q -- "$pdf"
      side="${pdf%.[pP][dD][fF]}.json"; [ -f "$side" ] && git rm -q -- "$side"
      ;;
  esac
  git add -A -- "$bookdir"
  if git diff --cached --quiet; then echo "No changes to commit for $pdf"; continue; fi
  rel="${bookdir#"$PWD"/}"
  git commit -q -m "Audiobook: $rel (from $pdf, Actions run ${GITHUB_RUN_NUMBER:-?}) [skip ci]"
  git --no-pager show --stat --format='commit %H' HEAD | head -40
  if push_with_retry; then echo "PUSHED to $BRANCH: $rel"; else echo "FAILED: could not push to $BRANCH"; failed=1; fi
done

if [ "$DRY_RUN" = "true" ]; then
  echo; echo "################ push-permission test (throwaway branch, deleted afterwards)"
  tb="dryrun/run-${GITHUB_RUN_ID:-$$}"
  git checkout -q -b "$tb"
  git commit -q --allow-empty -m "dry-run push test [skip ci]"
  if git push origin "refs/heads/$tb:refs/heads/$tb"; then
    echo "push of $tb: OK"
    if git push origin --delete "$tb"; then echo "delete of $tb: OK"; else echo "FAILED: could not delete $tb"; failed=1; fi
  else
    echo "FAILED: the job token cannot push. Check Settings > Actions is enabled for the repo and the token has write access."
    failed=1
  fi
fi

[ $failed -eq 0 ] && echo "JOB OK" || echo "JOB FAILED"
exit $failed