-- Probe: the PACK opened from a real Gold battle menu, on a real overworld.
--
--   POKEPORT_GAME=gold POKEPORT_DRIVER=tests/drivers/gold_battle_pack_probe.lua love .
--
-- engine/items/pack.asm:627
-- ../pokegold/engine/items/pack.asm:810
local U = require("tests.drivers.util")

local Mon = require("src.battle.gen2.Mon")
local PackMenu = require("src.ui.gen2.PackMenu")

return function(game)
  local out = os.getenv("POKEPORT_SHOT_DIR") or "/tmp/gold-battle-pack"

  local function tap(button, frames)
    game.input.pressQueue[#game.input.pressQueue + 1] = button
    game.input.state[button] = true
    U.wait(2)
    game.input.state[button] = false
    U.wait(frames or 6)
  end

  U.wait(45)
  local world = game.world
  assert(world and world.map, "gold world did not boot")

  local player = Mon.new(game.data, "CYNDAQUIL", 12)
  game.save.party = { player }
  game.save.inventory = { POKE_BALL = 5, POTION = 3, ITEMFINDER = 1,
    NORMAL_BOX = 1 }

  local wild = Mon.new(game.data, "PIDGEY", 4)
  assert(world:startBattle({ wild = wild }), "startBattle failed")

  -- DoBattleTransition owns the screen first; wBattleMode is only set when the
  -- battle screen itself goes on the stack.
  local battle
  for _ = 1, 600 do
    local top = game.stack:top()
    if top and top.battle then battle = top break end
    U.wait(1)
  end
  assert(battle, "battle screen never came up")
  assert(world.battleActive, "the world is not marked as in a battle")
  for _ = 1, 240 do
    if battle.phase == "menu" then break end
    tap("a", 3)
  end
  assert(battle.phase == "menu", "never reached the battle menu")

  -- The 2x2 grid: DOWN puts the cursor on PACK.
  tap("down")
  tap("a")
  local pack = game.stack:top()
  assert(getmetatable(pack) == PackMenu, "PACK did not open the pack")
  assert(pack:inBattle(), "the battle pack is not flagged as BattlePack")

  -- KEY ITEMS, then A on the ITEMFINDER.
  tap("right")
  tap("right")
  assert(pack:pocket().id == "KEY_ITEM",
    "did not reach the KEY ITEMS pocket: " .. tostring(pack:pocket().id))
  assert(pack.rows[1], "the key items pocket is empty")
  print("[driver] key item row 1 " .. tostring(pack.rows[1].id))
  tap("a")
  U.wait(4)
  U.shot(game, out .. "/00-battle-pack-unusable.png")

  assert(pack.submenu, "the ITEMFINDER did not open ItemSubmenu")
  assert(#pack.submenu.rows == 1 and pack.submenu.rows[1] == "quit",
    "the key item was offered more than .UnusableMenuHeader's QUIT")
  assert(pack:submenuColumn() == 0,
    "Gold drew the battle submenu in column " .. tostring(pack:submenuColumn()))
  assert(pack.message == nil, "OakThisIsntTheTimeText printed inside a battle")
  assert(game.stack:top() == pack, "the pack left the stack")
  assert(world.battleActive, "battleActive was cleared by a field effect")
  assert(world.queuedScript == nil, "a field script was queued from a battle")
  assert(game.save.inventory.ITEMFINDER == 1, "the key item was spent")

  tap("a")
  assert(pack.submenu == nil, "QUIT did not close the submenu")
  assert(game.save.inventory.ITEMFINDER == 1, "QUIT spent the key item")

  tap("left")
  tap("left")
  assert(pack:pocket().id == "ITEM",
    "did not reach the ITEMS pocket: " .. tostring(pack:pocket().id))
  for _ = 1, #pack.rows do
    if pack.rows[pack.index] and pack.rows[pack.index].id == "POTION" then break end
    tap("down")
  end
  assert(pack.rows[pack.index] and pack.rows[pack.index].id == "POTION",
    "never landed on the POTION")
  tap("a")
  U.wait(4)
  U.shot(game, out .. "/01-battle-pack-usable.png")
  assert(pack.submenu and #pack.submenu.rows == 2
    and pack.submenu.rows[1] == "use",
    "the POTION did not get .UsableMenuHeader's USE / QUIT")
  assert(game.stack:top() == pack, "USE ran before it was chosen")

  tap("b")
  tap("b")
  for _ = 1, 120 do
    if battle.phase == "menu" then break end
    U.wait(1)
  end
  assert(game.stack:top() == battle, "the battle is not back on top")
  assert(battle.phase == "menu", "the battle menu did not come back")
  U.shot(game, out .. "/01-battle-menu-back.png")

  print("[driver] PASS gold battle pack in " .. out)
end