ArtifactPathFinder.lua 1.4.2 (modified)

--[[

Copyright (C) 2016 Yossa <yossa.addons@googlemail.com>
All Rights Reserved unless otherwise explicitly stated.

]]--

ArtifactPathFinder = LibStub("AceAddon-3.0"):NewAddon("ArtifactPathFinder", "AceEvent-3.0")

function ArtifactPathFinder:OnInitialize()
    if not ArtifactPathFinderCDB then
        ArtifactPathFinderCDB = {}
    end
    if not ArtifactPathFinderCDB.selectedPathBySpec then
        ArtifactPathFinderCDB.selectedPathBySpec = {}
    end
end

function ArtifactPathFinder:OnEnable()
    self:RegisterEvent("ARTIFACT_UPDATE");
    self:RegisterEvent("ARTIFACT_MAX_RANKS_UPDATE");
    self:RefreshPerkRanks();
end

function ArtifactPathFinder:OnDisable()
    self:UnregisterAllEvents()
    local powers = C_ArtifactUI.GetPowers();
    for i, powerID in ipairs(powers) do
        local powerButton = ArtifactPerksMixin.powerIDToPowerButton[powerID]
        if powerButton and powerButton.ArtifactPathIndex then
            powerButton.ArtifactPathIndex:Hide();
        end
    end
end

function ArtifactPathFinder:ARTIFACT_UPDATE()
    self:RefreshPerkRanks();
end

function ArtifactPathFinder:ARTIFACT_MAX_RANKS_UPDATE()
    self:RefreshPerkRanks();
end

function ArtifactPathFinder:GetCurrentPaths()
    local _, className = UnitClass("player");
    local artifactItemId = C_ArtifactUI.GetArtifactInfo()
    local pathTable = self.Data

    local currentPath = nil
    local currentPathName = nil

    if pathTable[className] and pathTable[className][artifactItemId] then
        return pathTable[className][artifactItemId]
    end
    return nil
end

local function GetPowerInfoMaps()
    local powers = C_ArtifactUI.GetPowers();
    if not powers then return nil end
    local powerInfo, spellInfo = {}, {}
    -- get powers info
    for i, powerID in ipairs(powers) do
        local spellID, _, currentRank, maxRank, bonusRanks, x, y = C_ArtifactUI.GetPowerInfo(powerID);
        powerInfo[powerID] = {}
        powerInfo[powerID].spellID = spellID
        powerInfo[powerID].powerID = powerID
        powerInfo[powerID].buttonIndex = i
        powerInfo[powerID].currentRank = currentRank
        powerInfo[powerID].maxRank = maxRank
        powerInfo[powerID].bonusRanks = bonusRanks
        powerInfo[powerID].x = x
        powerInfo[powerID].y = y
        powerInfo[powerID].isFinished = currentRank == maxRank;
        spellInfo[spellID] = powerInfo[powerID]
    end
    return powers, powerInfo, spellInfo
end

local brokenSpellId = {}

local function CreateIndexesForPowers(currentPath, spellInfo)
    local selectionPath = {}
    local isNextChecking = true
    --set powers indexes and next selection
    for i,spellID in ipairs(currentPath) do
        -- Start multi-spellID fix (Added by SethAluma)
        if (type(spellID) == "table") then
            for j,subSpellID in ipairs(spellID) do
                if spellInfo[subSpellID] then
                    spellID = subSpellID
                    break
                end
            end
        end
        -- End multi-spellID fix

        if not spellInfo[spellID] then
            if not brokenSpellId[spellID] then
                local _, className = UnitClass("player");
                local artifactItemId = C_ArtifactUI.GetArtifactInfo()
                DEFAULT_CHAT_FRAME:AddMessage("Artifact spell not found. Please report a bug in Path data: "..tostring(spellID).." "..className.."/"..tostring(artifactItemId))
                brokenSpellId[spellID] = 1
            end
        else
            local isNext = false
            if isNextChecking then
                if not spellInfo[spellID].isFinished then
                    isNext = true
                    isNextChecking = false
                end
            end
            local powerID = spellInfo[spellID].powerID
            selectionPath[powerID] = {}
            selectionPath[powerID].index = i
            selectionPath[powerID].isNext = isNext
        end
    end
    return selectionPath
end

local function CreateButtonOverlay(powerButton)
    local button = powerButton:CreateFontString(nil, "OVERLAY")
    button:SetFont("Fonts\\FRIZQT__.TTF", 22, "OUTLINE")
    button:SetPoint("CENTER",powerButton.Icon,"CENTER",0,0)
    return button
end

function ArtifactPathFinder:RefreshSelectedPath()
    local currentPaths = ArtifactPathFinder:GetCurrentPaths()
    local pathCount = 0
    if not currentPaths then return 0 end
    for _ in pairs(currentPaths) do pathCount = pathCount + 1 end
    local artifactItemId = C_ArtifactUI.GetArtifactInfo()

    if pathCount == 0 then
        self.selectedName = nil
        self.selectedPath = nil
    elseif pathCount == 1 or not ArtifactPathFinderCDB.selectedPathBySpec[artifactItemId] then
        self.selectedName, self.selectedPath = next(currentPaths)
    else
        self.selectedName = ArtifactPathFinderCDB.selectedPathBySpec[artifactItemId]
        if currentPaths[self.selectedName] then
            self.selectedPath = currentPaths[self.selectedName]
        else
            self.selectedName, self.selectedPath = next(currentPaths)
        end
    end
    ArtifactPathFinderCDB.selectedPathBySpec[artifactItemId] = self.selectedName
    return pathCount
