local _,rematch = ...
local L = rematch.localization
local C = rematch.constants
local settings = rematch.settings
rematch.toolbar = rematch.frame.ToolBar
rematch.frame:Register("toolbar")

-- toolbarLayout is the toolbar layout used by each mode (0-minimized, 1-single, etc.). Notes:
-- These are listed in the displayed order but put in reverse in workingLayout because buttons are built from right to left.
-- If an entry is a string ("HealButton") it refers to the rematch.toolbar[String] button: rematch.toolbar.HealButton
-- If an entry is a number, it refers to an index into a PetSatchelButtons record. (should be 1-2 for now, but may expand later)
local toolbarLayouts = {
    [0] = {"HealButton","BandageButton","SafariHatButton",1,2,"PetSatchelButton","SummonPetButton","FindBattleButton"},
    [1] = {"HealButton","BandageButton","SafariHatButton",1,2,"PetSatchelButton","SummonPetButton"},
    [2] = {"HealButton","BandageButton","SafariHatButton","LesserPetTreatButton","PetTreatButton","LevelingStoneButton","RarityStoneButton","ImportTeamButton","ExportTeamButton","RandomTeamButton","SummonPetButton"},
    [3] = {"HealButton","BandageButton","SafariHatButton","LesserPetTreatButton","PetTreatButton","LevelingStoneButton","RarityStoneButton","ImportTeamButton","ExportTeamButton","RandomTeamButton","SummonPetButton"}
}

-- the toolbar layout currently being used, based on the above layouts and adjusted for PetSatchelIndex and ReverseToolbar
local displayLayout = {}

-- for toolbars that have the PetSatchelButton, the button will cycle through these buttons
rematch.toolbar.petSatchelButtons = {
    {"LesserPetTreatButton","PetTreatButton"},
    {"LevelingStoneButton","RarityStoneButton"},
    {"ImportTeamButton","ExportTeamButton"},
    {"RandomTeamButton","SaveAsButton"},
}

rematch.events:Register(rematch.toolbar,"PLAYER_LOGIN",function(self)
    -- steal the TopTileStreaks inherited from BasicFrameTemplate and put them behind the toolbar
    rematch.frame.TopTileStreaks:SetParent(rematch.toolbar)
    rematch.frame.TopTileStreaks:SetPoint("TOPLEFT")
    rematch.frame.TopTileStreaks:SetPoint("TOPRIGHT")

    rematch.events:Register(self,"REMATCH_TEAM_LOADED",self.REMATCH_TEAM_LOADED)
    rematch.timer:Start(0.1,rematch.toolbar.CacheSafariHat)
end)

function rematch.toolbar:Configure()
    local layout = self:GetToolbarLayout()
    local mode = rematch.layout:GetMode(C.CURRENT)
    local leftMostButton

    for _,button in ipairs(rematch.toolbar.Buttons) do
        button:Hide()
    end

    -- reversed toolbar changes the order the layout is traversed
    local from,to,step = 1,#layout,1
    if settings.ReverseToolbar then
        from,to,step = #layout,1,-1
    end

    if layout then
        local xoff = -2
        for i=from,to,step do
            local button = rematch.toolbar[layout[i]]
            button:SetPoint("RIGHT",xoff,0)
            button:Show()
            button.Border:SetShown(i~=to or mode~=0) -- hide leftmost left border when minimized, show otherwise
            xoff = xoff - C.TOOLBAR_BUTTON_SIZE
            leftMostButton = button
        end
    end

    self.TotalsButton:SetShown(mode~=0) -- show totals button in all but minimized mode
    self.TotalsButton.Border:SetShown(mode>1) -- show border to right of totals button in 2- and 3-panel modes
    self.TotalsButton:SetPoint("LEFT",rematch.journal:IsActive() and 56 or 3,0) -- need to nudge totals to right in journal

    self.AchievementTotal:SetShown(mode>1)
    self.AchievementTotal:SetPoint("LEFT",self.TotalsButton,"RIGHT",2,0)
    self.AchievementTotal:SetPoint("RIGHT",leftMostButton,"LEFT",-2,0)
    local showFlair = mode==3 or (settings.AlwaysUsePetSatchel and mode==2)
    self.AchievementTotal.LeftFlair:SetShown(showFlair)
    self.AchievementTotal.RightFlair:SetShown(showFlair)
