Gladdy-TBC/Modules/ArenaCountDown.lua

local select, floor, str_len, tostring, str_sub, str_find = select, math.floor, string.len, tostring, string.sub, string.find
local IsInInstance = IsInInstance

local Gladdy = LibStub("Gladdy")
local L = Gladdy.L
local ACDFrame = Gladdy:NewModule("Countdown", nil, {
    countdown = true,
    arenaCountdownSize = 256
})

function ACDFrame:Initialize()
    self.hidden = false
    self.countdown = -1
    self.texturePath = "Interface\\AddOns\\Gladdy\\Images\\Countdown\\";
    ACDFrame:RegisterEvent("CHAT_MSG_BG_SYSTEM_NEUTRAL")
    ACDFrame:RegisterEvent("PLAYER_ENTERING_WORLD")
end

function ACDFrame:OnEvent(event, ...)
    -- functions created in "object:method"-style have an implicit first parameter of "self", which points to object
    self[event](self, ...) -- route event parameters to LoseControl:event methods
end
ACDFrame:SetScript("OnEvent", ACDFrame.OnEvent)


local ACDNumFrame = CreateFrame("Frame", "ACDNumFrame", UIParent)
ACDNumFrame:SetHeight(256)
ACDNumFrame:SetWidth(256)
ACDNumFrame:SetPoint("CENTER", 0, 128)
ACDNumFrame:Show()

local ACDNumTens = ACDNumFrame:CreateTexture("ACDNumTens", "HIGH")
ACDNumTens:SetWidth(256)
ACDNumTens:SetHeight(128)
ACDNumTens:SetPoint("CENTER", ACDNumFrame, "CENTER", -48, 0)

local ACDNumOnes = ACDNumFrame:CreateTexture("ACDNumOnes", "HIGH")
ACDNumOnes:SetWidth(256)
ACDNumOnes:SetHeight(128)
ACDNumOnes:SetPoint("CENTER", ACDNumFrame, "CENTER", 48, 0)

local ACDNumOne = ACDNumFrame:CreateTexture("ACDNumOne", "HIGH")
ACDNumOne:SetWidth(256)
ACDNumOne:SetHeight(128)
ACDNumOne:SetPoint("CENTER", ACDNumFrame, "CENTER", 0, 0)

function ACDFrame:PLAYER_ENTERING_WORLD()

    local instanceType = select(2, IsInInstance())
    if (( instanceType == "arena" )) then
        ACDFrame:SetScript("OnUpdate", function(self, elapse)
            if (self.countdown > 0 and Gladdy.db.countdown) then
                self.hidden = false;

                if ((floor(self.countdown) ~= floor(self.countdown - elapse)) and (floor(self.countdown - elapse) >= 0)) then
                    local str = tostring(floor(self.countdown - elapse));

                    if (floor(self.countdown - elapse) == 0) then
                        ACDNumTens:Hide();
                        ACDNumOnes:Hide();
                        ACDNumOne:Hide();
                    elseif (str_len(str) == 2) then
                        -- Display has 2 digits
                        ACDNumOne:Hide();
                        ACDNumTens:Show();
                        ACDNumOnes:Show();

                        ACDNumTens:SetTexture(self.texturePath .. str_sub(str, 0, 1));
                        ACDNumOnes:SetTexture(self.texturePath .. str_sub(str, 2, 2));
                        ACDNumFrame:SetScale(0.7)
                    elseif (str_len(str) == 1) then
                        -- Display has 1 digit
                        ACDNumOne:Show();
                        ACDNumOne:SetTexture(self.texturePath .. str_sub(str, 0, 1));
                        ACDNumOnes:Hide();
                        ACDNumTens:Hide();
                        ACDNumFrame:SetScale(1.0)
                    end
                end
                self.countdown = self.countdown - elapse;
            else
                self.hidden = true;
                ACDNumTens:Hide();
                ACDNumOnes:Hide();
                ACDNumOne:Hide();
            end

        end)
    end
end

function ACDFrame:CHAT_MSG_BG_SYSTEM_NEUTRAL(msg)
    if (str_find(msg, "L'ar\195\168ne ouvre ses portes dans 60 secondes !")) then
        self.countdown = 61
        return
    end
    if (str_find(msg, "L'ar\195\168ne ouvre ses portes dans 30 secondes !")) then
        self.countdown = 31
        return
    end
    if (str_find(msg, "L'ar\195\168ne ouvre ses portes dans 15 secondes !")) then
        self.countdown = 16
        return
    end
    if (str_find(msg, "L'ar\195\168ne ouvre ses portes dans 10 secondes !")) then
        self.countdown = 11
        return
    end
    if (str_find(msg, "One minute until the Arena battle begins!")) then
        self.countdown = 61
        return
    end
    if (str_find(msg, "Thirty seconds until the Arena battle begins!")) then
        self.countdown = 31
        return
    end
    if (str_find(msg, "Fifteen seconds until the Arena battle begins!")) then
        self.countdown = 16
        return
    end
    if (str_find(msg, "Ten seconds until the Arena battle begins!")) then
        self.countdown = 10
        return
    end
    if (str_find(msg, "The Arena battle has begun!")) then
        ACDFrame:SetScript("OnUpdate", nil)
        return
    end
end

function ACDFrame:UpdateFrame()
    ACDNumFrame:SetHeight(Gladdy.db.arenaCountdownSize)
    ACDNumFrame:SetWidth(Gladdy.db.arenaCountdownSize)
    ACDNumFrame:SetPoint("CENTER", 0, 128)

    ACDNumTens:SetWidth(Gladdy.db.arenaCountdownSize)
    ACDNumTens:SetHeight(Gladdy.db.arenaCountdownSize/2)
    ACDNumTens:SetPoint("CENTER", ACDNumFrame, "CENTER", -(Gladdy.db.arenaCountdownSize/8 + Gladdy.db.arenaCountdownSize/8/2), 0)

    ACDNumOnes:SetWidth(Gladdy.db.arenaCountdownSize)
    ACDNumOnes:SetHeight(Gladdy.db.arenaCountdownSize/2)
    ACDNumOnes:SetPoint("CENTER", ACDNumFrame, "CENTER", (Gladdy.db.arenaCountdownSize/8 + Gladdy.db.arenaCountdownSize/8/2), 0)

    ACDNumOne:SetWidth(Gladdy.db.arenaCountdownSize)
    ACDNumOne:SetHeight(Gladdy.db.arenaCountdownSize/2)
    ACDNumOne:SetPoint("CENTER", ACDNumFrame, "CENTER", 0, 0)
end

function ACDFrame:Test()
    self.countdown = 30
end

function ACDFrame:Reset()
    self.countdown = 0
end

function ACDFrame:GetOptions()
    return {
        headerArenaCountdown = {
            type = "header",
            name = L["Arena Countdown"],
            order = 2,
        },
        countdown = Gladdy:option({
            type = "toggle",
            name = L["Turn on/off"],
            desc = L["Turns countdown before the start of an arena match on/off."],
            order = 3,
            width = "full",
        }),
        arenaCountdownSize = Gladdy:option({
            type = "range",
            name = L["Size"],
            order = 4,
            min = 64,
            max = 512,
            step = 16,
        }),
    }
end