--## Version: 1.5 ## Author: Siku
-- Reward Types
local MOUNT = MOUNT or "Mount"
local PET = PET or "Pet"
local TOY = TOY or "Toy"
local HEIRLOOM = ITEM_QUALITY7_DESC or "Heirloom"
local ESSENCE = AZERITE_ESSENCE_ITEM_TYPE or "Essence"
-- --local RECIPE = TRANSMOG_SOURCE_6

-- Others
local REPUTATION_PROGRESS_FORMAT = REPUTATION_PROGRESS_FORMAT or "%s / %s"
local RETRIEVING_ITEM_INFO = RETRIEVING_ITEM_INFO or "Retrieving item information"
local MISSING = ADDON_MISSING or "Missing"
local KNOWN = ITEM_SPELL_KNOWN or "Already Known"
local REWARDS = REWARDS or "Rewards"

local MEDALS_ID = {Alliance = 1717, Horde = 1716}
local CALLINGS = CALLINGS_QUESTS or "Callings"

local ASSAULTS_HEADER = {
    [63543] = "Necrolord Assault",
    [63823] = "Night Fae Assault",
    [63824] = "Kyrian Assault",
    [64554] = "Venthyr Assault"
}
local localizedQuestNames = {}

local function Known() return string.format('(|cFF00FF00%s|r)', KNOWN) end
local function Missing() return string.format('(|cFFFF0000%s|r)', MISSING) end

-------------------------------------------------------------------------------
----------------------------------- Mixin -------------------------------------
-------------------------------------------------------------------------------

---@class Reward
---@field rewardType string
---@field cost number # If given, will be listed next to reward type.
---@field itemLink string
---@field itemIcon number
---@field item number # ItemID of the item giving the reward
---@field id number # Type-specific ID to check if reward has been collected
local RewardMixin = {}

function RewardMixin:IsCollected()
    if self.rewardType == MOUNT then
        return select(11, C_MountJournal.GetMountInfoByID(self.id))
        
    elseif self.rewardType == PET then
        return C_PetJournal.GetNumCollectedInfo(self.id) > 0
        
    elseif self.rewardType == TOY then
        return PlayerHasToy(self.item)
        
    elseif self.rewardType == HEIRLOOM then
        return C_Heirloom.PlayerHasHeirloom(self.item)
        
    elseif self.rewardType == ESSENCE then
        local info = C_AzeriteEssence.GetEssenceInfo(self.id)
        return info and info.rank == 4
        
    end
end

--- Find if a character can use a given reward.
---@return boolean
function RewardMixin:IsUsable()
    if self.rewardType == ESSENCE then
        local info = C_AzeriteEssence.GetEssenceInfo(self.id)
        return type(info) == "table"
        -- --elseif self.rewardType == RECIPE then
    end
    return true
end

function RewardMixin:GetRewardText()
    if self.cost then
        return string.format("%s (%s, %d)", self.itemLink, self.rewardType, self.cost)
    end
    return string.format("%s (%s)", self.itemLink, self.rewardType)
end

--- Main function to turn the Reward data into user-facing text.
---@param tooltip GameTooltip
function RewardMixin:Render(tooltip)
    if self:IsUsable() then
        local collected = self:IsCollected()
        local status = collected and Known() or Missing()
        tooltip:AddDoubleLine(self:GetRewardText(), status)
        tooltip:AddTexture(self.itemIcon) --  {margin={left=10, top=2, right=2}}
    end
end

--- Reward constructor
---@param rewardType string
---@param data table
---@return Reward
local function AddReward(rewardType, data)
    data.rewardType = rewardType
    data.itemLink = RETRIEVING_ITEM_INFO
    data.itemIcon = 'Interface\\Icons\\Inv_misc_questionmark'
    local item = Item:CreateFromItemID(data.item)
    item:ContinueOnItemLoad(function()
        data.itemLink = item:GetItemLink()
        data.itemIcon = item:GetItemIcon()
    end)
    return Mixin(data, RewardMixin)
end

-------------------------------------------------------------------------------
---------------------------------- REWARDS ------------------------------------
-------------------------------------------------------------------------------
--To gather ID from mounts and Pets:
-- function PR_GetPet(name) print(C_PetJournal.FindPetIDByName(name),"=",name) end
-- function PR_GetMount(name) for i, m in ipairs(C_MountJournal.GetMountIDs()) do
--     local mountName = C_MountJournal.GetMountInfoByID(m)
--     if strfind(mountName, name) then print(m, "=", mountName) end
-- end end