end

local function DropDown_Initialize(self, level)
    local info = UIDropDownMenu_CreateInfo()
    local num = 1
    local selectedNum = nil
    for k, v in pairs(ArtifactPathFinder:GetCurrentPaths()) do
        info = UIDropDownMenu_CreateInfo()
        info.text = k
        info.value = num
        info.func = function (self)
            UIDropDownMenu_SetSelectedID(ArtifactFrame.PerksTab.ArtifactPathFinderFilter, self:GetID())
            ArtifactPathFinder.selectedPath = v
            ArtifactPathFinder.selectedName = k
            ArtifactPathFinderCDB.selectedPathBySpec[C_ArtifactUI.GetArtifactInfo()] = k
            ArtifactPathFinder:RefreshPerkRanks()
        end
        info.checked = ArtifactPathFinder.selectedName == k
        if info.checked then selectedNum = num end
        num = num + 1
        UIDropDownMenu_AddButton(info, level)
    end
    if selectedNum then
        UIDropDownMenu_SetSelectedID(ArtifactFrame.PerksTab.ArtifactPathFinderFilter, selectedNum)
    end
end

function ArtifactPathFinder:RefreshFilter()
    local pathCount = ArtifactPathFinder:RefreshSelectedPath()
    if pathCount < 2 then
        if ArtifactFrame.PerksTab.ArtifactPathFinderFilter then
            ArtifactFrame.PerksTab.ArtifactPathFinderFilter:Hide()
        end
        return
    end

    if not ArtifactFrame.PerksTab.ArtifactPathFinderFilter then
        ArtifactFrame.PerksTab.ArtifactPathFinderFilter = CreateFrame("Frame", "ArtifactPathFinderFilter", ArtifactFrame.PerksTab, "UIDropDownMenuTemplate")
    else
        ArtifactFrame.PerksTab.ArtifactPathFinderFilter:Show()
    end
    local FilterDropDown = ArtifactFrame.PerksTab.ArtifactPathFinderFilter
    UIDropDownMenu_Initialize(FilterDropDown, DropDown_Initialize)
    UIDropDownMenu_SetWidth(FilterDropDown, 200);
    UIDropDownMenu_SetButtonWidth(FilterDropDown, 144)
    UIDropDownMenu_JustifyText(FilterDropDown, "LEFT")
    FilterDropDown:SetPoint("TOPRIGHT", ArtifactFrame.PerksTab, "TOPRIGHT", 0, -50)
end

function ArtifactPathFinder:HideAll()
    local powers = C_ArtifactUI.GetPowers();
    for i, powerID in ipairs(powers) do
        local powerButton = ArtifactFrame.PerksTab:GetOrCreatePowerButton(i)
        if powerButton and powerButton.ArtifactPathIndex then
            powerButton.ArtifactPathIndex:Hide()
            powerButton.StarBurst:SetAlpha(0)
        end
    end
    if ArtifactFrame.PerksTab.ArtifactPathFinderFilter then
        ArtifactFrame.PerksTab.ArtifactPathFinderFilter:Hide()
    end
end

function ArtifactPathFinder:RefreshPerkRanks()
    if not ArtifactFrame or not ArtifactFrame.PerksTab then return end

    self:RefreshFilter()

    local currentPath = self.selectedPath
    if currentPath then
        local powers, powerInfo, spellInfo = GetPowerInfoMaps();
        if not powers then return end
        local selectionPath = CreateIndexesForPowers(currentPath, spellInfo)

        --fill all buttons
        for i, powerID in ipairs(powers) do
            local powerButton = ArtifactFrame.PerksTab:GetOrCreatePowerButton(i)
            if powerButton then
                if not powerButton.ArtifactPathIndex then
                    powerButton.ArtifactPathIndex = CreateButtonOverlay(powerButton)
                else
                    powerButton.ArtifactPathIndex:Show();
                end
                powerButton.StarBurst:SetAlpha(0)
                powerButton.ArtifactPathIndex:SetTextColor(1, 1, 1, 1)
                powerButton.ArtifactPathIndex:SetText("")

                if selectionPath[powerID] then
                    if (C_ArtifactUI.GetTotalPurchasedRanks() == 0 and selectionPath[powerID].index>2) or
                       (selectionPath[powerID].index == 18 and not selectionPath[powerID].isNext)
                    then
                        powerButton.ArtifactPathIndex:Hide()
                    else
                        powerButton.ArtifactPathIndex:SetText(tostring(selectionPath[powerID].index));
                        if selectionPath[powerID].isNext then
                            powerButton.ArtifactPathIndex:SetTextColor(1, 0, 0, 1)
                            powerButton.StarBurst:SetAlpha(0.9)
                        end
                    end
                end
            end
        end
    end
end