#!/bin/sh
set -eu

BASE_DIR="$(cd "$(dirname "$0")" && pwd)"
PYTHON_BIN="${PYTHON_BIN:-python3}"
FLASHER_PY="$BASE_DIR/flash-edl.py"

for arg in "$@"; do
  if [ "$arg" = "-h" ] || [ "$arg" = "--help" ]; then
    exec "$PYTHON_BIN" "$FLASHER_PY" "$@"
  fi
done

if [ "$(id -u)" -ne 0 ]; then
  if command -v sudo >/dev/null 2>&1; then
    echo "[*] Elevating privileges via sudo for raw USB EDL (05c6:9008) access..."
    exec sudo "$0" "$@"
  else
    echo "[!] Warning: Not running as root and 'sudo' not found. Continuing..."
  fi
fi

MM_WAS_ACTIVE=0
if command -v systemctl >/dev/null 2>&1; then
  if systemctl is-active --quiet ModemManager 2>/dev/null; then
    echo "[*] Temporarily stopping host ModemManager service during EDL flash..."
    systemctl stop ModemManager 2>/dev/null || true
    MM_WAS_ACTIVE=1
  fi
elif command -v sv >/dev/null 2>&1 && [ -e /var/service/ModemManager ]; then
  if sv status ModemManager 2>/dev/null | grep -q "^run:"; then
    echo "[*] Temporarily stopping host ModemManager service during EDL flash..."
    sv stop ModemManager 2>/dev/null || true
    MM_WAS_ACTIVE=2
  fi
fi

QCSERIAL_CONF="/run/modprobe.d/bananarchy-edl.conf"
if [ "$(id -u)" -eq 0 ]; then
  mkdir -p /run/modprobe.d 2>/dev/null || true
  printf "blacklist qcserial\ninstall qcserial /bin/true\n" > "$QCSERIAL_CONF" 2>/dev/null || true
  if lsmod 2>/dev/null | grep -q "^qcserial "; then
    echo "[*] Unloading host qcserial kernel module so it does not claim 05c6:9008..."
    modprobe -r qcserial 2>/dev/null || rmmod qcserial 2>/dev/null || true
  fi
  if [ -w /sys/module/usbcore/parameters/autosuspend ]; then
    echo -1 > /sys/module/usbcore/parameters/autosuspend 2>/dev/null || true
  fi
  for pctrl in /sys/bus/usb/devices/usb*/power/control; do
    [ -w "$pctrl" ] && echo on > "$pctrl" 2>/dev/null || true
  done
  udevadm control --reload 2>/dev/null || true
fi

cleanup() {
  if [ -f "$QCSERIAL_CONF" ]; then
    rm -f "$QCSERIAL_CONF" 2>/dev/null || true
    udevadm control --reload 2>/dev/null || true
  fi
  if [ -n "${SUDO_UID:-}" ] && [ -n "${SUDO_GID:-}" ] && [ -d "$BASE_DIR/backup" ]; then
    chown -R "$SUDO_UID:$SUDO_GID" "$BASE_DIR/backup" 2>/dev/null || true
  fi
  if [ "$MM_WAS_ACTIVE" -eq 1 ]; then
    echo "[*] Restarting host ModemManager service..."
    systemctl start ModemManager 2>/dev/null || true
  elif [ "$MM_WAS_ACTIVE" -eq 2 ]; then
    echo "[*] Restarting host ModemManager service..."
    sv start ModemManager 2>/dev/null || true
  fi
}
trap cleanup EXIT INT TERM

"$PYTHON_BIN" "$FLASHER_PY" "$@"