HandyNotes_Achievements.lua

-- Copyright 2015-2020, r. brian harrison.  all rights reserved.

local ADDON_NAME = ...
local HNA = LibStub("AceAddon-3.0"):NewAddon(ADDON_NAME, "AceEvent-3.0", "AceTimer-3.0")
if not HNA then return end

local AceLocale = LibStub:GetLibrary("AceLocale-3.0")
assert(AceLocale, string.format("%s requires %s", ADDON_NAME, "AceLocale-3.0")) -- not localizable

local L = AceLocale:GetLocale(ADDON_NAME)
assert(L, string.format("%s requires %s", ADDON_NAME, GetLocale())) -- not localizable

local AceDB = LibStub:GetLibrary("AceDB-3.0")
assert(AceDB, string.format(L["%s requires %s"], ADDON_NAME, "AceDB-3.0"))

local AchievementLocations = LibStub:GetLibrary("AchievementLocations-1.0")
assert(AchievementLocations, string.format(L["%s requires %s"], ADDON_NAME, "AchievementLocations-1.0"))

local InstanceLocations = LibStub:GetLibrary("InstanceLocations-1.0")
assert(InstanceLocations, string.format(L["%s requires %s"], ADDON_NAME, "InstanceLocations-1.0"))

local InSeason = LibStub:GetLibrary("InSeason-1.1")
assert(InSeason, string.format(L["%s requires %s"], ADDON_NAME, "InSeason-1.1"))

local HandyNotes = LibStub("AceAddon-3.0"):GetAddon("HandyNotes", true)
assert(HandyNotes, string.format(L["%s requires %s"], ADDON_NAME, "HandyNotes"))

local QTip = LibStub:GetLibrary("LibQTip-1.0")
assert(QTip, string.format(L["%s requires %s"], ADDON_NAME, "LibQTip-1.0"))

HNA.ICON_PATH = "Interface/AchievementFrame/UI-Achievement-TinyShield"
HNA.NEAR = 0.03
HNA.DEFAULT_COORD = 50005000
HNA.ZONE_COORD = 50005000

local EMPTY = {}
local visible = {}
local tooltip
local activeNodes = {}

function HNA:GetFactionInfoByID(factionID)
    if not factionID then return nil end

    local fi = C_Reputation.GetFactionDataByID(factionID)
    if fi then
        return fi.name, fi.description, fi.reaction,
            fi.currentReactionThreshold, fi.nextReactionThreshold,
            fi.currentStanding, fi.atWarWith, fi.canToggleAtWar, fi.isHeader,
            fi.isCollapsed, nil, fi.isWatched, fi.isChild, fi.factionID,
            fi.hasBonusRepGain, nil,
            -- 11.0 and above
            fi.isAccountWide, fi.isHeaderWithRep, fi.canSetInactive
    end
end

HNA.FACTION_STANDING_LABELS = {
    [1] = "Hated",
    [2] = "Hostile",
    [3] = "Unfriendly",
    [4] = "Neutral",
    [5] = "Friendly",
    [6] = "Honored",
    [7] = "Revered",
    [8] = "Exalted"
}

function HNA:GetAchievementCriteriaInfoByDescription(achievementID, description)
    for i = 1, GetAchievementNumCriteria(achievementID) do
        local retval = {GetAchievementCriteriaInfo(achievementID, i)}
        if description == retval[1] then
            return unpack(retval, 1, 10)
        end
    end
end


function HNA:RGBToColorCode(rgba)
    local a = rgba.a
    if a == nil then a = 1.0 end
    return string.format("|c%02x%02x%02x%02x", a * 255, rgba.r * 255, rgba.g * 255, rgba.b * 255)
end

function HNA:HandyNotesCoordsNear(c, coord)
    -- within 3% of the map
    local x, y = HandyNotes:getXY(c)
    local x2, y2 = HandyNotes:getXY(coord)
    local dx = x - x2
    local dy = y - y2
    return dx * dx + dy * dy < self.NEAR * self.NEAR
end


