pfUI/modules/totems.lua

pfUI:RegisterModule("totems", function ()
  local slots = {
    [FIRE_TOTEM_SLOT]  = { r = .5, g = .2, b = .1 },
    [EARTH_TOTEM_SLOT] = { r = .2, g = .4, b = .1 },
    [WATER_TOTEM_SLOT] = { r = .1, g = .4, b = .6 },
    [AIR_TOTEM_SLOT]   = { r = .4, g = .1, b = .7 },
  }

  local totems = CreateFrame("Frame", "pfTotems", UIParent)
  totems:RegisterEvent("PLAYER_TOTEM_UPDATE")
  totems:RegisterEvent("PLAYER_ENTERING_WORLD")
  totems:SetScript("OnEvent", function(self)
    totems:RefreshList()
  end)

  totems.OnEnter = function(self)
    local id = this:GetID()
    local spellID = select(7, GetTotemInfo(id))
    if not spellID or spellID == 0 then return end
    GameTooltip:SetOwner(this, "ANCHOR_LEFT")
    GameTooltip:SetSpell(FindSpellBookSlotByID(spellID))
    GameTooltip:AddDoubleLine(T["Left Click"], "|cffffffff" .. T["Recast Totem"])
    GameTooltip:AddDoubleLine(T["Right Click"], "|cffffffff" .. T["Target Totem"])
    GameTooltip:Show()
  end

  totems.OnLeave = GameTooltip_Hide

  totems.OnClick = function(self)
    local id = this:GetID()
    if arg1 == "LeftButton" then
      local _, name = GetTotemInfo(id)
      if name and name ~= "" then CastSpellByName(name) end
    elseif arg1 == "RightButton" then
      TargetTotem(id)
    end
  end

  totems.RefreshList = function(self)
    local count = 0
    for i = 1, MAX_TOTEMS do
      local _, _, start, duration, icon = GetTotemInfo(i)

      if start and start > 0 and icon and icon ~= "" then
        count = count + 1
        local color = slots[i]

        self.bar[count]:Show()
        self.bar[count]:SetBackdropBorderColor(color.r, color.g, color.b)
        self.bar[count].icon:SetTexture(icon)
        self.bar[count]:SetID(i)

        CooldownFrame_SetTimer(self.bar[count].cd, start, duration, 1)
      end
    end

    self:UpdateSize(count)
  end

  totems.UpdateSize = function(self, count)
    if not count or count == 0 then
      -- hide entire panel
      self:Hide()
    else
      -- hide remaining totems and show panel
      for i = count + 1, MAX_TOTEMS do self.bar[i]:Hide() end
      self:Show()
    end

    count = count and count > 0 and count or MAX_TOTEMS

    if pfUI_config.totems.direction == "HORIZONTAL" then
      self:SetHeight(self.iconsize + self.spacing*2)
      self:SetWidth(self.spacing*2 + self.iconsize + (count-1)*(self.iconsize + self.spacing*2))
    else
      self:SetWidth(self.iconsize + self.spacing*2)
      self:SetHeight(self.spacing*2 + self.iconsize + (count-1)*(self.iconsize + self.spacing*2))
    end
  end

  totems.UpdateConfig = function(self)
    local rawborder, border = GetBorderSize()
    self.iconsize = pfUI_config.totems.iconsize
    self.direction = pfUI_config.totems.direction
    self.spacing = tonumber(pfUI_config.totems.spacing) * GetPerfectPixel()
    self.showbg = pfUI_config.totems.showbg == "1" and true or nil

    for i = 1, MAX_TOTEMS do
      self.bar = self.bar or {}
      self.bar[i] = self.bar[i] or CreateFrame("Button", "pfTotemsBar"..i, totems)
      self.bar[i]:ClearAllPoints()

      if pfUI_config.totems.direction == "HORIZONTAL" then
        if self.bar[i-1] then
          self.bar[i]:SetPoint("LEFT", self.bar[i-1], "RIGHT", self.spacing*2, 0)
        else
          self.bar[i]:SetPoint("TOPLEFT", self, "TOPLEFT", self.spacing, -self.spacing)
        end
      else
        if self.bar[i-1] then
          self.bar[i]:SetPoint("TOP", self.bar[i-1], "BOTTOM", 0, -self.spacing*2)
        else
          self.bar[i]:SetPoint("TOPLEFT", self, "TOPLEFT", self.spacing, -self.spacing)
        end
      end

      self.bar[i]:SetSize(self.iconsize, self.iconsize)
      CreateBackdrop(self.bar[i], nil, true)

      self.bar[i].icon = self.bar[i].icon or self.bar[i]:CreateTexture(nil, "ARTWORK")
      self.bar[i].icon:SetTexCoord(.08, .92, .08, .92)
      SetAllPointsOffset(self.bar[i].icon, self.bar[i], 2,-2)

      self.bar[i].cdbg = self.bar[i].cdbg or CreateFrame("Frame", nil, self.bar[i])
      self.bar[i].cdbg:SetSize(self.iconsize - 3, self.iconsize - 3)
      self.bar[i].cdbg:SetPoint("CENTER", self.bar[i], "CENTER", 0, 0)
      self.bar[i].cd = self.bar[i].cd or CreateFrame(COOLDOWN_FRAME_TYPE, "pfTotemsBar"..i.."Cooldown", self.bar[i].cdbg, "CooldownFrameTemplate")
      self.bar[i].cd.pfCooldownStyleAnimation = 1
      self.bar[i].cd.pfCooldownType = "ALL"

      self.bar[i]:RegisterForClicks("LeftButtonUp", "RightButtonUp")
      self.bar[i]:SetScript("OnClick", self.OnClick)
      self.bar[i]:SetScript("OnEnter", self.OnEnter)
      self.bar[i]:SetScript("OnLeave", self.OnLeave)
    end

    self:RefreshList()
    self:ClearAllPoints()
    self:SetPoint("BOTTOM", 0, 75)
    UpdateMovable(self, true)

    if self.showbg then
      CreateBackdrop(self)
      self.backdrop:Show()
    elseif self.backdrop then
      self.backdrop:Hide()
    end
  end

  -- add totems to the pfUI global space
  pfUI.totems = totems
  pfUI.totems:UpdateConfig()
end)