end

function rematch.toolbar:Update(fromEvent)
    -- go through each button in the present layout and run their update if they have one
    local layout = self:GetToolbarLayout()
    for _,buttonName in ipairs(layout) do
        local button = rematch.toolbar[buttonName]
        if button and button.Update then
            button:Update(fromEvent)
        end
        if GetMouseFocus()==button then
            button:GetScript("OnEnter")(button)
        end
    end
    -- update cooldowns
    --self:SPELL_UPDATE_COOLDOWN()
    -- update totals button
    if settings.DisplayUniqueTotal then
        self.TotalsButton.Text:SetText(format("%s \124cffffffff%d",L["UNIQUE_PETS"],rematch.roster:GetNumUniqueOwned()))
    else
        self.TotalsButton.Text:SetText(format("%s \124cffffffff%d",L["TOTAL_PETS"],(select(2,C_PetJournal.GetNumPets()))))
    end
    -- update the achievement total
    self.AchievementTotal.Text:SetText(GetCategoryAchievementPoints(C.PET_ACHIEVEMENT_CATEGORY,true))
end

-- builds a list of parentKeys for toolbar buttons to be shown depending on mode and pet satchel
function rematch.toolbar:GetToolbarLayout()
    wipe(displayLayout)
    local mode = rematch.layout:GetMode(C.CURRENT)
    -- if Always Use Pet Satchel is enabled, then dual and triple-panel modes should use single-panel layout
    if mode>1 and settings.AlwaysUsePetSatchel then
        mode=1
    end
    local layout = mode and toolbarLayouts[mode]
    assert(layout,"No toolbar layout for mode "..(mode or "nil"))

    for i=#layout,1,-1 do
        if type(layout[i])=="string" then
            tinsert(displayLayout,layout[i])
        elseif type(layout[i])=="number" then
            local satchelIndex = settings.PetSatchelIndex
            if not satchelIndex or satchelIndex<1 or satchelIndex > #rematch.toolbar.petSatchelButtons then
                satchelIndex = 1
                settings.PetSatchelIndex = 1
            end
            tinsert(displayLayout,rematch.toolbar.petSatchelButtons[satchelIndex][layout[i]])
        end
    end
    return displayLayout
end

function rematch.toolbar:OnShow()
    rematch.events:Register(self,"COMPANION_UPDATE",self.COMPANION_UPDATE)
    rematch.events:Register(self,"SPELL_UPDATE_COOLDOWN",self.SPELL_UPDATE_COOLDOWN)
    rematch.events:Register(self,"UNIT_AURA",self.UNIT_AURA)
    rematch.events:Register(self,"BAG_UPDATE_DELAYED",self.BAG_UPDATE_DELAYED)
    rematch.events:Register(self,"PET_BATTLE_QUEUE_STATUS",self.PET_BATTLE_QUEUE_STATUS)
    rematch.events:Register(self,"REMATCH_LOADOUTS_CHANGED",self.REMATCH_LOADOUTS_CHANGED)
    rematch.events:Unregister(self,"REMATCH_TEAM_LOADED") -- only register while toolbar not shown
    rematch.toolbar:SPELL_UPDATE_COOLDOWN()
end

function rematch.toolbar:OnHide()
    rematch.events:Unregister(self,"COMPANION_UPDATE")
    rematch.events:Unregister(self,"SPELL_UPDATE_COOLDOWN")
    rematch.events:Unregister(self,"UNIT_AURA")
    rematch.events:Unregister(self,"BAG_UPDATE_DELAYED")
    rematch.events:Unregister(self,"PET_BATTLE_QUEUE_STATUS")
    rematch.events:Unregister(self,"REMATCH_LOADOUTS_CHANGED")
    rematch.events:Register(self,"REMATCH_TEAM_LOADED",self.REMATCH_TEAM_LOADED)
