management.lua rewritten, v2

--[[    $Id: management.lua 3447 2013-08-02 09:29:57Z [email protected] $       ]]

local _, caelCore = ...

------------------------------------------------------------------------
--  List of characters to run the addon for.
-- I cleaned up the table format, as I couldn't see any code that
-- depended on having so many extra table levels.
------------------------------------------------------------------------

local myToons = {
    ["ServerName"] = {
        ["CharacterA"]  = "DEATHKNIGHT",
        ["CharacterB"] = "DRUID",
        ["CharacterC"] = "HUNTER",
        ["CharacterD"]  = "MAGE",
        ["CharacterE"] = "MONK",
        ["CharacterF"] = "WARLOCK",
        ["CharacterG"]  = "WARRIOR",
    }
}

local herToons = {
    ["ServerName"] = {
        ["CharacterH"] = "DRUID",
        ["CharacterI"] = "HUNTER",
        ["CharacterJ"] = "MONK",
        ["CharacterK"] = "PRIEST",
        ["CharacterL"] = "ROGUE",
        ["CharacterM"] = "SHAMAN",
        ["CharacterN"] = "WARRIOR",
    }
}

local playerName = UnitName("player")

-- Don't make these global.
local myChars = myToons[caelLib.playerRealm] and myToons[caelLib.playerRealm][playerName]
local herChars = herToons[caelLib.playerRealm] and herToons[caelLib.playerRealm][playerName]

if not (myChars or herChars) then return end

------------------------------------------------------------------------
--  Basic initialization stuff.
------------------------------------------------------------------------

local Management = caelCore.createModule("Management")
caelCore.management = Management

-- Avoid giant if-else chains by passing off events to individual functions.
Management:SetScript("OnEvent", function(self, event, ...) return self[event](self, ...) end)
Management:RegisterEvent("PLAYER_LOGIN")

function Management:PLAYER_LOGIN()
    self:UnregisterEvent("PLAYER_LOGIN")

    -- If the GuildUI is already loaded, do it now, otherwise wait.
    if IsAddOnLoaded("Blizzard_GuildUI") then
        self:ADDON_LOADED("Blizzard_GuildUI")
    else
        self:RegisterEvent("ADDON_LOADED")
    end

    self:RegisterEvent("CALENDAR_UPDATE_EVENT_LIST")

    self:RegisterEvent("GUILD_ROSTER_UPDATE")
    GuildRoster() -- Forces a GUILD_ROSTER_UPDATE to fire soon.
end

------------------------------------------------------------------------
-- Let the GuildUI load normally.
------------------------------------------------------------------------

function Management:ADDON_LOADED(addon)
    if addon == "Blizzard_GuildUI" then
        GuildRosterShowOfflineButton:Disable()

        local function GuildFrameHook(GuildFrame)
            GuildFrameTab2:Click()
            GuildRoster_SetView("guildStatus")
            -- Currently not performed in _SetView
            UIDropDownMenu_SetSelectedValue(GuildRosterViewDropdown, "guildStatus")
        end
        GuildFrame:HookScript("OnShow", GuildFrameHook)
        GuildFrame:HookScript("OnHide", GuildFrameHook)

        self:UnregisterEvent("ADDON_LOADED")
    end
end

------------------------------------------------------------------------
--  List of calendar events changed.
------------------------------------------------------------------------

function Management:CALENDAR_UPDATE_EVENT_LIST()
    self:ManageCalendar()
end

------------------------------------------------------------------------
--  Guild roster changed.
------------------------------------------------------------------------

function Management:GUILD_ROSTER_UPDATE()
    -- Are we in the right guild? Do we have any permissions?
    if GetGuildInfo("player") == "We Did It" and (CanGuildPromote() or CanGuildRemove()) then
        -- Should we defer to the other manager?
        if herChars then
            -- Is the other manager online?
            local chars = myToons[caelLib.playerRealm]
            for i = 1, GetNumGuildMembers() do
                local name = GetGuildRosterInfo(i)
                if chars[name] then
                    -- Other manager is online. Let them handle it.
                    return
                end
            end
        end

        -- Pass "true" to do it for real, eg. self:ManageGuild(true)
        self:ManageGuild()
    end
end

------------------------------------------------------------------------
--  Actual calendar management logic.
------------------------------------------------------------------------