local RewardList = {
    
    -- Shadowlands Paragon
    [2407] = {
        -- The Ascended
        AddReward(PET, {item = 184399, id = 3064}),
        AddReward(TOY, {item = 184396}),
    },
    [2410] = {
        -- The Undying Army
        AddReward(MOUNT, {item = 182081, id = 1350}),
        AddReward(PET, {item = 181269, id = 2959}),
        AddReward(TOY, {item = 184495}),
    },
    [2413] = {
        -- Court of Harvesters
        AddReward(PET, {item = 180601, id = 3006}),
    },
    [2465] = {
        -- The Wild Hunt
        AddReward(MOUNT, {item = 183800, id = 1428}),
        AddReward(PET, {item = 180635, id = 2916}),
    },
    [2432] = {
        -- Ve'nari
        AddReward(MOUNT, {item = 186657, id = 1501}),
        AddReward(PET, {item = 186552, id = 3133}),
    },
    [2470] = {
        -- Death Advance
        AddReward(MOUNT, {item = 186644, id = 1455}),
        AddReward(MOUNT, {item = 186649, id = 1508}),
        AddReward(PET, {item = 186541, id = 3137}),
    },
    [2472] = {
        -- Archivist Codex
        AddReward(MOUNT, {item = 186641, id = 1454}),
        AddReward(PET, {item = 186538, id = 3140}),
    },
    [2478] = {
        -- The Enlightened
        AddReward(MOUNT, {item = 188810, id = 1571}),
        AddReward(TOY, {item = 190177}),
    },

    -- BFA Alliance
    [2160] = {
        -- Proudmoore Admiralty
        AddReward(TOY, {item = 166702}),
        AddReward(PET, {item = 166714, id = 2566}),
    },
    [2161] = {
        -- Order of Embers
        AddReward(TOY, {item = 166808}),
        AddReward(PET, {item = 166718, id = 2568}),
    },
    [2162] = {
        -- Storm's Wake
        AddReward(PET, {item = 166719, id = 2569}),
    },
    [2159] = {
        -- 7th Legion
        AddReward(TOY, {item = 166879}),
        -- AddReward(RECIPE, {item = 166279}),
    },
    [2400] = {
        -- Waveblade Ankoan
        AddReward(TOY, {item = 170203}),
        AddReward(TOY, {item = 170469}),
        -- AddReward(RECIPE, {item = 170169}),
        AddReward(MOUNT, {item = 169198, id = 1237}),
        AddReward(ESSENCE, {item = 168866, id = 28}),
        AddReward(ESSENCE, {item = 168931, id = 17}),
        AddReward(ESSENCE, {item = 168840, id = 25}),
    },
    
    -- BFA Horde
    [2103] = {
        -- -Zandalari Empire
        AddReward(TOY, {item = 166701}),
    },
    [2156] = {
        -- Talanji's Expedition
        AddReward(TOY, {item = 166308}),
        AddReward(PET, {item = 166716, id = 2567}),
    },
    [2158] = {
        -- Voldunai
        AddReward(TOY, {item = 166703}),
        AddReward(TOY, {item = 165021}),
        AddReward(TOY, {item = 166880}),
    },
    [2157] = {
        -- Honorbound
        AddReward(TOY, {item = 166879}),
        -- AddReward(RECIPE, {item = 166311}),
    },
    [2373] = {
        -- The Unshackled
        AddReward(TOY, {item = 170203}),
        AddReward(TOY, {item = 170469}),
        -- AddReward(RECIPE, {item = 170169}),
        AddReward(MOUNT, {item = 169198, id = 1237}),
        AddReward(ESSENCE, {item = 168866, id = 28}),
        AddReward(ESSENCE, {item = 168931, id = 17}),
        AddReward(ESSENCE, {item = 168840, id = 25}),
    },
    
    -- BFA Neutral
    [2163] = {
        -- Tortollan Seekers
        AddReward(TOY, {item = 166704}),
        -- AddReward(RECIPE, {item = 166264}),
    },
    [2164] = {
        -- Champions of Azeroth
        AddReward(TOY, {item = 166877}),
    },
    [2391] = {
        -- Rustbolt Resistance
        AddReward(ESSENCE, {item = 168861, id = 6}),
        AddReward(ESSENCE, {item = 168935, id = 19}),
        AddReward(ESSENCE, {item = 168569, id = 13}),
    },
    [2417] = {
        -- Uldum Accord
        AddReward(PET, {item = 174481, id = 2850}),
        AddReward(ESSENCE, {item = 173283, id = 35}),
    },
    [2415] = {
        -- Rajani
        AddReward(PET, {item = 174479, id = 2852}),
        AddReward(ESSENCE, {item = 168816, id = 24}),
        AddReward(ESSENCE, {item = 168576, id = 33}),
    },
    
    -- Legion Paragons
    [1900] = {
        -- Court of Farondis
        AddReward(MOUNT, {item = 147806, id = 943}),
    },
    [1828] = {
        -- Highmountain Tribe
        AddReward(MOUNT, {item = 147807, id = 941}),
    },
    [1883] = {
        -- Dreamweavers
        AddReward(MOUNT, {item = 147804, id = 942}),
    },
    [1948] = {
        -- Valarjar
        AddReward(MOUNT, {item = 147805, id = 944}),
    },
    [1859] = {
        -- The Nightfallen
        AddReward(MOUNT, {item = 143764, id = 905}),
    },
    [1894] = {
        -- The Wardens
        AddReward(TOY, {item = 147843}),
    },
    [2045] = {
        -- Armies of Legionfall
        AddReward(PET, {item = 147841, id = 2050}),
    },
    [2165] = {
        -- Army of the Light
        AddReward(TOY, {item = 153182}),
        AddReward(MOUNT, {item = 153044, id = 985}),
        AddReward(MOUNT, {item = 153043, id = 984}),
        AddReward(MOUNT, {item = 153042, id = 983}),
    },
}