end

-- safari hat is a toy and needs to be applied by name; forcing a cache on login to get the name
function rematch.toolbar:CacheSafariHat()
    if not GetItemInfo(C.SAFARI_HAT_ITEM_ID) then
        rematch.timer:Start(0.1,rematch.toolbar.CacheSafariHat)
    end
end

--[[ events ]]

function rematch.toolbar:SPELL_UPDATE_COOLDOWN()
    self.HealButton.Cooldown:SetCooldown(GetSpellCooldown(C.REVIVE_SPELL_ID))
    CooldownFrame_Set(self.SafariHatButton.Cooldown,GetItemCooldown(C.SAFARI_HAT_ITEM_ID))
    CooldownFrame_Set(self.BandageButton.Cooldown,GetItemCooldown(C.BANDAGE_ITEM_ID))
    --self.SafariHatButton.Cooldown:SetCooldown(GetItemCooldown(C.SAFARI_HAT_ITEM_ID))
    --self.BandageButton.Cooldown:SetCooldown(GetItemCooldown(C.BANDAGE_ITEM_ID))
    self.SummonPetButton.Cooldown:SetCooldown(GetSpellCooldown(C.GCD_SPELL_ID))
    if self.LevelingStoneButton:IsVisible() then
        CooldownFrame_Set(self.LevelingStoneButton.Cooldown,GetItemCooldown(self.LevelingStoneButton:GetAttribute("item")))
        CooldownFrame_Set(self.RarityStoneButton.Cooldown,GetItemCooldown(self.RarityStoneButton:GetAttribute("item")))
        --self.LevelingStoneButton.Cooldown:SetCooldown(GetItemCooldown(self.LevelingStoneButton:GetAttribute("item")))
        --self.RarityStoneButton.Cooldown:SetCooldown(GetItemCooldown(self.RarityStoneButton:GetAttribute("item")))
    end
end

function rematch.toolbar:COMPANION_UPDATE()
    self.SummonPetButton.needsUpdate = true
    self.LevelingStoneButton.needsUpdate = true
    self.RarityStoneButton.needsUpdate = true
    self:Update(true)
end

function rematch.toolbar:UNIT_AURA(unit)
    if unit=="player" then
        self.SafariHatButton.needsUpdate = true
        self.LesserPetTreatButton.needsUpdate = true
        self.PetTreatButton.needsUpdate = true
        self:Update(true)
    end
end

function rematch.toolbar:BAG_UPDATE_DELAYED()
    self.BandageButton.needsUpdate = true
    self.LesserPetTreatButton.needsUpdate = true
    self.PetTreatButton.needsUpdate = true
    self.LevelingStoneButton.needsUpdate = true
    self.RarityStoneButton.needsUpdate = true
    self:Update(true)
end

function rematch.toolbar:PET_BATTLE_QUEUE_STATUS()
    self.FindBattleButton.needsUpdate = true
    rematch.frame:Update() -- update loadouts too
    self:Update(true)
end

-- for Safari Hat Reminder, update safari hat button when loadouts change
function rematch.toolbar:REMATCH_LOADOUTS_CHANGED()
    self.SafariHatButton.needsUpdate = true
    self:Update()
end

-- for Safari Hat Reminder, should only fire when frame is not shown
function rematch.toolbar:REMATCH_TEAM_LOADED()
    if settings.SafariHatShine and not rematch.frame:IsVisible() and not rematch.utils:GetItemBuff(C.SAFARI_HAT_ITEM_ID) and PlayerHasToy(C.SAFARI_HAT_ITEM_ID) and rematch.loadouts:NotAllMaxLevel() then
        rematch.frame:Toggle(true)
    end
end

