-- SwitchPlayerMon (core.asm:2419-2423), AnimateRetreatingPlayerMon (:1769-1796)
-- (#1563); pokeyellow core.asm:1862-1866 (#1545); core.asm:1471-1488 (#1608)
package.path = "./?.lua;./?/init.lua;" .. package.path

local T = require("tests.modkit")
local Data = T.fixtures.fresh()
require("src.render.Font").load(Data)
local BattleState = require("src.battle.BattleState")
local Pokemon = require("src.pokemon.Pokemon")
local SaveData = require("src.core.SaveData")
local TypeChart = require("src.battle.TypeChart")
TypeChart.load(Data)

local function newBattle()
  local save = SaveData.newGame()
  save.party = { Pokemon.new(Data, "FIXMON_A", 40), Pokemon.new(Data, "FIXMON_A", 40) }
  local game = { data = Data, save = save,
                 stack = { top = function() return nil end, push = function() end } }
  local battle = BattleState.newWild(game, "FIXMON_C", 40)
  battle.rng = function(a) if a then return a end return 0 end
  return battle
end

-- ---------------------------------------------------------------------
-- the shrink stages and their frame budget
-- ---------------------------------------------------------------------
do
  local battle = newBattle()
  battle.queue, battle.nextInsert = {}, 0
  battle:queueRetreatAnim()
  T.eq(#battle.queue, 2, "the retreat queues its start act plus the hold")
  T.eq(battle.queue[2].wait, 7, "4 frames at 5x5 then Delay3 at 3x3")
  T.check(battle:shrinkOutScale(battle.player) == nil, "nothing shrinks yet")
  battle.queue[1].fn()
  T.eq(battle.shrinkOut.battler, battle.player, "the outgoing pic is the one drawn")
  T.eq(battle:shrinkOutScale(battle.player), 5 / 7, "wDownscaledMonSize 0 -> 5x5")
  battle.shrinkOut.frame = 3
  T.eq(battle:shrinkOutScale(battle.player), 5 / 7, "for four frames")
  battle.shrinkOut.frame = 4
  T.eq(battle:shrinkOutScale(battle.player), 3 / 7, "wDownscaledMonSize 1 -> 3x3")
  battle.shrinkOut.frame = 6
  T.eq(battle:shrinkOutScale(battle.player), 3 / 7, "through Delay3")
  battle.shrinkOut.frame = 7
  T.eq(battle:shrinkOutScale(battle.player), 0,
    "then the 7x7 area holds cleared: no full-size flash before the swap")
  T.check(battle:shrinkOutScale({}) == nil,
    "the swapped-in battler draws normally")
  T.check(battle:shrinkOutScale(battle.enemy) == nil,
    "AnimateRetreatingPlayerMon is player-side only")
end

-- ---------------------------------------------------------------------
-- the Yellow starter slides off instead of shrinking
-- ---------------------------------------------------------------------
do
  local battle = newBattle()
  battle.starterPikachuSendOut = function() return true end
  battle.queue, battle.nextInsert = {}, 0
  battle:queueRetreatAnim()
  T.eq(#battle.queue, 3, "start act, the slide's hold, then the slot clear")
  battle.queue[1].fn()
  local slide = battle.picOff and battle.picOff.playerMon
  T.check(slide ~= nil, "the back pic slot is sliding")
  T.eq(slide.to, -64, "8 tiles off the left edge")
  T.eq(slide.hold, 3, "wSlideMonDelay 3 V-blanks per tile")
  T.eq(battle.queue[2].wait, 24, "8 tiles x 3 frames")
  T.check(battle:shrinkOutScale(battle.player) == nil, "and no downscale stage")
  battle.queue[3].fn()
  T.check((battle.picOff or {}).playerMon == nil, "the slot clears afterwards")
  T.eq(battle.sendingOut, true,
    "and stays hidden until the swap's send-out (#1545)")
end

-- ---------------------------------------------------------------------
-- resolveSwitch runs the retreat between the withdraw text and the swap
-- ---------------------------------------------------------------------
do
  local battle = newBattle()
  battle.queue, battle.nextInsert = {}, 0
  local outgoing = battle.player.mon
  battle:resolveSwitch(battle.game.save.party[2])
  local withdraw = table.remove(battle.queue, 1)
  battle.nextInsert = 0
  withdraw.fn()
  T.check(battle.queue[1] and battle.queue[1].text ~= nil, "RetreatMon text first")
  T.check(battle.queue[2] and battle.queue[2].fn ~= nil, "then the retreat start")
  T.eq(battle.queue[3] and battle.queue[3].wait, 7, "then its hold")
  T.eq(battle.player.mon, outgoing, "the party slot has not been swapped yet")
end

-- ---------------------------------------------------------------------
-- a fainted pick reopens the party list (#1608)
-- ---------------------------------------------------------------------
do
  local battle = newBattle()
  battle.queue, battle.nextInsert = {}, 0
  battle.buildScreen = function(_, _, opts) return opts end
  battle:openParty()
  local opts = battle.queue[1].ui()
  battle.queue, battle.nextInsert = {}, 0
  battle.game.save.party[2].hp = 0
  opts.onSwitch(battle.game.save.party[2])
  T.check(battle.queue[1] and battle.queue[1].text
          and battle.queue[1].text:find("no will", 1, true) ~= nil,
    "HasMonFainted prints NoWillText")
  T.check(battle.queue[2] and battle.queue[2].fn ~= nil, "and queues the reprompt")
  battle.queue[2].fn()
  T.check(battle.queue[3] and battle.queue[3].ui ~= nil,
    "GoBackToPartyMenu puts the list back up")

  battle.queue, battle.nextInsert = {}, 0
  opts.onSwitch(battle.player.mon)
  T.check(battle.queue[1] and battle.queue[1].text
          and battle.queue[1].text:find("already out", 1, true) ~= nil,
    "AlreadyOutText for the mon that is already out")
  T.check(battle.queue[2] and battle.queue[2].fn ~= nil, "which also reprompts")
end

-- ---------------------------------------------------------------------
-- the SHIFT prompt's picker reprompts on a dead or already-out pick too
-- (HasMonFainted's NoWillText, core.asm:1473-1488) (#1608)
-- ---------------------------------------------------------------------
do
  local save = SaveData.newGame()
  save.player.name = "RED"
  save.party = { Pokemon.new(Data, "FIXMON_A", 30),
                 Pokemon.new(Data, "FIXMON_B", 30) }
  save.options = { battleStyle = "shift" }
  local game = { data = Data, save = save,
                 stack = { top = function() return nil end,
                           push = function() end, pop = function() end } }
  local battle = BattleState.newTrainer(game, "OPP_FIX_YOUNGSTER", 1)
  battle.participants = {}
  battle.buildScreen = function(_, _, opts) return opts end
  -- capture the picker opts the SHIFT branch pushes
  local captured
  Data.screens = Data.screens or {}
  Data.screens.PartyMenu = function(_, opts) captured = opts; return {} end
  require("src.ui.Screens").invalidate()
  battle.enemyParty[1].hp = 0
  battle.enemy.mon = battle.enemyParty[1]
  battle:enemyMonFainted()
  local choiceRow
  for _, row in ipairs(battle.queue) do
    if row.choice then choiceRow = row end
  end
  T.check(choiceRow ~= nil, "SHIFT queues the change-POKeMON choice")
  choiceRow.choice(true)
  T.check(captured ~= nil and captured.forceSwitch == true,
    "YES opens the forced party picker")

  battle.queue, battle.nextInsert = {}, 0
  save.party[2].hp = 0
  captured.onSwitch(save.party[2])
  T.check(battle.queue[1] and battle.queue[1].text
          and battle.queue[1].text:find("no will", 1, true) ~= nil,
    "a fainted SHIFT pick prints NoWillText first")
  T.check(battle.queue[2] and battle.queue[2].ui ~= nil,
    "then the picker goes straight back up, ahead of the send-out")
  T.eq(battle.queue[2].ui(), captured, "with the same forced opts")

  battle.queue, battle.nextInsert = {}, 0
  captured.onSwitch(battle.player.mon)
  T.check(battle.queue[1] and battle.queue[1].text
          and battle.queue[1].text:find("already out", 1, true) ~= nil,
    "an already-out SHIFT pick prints AlreadyOutText")
  T.check(battle.queue[2] and battle.queue[2].ui ~= nil, "and reprompts too")

  battle.queue, battle.nextInsert = {}, 0
  save.party[2].hp = 10
  captured.onSwitch(save.party[2])
  T.eq(#battle.queue, 0, "a healthy pick queues no reprompt rows")
  Data.screens.PartyMenu = nil
  require("src.ui.Screens").invalidate()
end

T.finish("retreat animation and switch reprompt (#1563, #1545, #1608)")