pfUI/modules/unitframes.lua

pfUI.uf = CreateFrame("Frame",nil,UIParent)

function pfUI.uf:AddIcon(frame, pos, icon)
  local iconsize = 10
  if not frame.hp then return end
  local frame = frame.hp.bar
  if pos > floor(frame:GetWidth() / iconsize) then return end

  if not frame.icon then frame.icon = {} end

  for i=1,6 do
    if not frame.icon[i] then
      frame.icon[i] = CreateFrame("Frame", frame)
      frame.icon[i]:SetPoint("TOPLEFT", frame, "TOPLEFT", (i-1)*iconsize, 0)
      frame.icon[i]:SetWidth(iconsize)
      frame.icon[i]:SetHeight(iconsize)
      frame.icon[i]:SetAlpha(.7)
      frame.icon[i]:SetParent(frame)
      frame.icon[i].tex = frame.icon[i]:CreateTexture("OVERLAY")
      frame.icon[i].tex:SetAllPoints(frame.icon[i])
      frame.icon[i].tex:SetTexCoord(.08, .92, .08, .92)
    end
  end
  frame.icon[pos].tex:SetTexture(icon)
  pfUI.utils:CreateBackdrop(frame.icon[pos], nil, true)
  frame.icon[pos]:Show()
end

function pfUI.uf:HideIcon(frame, pos)
  if not frame or not frame.hp or not frame.hp.bar then return end

  local frame = frame.hp.bar
  if frame.icon and frame.icon[pos] then
    frame.icon[pos]:Hide()
  end
end

function pfUI.uf:SetupDebuffFilter()
  if pfUI.uf.debuffs then return end

  local _, myclass = UnitClass("player")
  pfUI.uf.debuffs = {}
  if pfUI_config.unitframes.raid.debuffs_enable == "1" then
    if myclass == "PALADIN" or myclass == "PRIEST" or pfUI_config.unitframes.raid.debuffs_class ~= "1" then
      table.insert(pfUI.uf.debuffs, "magic")
    end

    if myclass == "DRUID" or myclass == "PALADIN" or myclass == "SHAMAN" or pfUI_config.unitframes.raid.debuffs_class ~= "1" then
      table.insert(pfUI.uf.debuffs, "poison")
    end

    if myclass == "PRIEST" or myclass == "PALADIN" or myclass == "SHAMAN" or pfUI_config.unitframes.raid.debuffs_class ~= "1" then
      table.insert(pfUI.uf.debuffs, "disease")
    end

    if myclass == "DRUID" or myclass == "MAGE" or pfUI_config.unitframes.raid.debuffs_class ~= "1" then
      table.insert(pfUI.uf.debuffs, "curse")
    end
  end
end

function pfUI.uf:SetupBuffFilter()
  if pfUI.uf.buffs then return end

  local _, myclass = UnitClass("player")

  pfUI.uf.buffs = {}

  -- [[ DRUID ]]
  if myclass == "DRUID" and pfUI_config.unitframes.raid.buffs_buffs == "1" then
    -- Gift of the Wild
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_regeneration")

    -- Thorns
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_thorns")
  end

  if (pfUI_config.unitframes.raid.buffs_classonly ~= "1" or myclass == "DRUID") and pfUI_config.unitframes.raid.buffs_hots == "1" then
    -- Regrowth
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_resistnature")

    -- Rejuvenation
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_rejuvenation")
  end


  -- [[ PRIEST ]]
  if myclass == "PRIEST" and pfUI_config.unitframes.raid.buffs_buffs == "1" then
    -- Prayer Of Fortitude"
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_wordfortitude")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_prayeroffortitude")

    -- Prayer of Spirit
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_divinespirit")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_prayerofspirit")
  end

  if (pfUI_config.unitframes.raid.buffs_classonly ~= "1" or myclass == "PRIEST") and pfUI_config.unitframes.raid.buffs_hots == "1" then
    -- Renew
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_renew")
  end

  if (pfUI_config.unitframes.raid.buffs_classonly ~= "1" or myclass == "PRIEST") and pfUI_config.unitframes.raid.buffs_procs == "1" then
    -- Inspiration
    table.insert(pfUI.uf.buffs, "interface\\icons\\inv_shield_06")
  end


  -- [[ PALADIN ]]
  if myclass == "PALADIN" and pfUI_config.unitframes.raid.buffs_buffs == "1" then
    -- Blessing of Salvation
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_greaterblessingofsalvation")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_sealofsalvation")

    -- Blessing of Wisdom
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_sealofwisdom")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_greaterblessingofwisdom")

    -- Blessing of Sanctuary
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_lightningshield")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_greaterblessingofsanctuary")

    -- Blessing of Kings
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_magic_magearmor")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_magic_greaterblessingofkings")

    -- Blessing of Might
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_fistofjustice")
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_holy_greaterblessingofkings")
  end


  -- [[ SHAMAN ]]
  if (pfUI_config.unitframes.raid.buffs_classonly ~= "1" or myclass == "SHAMAN") and pfUI_config.unitframes.raid.buffs_procs == "1" then
    -- Ancestral Fortitude
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_undyingstrength")

    -- Healing Way
    table.insert(pfUI.uf.buffs, "interface\\icons\\spell_nature_healingway")
  end
