#!/usr/bin/env bash
#
# (Re)write /usr/local/bin/klonkt-update so it matches how this server runs.
#
# Why this exists: the updater is generated once at install time. A server
# that later migrated to the split layout (klonkt@<slug> units) kept its old
# updater, which still restarts the retired klonkt.service. Result: the code
# on disk updates, the restart quietly fails, and the old process keeps
# serving — half old routes, half new templates, which is how you get a 500
# on one page and nothing in the logs that says why.
#
# Idempotent; safe to run any time:
#
#     sudo bash /opt/klonkt/scripts/klonkt-refresh-updater.sh
#
set -euo pipefail

KLONKT_DIR="${KLONKT_DIR:-/opt/klonkt}"
KLONKT_USER="${KLONKT_USER:-klonkt}"
DATA_ROOT="${KLONKT_DATA_ROOT:-/var/lib/klonkt}"
# Follow whatever branch the checkout is on (stable for most self-hosters).
BRANCH="${KLONKT_BRANCH:-$(git -C "$KLONKT_DIR" rev-parse --abbrev-ref HEAD 2>/dev/null || echo stable)}"

[ "$(id -u)" = 0 ] || { echo "run this as root (sudo)." >&2; exit 1; }
[ -d "$KLONKT_DIR/.git" ] || { echo "no git checkout at $KLONKT_DIR" >&2; exit 1; }

cat > /usr/local/bin/klonkt-update <<EOF
#!/usr/bin/env bash
set -euo pipefail
D="${KLONKT_DIR}"
B=\$(runuser -u ${KLONKT_USER} -- git -C "\$D" rev-parse HEAD 2>/dev/null || true)
# Deepen a shallow checkout ONCE. \`fetch --depth 1\` keeps only the newest commit
# and \`checkout -f\` throws away the tree that was there, so after an update the
# previous version existed nowhere on the machine: a bad release could not be
# undone without the network, and only if you knew which commit to ask for.
# One deepening buys back the history; every later fetch keeps it.
if [ "\$(runuser -u ${KLONKT_USER} -- git -C "\$D" rev-parse --is-shallow-repository 2>/dev/null)" = "true" ]; then
  echo "deepening the checkout once so updates can be rolled back..."
  runuser -u ${KLONKT_USER} -- git -C "\$D" fetch --unshallow origin ${BRANCH} || true
fi
runuser -u ${KLONKT_USER} -- git -C "\$D" fetch origin ${BRANCH}
runuser -u ${KLONKT_USER} -- git -C "\$D" checkout -qf -B ${BRANCH} FETCH_HEAD
A=\$(runuser -u ${KLONKT_USER} -- git -C "\$D" rev-parse HEAD)
if [ "\$B" = "\$A" ]; then
  echo "Klonkt is already up to date (\$A) — nothing to do."
  exit 0
fi
if ! runuser -u ${KLONKT_USER} -- git -C "\$D" diff --quiet "\$B" "\$A" -- package-lock.json 2>/dev/null; then
  runuser -u ${KLONKT_USER} -- env HOME="\$D" bash -c "cd '\$D' && npm ci --omit=dev"
fi
# Restart every instance sharing this checkout: one directory with an .env
# under the data root per instance. No instances there = the pre-split
# single-service layout, which still runs plain klonkt.service.
N=0
for d in ${DATA_ROOT}/*/; do
  [ -f "\$d/.env" ] || continue
  s=\$(basename "\$d")
  systemctl restart "klonkt@\$s" && N=\$((N+1))
done
if [ "\$N" = 0 ]; then
  systemctl restart klonkt
  echo "Klonkt updated (\$A) + restarted."
else
  echo "Klonkt updated (\$A) + restarted \$N instance(s)."
fi
# History alone is not a rollback; at three in the morning you also need the
# command. Print it while the previous commit is still known.
if [ -n "\$B" ]; then
  echo
  echo "Previous version: \$B"
  echo "To go back:"
  echo "  runuser -u ${KLONKT_USER} -- git -C \$D checkout -qf -B ${BRANCH} \$B"
  echo "  then restart the instances (systemctl restart 'klonkt@*')"
fi
EOF
chmod +x /usr/local/bin/klonkt-update
echo "klonkt-update rewritten: branch ${BRANCH}, code ${KLONKT_DIR}, instances under ${DATA_ROOT}"

# On a split install the old single unit must not be startable. `disable` alone
# does not stop `systemctl restart klonkt` from starting it, and a resurrected
# klonkt.service has no .env (it moved with the data): it falls back to the
# defaults and writes a fresh empty database into the checkout.
SPLIT=0
for d in "${DATA_ROOT}"/*/; do [ -f "$d/.env" ] && SPLIT=1 && break; done
if [ "$SPLIT" = 1 ] && systemctl list-unit-files klonkt.service >/dev/null 2>&1; then
  if ! systemctl is-enabled klonkt.service 2>/dev/null | grep -q masked; then
    systemctl stop klonkt.service 2>/dev/null || true
    systemctl disable klonkt.service 2>/dev/null || true
    systemctl mask klonkt.service
    echo "retired klonkt.service: stopped, disabled and masked (unmask to roll back)"
  fi
fi

# A leftover storage/ in the checkout means something ran without the instance
# config. Report it; never delete it unattended — only its owner can tell
# whether it holds anything.
if [ "$SPLIT" = 1 ] && [ -e "${KLONKT_DIR}/storage" ]; then
  echo
  echo "WARNING: ${KLONKT_DIR}/storage exists while instance data lives in ${DATA_ROOT}."
  echo "         Something ran without the instance .env and wrote here. Check with:"
  echo "             sqlite3 ${KLONKT_DIR}/storage/database.sqlite 'select count(*) from posts;'"
  echo "         If it is empty, it is a stray from a resurrected klonkt.service and"
  echo "         can be removed. If it is NOT empty, do not delete it: ask first."
fi