prepare LibGroupTalents, fix broken rebase

2025-01-27 18:40:55 +01:00
parent 31bcef0dc7
commit a99f7bd323
11 changed files with 2893 additions and 8 deletions
@@ -0,0 +1,166 @@
+if not WeakAuras.IsLibsOK() then return end
+
+local AddonName, Private = ...
+
+local timer = WeakAuras.timer;
+local eventLock = false
+
+local nameToGlyphs = {}
+local nameToSpecMap = {}
+local nameToUnitMap = {
+  [UnitName("player")] = "player"
+}
+
+local subscribers = {}
+
+Private.LibGroupTalentsWrapper = {
+  Register = function(callback) end,
+  SpecForUnit = function(unit) end,
+  SpecRolePositionForUnit = function(unit) end,
+  CheckTalentForUnit = function(unit, talentId) end,
+  CheckGlyphForUnit = function(unit, glyphId) end,
+}
+
+local LibGroupTalents = LibStub("LibGroupTalents-1.0")
+if LibGroupTalents then
+  local frame = CreateFrame("Frame")
+  frame:RegisterEvent("PLAYER_LOGIN")
+  frame:RegisterEvent("RAID_ROSTER_UPDATE")
+  frame:RegisterEvent("PARTY_MEMBERS_CHANGED")
+
+  local function ProcessEvent()
+    local ownName = UnitName("player")
+
+    nameToUnitMap = {}
+    nameToUnitMap[ownName] = "player"
+
+    local numMembers
+    local units
+    if IsInRaid() then
+      numMembers = GetNumGroupMembers()
+      units = WeakAuras.raidUnits
+    else
+      numMembers = GetNumPartyMembers()
+      units = WeakAuras.partyUnits
+    end
+
+    for i = 1, numMembers do
+      local unit = units[i]
+      local name = UnitName(unit)
+      nameToUnitMap[name] = unit
+    end
+
+    for name in pairs(nameToSpecMap) do
+      if not nameToUnitMap[name] then
+        nameToSpecMap[name] = nil
+        nameToGlyphs[name] = nil
+      end
+    end
+
+    eventLock = false
+  end
+
+  frame:SetScript("OnEvent", function()
+    if not eventLock then
+      eventLock = true
+      timer:ScheduleTimer(ProcessEvent, 1)
+    end
+  end)
+
+  --- LibGroupTalents callback for talents and glyphs
+  local function LibGroupTalentsCallback(guid, unit)
+    local unitName = UnitName(unit)
+
+    -- Update specialization data
+    local specInfo = { LibGroupTalents:GetUnitTalentSpec(unit) }
+    if specInfo and #specInfo > 0 then
+      nameToSpecMap[unitName] = specInfo
+    end
+
+    -- Update glyphs
+    local glyphs = { LibGroupTalents:GetUnitGlyphs(unit) }
+    if glyphs and #glyphs > 0 then
+      nameToGlyphs[unitName] = {}
+      for _, glyphId in ipairs(glyphs) do
+        nameToGlyphs[unitName][glyphId] = true
+      end
+    end
+
+    -- Notify subscribers
+    for _, f in ipairs(subscribers) do
+      f(nameToUnitMap[unitName])
+    end
+  end
+
+  LibGroupTalents:RegisterCallback("LibGroupTalents_Update", LibGroupTalentsCallback)
+
+  function Private.LibGroupTalentsWrapper.Register(f)
+    table.insert(subscribers, f)
+  end
+
+  function Private.LibGroupTalentsWrapper.SpecForUnit(unit)
+    if UnitIsUnit(unit, "player") then
+      return LibGroupTalents:GetUnitTalentSpec(unit)
+    end
+
+    local unitName = UnitName(unit)
+    return nameToSpecMap[unitName] and nameToSpecMap[unitName][1] or nil
+  end
+
+  function Private.LibGroupTalentsWrapper.SpecRolePositionForUnit(unit)
+    if UnitIsUnit(unit, "player") then
+      return LibGroupTalents:GetUnitTalentSpec(unit)
+    end
+
+    local data = nameToSpecMap[UnitName(unit)]
+    if data then
+      return unpack(data)
+    else
+      return nil
+    end
+  end
+
+  function Private.LibGroupTalentsWrapper.CheckTalentForUnit(unit, talentId)
+    if UnitIsUnit(unit, "player") then
+      return select(4, WeakAuras.GetTalentById(talentId))
+    end
+
+    local unitName = UnitName(unit)
+    if not nameToSpecMap[unitName] then
+      return nil
+    end
+
+    return LibGroupTalents:UnitHasTalent(unit, talentId) and true or false
+  end
+
+  function Private.LibGroupTalentsWrapper.CheckGlyphForUnit(unit, glyphId)
+    if UnitIsUnit(unit, "player") then
+      local glyphs = { LibGroupTalents:GetUnitGlyphs(unit) }
+      for _, id in ipairs(glyphs) do
+        if id == glyphId then
+          return true
+        end
+      end
+      return false
+    end
+
+    local unitName = UnitName(unit)
+    if nameToGlyphs[unitName] then
+      return nameToGlyphs[unitName][glyphId] or false
+    end
+
+    return false
+  end
+else
+  function Private.LibGroupTalentsWrapper.Register(f) end
+  function Private.LibGroupTalentsWrapper.SpecForUnit(unit) return nil end
+  function Private.LibGroupTalentsWrapper.SpecRolePositionForUnit(unit) return nil end
+  function Private.LibGroupTalentsWrapper.CheckTalentForUnit(unit) return nil end
+  function Private.LibGroupTalentsWrapper.CheckGlyphForUnit(unit) return nil end
+end
+
+-- Export for GenericTrigger
+WeakAuras.SpecForUnit = Private.LibGroupTalentsWrapper.SpecForUnit
+WeakAuras.SpecRolePositionForUnit = Private.LibGroupTalentsWrapper.SpecRolePositionForUnit
+WeakAuras.CheckTalentForUnit = Private.LibGroupTalentsWrapper.CheckTalentForUnit
+WeakAuras.CheckGlyphForUnit = Private.LibGroupTalentsWrapper.CheckGlyphForUnit