Class Icons + Druid Forms

local forms = {}
local pclass = select(2, UnitClass("player"))
hooksecurefunc("UnitFramePortrait_Update", function(self)
        if self.portrait then
                        local form = GetShapeshiftForm()
                        local class = select(2, UnitClass(self.unit))
                        local tex = CLASS_ICON_TCOORDS[select(2,UnitClass(self.unit))]
                        if tex then
                                if (self.unit == "player") and (pclass == "DRUID") and (form > 0) then
                                        SetPortraitToTexture(PlayerPortrait, forms[form])
                                        PlayerPortrait:SetTexCoord(0, 1, 0, 1)
                                else
                                        self.portrait:SetTexture("Interface\\TargetingFrame\\UI-Classes-Circles")
                                        self.portrait:SetTexCoord(unpack(tex))
                        end
                end
        end
end)

local f = CreateFrame'Frame'
f:RegisterEvent'UPDATE_SHAPESHIFT_FORMS'
f:RegisterEvent'PLAYER_ENTERING_WORLD'
f:SetScript('OnEvent', function()
        if pclass ~= 'DRUID' then return end
        if GetNumShapeshiftForms() == 0 then return end
        for i = 1, GetNumShapeshiftForms() do
                forms[i] = (GetShapeshiftFormInfo(i))
        end
end)