#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck disable=SC1091
source "${SCRIPT_DIR}/common.sh"

STEP="disable"
SYSTEMD_DEST_DIR="/etc/systemd/system"
UNITS=(
  "blitz-watchdog.service"
  "blitz-5g-link-logger.service"
  "blitz-b-side-omnid.service"
  "blitz-ros-receiver.service"
  "blitz-5g-dial.service"
  "blitz-run-context.service"
  "blitz-boot-gate.service"
  "blitz-robot.target"
)

stop_unit_if_present() {
  local unit_name="$1"
  local unit_path="${SYSTEMD_DEST_DIR}/${unit_name}"

  if [[ ! -f "${unit_path}" ]]; then
    return 0
  fi
  blitz_run "${STEP}" "stop-unit" systemctl stop "${unit_name}" || true
}

disable_unit_if_present() {
  local unit_name="$1"
  local unit_path="${SYSTEMD_DEST_DIR}/${unit_name}"

  if [[ ! -f "${unit_path}" ]]; then
    return 0
  fi
  blitz_run "${STEP}" "disable-unit" systemctl disable "${unit_name}" || true
}

blitz_load_boot_env
blitz_require_root "${STEP}"
blitz_require_command systemctl "${STEP}"

for unit_name in "${UNITS[@]}"; do
  stop_unit_if_present "${unit_name}"
done

for unit_name in "${UNITS[@]}"; do
  disable_unit_if_present "${unit_name}"
done

blitz_log "${STEP}" "complete" "success" "boot chain stopped and disabled; next reboot will not auto-start blitz services" 0