#!/usr/bin/env bash
# Set up headless streaming mode: install prep-cmd hook scripts into a stable
# location and ensure they're executable. The actual sunshine.conf entries
# (capture=wlr, output_name=HEADLESS-1, global_prep_cmd=[...]) are written
# by lib/config.sh.

HEADLESS_BIN_DIR="$HOME/.local/share/omarchy-moonlight/bin"

DO_SCRIPT="$HEADLESS_BIN_DIR/sunshine-stream-do.sh"
UNDO_SCRIPT="$HEADLESS_BIN_DIR/sunshine-stream-undo.sh"
export DO_SCRIPT UNDO_SCRIPT

install_headless_hooks() {
  # Install hook scripts to ~/.local/share so they don't disappear if the
  # repo gets moved or deleted. Sunshine's config will reference these stable paths.
  mkdir -p "$HEADLESS_BIN_DIR"
  install -m 0755 "$SCRIPT_DIR/bin/sunshine-stream-do.sh"   "$DO_SCRIPT"
  install -m 0755 "$SCRIPT_DIR/bin/sunshine-stream-undo.sh" "$UNDO_SCRIPT"
  ok "Installed prep-cmd hooks to $HEADLESS_BIN_DIR"
}

# Install a systemd-user drop-in that pre-creates HEADLESS-1 before Sunshine
# starts, so the encoder probe at startup sees a valid Wayland output. Without
# this, Sunshine reports a fatal "Unable to find display or encoder during
# startup" on every restart, even though streaming works once a client connects.
install_headless_prestart_dropin() {
  local dropin_src="$SCRIPT_DIR/files/headless-prestart.conf"
  if [[ ! -f "$dropin_src" ]]; then
    err "Drop-in source missing: $dropin_src"
    return 1
  fi

  # Resolve the actual unit name. Prefer sunshine.service when present (alias
  # or sunshine-bin); fall back to the AUR source pkg's reverse-DNS name.
  local unit=""
  for u in sunshine.service app-dev.lizardbyte.app.Sunshine.service; do
    if systemctl --user list-unit-files "$u" >/dev/null 2>&1 \
       && systemctl --user cat "$u" >/dev/null 2>&1; then
      unit="$u"
      break
    fi
  done
  if [[ -z "$unit" ]]; then
    warn "Could not resolve a sunshine unit to attach the drop-in to; skipping."
    return 0
  fi

  local dropin_dir="$HOME/.config/systemd/user/${unit}.d"
  mkdir -p "$dropin_dir"
  install -m 0644 "$dropin_src" "$dropin_dir/headless-prestart.conf"
  systemctl --user daemon-reload
  ok "Installed headless prestart drop-in at $dropin_dir/headless-prestart.conf"
}