#!/bin/bash

set -euo pipefail

source "$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)/base-test.sh"

test_tmp=$(mktemp -d)
trap 'rm -rf "$test_tmp"' EXIT

stub_bin="$test_tmp/bin"
acpi_lid="$test_tmp/acpi/button/lid"
input_class="$test_tmp/input"
dmi_chassis="$test_tmp/chassis_type"
mkdir -p "$stub_bin" "$acpi_lid/LID0" "$input_class"

busctl_log="$test_tmp/busctl.log"
cat >"$stub_bin/busctl" <<'SH'
#!/bin/bash
printf '%s\n' "$*" >>"$BUSCTL_LOG"
[[ ${OMARCHY_TEST_LID_STATE:-} != "fail" ]] || exit 1
printf '%s\n' "${OMARCHY_TEST_LID_STATE:-}"
SH
chmod +x "$stub_bin/busctl"

run_laptop() {
  OMARCHY_DMI_CHASSIS_TYPE_PATH="$dmi_chassis" \
    OMARCHY_ACPI_LID_PATH="$acpi_lid" \
    OMARCHY_INPUT_CLASS_PATH="$input_class" \
    PATH="$stub_bin:$PATH" \
    "$ROOT/bin/omarchy-hw-laptop"
}

run_lid_closed() {
  OMARCHY_ACPI_LID_PATH="$acpi_lid" \
    OMARCHY_INPUT_CLASS_PATH="$input_class" \
    BUSCTL_LOG="$busctl_log" \
    PATH="$stub_bin:$PATH" \
    OMARCHY_TEST_LID_STATE="$1" \
    "$ROOT/bin/omarchy-hw-laptop-closed"
}

# Input devices as sysfs shows them: a name and a switch capability bitmap
# whose last word holds SW_LID in bit 0.
input_device() {
  mkdir -p "$input_class/$1/capabilities"
  printf '%s\n' "$2" >"$input_class/$1/name"
  printf '%s\n' "$3" >"$input_class/$1/capabilities/sw"
}

# No ACPI lid, no lid switch, desktop chassis: a Mac mini, a Studio or a tower.
: >"$acpi_lid/LID0/state"
rm -f "$acpi_lid/LID0/state"
printf '3\n' >"$dmi_chassis"
input_device input0 "Power Button" 0
input_device input1 "Apple SMC power/lid events" 0
input_device input2 "MacBook Pro J416 Headphone Jack" 14
if run_laptop; then
  fail "a desktop without a lid is not classified as a laptop"
fi
pass "a desktop without a lid is not classified as a laptop, even with the SMC's power device"

# Apple Silicon MacBook: no ACPI lid, but the SMC reports a lid switch.
input_device input1 "Apple SMC power/lid events" 1
run_laptop || fail "an Apple SMC lid switch is classified as a laptop"
pass "an Apple SMC lid switch is classified as a laptop"

input_device input1 "Lid Switch" "0 1"
run_laptop || fail "a lid switch in the low word of a longer bitmap is classified as a laptop"
pass "a lid switch in the low word of a longer bitmap is classified as a laptop"

input_device input1 "Lid Switch" "1 0"
if run_laptop; then
  fail "a switch bit in a higher word is not a lid"
fi
pass "a switch bit in a higher word is not a lid"

# x86 laptop via DMI when ACPI and evdev lids are absent.
input_device input1 "Power Button" 0
printf '9\n' >"$dmi_chassis"
run_laptop || fail "DMI laptop chassis is recognized when ACPI is absent"
pass "DMI laptop chassis is recognized when ACPI is absent"

printf '3\n' >"$dmi_chassis"
if run_laptop; then
  fail "a desktop DMI chassis is not classified as a laptop"
fi
pass "a desktop DMI chassis is not classified as a laptop"

printf 'closed\n' >"$acpi_lid/LID0/state"
run_laptop || fail "an ACPI lid is classified as a laptop"
pass "an ACPI lid is classified as a laptop"

# An ACPI lid answers for itself, as it always has, without asking logind over
# D-Bus: this runs in sudo's PAM stack where fingerprint login is set up.
: >"$busctl_log"
run_lid_closed 'b false' ||
  fail "an ACPI lid reports itself closed"
pass "an ACPI lid reports itself closed"

printf 'open\n' >"$acpi_lid/LID0/state"
if run_lid_closed 'b true'; then
  fail "an ACPI lid reports itself open"
fi
pass "an ACPI lid reports itself open"
[[ ! -s $busctl_log ]] || fail "an ACPI lid leaves logind unasked" "$(cat "$busctl_log")"
pass "an ACPI lid leaves logind unasked"

# Without an ACPI lid, logind answers only for a lid switch it can track: it
# says "open" of a machine it tracks no lid on.
rm -f "$acpi_lid/LID0/state"
input_device input1 "Power Button" 0
if run_lid_closed 'b true'; then
  fail "a machine without a lid is never closed"
fi
[[ ! -s $busctl_log ]] || fail "a machine without a lid leaves logind unasked" "$(cat "$busctl_log")"
pass "a machine without a lid is never closed and leaves logind unasked"

input_device input1 "Apple SMC power/lid events" 1
run_lid_closed 'b true' ||
  fail "logind recognizes a closed lid switch"
pass "logind recognizes a closed lid switch"
[[ $(head -1 "$busctl_log") == "--timeout=2 "*" LidClosed" ]] ||
  fail "logind gets a short timeout, not busctl's 25 s" "$(cat "$busctl_log")"
pass "logind gets a short timeout, not busctl's 25 s"

if run_lid_closed 'b false'; then
  fail "logind recognizes an open lid switch"
fi
pass "logind recognizes an open lid switch"

input_device input1 "Lid Switch" "0 1"
run_lid_closed 'b true' ||
  fail "a lid switch in the low word of a longer bitmap is asked about"
pass "a lid switch in the low word of a longer bitmap is asked about"

for answer in '' 'not-a-property' 'fail'; do
  if run_lid_closed "$answer"; then
    fail "a logind that does not answer leaves the lid open" "$answer"
  fi
done
pass "a logind that does not answer leaves the lid open"