gen1recomp/tools/save-editor/panels/SpeciesPicker.lua

-- Type-to-search species picker (#541).  The inspector used to change species
-- with a pair of arrows, which walked the catalog one entry at a time -- 151
-- taps to cross the dex -- and ran a full MonOps recalculation on whatever
-- record happened to be next, including records the Gen1 formulas cannot use.
-- This is the replacement: one modal list, filtered as you type, committing
-- through Ops.setSpecies so an unusable record refuses instead of crashing.
--
-- Modal is literal.  Kit hit-tests without a z-order, so App.draw raises
-- Kit.blockClicks over the chrome and the panel while this is open and lowers
-- it only for this overlay; nothing underneath can take the same tap.

local Theme = require("Theme")
local Ops = require("Ops")
local MonEditor = require("MonEditor")
local PickerChrome = require("PickerChrome")
local PAL = Theme.PAL

local Picker = {}

local FIELD_ID = "species-picker"

function Picker.results(S)
  local p = S.speciesPicker
  return Ops.speciesSearch(S, p and p.query or "")
end

-- One commit funnel for both of the picker's jobs: changing the inspected
-- mon's species, and the Boxes panel's add flow (mode "box-add"), which
-- creates a fresh Lv5 mon in the selected box instead (#715).  Either way an
-- unusable record refuses in the status bar rather than crashing (#541).
local function commit(S, id)
  local p = S.speciesPicker
  if p and p.mode == "box-add" then
    return Ops.boxAddSpecies(S, id)
  end
  return Ops.setSpecies(S, S.editingMon, id)
end

-- Enter commits the top match, which is the whole point of a search field.
function Picker.commitFirst(S, Kit)
  local hits = Picker.results(S)
  if not hits[1] then
    return Ops.say(S, "No species matches that")
  end
  local ok = commit(S, hits[1])
  if ok then
    Ops.closeSpeciesPicker(S, Kit)
  end
  return ok
end

function Picker.draw(S, Kit, width, height)
  local p = S.speciesPicker
  if not p then
    return
  end
  local s = Kit.scale

  -- The click that opened the picker is still the frame's click: the
  -- inspector dispatches earlier in App.draw than this overlay does, so
  -- without swallowing it the scrim below would read it as a tap outside and
  -- shut the picker in the same frame it went up.  App re-raises the shield
  -- at the top of the next frame, so leaving it up here is safe.
  if p.opened then
    p.opened = nil
    Kit.blockClicks = true
  end

  -- the scrim doubles as the "tap outside to cancel" target; it covers the
  -- full window so unsafe bands (notch / home indicator) stay dimmed too
  Theme.col(PAL.bgBot, 0.72)
  love.graphics.rectangle("fill", 0, 0, width, height)

  -- Card fills / centres in SafeArea so phones and RGxxx landscapes keep a
  -- usable list (#917 / #715).
  local x, y, w, h, pad = PickerChrome.card(Kit, width, height)
  if Kit.press(0, 0, width, height) and not Kit.hit(x, y, w, h) then
    Ops.closeSpeciesPicker(S, Kit)
    return
  end

  Kit.card(x, y, w, h)
  local cx, cy = x + pad, y + pad
  local inner = w - 2 * pad

  local closeW = PickerChrome.closeSize(Kit)
  local captionH = Kit.textHeight("caption")
  local headH = math.max(captionH, closeW)
  Kit.caption(
    cx,
    cy + (headH - captionH) / 2,
    p.mode == "box-add" and ("ADD TO BOX %d"):format(S.selectedBox or 1) or "CHOOSE A SPECIES"
  )
  if
    Kit.iconButton(
      x + w - pad - closeW,
      cy + (headH - closeW) / 2,
      closeW,
      closeW,
      "x",
      "Close picker"
    )
  then
    Ops.closeSpeciesPicker(S, Kit)
    return
  end
  cy = cy + headH + 10 * s

  local fieldH = PickerChrome.fieldH(Kit)
  p.query =
    Kit.textfield(FIELD_ID, cx, cy, inner, fieldH, p.query, "type a name, an id, or a dex number")
  cy = cy + fieldH + 10 * s

  local hits = Picker.results(S)
  local listH, rowH, rowGap, pagerH = PickerChrome.listMetrics(Kit, y, h, pad, cy)
  local perPage = math.max(1, math.floor((listH + rowGap) / (rowH + rowGap)))
  -- wheel / touch drag scroll the modal list too; the shield is already
  -- lowered for this layer, so Kit.scroll works here and only here (#715)
  local drawn, shift = Kit.list(p, "offset", cx, cy, inner, listH, #hits, rowH + rowGap)

  if #hits == 0 then
    Kit.emptyBox(cx, cy, inner, listH, "Nothing matches that.")
  else
    Kit.pushClip(cx, cy, inner, listH)
    for i = 1, drawn do
      local id = hits[p.offset + i]
      if not id then
        break
      end
      local ry = cy + (i - 1) * (rowH + rowGap) - shift
      local def = S.data.pokemon[id]
      -- A record the formulas cannot use still lists, greyed: hiding it would
      -- make a modded species look like it never registered (#541).
      local usable = Ops.speciesUsable(S, id)
      -- box-add has no "current" species: nothing is being replaced
      local current = p.mode ~= "box-add"
          and (S.editingMon and (S.editingMon.species == id or (def and (S.editingMon.species == def.speciesId or S.editingMon.speciesId == def.speciesId or S.editingMon.species == def.id))))
        or false
      if Kit.row(cx, ry, inner, rowH, current, PAL.green, 9 * s) then
        if commit(S, id) then
          Ops.closeSpeciesPicker(S, Kit)
          Kit.popClip()
          return
        end
      end
      local icon = rowH - 4 * s
      MonEditor.drawSprite(S, Kit, id, cx + 6 * s, ry + 2 * s, icon)
      local tx = cx + 6 * s + icon + 8 * s
      local tail = usable and ("#%03d"):format(tonumber(def and def.dex) or 0) or "no data"
      local tailW = Kit.textWidth("tiny", tail)
      Kit.text(
        "monoRow",
        Kit.ellipsize("monoRow", (def and def.name) or id, inner - (tx - cx) - tailW - 20 * s),
        tx,
        ry + (rowH - Kit.textHeight("monoRow")) / 2,
        usable and PAL.text or PAL.faint
      )
      Kit.textRight(
        "tiny",
        tail,
        cx + inner - 10 * s,
        ry + (rowH - Kit.textHeight("tiny")) / 2,
        PAL.caption
      )
    end
    Kit.popClip()
    Kit.listScrollbar(p, "offset", cx, cy, inner, listH)
  end

  p.offset = Kit.pager(cx, y + h - pad - pagerH, inner, p.offset, #hits, perPage)
end

return Picker