function SortByTrackedAchievement(nodes)
    function getTrackedAchievements()
        id1, id2, id3, id4, id5, id6, id7, id8, id9, id10 = GetTrackedAchievements()

        local result = {}

        function add(id, result)
            if id ~= nil then
                result[tonumber(id)] = tonumber(id)
            end
        end

        add(id1, result)
        add(id2, result)
        add(id3, result)
        add(id4, result)
        add(id5, result)
        add(id6, result)
        add(id7, result)
        add(id8, result)
        add(id9, result)
        add(id10, result)

        return result
    end
    local trackedAchievements = getTrackedAchievements()
    local sortedNodes = {}
    local notTrackedNodes = {}

    for nodeIndex = 1, #nodes, 2 do
        local row = nodes[nodeIndex + 1]
        local achievementID = row[2]

        if trackedAchievements[achievementID] then
            table.insert(sortedNodes, nodes[nodeIndex])
            table.insert(sortedNodes, nodes[nodeIndex + 1])
        else
            table.insert(notTrackedNodes, nodes[nodeIndex])
            table.insert(notTrackedNodes, nodes[nodeIndex + 1])
        end
    end

    for nodeIndex = 1, #notTrackedNodes, 2 do
        table.insert(sortedNodes, notTrackedNodes[nodeIndex])
        table.insert(sortedNodes, notTrackedNodes[nodeIndex + 1])
    end

    return sortedNodes
end

function HNA:OnEnter(mapFile, nearCoord)
    tooltip = QTip:Acquire(ADDON_NAME, 2, "LEFT", "RIGHT")
    local firstRow = true
    local previousAchievementID

    local nodes = activeNodes[mapFile]

    if HNA.db.profile.sort_by_tracked == true then
        nodes = SortByTrackedAchievement(nodes)
    end

    for nodeIndex = 1, #nodes, 2 do
        local coord, row = nodes[nodeIndex], nodes[nodeIndex + 1]
        if HNA:HandyNotesCoordsNear(coord, nearCoord) and HNA:Valid(row) then
            local achievementID = row[2]
            local criterion = row.criterion

            local _, name, points, completed, _, _, _, description, _, _, _, _, _, _, _ = GetAchievementInfo(achievementID)

            if achievementID ~= previousAchievementID then
                if not firstRow then
                    tooltip:AddSeparator(2, 0, 0, 0, 0)
                    tooltip:AddSeparator(1, 1, 1, 1, 0.5)
                    tooltip:AddSeparator(2, 0, 0, 0, 0)
                end
                firstRow = false

                tooltip:SetHeaderFont(GameFontGreenLarge)
                tooltip:AddHeader(name)

                tooltip:SetFont(GameTooltipTextSmall)
                tooltip:AddLine(description)
                previousAchievementID = achievementID
            end

            if row.faction then
                local name, _, standing = HNA:GetFactionInfoByID(row.faction)
                tooltip:AddSeparator(2, 0, 0, 0, 0)
                tooltip:SetFont(GameTooltipTextSmall)
                local reputation = HNA:RGBToColorCode(FACTION_BAR_COLORS[standing]) .. (HNA.FACTION_STANDING_LABELS[standing] or "Unknown") .. "|r"
                tooltip:AddLine(name, reputation)
            end

            if criterion then
                local criterionDescription, quantityString
                if type(criterion) == "number" then
                    criterionDescription, _, _, _, _, _, _, _, quantityString, _ = GetAchievementCriteriaInfoByID(achievementID, criterion)
                else
                    criterionDescription, _, _, _, _, _, _, _, quantityString, _ = HNA:GetAchievementCriteriaInfoByDescription(achievementID, criterion)
                end

                if quantityString == "0" then
                    quantityString = ""
                end

                if criterionDescription then
                    tooltip:AddSeparator(2, 0, 0, 0, 0)
                    tooltip:SetFont(GameTooltipTextSmall)
                    tooltip:AddLine(criterionDescription, quantityString)
                end
            end

        end
    end

    tooltip:SmartAnchorTo(self)
    tooltip:Show()
end


function HNA:OnLeave(mapFile, coord)
    QTip:Release(tooltip)
end