local ServiceMedals = {
    ["Alliance"] = {
        AddReward(HEIRLOOM, {item = 166770, cost = 75}),
        AddReward(HEIRLOOM, {item = 166766, cost = 75}),
        AddReward(HEIRLOOM, {item = 166767, cost = 75}),
        AddReward(HEIRLOOM, {item = 166768, cost = 75}),
        AddReward(HEIRLOOM, {item = 166769, cost = 75}),
        AddReward(PET, {item = 166346, id = 2539, cost = 100}),
        AddReward(TOY, {item = 166744, cost = 125}),
        AddReward(MOUNT, {item = 166465, id = 1214, cost = 300}),
        AddReward(MOUNT, {item = 166463, id = 1216, cost = 750}),
        AddReward(MOUNT, {item = 166436, id = 1204, cost = 350}),
    },
    ["Horde"] = {
        AddReward(HEIRLOOM, {item = 166752, cost = 75}),
        AddReward(HEIRLOOM, {item = 166756, cost = 75}),
        AddReward(HEIRLOOM, {item = 166755, cost = 75}),
        AddReward(HEIRLOOM, {item = 166754, cost = 75}),
        AddReward(HEIRLOOM, {item = 166753, cost = 75}),
        AddReward(PET, {item = 166347, id = 2540, cost = 100}),
        AddReward(TOY, {item = 166743, cost = 125}),
        AddReward(MOUNT, {item = 166464, id = 1215, cost = 300}),
        AddReward(MOUNT, {item = 166469, id = 1210, cost = 750}),
        AddReward(MOUNT, {item = 166436, id = 1204, cost = 350}),
    },
}

local CallingRewards = {
    [2407] = {
        -- Bastion
        AddReward(TOY, {item = 187419}),
    },
    [2410] = {
        -- Necrolords
        AddReward(MOUNT, {item = 184160, id = 1438}),
        AddReward(MOUNT, {item = 184161, id = 1439}),
        AddReward(MOUNT, {item = 184162, id = 1440}),
        AddReward(TOY, {item = 187913}),
    },
    [2413] = {
        -- Court of Harvesters
        AddReward(TOY, {item = 187512}),
    },
    [2465] = {
        -- The Wild Hunt
        AddReward(TOY, {item = 187840}),
    },

}

