## Pure unit guard for the `TR_VBULLET_ADMIT_ONLY` spawn-gate predicate.
##
## No Java, no battle, no virtual bullets: `vBulletAdmitted` is a pure function of
## (gunId, rack mode, membership table, admitOnly) over the SHIPPED
## `DefaultRackMembership`, so the live gate can be pinned without running the
## bot. Covers:
##   * the shipped default rack admits Pattern (5) and nothing else;
##   * `admitOnly=false` (TR_VBULLET_ADMIT_ONLY=0) restores "spawn for every gun";
##   * mode-restricted membership (`rmOnly1v1` / `rmOnlyMelee`) flips with mode;
##   * an empty membership table filters nothing (pre-change callers);
##   * out-of-range ids are admitted, negative ids are not;
##   * the TMPATTERN special case: forcing `admitOnly=true` keeps gun 14 gated
##     even when the knob is off — the exact composition the live loop uses.
##
## Run with plain:
##   nim c -r common_libs/tests/test_vbullet_admit_gate.nim

import gun_harness/virtual_bullets
import gun_harness/selector

var failures = 0
proc check(name: string, ok: bool) =
  if ok: echo "PASS: ", name
  else: echo "FAIL: ", name; inc failures

proc testShippedDefaultRack() =
  ## The shipped onlyPattern table: Pattern (5) admitted in both modes, every
  ## other registered gun (0..14) off.
  check "default: Pattern admitted in 1v1",
        vBulletAdmitted(5, rm1v1, DefaultRackMembership, true)
  check "default: Pattern admitted in melee",
        vBulletAdmitted(5, rmMelee, DefaultRackMembership, true)
  var allOthersOff = true
  for gi in 0..<len(DefaultRackMembership):
    if gi == 5: continue
    if vBulletAdmitted(gi, rm1v1, DefaultRackMembership, true): allOthersOff = false
    if vBulletAdmitted(gi, rmMelee, DefaultRackMembership, true): allOthersOff = false
  check "default: every other gun (incl. TMPATTERN 14) is NOT admitted",
        allOthersOff

proc testKnobOffRestoresFullRack() =
  ## TR_VBULLET_ADMIT_ONLY=0: the gate is a no-op for the general guns, i.e.
  ## every id spawns regardless of rack membership.
  var allAdmitted = true
  for gi in 0..<len(DefaultRackMembership):
    if not vBulletAdmitted(gi, rm1v1, DefaultRackMembership, false): allAdmitted = false
    if not vBulletAdmitted(gi, rmMelee, DefaultRackMembership, false): allAdmitted = false
  check "knob=0: vBulletAdmitted is true for EVERY gun in both modes", allAdmitted

proc testModeRestrictedMembership() =
  # 0: both, 1: only1v1, 2: onlyMelee, 3: off
  let mem = [rmBoth, rmOnly1v1, rmOnlyMelee, rmOff]
  check "1v1: both + only1v1 admitted, onlyMelee/off not",
        vBulletAdmitted(0, rm1v1, mem, true) and
        vBulletAdmitted(1, rm1v1, mem, true) and
        (not vBulletAdmitted(2, rm1v1, mem, true)) and
        (not vBulletAdmitted(3, rm1v1, mem, true))
  check "melee: both + onlyMelee admitted, only1v1/off not",
        vBulletAdmitted(0, rmMelee, mem, true) and
        vBulletAdmitted(2, rmMelee, mem, true) and
        (not vBulletAdmitted(1, rmMelee, mem, true)) and
        (not vBulletAdmitted(3, rmMelee, mem, true))
  check "knob=0 overrides even rmOff / mode restrictions",
        vBulletAdmitted(3, rm1v1, mem, false) and
        vBulletAdmitted(1, rmMelee, mem, false)

proc testEmptyAndOutOfRange() =
  let empty: seq[RackMembership] = @[]
  check "empty membership filters nothing (pre-change callers)",
        vBulletAdmitted(0, rm1v1, empty, true) and
        vBulletAdmitted(13, rmMelee, empty, true)
  let mem = [rmOff, rmOff]
  check "id past the membership table is admitted",
        vBulletAdmitted(7, rm1v1, mem, true)
  check "negative id is not admitted when the gate is on",
        (not vBulletAdmitted(-1, rm1v1, mem, true))

proc testTmPatternStaysGatedWhenKnobOff() =
  ## The live loop composes the TMPATTERN gate as
  ##   vBulletAdmitted(14, mode, mem, VBulletAdmitOnly or id == TmPatternId)
  ## so gun 14 stays gated even when TR_VBULLET_ADMIT_ONLY=0.
  let knobOff = false
  for mode in [rm1v1, rmMelee]:
    let composed = vBulletAdmitted(14, mode, DefaultRackMembership,
                                   knobOff or 14 == 14)
    let alwaysGated = vBulletAdmitted(14, mode, DefaultRackMembership, true)
    check "knob=0 keeps TMPATTERN (14) gated exactly as when gated",
          composed == alwaysGated and composed == false

testShippedDefaultRack()
testKnobOffRestoresFullRack()
testModeRestrictedMembership()
testEmptyAndOutOfRange()
testTmPatternStaysGatedWhenKnobOff()

if failures > 0:
  echo "\n", failures, " check(s) FAILED"
  quit(1)
echo "\nAll vbullet-admit-gate checks passed."