function Management:ManageCalendar()
    for month = 0, 12 do
        for day = 1, 31 do
            for eventIndex = 1, CalendarGetNumDayEvents(month, day) do
                local _, _, _, calendarType, _, _, _, _, _, invitedBy = CalendarGetDayEvent(month, day, eventIndex)
                if calendarType == "PLAYER" and CalendarContextInviteIsPending(month, day, eventIndex)
                    if caelLib.isGuild(invitedBy) then -- Check this first so you don't have to check it twice.
                        CalendarContextInviteAvailable(month, day, eventIndex)
                    elseif invitedBy == "" then -- or (invitedBy ~= caelLib.playerName and not caelLib.isFriend(invitedBy) and not not caelLib.isInGroup(invitedBy)) then
                        CalendarContextInviteRemove(month, day, eventIndex)
                    end
                end
            end
        end
    end
end

------------------------------------------------------------------------
--  Actual guild management logic, all wrapped up in a do-end block to
-- limit scoping.
------------------------------------------------------------------------

do
    local ignoreList = {
        -- Use a hash table here so ignoreList[name] actually works.
        ["Byleths"] = true,
        ["Prøsper"] = true,
        ["Trickaz"] = true,
        ["Trøyà"] = true,
        ["Winnø"] = true,
    }

    local testMode, lastAction, lastName = true

    local group = Management:CreateAnimationGroup()
    group.anim = ManageGuild:CreateAnimation()
    group.anim:SetDuration(1)
    group.anim:SetOrder(1)
    group.parent = Management

    group:SetScript("OnFinished", function(self, forced)
        if forced then
            return
        end

        local didAction
        for playerIndex = 1, GetNumGuildMembers() do
            local name, _, rankIndex, level, _, _, _, _, _, _, classFileName = GetGuildRosterInfo(playerIndex)
            -- Make it consistent with values for SetGuildMemberRank() for sanity's sake.
            rankIndex = rankIndex + 1

            if rankIndex > 3 then
                if CanGuildPromote() then
                    local targetRank
                    if level >= 1 and level <= 79 and rankIndex ~= 7 then
                        targetRank = 7
                    elseif level >= 80 and level <= 84 and rankIndex ~= 6 then
                        targetRank = 6
                    elseif level >= 85 and level <= 89 and rankIndex ~= 5 then
                        targetRank = 5
                    elseif level == 90 and rankIndex ~= 4 then
                        targetRank = 4
                    end
                    if targetRank then
                        if testMode then
                            print("PROMOTE", name, targetRank)
                        else
                            if lastAction == "PROMOTE" and lastName == name then
                                return print("Action failed:", lastAction, lastName, targetRank)
                            end
                            didAction, lastAction, lastName = true, "PROMOTE", name
                            SetGuildMemberRank(playerIndex, targetRank)
                        end
                        break
                    end
                end

                if not ignoreList[name] and CanGuildRemove() then
                    local year, month, day = GetGuildRosterLastOnline(playerIndex)
                    if year and (
                        (rankIndex == 7
                        -- level 1 for more than 1 day.
                            and (day >= 1 and level == 1)
                        -- level 55 DK for more than 2 days.
                            or (day >= 2 and level == 55 and classFileName == "DEATHKNIGHT")
                        -- level 2-5 and DK level 56-58 for more than 3 days.
                            or (day >= 3 and ((level > 1 and level <= 5) or (level > 55 and level <= 58 and classFileName == "DEATHKNIGHT")))
                        -- level 6-79 for more than 14 days.
                            or (day >= 14 and level > 5 and level <= 79)
                        )
                        -- level 80-84 for more than 21 days.
                        or (rankIndex == 6 and day >= 21)
                        -- level 85-90 for more than 28 days.
                        or ((rankIndex == 4 or rankIndex == 5) and day >= 28)
                    ) then
                        if testMode then
                            print("REMOVE", name)
                        else
                            if lastAction == "REMOVE" and lastName == name then
                                -- Avoid infinite loops if the action is failing.
                                return print("Action failed:", lastAction, lastName)
                            end
                            didAction, lastAction, lastName = true, "REMOVE", name
                            GuildUninvite(name)
                        end
                        break
                    end
                end
            end
        end

        if didAction then
            -- Did something, queue the next action.
            self:Play()
        else
            -- Did nothing, we're done.
            Management:RegisterEvent("GUILD_ROSTER_UPDATE")
        end
    end)

    function Management:ManageGuild(doItForReal)
        self:UnregisterEvent("GUILD_ROSTER_UPDATE")

        SortGuildRoster("online")
        SetGuildRosterShowOffline(true)

        if group:IsPlaying() then
            group:Finish()
        end

        testMode = not not doItForReal -- force boolean
        group:Play()
    end
end