WoWInterface/Interface/AddOns/tdBattlePetScript/Rematch/UI.lua

--[[
UI.lua
@Author  : DengSir (tdaddon@163.com)
@Link    : https://dengsir.github.io
]]

local ns = select(2, ...)

local RematchPlugin = ns.RematchPlugin
local L     = LibStub('AceLocale-3.0'):GetLocale('PetBattleScripts')

local scriptMenuItem = {
    text = function(_, key, ...)
        return RematchPlugin:GetScript(key) and L.EDITOR_EDIT_SCRIPT or L.EDITOR_CREATE_SCRIPT
    end,
    func = function(_, key, ...)
        RematchPlugin:OpenScriptEditor(key, RematchPlugin:GetTitleByKey(key))
    end
}

local scriptButtonIcon = 'Interface/AddOns/tdBattlePetScript/Rematch/Textures/ScriptIcon-Rematch{{REMATCHVERSION}}.png'
-- Rematch4 only
local scriptButtons = setmetatable({}, {
    __index = function(t, parent)
        local button = CreateFrame('Button', nil, parent, 'RematchFootnoteButtonTemplate')
        do
            if parent.slim then
                button:SetSize(18, 18)
            end
            button:SetPoint('CENTER')
            button:SetNormalTexture(scriptButtonIcon)
            button:SetPushedTexture(scriptButtonIcon)
            button:SetScript('OnClick', function(button)
                RematchPlugin:OpenScriptEditor(button.key, Rematch:GetTeamTitle(button.key))
            end)
            button:SetScript('OnEnter', function(button)
                GameTooltip:SetOwner(button, 'ANCHOR_RIGHT')
                GameTooltip:SetText(L.ADDON_NAME)
                GameTooltip:AddLine(L.EDITOR_CREATE_SCRIPT, HIGHLIGHT_FONT_COLOR:GetRGB())
                GameTooltip:Show()
            end)
            button:SetScript('OnLeave', GameTooltip_Hide)
        end
        t[parent] = button
        return button
    end
})

