-- Party menu: list the party, choose a member.
-- Modes:
--   default: A -> submenu (STATS / SWITCH order / CANCEL)
--   opts.onSwitch: A -> hand the chosen mon to the callback (battle
--                  switch, item targeting via opts.pickOnly)
--   opts.onCancel: fired when the menu closes without a pick (B)
-- Pops itself on B.

local Font = require("src.render.Font")

local PartyMenu = {}
PartyMenu.__index = PartyMenu
PartyMenu.isOpaque = true

-- SGB: generic whole-screen palette (SET_PAL_GENERIC)
function PartyMenu:sgbPalettes(game)
  return require("src.render.PaletteFX").wholeNamed(game.data, "MEWMON")
end

local CURSOR = 0xED

-- where DIG escapes work: escape_rope_tilesets.asm (Agatha's room is
-- excluded by map id in ItemUseEscapeRope)
local DIG_TILESETS = { FOREST = true, CEMETERY = true, CAVERN = true,
                       FACILITY = true, INTERIOR = true }

-- Party mon icons (engine/gfx/mon_icons.asm AnimatePartyMon): only the
-- SELECTED mon's icon animates, at a speed set by its HP bar color --
-- 5 / 16 / 32 frames per phase for green / yellow / red (the famous
-- health-speed detail).  BALL and HELIX icons nudge one pixel down
-- instead of switching frames; every other icon swaps to a real second
-- frame (+ICONOFFSET).

-- Rest/alt frame per icon (data/icon_pointers.asm
-- MonPartySpritePointers): the base entries are the RESTING frame,
-- the +ICONOFFSET entries the animated alternate.  The 16x32 icon
-- sheets stack Frame1 (index 0) over Frame2 (index 1, INC_FRAME_2):
-- BUG/GRASS rest on BugIconFrame2/PlantIconFrame2 and animate to
-- Frame1; SNAKE/QUADRUPED are the reverse.  Sprite-reused icons draw
-- from 16x16x6 overworld sheets where index 3 is walk-down (tile 12):
-- MON/FAIRY/BIRD rest on the walk frame and animate to standing
-- (tile 0); WATER (Seel) is the reverse.
PartyMenu.iconFrames = {
  BUG       = { rest = 1, alt = 0 }, -- BugIconFrame2 <-> BugIconFrame1
  GRASS     = { rest = 1, alt = 0 }, -- PlantIconFrame2 <-> PlantIconFrame1
  SNAKE     = { rest = 0, alt = 1 }, -- SnakeIconFrame1 <-> SnakeIconFrame2
  QUADRUPED = { rest = 0, alt = 1 }, -- QuadrupedIconFrame1 <-> Frame2
  MON       = { rest = 3, alt = 0 }, -- MonsterSprite tile 12 <-> tile 0
  FAIRY     = { rest = 3, alt = 0 }, -- FairySprite tile 12 <-> tile 0
  BIRD      = { rest = 3, alt = 0 }, -- BirdSprite tile 12 <-> tile 0
  WATER     = { rest = 0, alt = 3 }, -- SeelSprite tile 0 <-> tile 12
}

-- Which 16x16 frame of `name`'s sheet to draw; `ih` (sheet pixel
-- height) only matters for the fallback, which keeps the old uniform
-- behavior for icons outside the table (BALL/HELIX y-bob instead).
function PartyMenu.frameFor(name, alt, ih)
  local m = PartyMenu.iconFrames[name]
  if m then return alt and m.alt or m.rest end
  return alt and ((ih or 0) >= 64 and 3 or 1) or 0
end