local AssaultRewards = {
    -- Necrolord = 63543, NightFae = 63823, Kyrian = 63824, Venthyr = 64554
    [63543] = {
        AddReward(MOUNT, {item = 186103, id = 1477}),
        AddReward(PET, {item = 185992, id = 3114}),
    },
    [63823] = {
        AddReward(MOUNT, {item = 186000, id = 1476}),
        AddReward(PET, {item = 186547, id = 3116}),
    },
    [63824] = {
        AddReward(PET, {item = 186546, id = 3103}),
        AddReward(TOY, {item = 187185}),
    },
    [64554] = {
        AddReward(MOUNT, {item = 185996, id = 1378}),
    },

}

-------------------------------------------------------------------------------
--------------------------------- REPUTATION ----------------------------------
-------------------------------------------------------------------------------

local function GetParagonBarValues(factionID)
    local currentValue, rewardThreshold, _,  rewardPending = C_Reputation.GetFactionParagonInfo(factionID)
    currentValue = (currentValue - rewardThreshold) % rewardThreshold
    
    if rewardPending then
        return currentValue + rewardThreshold, rewardThreshold
    else
        return currentValue, rewardThreshold
    end
end

local function DisplayServiceMedalsRewards(tooltip)
    local faction = UnitFactionGroup("player")
    local rewards = ServiceMedals[faction]
    local currencyInfo = C_CurrencyInfo.GetCurrencyInfo(MEDALS_ID[faction])
    tooltip:AddLine(" ")
    tooltip:AddLine(string.format("%s %s", currencyInfo.name, REWARDS))
    for i = 1, #rewards do
        rewards[i]:Render(tooltip)
    end
end

local function DisplayCovenantCallings(tooltip, faction)
    local rewards = CallingRewards[faction]
    tooltip:AddLine(" ")
    tooltip:AddLine(string.format("%s %s", CALLINGS, REWARDS))
    for i = 1, #rewards do
        rewards[i]:Render(tooltip)
    end
end

local function DisplayCovenantAssaults(tooltip)
    tooltip:AddLine(" ")
    for id, name in pairs(ASSAULTS_HEADER) do
        local rewards = AssaultRewards[id]
        
        if not localizedQuestNames[id] then
            local title = C_QuestLog.GetTitleForQuestID(id)
            if title then localizedQuestNames[id] = title end
        end
        tooltip:AddLine(localizedQuestNames[id] or name)
        for i = 1, #rewards do
            rewards[i]:Render(tooltip)
        end
    end
end

local function UpdateParagonRewards(frame)
    local tooltip = EmbeddedItemTooltip
    local rewards = RewardList[frame.factionID]
    if rewards and #rewards > 0 then
        for i = 1, #rewards do
            rewards[i]:Render(tooltip)
        end
        
        if frame.factionID == 2159 or frame.factionID == 2157 then
            DisplayServiceMedalsRewards(tooltip)
        elseif frame.factionID == 2470 then
            DisplayCovenantAssaults(tooltip)
        elseif frame.factionID == 2407 or frame.factionID == 2410 or frame.factionID == 2413 or frame.factionID == 2465 then
            DisplayCovenantCallings(tooltip, frame.factionID)
        end
        
        GameTooltip_AddBlankLineToTooltip(tooltip)
        tooltip:Show()
    end
end

--- Update the Paragon Bars
---@param entry ReputationEntryMixin
local function UpdateNewParagonBars(entry)
    local factionBar = entry.Content.ReputationBar
    local factionID = entry.factionID
    local factionIndex = entry.factionIndex
    if ( factionID and C_Reputation.IsFactionParagon(factionID) and C_Reputation.IsFactionActive(factionIndex) ) then
        local barValue, barMax = GetParagonBarValues(factionID)
        factionBar:SetMinMaxValues(0, barMax)
        factionBar:SetValue(barValue)
        factionBar.barProgressText = string.format(REPUTATION_PROGRESS_FORMAT, BreakUpLargeNumbers(barValue), BreakUpLargeNumbers(barMax))
        factionBar.reputationStandingText = "⭐"
    end
end

hooksecurefunc(ReputationEntryMixin, "TryInitParagonDisplay", UpdateNewParagonBars)
hooksecurefunc(ReputationSubHeaderMixin, "TryInitParagonDisplay", UpdateNewParagonBars)
hooksecurefunc("ReputationParagonFrame_SetupParagonTooltip", UpdateParagonRewards)