function RematchPlugin:SetupUI()
    local rematchVersion = ns.Version:Current('Rematch')

    -- Add menu to edit script
    if rematchVersion >= ns.Version:New(5, 0, 0, 0) then
        local afterText = Rematch.localization['Set Notes'] -- Use Rematch's locale string.
        Rematch.menus:AddToMenu('TeamMenu', scriptMenuItem, afterText)
        Rematch.menus:AddToMenu('LoadedTeamMenu', scriptMenuItem, afterText)
    else
        tinsert(Rematch:GetMenu('TeamMenu'), 6, scriptMenuItem)
    end

    -- When a script is added/removed, refresh the teams list.
    self.updateFrames = nil
    if rematchVersion >= ns.Version:New(5, 0, 0, 0) then
        self.updateFrames = function()
            if Rematch.frame:IsVisible() then
                Rematch.frame:Update()
            end
        end
    else
        self.updateFrames = function()
            if RematchLoadedTeamPanel:IsVisible() then
                RematchLoadedTeamPanel:Update()
            end
            if RematchTeamPanel:IsVisible() then
                if RematchTeamPanel.UpdateList then
                    RematchTeamPanel:UpdateList()
                elseif RematchTeamPanel.List then
                    RematchTeamPanel.List:Update()
                end
            end
        end
    end
    self:RegisterMessage('PET_BATTLE_SCRIPT_SCRIPT_LIST_UPDATE', self.updateFrames)

    -- Button to indicate a script exists
    scriptButtonIcon = scriptButtonIcon:gsub('{{REMATCHVERSION}}', rematchVersion.major)
    if rematchVersion >= ns.Version:New(5, 0, 0, 0) then
        -- TODO: restore tooltip/button once Rematch supports that again
        Rematch.badges:RegisterBadge('teams', 'PetBattleScripts', scriptButtonIcon, nil, function(parent, teamID)
            if rematchVersion < ns.Version:New(5, 1, 0, 0) then
                teamID = parent
            end
            return teamID and self:GetScript(teamID)
        end)
    elseif rematchVersion >= ns.Version:New(4, 8, 10, 5) then
        self:SecureHook(RematchTeamPanel.List, 'callback', function(button, key)
            local script = scriptButtons[button]
            if self:GetScript(key) then
                script.key = key
                script:Show()
                script:ClearAllPoints()

                local relative = button.Preferences:IsShown()                      and button.Preferences or
                                 button.Notes:IsShown()                            and button.Notes or
                                 button.compact and button.WinRecordBack:IsShown() and button.WinRecordBack

                if relative then
                    script:SetPoint('RIGHT', relative, 'LEFT', relative == button.WinRecordBack and -3 or 0, 0)
                else
                    script:SetPoint('TOPRIGHT', -2, -3)
                end

                button.Name:SetPoint('TOPRIGHT', script:GetLeft() - button:GetRight() , -4)
            else
                script:Hide()
            end
        end)
    else
        self:SecureHook(RematchTeamPanel, 'FillTeamButton', function(_, button, key)
            local script = scriptButtons[button]
            if self:GetScript(key) then
                script.key = key
                script:Show()
                script:ClearAllPoints()

                local xoffset = -3
                local yoffset = 0

                if button.WinRecord:IsShown() then
                    if button.slim then
                        xoffset = xoffset - button.WinRecord:GetWidth()
                    else
                        yoffset = 8
                    end
                    button.WinRecord:SetPoint('BOTTOMRIGHT',-3,4)
                    button.WinRecord:SetHitRectInsets(0,0,0,-2)
                end

                if button.Preferences:IsShown() then
                    xoffset = xoffset - button.Preferences:GetWidth()
                end
                if button.Notes:IsShown() then
                    xoffset = xoffset - button.Notes:GetWidth()
                end

                script:SetPoint('RIGHT', xoffset, yoffset)

                xoffset = xoffset - script:GetWidth()

                if button.slim then
                    button.Name:SetPoint('RIGHT',xoffset-2,0)
                else
                    button.Name:SetPoint('TOPRIGHT',xoffset-1,-6)
                end
            else
                script:Hide()
            end
        end)
    end

    if rematchVersion < ns.Version:New(5, 0, 0, 0) then
        local function move(button, add)
            if not button:IsShown() then
                return 0
            end

            local point, relative, relativePoint, x, y = button:GetPoint()
            button:SetPoint(point, relative, relativePoint, x + 21, y)
            return add
        end

        self:SecureHook(RematchLoadedTeamPanel, 'Update', function(panel)
            local footnotes = panel.Footnotes
            local script = scriptButtons[footnotes]

            if self:GetScript(RematchSettings.loadedTeam) then
                script.key = RematchSettings.loadedTeam
                script:Show()
                script:ClearAllPoints()
                script:SetPoint('LEFT', 5, -0.5)

                local fx = 5 + 21
                fx = fx + move(footnotes.Preferences, 21)
                fx = fx + move(footnotes.Notes, 21)
                fx = fx + move(footnotes.WinRecord, footnotes.WinRecord:GetWidth())
                fx = fx + move(footnotes.Maximize, 21)
                fx = fx + move(footnotes.Close, 21)

                local footnoteWidth = fx + 4
                local panelWidth = panel.maxWidth or 280

                footnotes:SetWidth(footnoteWidth)
                footnotes:Show()
                panel:SetWidth(panelWidth-footnoteWidth-3)
            else
                script:Hide()
            end
        end)
    end

    if rematchVersion < ns.Version:New(5, 0, 0, 0) then
        self:HookScript(RematchJournal, 'OnShow', function(self)
            CollectionsJournal:SetAttribute('UIPanelLayout-width', 870)
            UpdateUIPanelPositions()
        end)
        self:HookScript(RematchJournal, 'OnHide', function(self)
            CollectionsJournal:SetAttribute('UIPanelLayout-width', 710)
            UpdateUIPanelPositions()
        end)
    end

    self.updateFrames()
end

function RematchPlugin:TeardownUI()
    local rematchVersion = ns.Version:Current('Rematch')

    self:UnregisterMessage('PET_BATTLE_SCRIPT_SCRIPT_LIST_UPDATE')

    if rematchVersion >= ns.Version:New(5, 0, 0, 0) then
        tDeleteItem(Rematch.menus:GetDefinition('TeamMenu'), scriptMenuItem)
        tDeleteItem(Rematch.menus:GetDefinition('LoadedTeamMenu'), scriptMenuItem)

        Rematch.badges:UnregisterBadge('teams', 'PetBattleScripts')
    else
        self:UnhookAll()

        tDeleteItem(Rematch:GetMenu('TeamMenu'), scriptMenuItem)

        for _, frame in pairs(scriptButtons) do
            frame:ClearAllPoints()
            frame:Hide()
        end
        if RematchLoadedTeamPanel:IsVisible() then
            RematchLoadedTeamPanel:Update()
        end
        if RematchTeamPanel:IsVisible() then
            if RematchTeamPanel.UpdateList then
                RematchTeamPanel:UpdateList()
            elseif RematchTeamPanel.List then
                RematchTeamPanel.List:Update()
            end
        end
    end

    if self.updateFrames then
        self.updateFrames()
    end
end