-- Driver: reworked battle flows.
--  A) Old man catch tutorial (DisplayBattleMenu's old-man script):
--     scripted cursor FIGHT(80f) -> ITEM(50f), forced item menu with
--     one POKé BALL x50 -- itself scripted (list_menu.asm:65-80):
--     '▶' hover 80f, auto-A leaves the hollow '▷', always-caught throw.
--  B) Mimic's MID-move copy menu (MimicEffect): the chooser opens only
--     after the hit test, at (0,7) like MoveSelectionMenu .mimicmenu.
--  C) #162 trainer SHIFT offer: after KO'ing the first enemy mon,
--     "about to use" + "Will … change POKéMON?" with YES/NO.
return function(game)
  local U = dofile("tests/drivers/util.lua")
  local DIR = os.getenv("SHOT_DIR") or "/tmp/shots"
  local Pokemon = require("src.pokemon.Pokemon")
  local mon = Pokemon.new(game.data, "CHARMANDER", 12)
  mon.moves[1] = { id = "MIMIC", pp = 10 }
  table.insert(game.save.party, 1, mon)
  U.teleport(game, "ROUTE_1", 5, 5, "down")
  local ow = game.overworld
  local BattleState = require("src.battle.BattleState")

  local function waitFor(cond, max)
    for _ = 1, max or 600 do
      if cond() then return true end
      U.wait(1)
    end
    return false
  end
  local function mashUntil(cond, max)
    for _ = 1, max or 80 do
      if cond() then return true end
      U.tap(game, "a")
      U.wait(4)
    end
    return false
  end

  -- ------------------------------------------------ A) old man demo
  local demo = BattleState.newWild(game, "WEEDLE", 5)
  demo:makeOldManDemo()
  demo.onFinish = function() end
  ow:pushBattle(demo)
  -- through "Wild WEEDLE appeared!" to the scripted battle menu
  mashUntil(function() return demo.phase == "menu" and (demo.demoTimer or 0) > 5 end)
  U.shot(game, DIR .. "/oldman_0_cursor_fight.png") -- hand on FIGHT
  waitFor(function() return (demo.demoTimer or 131) > 95 end)
  U.shot(game, DIR .. "/oldman_1_cursor_item.png")  -- hand on ITEM
  waitFor(function() return game.stack:top() ~= demo end)
  U.wait(3)
  U.shot(game, DIR .. "/oldman_2_bag.png")          -- POKé BALL x50 list
  local bag = game.stack:top()
  U.tap(game, "b")                                   -- ignored: no backing out
  waitFor(function() return bag.hollowIndex ~= nil end)
  U.shot(game, DIR .. "/oldman_3_hollow_cursor.png") -- auto-A: hollow '▷'
  waitFor(function() return game.stack:top() == demo end) -- list down, throw
  U.wait(5)
  U.shot(game, DIR .. "/oldman_4_throw.png")        -- "OLD MAN used POKé BALL!"
  U.tap(game, "a")
  U.wait(25)
  for i = 5, 7 do
    U.wait(55)
    U.shot(game, ("%s/oldman_%d_catch.png"):format(DIR, i))
  end
  for _ = 1, 20 do U.tap(game, "a"); U.wait(6) end
  while game.stack:top() ~= ow do game.stack:pop() end
  U.wait(5)

  -- ------------------------------------------------ B) Mimic mid-move
  local battle = BattleState.newWild(game, "PIDGEY", 8)
  battle.onFinish = function() end
  battle.rng = function(a, b) return a end -- Mimic hits
  ow:pushBattle(battle)
  mashUntil(function() return battle.phase == "menu" end)
  U.shot(game, DIR .. "/mimic_0_menu.png")
  U.tap(game, "a"); U.wait(8)                        -- FIGHT
  U.shot(game, DIR .. "/mimic_1_moves.png")
  U.tap(game, "a"); U.wait(4)                        -- MIMIC
  mashUntil(function() return battle.phase == "mimicSelect" end)
  U.shot(game, DIR .. "/mimic_2_chooser.png")        -- copy menu at (0,7)
  U.tap(game, "down"); U.wait(4)
  U.shot(game, DIR .. "/mimic_3_chooser_down.png")
  U.tap(game, "a"); U.wait(30)
  U.shot(game, DIR .. "/mimic_4_learned.png")        -- anim + learned text
  mashUntil(function() return battle.phase == "moveSelect" end, 40)
  U.wait(2)
  U.shot(game, DIR .. "/mimic_5_moves_after.png")    -- slot now holds the copy
  while game.stack:top() ~= ow do game.stack:pop() end
  U.wait(5)

  -- ------------------------------------------------ C) SHIFT about-to-use
  -- Inject the foe-faint path at the menu so screenshots don't depend on
  -- accuracy rolls; party has 2 slots so TrainerAboutToUseText fires.
  local ChoiceBox = require("src.ui.ChoiceBox")
  local PartyMenu = require("src.ui.PartyMenu")
  game.save.options = game.save.options or {}
  game.save.options.battleStyle = "shift"
  game.save.options.animations = false
  game.save.party = {
    Pokemon.new(game.data, "CHARIZARD", 50),
    Pokemon.new(game.data, "BLASTOISE", 50),
  }
  local trainer = BattleState.newTrainer(game, "OPP_YOUNGSTER", 1)
  trainer.onFinish = function() end
  ow:pushBattle(trainer)
  mashUntil(function() return trainer.phase == "menu" end)
  U.shot(game, DIR .. "/shift_0_menu.png")
  trainer:markParticipant()
  trainer.enemy.mon.hp = 0
  trainer.enemyParty[1].hp = 0
  trainer.phase = "messages"
  trainer.afterQueue = "menu"
  trainer:onFaint(trainer.enemy)
  local function topIs(cls)
    return getmetatable(game.stack:top()) == cls
  end
  local function textDone(needle)
    local t = trainer.current and trainer.current.text
    return t and t:find(needle, 1, true)
      and (trainer.charIndex or 0) >= (trainer.total or 0)
  end
  mashUntil(function() return textDone("about to use") end, 200)
  U.shot(game, DIR .. "/shift_1_about_to_use.png")
  mashUntil(function() return textDone("change POKéMON") end, 80)
  U.wait(20) -- ChoiceBox auto-opens once the page has typed out
  mashUntil(function() return topIs(ChoiceBox) end, 40)
  U.shot(game, DIR .. "/shift_2_will_change.png")
  U.shot(game, DIR .. "/shift_3_yes_no.png")
  U.tap(game, "a"); U.wait(8)                        -- YES
  mashUntil(function() return topIs(PartyMenu) end, 40)
  U.shot(game, DIR .. "/shift_4_party.png")
  U.tap(game, "down"); U.wait(4)
  U.tap(game, "a"); U.wait(8)                        -- BLASTOISE
  mashUntil(function() return textDone("sent") end, 120)
  U.shot(game, DIR .. "/shift_5_enemy_sent.png")
  mashUntil(function() return textDone("BLASTOISE") end, 120)
  U.shot(game, DIR .. "/shift_6_player_sent.png")
  while game.stack:top() ~= ow do game.stack:pop() end
  U.wait(5)
  love.event.quit()
end