ShaguTweaks-Extras-ClassicAPI/mods/macro-icons.lua

local _G = ShaguTweaks.GetGlobalEnv()
local T = ShaguTweaks.T
local API = ShaguTweaks.API
local libspell = ShaguTweaks.libspell

local module = ShaguTweaks:register({
  title = T["Macro Icons"],
  description = T["Detect showtooltip and spells in macros to use them on action buttons."],
  expansions = { ["vanilla"] = true, ["tbc"] = false },
  maintainer = "@shagu (GitHub)",
  category = T["Macro"],
  enabled = true,
})

module.enable = function(self)
  local gfind = string.gmatch or string.gfind

  local function BuildMacroCache()
    local cache = {}
    for macroSlot = 1, 36 do
      local name, _, body = GetMacroInfo(macroSlot)
      if name and body then
        cache[name] = { id = macroSlot, body = body }
      end
    end
    return cache
  end

  local function GetMacroData(actionSlot, cache)
    local actionType, id = API.GetActionInfo(actionSlot)
    if actionType == "macro" and id then
      local _, _, body = GetMacroInfo(id)
      if body then return id, body end
    end

    local macroName = GetActionText(actionSlot)
    local entry = macroName and cache[macroName]
    if entry then return entry.id, entry.body end
  end

  local function ParseSpellText(match)
    if not match then return end
    local _, _, spell, rank = string.find(match, "(.+)%((.+)%)")
    return spell or match, rank
  end

  local function ResolveSpell(body, macroSlot)
    if not body then return end

    local customTooltip
    local fallback

    for line in gfind(body, "[^%\n]+") do
      -- Explicit #showtooltip always wins.
      local _, _, match = string.find(line, "^#showtooltip%s+(.+)")
      if match then
        return ParseSpellText(match)
      end

      -- Preserve the historical ShaguTweaks custom override.
      if not customTooltip then
        _, _, customTooltip = string.find(line, "%-%-showtooltip%s+(.+)")
      end

      -- Keep text parsing as a fallback for old/stale macro caches and /pfcast.
      if not fallback then
        _, _, fallback = string.find(line, "^/cast%s+(.+)")
      end
      if not fallback then
        _, _, fallback = string.find(line, "^/pfcast%s+(.+)")
      end
      if not fallback then
        _, _, fallback = string.find(line, "CastSpellByName%(%\"(.+)%\"%)")
      end
    end

    if customTooltip then
      return ParseSpellText(customTooltip)
    end

    -- ClassicAPI already resolves and caches the macro's primary spell.
    -- This is the correct modern behavior for a bare "#showtooltip" line and
    -- also covers /castsequence, CastSpellByName and CastSpellNoToggle.
    if macroSlot then
      local spell, rank = API.GetMacroSpell(macroSlot)
      if spell then
        return spell, rank ~= "" and rank or nil
      end
    end

    return ParseSpellText(fallback)
  end

  local function ButtonMacroScan(bar, macroCache)
    if not bar:IsVisible() then return end

    local prefix = bar:GetName()
    prefix = bar == MainMenuBar and "Action" or prefix
    prefix = bar == BonusActionBarFrame and "BonusAction" or prefix

    for index = 1, 12 do
      local button = _G[prefix.."Button"..index]
      local icon = _G[prefix.."Button"..index.."Icon"]
      if not button then break end

      local actionSlot = ActionButton_GetPagedID(button)
      local texture = GetActionTexture(actionSlot)
      button.spellslot, button.booktype = nil, nil

      local macroSlot, body = GetMacroData(actionSlot, macroCache)
      local spell, rank = ResolveSpell(body, macroSlot)
      if spell then
        button.spellslot, button.booktype = libspell.GetSpellIndex(spell, rank)

        if button.spellslot and button.booktype then
          texture = GetSpellTexture(button.spellslot, button.booktype) or texture
        end
      end

      if icon and texture and texture ~= icon:GetTexture() then
        icon:SetTexture(texture)
      end
    end
  end

  local bars = {
    MainMenuBar, BonusActionBarFrame, MultiBarRight, MultiBarLeft,
    MultiBarBottomRight, MultiBarBottomLeft
  }

  local macroicons = CreateFrame("Frame", "ShaguTweaksMacroIcons", UIParent)
  macroicons:RegisterEvent("PLAYER_ENTERING_WORLD")
  macroicons:RegisterEvent("ACTIONBAR_SLOT_CHANGED")
  macroicons:RegisterEvent("ACTIONBAR_PAGE_CHANGED")
  macroicons:RegisterEvent("UPDATE_BONUS_ACTIONBAR")
  macroicons:RegisterEvent("ACTIONBAR_SHOWGRID")
  macroicons:RegisterEvent("UPDATE_MACROS")
  macroicons:SetScript("OnEvent", function()
    local macroCache = BuildMacroCache()
    for _, bar in ipairs(bars) do
      ButtonMacroScan(bar, macroCache)
    end
  end)
end