end

function pfUI.uf:RefreshUnit(unit)
  if not unit.cache then unit.cache = {} end
  if not unit.id then unit.id = "" end

  unit.cache.hp = UnitHealth(unit.label..unit.id)
  unit.cache.hpmax = UnitHealthMax(unit.label..unit.id)
  unit.cache.power = UnitMana(unit.label .. unit.id)
  unit.cache.powermax = UnitManaMax(unit.label .. unit.id)

  if this.label == "target" and MobHealth3 then
    unit.cache.hp, unit.cache.hpmax = MobHealth3:GetUnitHealth(this.label)
  end

  unit.hp.bar:SetMinMaxValues(0, unit.cache.hpmax)
  unit.power.bar:SetMinMaxValues(0, unit.cache.powermax)

  pfUI.uf:SetupDebuffFilter()
  if table.getn(pfUI.uf.debuffs) > 0 and
    ((pfUI_config.unitframes.group.raid_debuffs == "1" and this.label == "party") or
    this.label == "raid")
  then
    local infected = false
    for i=1,32 do
      local _,_,dtype = UnitDebuff(unit.label .. unit.id, i)
      if dtype then
        for _, filter in pairs(pfUI.uf.debuffs) do
          if filter == string.lower(dtype) then
            if dtype == "Magic" then
              if not unit.hp.bar.magic then
                unit.hp.bar.magic = CreateFrame("Frame", unit.hp.bar)
                unit.hp.bar.magic:SetAllPoints(unit)
                unit.hp.bar.magic:SetParent(unit.hp.bar)
                unit.hp.bar.magic.tex = unit.hp.bar.magic:CreateTexture("OVERLAY")
                unit.hp.bar.magic.tex:SetAllPoints(unit.hp.bar.magic)
                unit.hp.bar.magic.tex:SetTexture(.2,.8,.8,.4)
              end
              unit.hp.bar.magic:Show()
              infected = true

            elseif dtype == "Poison" then
              if not unit.hp.bar.poison then
                unit.hp.bar.poison = CreateFrame("Frame", unit.hp.bar)
                unit.hp.bar.poison:SetAllPoints(unit)
                unit.hp.bar.poison:SetParent(unit.hp.bar)
                unit.hp.bar.poison.tex = unit.hp.bar.poison:CreateTexture("OVERLAY")
                unit.hp.bar.poison.tex:SetAllPoints(unit.hp.bar.poison)
                unit.hp.bar.poison.tex:SetTexture(.2,.8,.2,.4)
              end
              unit.hp.bar.poison:Show()
              infected = true

            elseif dtype == "Curse" then
              if not unit.hp.bar.curse then
                unit.hp.bar.curse = CreateFrame("Frame", unit.hp.bar)
                unit.hp.bar.curse:SetAllPoints(unit)
                unit.hp.bar.curse:SetParent(unit.hp.bar)
                unit.hp.bar.curse.tex = unit.hp.bar.curse:CreateTexture("OVERLAY")
                unit.hp.bar.curse.tex:SetAllPoints(unit.hp.bar.curse)
                unit.hp.bar.curse.tex:SetTexture(.8,.2,.8,.4)
              end
              unit.hp.bar.curse:Show()
              infected = true

            elseif dtype == "Disease" then
              if not unit.hp.bar.disease then
                unit.hp.bar.disease = CreateFrame("Frame", unit.hp.bar)
                unit.hp.bar.disease:SetAllPoints(unit)
                unit.hp.bar.disease:SetParent(unit.hp.bar)
                unit.hp.bar.disease.tex = unit.hp.bar.disease:CreateTexture("OVERLAY")
                unit.hp.bar.disease.tex:SetAllPoints(unit.hp.bar.disease)
                unit.hp.bar.disease.tex:SetTexture(.8,.8,.2,.4)
              end
              unit.hp.bar.disease:Show()
              infected = true
            end
          end
        end
      end
    end
    if infected == false then
      if unit.hp.bar.magic then unit.hp.bar.magic:Hide() end
      if unit.hp.bar.poison then unit.hp.bar.poison:Hide() end
      if unit.hp.bar.curse then unit.hp.bar.curse:Hide() end
      if unit.hp.bar.disease then unit.hp.bar.disease:Hide() end
    end
  end

  pfUI.uf:SetupBuffFilter()
  if table.getn(pfUI.uf.buffs) > 0 and
    ((pfUI_config.unitframes.group.raid_buffs == "1" and this.label == "party") or
    this.label == "raid")
  then
    local active = {}

    for i=1,32 do
      local texture = UnitBuff(unit.label .. unit.id,i)

      if texture then
        -- match filter
        for _, filter in pairs(pfUI.uf.buffs) do
          if filter == string.lower(texture) then
            table.insert(active, texture)
            break
          end
        end

        -- add icons for every found buff
        for pos, icon in pairs(active) do
          pfUI.uf:AddIcon(this, pos, icon)
        end
      end
    end

    -- hide unued icon slots
    for pos=table.getn(active)+1, 6 do
      pfUI.uf:HideIcon(this, pos)
    end
  end

  _, class = UnitClass(unit.label..unit.id)
  local color = RAID_CLASS_COLORS[class]

  local r, g, b = .2, .2, .2
  if pfUI_config.unitframes.dark == "1" and color then
    unit.hp.bar:SetStatusBarColor(r, g, b)
    if pfUI_config.unitframes.pastel == "1" then
      r, g, b = (color.r + .5) * .5, (color.g + .5) * .5, (color.b + .5) * .5
    else
      r, g, b = color.r, color.g, color.b
    end
  elseif color then
    if pfUI_config.unitframes.pastel == "1" then
      r, g, b = (color.r + .5) * .5, (color.g + .5) * .5, (color.b + .5) * .5
    else
      r, g, b = color.r, color.g, color.b
    end
    unit.hp.bar:SetStatusBarColor(r, g, b)
  end
  unit.caption:SetTextColor(r,g,b)

  local p = ManaBarColor[UnitPowerType(unit.label..unit.id)]
  local pr, pg, pb = 0, 0, 0
  if p then pr, pg, pb = p.r + .5, p.g +.5, p.b +.5 end
  unit.power.bar:SetStatusBarColor(pr, pg, pb)

  if UnitName(unit.label..unit.id) then
    if this.label == "raid" and unit.cache.hpmax ~= unit.cache.hp and pfUI_config.unitframes.raid.show_missing == "1" then
      unit.caption:SetText("-" .. unit.cache.hpmax - unit.cache.hp)
      unit.caption:SetTextColor(1,.3,.3)
    else
      unit.caption:SetText(UnitName(unit.label..unit.id))
      if pfUI_config.unitframes.dark ~= "1" then
        unit.caption:SetTextColor(1,1,1)
      end
    end
    unit.caption:SetAllPoints(unit.hp.bar)
  end