local iconImages = {}
local function drawIcon(game, mon, x, y, selected, counter)
  local icons = game.data.icons
  if not icons then return end
  local def = game.data.pokemon[mon.species]
  local name = def and def.dex and icons.byDex[def.dex]
  local path = name and icons.icons[name]
  if not path then return end
  if iconImages[path] == nil then
    local ok, img = pcall(love.graphics.newImage, path)
    iconImages[path] = ok and img or false
  end
  local img = iconImages[path]
  if not img then return end
  local alt = false
  if selected then
    local px = math.floor(mon.hp * 48 / math.max(1, mon.stats.hp))
    local speed = px >= 27 and 5 or px >= 10 and 16 or 32
    alt = math.floor(counter / speed) % 2 == 1
  end
  if alt and (name == "BALL" or name == "HELIX") then
    y = y + 1
    alt = false
  end
  local iw, ih = img:getDimensions()
  if ih > 16 then
    local frame = PartyMenu.frameFor(name, alt, ih)
    love.graphics.draw(img, love.graphics.newQuad(0, frame * 16, 16, 16, iw, ih), x, y)
  else
    love.graphics.draw(img, x, y)
  end
end

function PartyMenu.new(game, opts)
  opts = opts or {}
  local self = setmetatable({}, PartyMenu)
  self.game = game
  self.index = 1
  self.onSwitch = opts.onSwitch
  self.onCancel = opts.onCancel
  self.pickOnly = opts.pickOnly
  self.battle = opts.battle
  self.party = opts.party -- link battles pass their clamped copies
  self.swapFrom = nil
  self.submenu = nil
  self.subIndex = 1
  self.blink = 0
  return self
end