-- picks the best suited leveling or rarity stone depending on what pet is summoned
-- stoneList is either C.RARITY_STONES or C.LEVELING_STONES
-- defaultStone is either C.DEFAULT_LEVELING_STONE_ITEM_ID or C.DEFAULT_RARITY_STONE_ITEM_ID
-- if a pet is summoned, it will look for any itemIDs in the list for that type (first 10 entries)
-- if a pet is not summoned or one of the type-specific stones is not in inventory, it will display
-- the first itemID found in inventory in the remainer of the stoneList, or the given default if none
function rematch.toolbar:PickBestStone(stoneList,defaultStone)
    local petID = C_PetJournal.GetSummonedPetGUID()
    local petInfo = rematch.petInfo:Fetch(petID)
    local numTypes = C_PetJournal.GetNumPetTypes() -- going to be 10 until they add a pet type
    if petInfo.isValid then -- this pet is summoned, look for a battle-training stone for the summoned type
        local itemID = stoneList[petInfo.petType]
        if itemID and GetItemCount(itemID)>0 then
            return itemID
        end
    end
    -- if here, either no pet was summoned or we ran out of pet type-specific battle-training stones
    -- look for one of the generic ones
    for i=numTypes+1,#stoneList do
        local itemID = stoneList[i]
        if GetItemCount(itemID)>0 then
            return itemID
        end
    end
    -- none of the generic stones are in inventory either, return the "none found" default stone itemID given as an argument
    return defaultStone
end

--[[ achievment total ]]

function rematch.toolbar.AchievementTotal:OnClick()
    ToggleAchievementFrame()
    if AchievementFrame:IsVisible() then
        -- some of this is lifted out of AchievementFrameCategories_SelectDefaultElementData() in Blizzard_AchievementUI.lua
        if not AchievementFrameCategories.ScrollBox:HasDataProvider() then
            AchievementFrameCategories_UpdateDataProvider()
        end
        -- find the Pet Battles index in the data provider
        local categoryIndex
        for index,info in AchievementFrameCategories.ScrollBox:GetDataProvider():Enumerate() do
            if info.id==C.PET_ACHIEVEMENT_CATEGORY then
                categoryIndex = index
            end
        end
        -- go to that category if it exists
        if categoryIndex then
            local elementData = AchievementFrameCategories.ScrollBox:ScrollToElementDataIndex(categoryIndex, ScrollBoxConstants.AlignCenter, ScrollBoxConstants.NoScrollInterpolation);
            if elementData then
                AchievementFrameCategories_SelectElementData(elementData)
            end
        end
    end
end

--[[ totals button ]]

function rematch.toolbar.TotalsButton:OnEnter()
    self.Highlight:Show()
    local stats = rematch.collectionInfo:GetCollectionStats()
    local tooltipBody = format(L["Unique Pets: %s%d\124r\nTotal Pets: %s%d\124r\nUncollected Pets: %s%d\124r\nAverage Level: %s%.1f\124r%s"],C.HEX_WHITE,stats.numCollectedUnique,C.HEX_WHITE,stats.numCollectedTotal,C.HEX_WHITE,stats.numUncollected,C.HEX_WHITE,stats.averageLevel,settings.HideMenuHelp and "" or format(L["\n\n%s Click for details"],C.LMB_TEXT_ICON))
    rematch.tooltip:ShowSimpleTooltip(self,nil,tooltipBody)
end

function rematch.toolbar.TotalsButton:OnLeave()
    self.Highlight:Hide()
    rematch.tooltip:Hide()
end

function rematch.toolbar.TotalsButton:OnMouseDown()
    self.Back:SetTexCoord(0,0.421875,0.125,0.234375)
    self.Text:SetPoint("CENTER",-1,-2)
end

function rematch.toolbar.TotalsButton:OnMouseUp()
    if GetMouseFocus()==self then
        self.Highlight:Show()
    end
    self.Back:SetTexCoord(0,0.421875,0,0.109375)
    self.Text:SetPoint("CENTER")
end

function rematch.toolbar.TotalsButton:OnClick()
    rematch.dialog:ToggleDialog(settings.MinimizePetSummary and "PetSummaryMinimized" or "PetSummary")
end