end

function pfUI.uf:CreateUnit(unit)
  unit:RegisterEvent("UNIT_AURA")
  unit:RegisterEvent("UNIT_HEALTH")
  unit:RegisterEvent("UNIT_MAXHEALTH")
  unit:RegisterEvent("UNIT_DISPLAYPOWER")
  unit:RegisterEvent("UNIT_MANA")
  unit:RegisterEvent("UNIT_MAXMANA")
  unit:RegisterEvent("UNIT_RAGE")
  unit:RegisterEvent("UNIT_MAXRAGE")
  unit:RegisterEvent("UNIT_ENERGY")
  unit:RegisterEvent("UNIT_MAXENERGY")
  unit:RegisterEvent("UNIT_FOCUS")

  unit:RegisterEvent("RAID_ROSTER_UPDATE")
  unit:SetScript("OnShow", function ()
    pfUI.uf:RefreshUnit(this)
  end)

  unit:SetScript("OnEvent", function ()
    if this.label == "raid" and event == "RAID_ROSTER_UPDATE" then
      pfUI.uf:RefreshUnit(this)
    end

    if arg1 == this.label .. this.id then
      pfUI.uf:RefreshUnit(this)
    end
  end)

  unit:SetScript("OnClick", function ()
    if ( SpellIsTargeting() and arg1 == "RightButton" ) then
      SpellStopTargeting()
      return
    end

    if ( arg1 == "LeftButton" ) then
      if ( SpellIsTargeting() ) then
        SpellTargetUnit(this.label .. this.id)
      elseif ( CursorHasItem() ) then
        DropItemOnUnit(this.label .. this.id)
      else
        TargetUnit(this.label .. this.id)

        -- break here if party frame and no clickcast is activated
        if this.label == "party" and
          pfUI_config.unitframes.group.clickcast == "0" and
          pfUI_config.unitframes.globalclick == "0" then
            return
        end

        -- break here for non-party and non-raid frames without clickcast
        if this.label ~= "raid" and this.label ~= "party" and pfUI_config.unitframes.globalclick == "0" then
          return
        end

        -- clickcast: shift modifier
        if IsShiftKeyDown() then
          if pfUI_config.unitframes.raid.clickcast_shift ~= "" then
            CastSpellByName(pfUI_config.unitframes.raid.clickcast_shift)
            pfUI.uf.target.noanim = "yes"
            TargetLastTarget()
            return
          end

        -- clickcast: alt modifier
        elseif IsAltKeyDown() then
          if pfUI_config.unitframes.raid.clickcast_alt ~= "" then
            CastSpellByName(pfUI_config.unitframes.raid.clickcast_alt)
            pfUI.uf.target.noanim = "yes"
            TargetLastTarget()
            return
          end

        -- clickcast: ctrl modifier
        elseif IsControlKeyDown() then
          if pfUI_config.unitframes.raid.clickcast_ctrl ~= "" then
            CastSpellByName(pfUI_config.unitframes.raid.clickcast_ctrl)
            pfUI.uf.target.noanim = "yes"
            TargetLastTarget()
            return
          end

        -- clickcast: default
        else
          if pfUI_config.unitframes.raid.clickcast ~= "" then
            CastSpellByName(pfUI_config.unitframes.raid.clickcast)
            pfUI.uf.target.noanim = "yes"
            TargetLastTarget()
            return
          else
            -- no clickcast: default action
            TargetUnit(this.label .. this.id)
          end
        end
      end
    else
      if this.label == "party" then
        ToggleDropDownMenu(1, nil, getglobal("PartyMemberFrame" .. this.id .. "DropDown"), "cursor")
      elseif this.label == "raid" then
        ToggleDropDownMenu(1, nil, getglobal("RaidMemberFrame" .. this.id .. "DropDown"), "cursor")
        FriendsDropDown.initialize = RaidFrameDropDown_Initialize
        FriendsDropDown.displayMode = "MENU"
        ToggleDropDownMenu(1, nil, FriendsDropDown, "cursor")
      end
    end
  end)

  unit:SetScript("OnEnter", function()
    GameTooltip_SetDefaultAnchor(GameTooltip, this)
    GameTooltip:SetUnit(this.label .. this.id)
    GameTooltip:Show()
  end)

  unit:SetScript("OnLeave", function()
    GameTooltip:FadeOut()
  end)

  unit:SetScript("OnUpdate", function ()
    if CheckInteractDistance(this.label .. this.id, 4) or not UnitName(this.label .. this.id) then
      this:SetAlpha(1)
    else
      this:SetAlpha(.5)
    end

    if UnitIsConnected(this.label .. this.id) then
      if not this.cache then return end

      local hpDisplay = this.hp.bar:GetValue()
      local hpReal = this.cache.hp
      local hpDiff = abs(hpReal - hpDisplay)

      if this.label == "raid" and pfUI_config.unitframes.raid.invert_healthbar == "1" then
        hpDisplay = this.hp.bar:GetValue()
        hpReal = this.cache.hpmax - this.cache.hp
        hpDiff = abs(hpReal - hpDisplay)
      end

      if hpDisplay < hpReal then
        this.hp.bar:SetValue(hpDisplay + ceil(hpDiff / pfUI_config.unitframes.animation_speed))
      elseif hpDisplay > hpReal then
        this.hp.bar:SetValue(hpDisplay - ceil(hpDiff / pfUI_config.unitframes.animation_speed))
      else
        this.hp.bar:SetValue(hpReal)
      end

      local powerDisplay = this.power.bar:GetValue()
      local powerReal = this.cache.power
      local powerDiff = abs(powerReal - powerDisplay)

      if powerDisplay < powerReal then
        this.power.bar:SetValue(powerDisplay + ceil(powerDiff / pfUI_config.unitframes.animation_speed))
      elseif powerDisplay > powerReal then
        this.power.bar:SetValue(powerDisplay - ceil(powerDiff / pfUI_config.unitframes.animation_speed))
      else
        this.power.bar:SetValue(this.cache.power)
      end
    else
      this.hp.bar:SetMinMaxValues(0, 100)
      this.power.bar:SetMinMaxValues(0, 100)
      this.hp.bar:SetValue(0)
      this.power.bar:SetValue(0)
    end
  end)
end