function HNA:OnClick(button, down, mapFile, nearCoord)
    if not down then
        if not AchievementFrame then
            AchievementFrame_LoadUI()
        end

        local shown = {}

        for nodeIndex = 1, #activeNodes[mapFile], 2 do
            local nodes = activeNodes[mapFile]
            local coord, row = nodes[nodeIndex], nodes[nodeIndex + 1]
            if HNA:HandyNotesCoordsNear(coord, nearCoord) and HNA:Valid(row) then
                local achievementID = row[2]
                if not shown[achievementID] then
                    shown[achievementID] = true
                    print(GetAchievementLink(achievementID))
                end
                --ShowUIPanel(AchievementFrame)
                --AchievementFrame_SelectAchievement(achievementID)
            end
        end
    end
end


function HNA:OnInitialize()
    self:RegisterEvent("PLAYER_ENTERING_WORLD")

    local defaults = {
        profile = {
            enabled = true,
            icon_scale = 2.0,
            icon_alpha = 1.0,
            just_mine = false,
            season_warning = 14,
            clean_continents = true,
            completed = false,
            sort_by_tracked = false,
        }
    }

    self.db = AceDB:New("HandyNotesAchievementsDB", defaults, true)
end


local function notifyUpdate(frame, event)
    -- print(string.format("%s:%s()", ADDON_NAME, event))
    HNA:UpdateVisible()
    HNA:SendMessage("HandyNotes_NotifyUpdate", ADDON_NAME)
end


function HNA:PLAYER_ENTERING_WORLD(event)
    self:UnregisterEvent("PLAYER_ENTERING_WORLD")
    -- XXX causes low end systems to sieze
    --self:RegisterEvent("ACHIEVEMENT_EARNED")
    --self:RegisterEvent("CRITERIA_COMPLETE")
    --self:RegisterEvent("CRITERIA_EARNED")
    --self:RegisterEvent("CRITERIA_UPDATE")
    --self:RegisterEvent("QUEST_COMPLETE")
    -- XXX event for "... has been added to your pet journal"

    -- XXX ...
    local options = {
        type = "group",
        name = "Achievements",
        get = function(info)
            return self.db.profile[info.arg]
        end,
        set = function(info, v)
            self.db.profile[info.arg] = v
            notifyUpdate()
        end,
        args = {
            icon_scale = {
                type = "range",
                name = L["Icon Scale"],
                desc = L["The size of the icons."],
                min = 0.3, max = 5, step = 0.1,
                arg = "icon_scale",
                order = 1,
            },
            icon_alpha = {
                type = "range",
                name = L["Icon Alpha"],
                desc = L["The transparency of the icons."],
                min = 0, max = 1, step = 0.01,
                arg = "icon_alpha",
                order = 2,
            },
            completed = {
                type = "toggle",
                name = L["Show Completed"],
                desc = L["Show map pins for achievements you have completed."],
                width = "full",
                arg = "completed",
                order = 3,
            },
            clean_continents = {
                type = "toggle",
                name = L["Consolidate Zone Pins"],
                desc = L["Show fewer map pins."],
                width = "full",
                arg = "clean_continents",
                order = 4,
            },
            just_mine = {
                type = "toggle",
                name = L["Just Mine"],
                desc = L["Show more map pins by including achievements completed only by other characters."],
                width = "full",
                arg = "just_mine",
                order = 5,
            },
            season_warning = {
                type = "range",
                name = L["Season Warning"],
                desc = L["Days in advance to show pins for seasonal holiday achievements."],
                min = 0, max = 60, step = 1,
                arg = "season_warning",
                order = 6,
            },
            sort_by_tracked = {
                type = "toggle",
                name = L["Sort by tracked"],
                desc = L["Sort achievements by tracked"],
                width = "full",
                arg = "sort_by_tracked",
                order = 7
            }
        },
    }
    HandyNotes:RegisterPluginDB(ADDON_NAME, self, options)
    notifyUpdate(nil, event)
end


HNA.ACHIEVEMENT_EARNED = notifyUpdate
HNA.CRITERIA_COMPLETE  = notifyUpdate
HNA.CRITERIA_EARNED    = notifyUpdate
HNA.CRITERIA_UPDATE    = notifyUpdate
HNA.QUEST_COMPLETE     = notifyUpdate