function PartyMenu:update(dt)
  -- icon animation counter; 320 = a whole cycle at every HP speed
  self.blink = ((self.blink or 0) + 1) % 320
  local input = self.game.input
  local party = self.party or self.game.save.party

  if self.submenu then
    local n = #self.subItems
    if input:wasPressed("up") then
      self.subIndex = self.subIndex > 1 and self.subIndex - 1 or n
    elseif input:wasPressed("down") then
      self.subIndex = self.subIndex < n and self.subIndex + 1 or 1
    elseif input:wasPressed("b") then
      self.submenu = nil
    elseif input:wasPressed("a") then
      local mon = party[self.index]
      local action = self.subItems[self.subIndex].action
      if action == "stats" then
        local SummaryMenu = require("src.ui.SummaryMenu")
        self.game.stack:push(SummaryMenu.new(self.game, mon))
      elseif action == "switch" then
        self.swapFrom = self.index
      elseif action == "fly" then
        local FlyMenu = require("src.ui.FlyMenu")
        self.game.stack:pop() -- close the party menu
        self.game.stack:push(FlyMenu.new(self.game))
        return
      elseif action == "flash" then -- FLASH lights dark tunnels
        -- start_sub_menus.asm .flash: PrintText _FlashLightsAreaText, then
        -- GBPalWhiteOutWithDelay3 + jp .goBackToMap
        local ow = self.game.overworld
        local TextBox = require("src.render.TextBox")
        local Transition = require("src.render.Transition")
        self.game.stack:pop()
        ow.dark = false
        self.game.save.flashLit = true
        self.game.stack:push(TextBox.new(self.game,
          self.game.data.text._FlashLightsAreaText
          or "A blinding FLASH\nlights the area!", function()
            self.game.stack:push(Transition.whiteFlash(self.game))
          end))
        return
      elseif action == "surf" then
        -- start_sub_menus.asm .surf: SOULBADGE-gated (checked at list time
        -- above), then IsSurfingAllowed (the Cycling Road / Seafoam B4F
        -- current refusals, both of which loop back to the submenu), then
        -- ItemUseSurfboard: while surfing it tries to dismount instead;
        -- otherwise it mounts only if the FACING tile is water, else
        -- SurfingAttemptFailed (_NoSurfingHereText) loops back to the
        -- submenu.  useSurfFieldMove reports which; trySurf does the mount.
        local ow = self.game.overworld
        local reason = ow:useSurfFieldMove()
        local Transition = require("src.render.Transition")
        if reason == "ok" then
          self.game.stack:pop() -- close the party menu (jp .goBackToMap)
          local fx, fy = ow.player:facingCell()
          ow:trySurf(fx, fy)
          return
        end
        if reason == "dismount" then
          -- ItemUseSurfboard .stopSurfing: no text -- the walking state
          -- and music return first (PlayDefaultMusic +
          -- LoadWalkingPlayerSpriteGraphics), the menu closes with the
          -- GBPalWhiteOutWithDelay3 blink, and the simulated pad press
          -- steps the player forward onto land
          self.game.stack:pop()
          ow.player.surfing = false
          require("src.core.Music").setSurfing(self.game.data, false)
          self.game.stack:push(Transition.whiteFlash(self.game, nil, function()
            ow:scriptMove(ow.player, ow.player.facing, 1)
          end))
          return
        end
        local TextBox = require("src.render.TextBox")
        local def = self.game.data.pokemon[mon.species]
        local key = ({ no_badge = "_NewBadgeRequiredText",
                       forced_bike = "_CyclingIsFunText",
                       current = "_CurrentTooFastText",
                       no_place = "_SurfingNoPlaceToGetOffText" })[reason]
                    or "_NoSurfingHereText"
        local txt = (self.game.data.text[key] or "No SURFing here!")
                    :gsub("{RAM:wNameBuffer}", mon.nickname or def.name)
        if reason == "no_place" then
          -- .cannotStopSurfing prints _SurfingNoPlaceToGetOffText but
          -- never zeroes wActionResultOrTookBattleTurn, so unlike the
          -- other refusals the menu still closes afterwards
          -- (GBPalWhiteOutWithDelay3 + .goBackToMap)
          self.game.stack:pop()
          self.game.stack:push(TextBox.new(self.game, txt, function()
            self.game.stack:push(Transition.whiteFlash(self.game))
          end))
          return
        end
        self.game.stack:push(TextBox.new(self.game, txt))
        return -- .loop: submenu stays open behind the message
      elseif action == "cut" then
        -- start_sub_menus.asm .cut -> predef UsedCut (engine/overworld/cut.asm):
        -- CASCADEBADGE-gated (list time); _NothingToCutText loops back to the
        -- submenu when the FACING tile isn't a cuttable tree.
        local ow = self.game.overworld
        local reason = ow:useCutFieldMove()
        if reason == "ok" then
          self.game.stack:pop() -- close the party menu (CloseTextDisplay)
          local fx, fy = ow.player:facingCell()
          ow:tryCut(fx, fy)
          return
        end
        local TextBox = require("src.render.TextBox")
        local def = self.game.data.pokemon[mon.species]
        local key = (reason == "no_badge") and "_NewBadgeRequiredText"
                                            or "_NothingToCutText"
        local txt = (self.game.data.text[key] or "Nothing to CUT!")
                    :gsub("{RAM:wNameBuffer}", mon.nickname or def.name)
        self.game.stack:push(TextBox.new(self.game, txt))
        return -- .loop: submenu stays open behind the message
      elseif action == "strength" then
        -- start_sub_menus.asm .strength: RAINBOWBADGE-gated (list time);
        -- predef PrintStrengthText (field_move_messages.asm) sets
        -- BIT_STRENGTH_ACTIVE of wStatusFlags1 -- the sole gate
        -- push_boulder.asm reads -- then prints _UsedStrengthText (no
        -- prompt: after the text, the text_asm tail plays the chosen
        -- mon's cry, Delay3, and it auto-advances) and
        -- _CanMoveBouldersText (`prompt`: waits for A/B).  Back in
        -- .strength, GBPalWhiteOutWithDelay3 blinks the screen white
        -- before CloseTextDisplay returns to the map.
        local ow = self.game.overworld
        local TextBox = require("src.render.TextBox")
        local Transition = require("src.render.Transition")
        local def = self.game.data.pokemon[mon.species]
        local name = mon.nickname or def.name
        self.game.stack:pop() -- close the party menu (jp .goBackToMap)
        ow.strengthActive = true
        local t1 = (self.game.data.text._UsedStrengthText
          or "{RAM:wNameBuffer} used\nSTRENGTH."):gsub("{RAM:wNameBuffer}", name)
        local t2 = (self.game.data.text._CanMoveBouldersText
          or "{RAM:wNameBuffer} can\nmove boulders."):gsub("{RAM:wNameBuffer}", name)
        self.game.stack:push(TextBox.new(self.game, t1, function()
          self.game.stack:push(TextBox.new(self.game, t2, function()
            self.game.stack:push(Transition.whiteFlash(self.game))
          end))
        end, { auto = { sound = function()
          return require("src.core.Sound").playCry(self.game.data, mon.species)
        end } }))
        return
      elseif action == "softboiled" then
        -- field SOFTBOILED (StartMenu_Pokemon .softboiled): transfer
        -- 1/5 of the user's max HP to a chosen teammate
        self.softboiledFrom = self.index
      elseif action == "escape" then
        -- DIG / TELEPORT both warp to the last Pokémon Center town
        -- (wLastBlackoutMap, special_warps.asm escape warp); .dig/.teleport
        -- end with GBPalWhiteOutWithDelay3 + jp .goBackToMap
        local ow = self.game.overworld
        local heal = self.game.save.lastHeal
        local Transition = require("src.render.Transition")
        self.game.stack:pop()
        if ow and heal then
          self.game.stack:push(Transition.whiteFlash(self.game, nil, function()
            require("src.core.Sound").play(self.game.data, "Teleport_Exit1")
            ow:warpToHealPoint()
          end))
        end
        return
      end
      self.submenu = nil
    end
    return
  end

  if input:wasPressed("up") then
    self.index = self.index > 1 and self.index - 1 or math.max(1, #party)
  elseif input:wasPressed("down") then
    self.index = self.index < #party and self.index + 1 or 1
  elseif input:wasPressed("b") then
    self.game.stack:pop()
    if self.onCancel then self.onCancel() end
  elseif input:wasPressed("a") and #party > 0 then
    local mon = party[self.index]
    if self.softboiledFrom then
      local user = party[self.softboiledFrom]
      local heal = math.floor(user.stats.hp / 5)
      if mon == user or mon.hp <= 0 or mon.hp >= mon.stats.hp
         or user.hp <= heal then
        self.softboiledFrom = nil
        local TextBox = require("src.render.TextBox")
        self.game.stack:push(TextBox.new(self.game, "It won't have\nany effect."))
      else
        user.hp = user.hp - heal
        mon.hp = math.min(mon.stats.hp, mon.hp + heal)
        self.softboiledFrom = nil
        require("src.core.Sound").play(self.game.data, "Heal_HP")
        local def = self.game.data.pokemon[mon.species]
        local TextBox = require("src.render.TextBox")
        self.game.stack:push(TextBox.new(self.game,
          ("%s's HP\nwas restored!"):format(mon.nickname or def.name)))
      end
    elseif self.swapFrom then
      if self.swapFrom ~= self.index then
        party[self.swapFrom], party[self.index] = party[self.index], party[self.swapFrom]
        require("src.core.Sound").play(self.game.data, "Swap")
      end
      self.swapFrom = nil
    elseif self.onSwitch then
      self.game.stack:pop()
      self.onSwitch(mon)
    else
      self.submenu = true
      self.subIndex = 1
      -- STATS/SWITCH plus this mon's field moves (start_sub_menus.asm
      -- builds the same dynamic list)
      self.subItems = { { label = "STATS", action = "stats" },
                        { label = "SWITCH", action = "switch" } }
      local ow = self.game.overworld
      if not self.battle and ow and mon.hp > 0 then
        for _, mv in ipairs(mon.moves) do
          if mv.id == "FLY" and ow.map.def.tileset == "OVERWORLD"
             and self.game.save.inventory.THUNDERBADGE then
            table.insert(self.subItems, { label = "FLY", action = "fly" })
          elseif mv.id == "FLASH" and ow.dark
             and self.game.save.inventory.BOULDERBADGE then
            table.insert(self.subItems, { label = "FLASH", action = "flash" })
          elseif mv.id == "CUT" and self.game.save.inventory.CASCADEBADGE then
            -- CUT/SURF/STRENGTH are party-menu field moves too
            -- (start_sub_menus.asm .outOfBattleMovePointers); listed here
            -- with the same list-time badge filter this file already uses
            -- for FLY/FLASH.  The facing-tile/activation check happens on
            -- selection (useCutFieldMove/useSurfFieldMove).
            table.insert(self.subItems, { label = "CUT", action = "cut" })
          elseif mv.id == "SURF" and self.game.save.inventory.SOULBADGE then
            table.insert(self.subItems, { label = "SURF", action = "surf" })
          elseif mv.id == "STRENGTH" and self.game.save.inventory.RAINBOWBADGE then
            table.insert(self.subItems, { label = "STRENGTH", action = "strength" })
          elseif mv.id == "SOFTBOILED" then
            table.insert(self.subItems, { label = "SOFTBOILED", action = "softboiled" })
          elseif mv.id == "TELEPORT" and ow.map.def.tileset == "OVERWORLD" then
            -- TELEPORT works only OUTDOORS (start_sub_menus.asm
            -- .teleport -> CheckIfInOutsideMap); dark maps don't
            -- block it
            table.insert(self.subItems, { label = "TELEPORT", action = "escape" })
          elseif mv.id == "DIG" and DIG_TILESETS[ow.map.def.tileset]
             and ow.map.id ~= "AGATHAS_ROOM" then
            -- DIG runs ItemUseEscapeRope (.dig sets wCurItem =
            -- ESCAPE_ROPE): usable in the dungeon tilesets of
            -- escape_rope_tilesets.asm minus Agatha's room, even in
            -- the dark (Rock Tunnel)
            table.insert(self.subItems, { label = "DIG", action = "escape" })
          end
        end
      end
    end
  end
end

function PartyMenu:draw()
  love.graphics.setColor(1, 1, 1, 1)
  love.graphics.rectangle("fill", 0, 0, 160, 144)
  love.graphics.setColor(0, 0, 0, 1)
  local party = self.party or self.game.save.party
  if #party == 0 then
    Font.draw("No POKéMON!", 16, 64)
  end
  local HudTiles = require("src.render.HudTiles")
  for i, mon in ipairs(party) do
    local def = self.game.data.pokemon[mon.species]
    local y = (i - 1) * 16 + 12
    love.graphics.setColor(1, 1, 1, 1)
    drawIcon(self.game, mon, 8, y - 2, i == self.index, self.blink or 0)
    love.graphics.setColor(0, 0, 0, 1)
    Font.draw(mon.nickname or def.name, 24, y)
    -- level at column 13 (<LV> tile + digits, PrintLevel) AND the
    -- status/FNT text at column 17 (PrintStatusCondition), like the
    -- original rows -- statused mons keep their level display
    if mon.level < 100 then
      HudTiles.tile(0x6E, 104, y) -- <LV>
      Font.draw(tostring(mon.level), 112, y)
    else
      -- PrintLevel overwrites the <LV> tile with the third digit
      Font.draw(tostring(mon.level), 104, y)
    end
    if mon.hp <= 0 then
      Font.draw("FNT", 136, y)
    elseif mon.status then
      Font.draw(mon.status, 136, y)
    end
    -- the colored tile HP bar (DrawHP2 + SetPartyMenuHPBarColor)
    love.graphics.setColor(1, 1, 1, 1)
    HudTiles.drawHPBar(self.game.data, 5, (y + 8) / 8, mon)
    love.graphics.setColor(0, 0, 0, 1)
    Font.draw(("%3d/%3d"):format(mon.hp, mon.stats.hp), 104, y + 8)
    if i == self.index then
      Font.drawCode(CURSOR, 0, y)
    end
    if i == self.swapFrom or i == self.softboiledFrom then
      Font.drawCode(0xEC, 0, y) -- the unfilled swap arrow
    end
  end
  if self.swapFrom then
    Font.draw("Move to where?", 8, 136)
  elseif self.softboiledFrom then
    Font.draw("Use on which one?", 8, 136)
  elseif self.pickOnly then
    Font.draw("Use on which one?", 8, 136)
  end
  if self.submenu then
    local n = #self.subItems
    Font.drawBox(9, 17 - n * 2 - 1, 11, n * 2 + 1)
    local y0 = (17 - n * 2) * 8
    for si, entry in ipairs(self.subItems) do
      Font.draw(entry.label, 88, y0 + (si - 1) * 16)
    end
    Font.drawCode(CURSOR, 80, y0 + (self.subIndex - 1) * 16)
  end
  love.graphics.setColor(1, 1, 1, 1)
end

return PartyMenu