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