function HNA:UpdateVisible()
    for _, categoryID in ipairs(GetCategoryList()) do
        for i = 1, GetCategoryNumAchievements(categoryID) do
            local achievementID = GetAchievementInfo(categoryID, i)
            if achievementID then
                visible[achievementID] = true
            end
        end
    end
end


function HNA:Valid(row)
    local achievementID = row[2]
    if not visible[achievementID] then return false end

    local factionGroup = UnitFactionGroup("player")
    if row.side ~= nil and row.side ~= "both" and row.side ~= string.lower(factionGroup) then
        return false
    end

    if self.db.profile.completed then return true end

    local _, _, _, completed, _, _, _, _, _, _, _, _, earnedByMe, _ = GetAchievementInfo(achievementID)
    if completed and (earnedByMe or not self.db.profile.just_mine) then return false end

    if type(row.criterion) == "number" then
        local criteriaDescription, _, completed = GetAchievementCriteriaInfoByID(achievementID, row.criterion)
        if completed or not criteriaDescription then return false end
    elseif type(row.criterion) == "string" then
        local criteriaDescription, _, completed = HNA:GetAchievementCriteriaInfoByDescription(achievementID, row.criterion)
        if completed or not criteriaDescription then return false end
    end

    if row.quest then
        completed = C_QuestLog.IsQuestFlaggedCompleted(row.quest)
        if completed then return false end
    end

    if row.faction then
        local _, _, standing = C_Reputation.GetFactionDataByID(row.faction)
        completed = (standing == 8)
        if completed then return false end
    end

    if row.season then
        local days = InSeason:TimeUntilHoliday(row.season)
        if days ~= nil and days > self.db.profile.season_warning then return false end
    end

    return true
end


function HNA:GetNodes(mapFile, minimap, dungeonLevel)
    -- recursive function to generate valid achievements, sometimes projected into outer zones or consolidated pins
    local function validRows(mapFile, dungeonLevel, overrideMapFile, overrideCoord)
        local rows = AchievementLocations:Get(mapFile)
        for _, row in ipairs(rows or EMPTY) do
            if (dungeonLevel or row.floor) == (row.floor or dungeonLevel) then
                if self:Valid(row) then
                    local coord = row[3] and row[4] and HandyNotes:getCoord(row[3], row[4])
                    coroutine.yield(overrideMapFile or mapFile, overrideCoord or coord or self.DEFAULT_COORD, row)
                end
            end
        end

        local zones = HandyNotes:GetContinentZoneList(mapFile)
        for _, subMap in ipairs(zones or EMPTY) do
            local subMapFile = HandyNotes:GetMapIDtoMapFile(subMap)
            -- put this zone's achievements on the world map, all on one pin
            -- overrideMapFile and overrideCoord are likely nil
            validRows(subMapFile, nil, overrideMapFile, overrideCoord or (self.db.profile.clean_continents and self.ZONE_COORD))
        end

        local instances = InstanceLocations:GetBelow(mapFile)
        for _, instanceMapFile in ipairs(instances or EMPTY) do
            local overrideMapFile, instanceX, instanceY = unpack(InstanceLocations:GetLocation(instanceMapFile))
            local coord = instanceX and instanceY and HandyNotes:getCoord(instanceX, instanceY)
            validRows(instanceMapFile, nil, overrideMapFile, overrideCoord or coord)
        end
    end

    for _, nodes in pairs(activeNodes) do
        -- XXX skip minimap's
        wipe(nodes)
    end

    local rowsCo = coroutine.create(validRows)
    return function(state, value)
        local status, mF, coord, row = coroutine.resume(rowsCo, mapFile, dungeonLevel)
        if not status then
            print(string.format("|cffff0000%s Error:|r %s", ADDON_NAME, tostring(mF)))
            return nil
        end
        if not row then
            return nil
        end
        if not activeNodes[mF] then
            activeNodes[mF] = {}
        end
        table.insert(activeNodes[mF], coord)
        table.insert(activeNodes[mF], row)
        -- HandyNotes does iterators wrong: the first value should be a iterator variable (cursor), eliminating the need for a "value" closure or coroutine
        -- added row for OnEnter
        return coord, mF, self.ICON_PATH, self.db.profile.icon_scale, self.db.profile.icon_alpha, nil, row
    end, nil
end