ExtraButton.lua

local ADDON_NAME = ...
local PB = {}

_G.PartyButtons = PB

-- ============================================================
-- Configuration
-- ============================================================

local CONFIG = {
    buttonWidth  = 118,
    buttonHeight = 32,
    spacing      = 6,
    topOffset    = -48,

    -- Raid markers:
    -- 1 = Star
    -- 2 = Circle
    -- 3 = Diamond
    -- 4 = Triangle
    -- 5 = Moon
    -- 6 = Square
    -- 7 = Cross
    -- 8 = Skull

    tankMarker   = 6, -- Square
    healerMarker = 4, -- Triangle

    greetings = {
        "Hi",
        "yo",
        "o/",
    },

    bloodlustIcon =
        "Interface\\Icons\\Spell_Nature_BloodLust",

    -- ========================================================
    -- Lust spell IDs
    -- ========================================================

    bloodlustSpellID = 2825,
    heroismSpellID   = 32182,
    timeWarpSpellID  = 80353,
    primalRageSpellID = 264667,
    furyAspectsSpellID = 390386,
    harriersCrySpellID = 466904,
}

-- ============================================================
-- Frames
-- ============================================================

local mainFrame
local markButton
local hiButton
local heroButton

-- ============================================================
-- Secure update state
-- ============================================================

local needsMarkMacroUpdate = true
local needsHeroMacroUpdate = true

-- ============================================================
-- Hunter inspection state
-- ============================================================

local inspectQueue = {}
local inspectInProgress = false
local inspectUnit = nil

-- ============================================================
-- Utility
-- ============================================================

local function Print(msg)
    if DEFAULT_CHAT_FRAME then
        DEFAULT_CHAT_FRAME:AddMessage(
            "|cff55ccffPartyButtons|r: " ..
            tostring(msg)
        )
    end
end

-- ============================================================
-- Group units
-- ============================================================

local function GetGroupUnits()
    local units = {}

    -- Player
    if UnitExists("player") then
        table.insert(
            units,
            "player"
        )
    end

    -- Raid
    if IsInRaid() then

        local count = GetNumGroupMembers()

        for i = 1, count do
            local unit = "raid" .. i

            if UnitExists(unit) then
                table.insert(
                    units,
                    unit
                )
            end
        end

    -- Party
    elseif IsInGroup() then

        for i = 1, 4 do
            local unit = "party" .. i

            if UnitExists(unit) then
                table.insert(
                    units,
                    unit
                )
            end
        end
    end

    return units
end

-- ============================================================
-- Party role detection
-- ============================================================

local function GetPartyRoleUnits()
    local tank
    local healer

    for _, unit in ipairs(GetGroupUnits()) do

        if UnitExists(unit) then

            local role =
                UnitGroupRolesAssigned(unit)

            if role == "TANK"
                and not tank then

                tank = unit

            elseif role == "HEALER"
                and not healer then

                healer = unit
            end

            if tank and healer then
                break
            end
        end
    end

    return tank, healer
end

-- ============================================================
-- Unit validity
-- ============================================================

local function IsUnitValidForLust(unit)

    if not UnitExists(unit) then
        return false
    end

    if not UnitIsConnected(unit) then
        return false
    end

    if UnitIsDeadOrGhost(unit) then
        return false
    end

    return true
end

-- ============================================================
-- Spell link
-- ============================================================

local function GetSpellLinkSafe(spellID)

    if not spellID then
        return nil
    end

    local link =
        C_Spell.GetSpellLink(spellID)

    if link and link ~= "" then
        return link
    end

    local info =
        C_Spell.GetSpellInfo(spellID)

    if info and info.name then
        return info.name
    end

    return nil
end

-- ============================================================
-- Get spell name
-- ============================================================

local function GetSpellNameSafe(spellID)

    if not spellID then
        return nil
    end

    local name =
        C_Spell.GetSpellName(spellID)

    if name then
        return name
    end

    local info =
        C_Spell.GetSpellInfo(spellID)

    if info then
        return info.name
    end

    return nil
end

-- ============================================================
-- Player specialization
-- Modern C_SpecializationInfo API
-- ============================================================

local function GetPlayerSpecID()

    local specIndex =
        C_SpecializationInfo.GetSpecialization()

    if not specIndex then
        return nil
    end

    local specID =
        C_SpecializationInfo.GetSpecializationInfo(
            specIndex
        )

    return specID
end

-- ============================================================
-- Player's own Hunter pet specialization
-- ============================================================

local function GetOwnPetSpecID()

    local petSpecIndex =
        C_SpecializationInfo.GetSpecialization(
            false,
            true
        )

    if not petSpecIndex then
        return nil
    end

    local petSpecID =
        C_SpecializationInfo.GetSpecializationInfo(
            petSpecIndex,
            false,
            true
        )

    return petSpecID
end

-- ============================================================
-- Own Shaman lust
-- ============================================================

local function GetOwnShamanLustSpellID()

    if IsPlayerSpell(
        CONFIG.bloodlustSpellID
    ) then

        return CONFIG.bloodlustSpellID
    end

    if IsPlayerSpell(
        CONFIG.heroismSpellID
    ) then

        return CONFIG.heroismSpellID
    end

    return nil
end

-- ============================================================
-- Own Hunter lust
-- ============================================================

local function GetOwnHunterLustSpellID()

    local specID =
        GetPlayerSpecID()

    if not specID then
        return nil
    end

    -- --------------------------------------------------------
    -- Beast Mastery
    -- --------------------------------------------------------

    if specID == 253 then

        if not IsPlayerSpell(
            CONFIG.primalRageSpellID
        ) then

            return nil
        end

        if not UnitExists("pet") then
            return nil
        end

        if UnitIsDeadOrGhost("pet") then
            return nil
        end

        local petSpecID =
            GetOwnPetSpecID()

        -- 1 = Ferocity
        if petSpecID == 1 then
            return CONFIG.primalRageSpellID
        end

        return nil
    end

    -- --------------------------------------------------------
    -- Marksmanship
    -- --------------------------------------------------------

    if specID == 254 then

        if IsPlayerSpell(
            CONFIG.harriersCrySpellID
        ) then

            return CONFIG.harriersCrySpellID
        end

        return nil
    end

    -- --------------------------------------------------------
    -- Survival
    -- --------------------------------------------------------

    if specID == 255 then

        if not IsPlayerSpell(
            CONFIG.primalRageSpellID
        ) then

            return nil
        end

        if not UnitExists("pet") then
            return nil
        end

        if UnitIsDeadOrGhost("pet") then
            return nil
        end

        local petSpecID =
            GetOwnPetSpecID()

        if petSpecID == 1 then
            return CONFIG.primalRageSpellID
        end

        return nil
    end

    return nil
end

-- ============================================================
-- Get own lust
-- ============================================================

local function GetOwnLustSpellID()

    -- Shaman
    local shamanSpell =
        GetOwnShamanLustSpellID()

    if shamanSpell then
        return shamanSpell
    end

    -- Mage
    if IsPlayerSpell(
        CONFIG.timeWarpSpellID
    ) then

        return CONFIG.timeWarpSpellID
    end

    -- Evoker
    if IsPlayerSpell(
        CONFIG.furyAspectsSpellID
    ) then

        return CONFIG.furyAspectsSpellID
    end

    -- Hunter
    local _, classFile =
        UnitClass("player")

    if classFile == "HUNTER" then
        return GetOwnHunterLustSpellID()
    end

    return nil
end

-- ============================================================
-- Get pet unit for group member
-- ============================================================

local function GetPetUnitForGroupUnit(unit)

    if not unit then
        return nil
    end

    if unit:match("^party%d+$") then

        return unit:gsub(
            "^party",
            "partypet"
        )
    end

    if unit:match("^raid%d+$") then

        return unit:gsub(
            "^raid",
            "raidpet"
        )
    end

    return nil
end

-- ============================================================
-- Inspect queue
-- ============================================================

local function ProcessInspectQueue()

    if inspectInProgress then
        return
    end

    local nextUnit

    for unit in pairs(inspectQueue) do
        nextUnit = unit
        break
    end

    if not nextUnit then
        return
    end

    inspectQueue[nextUnit] = nil

    if not UnitExists(nextUnit) then
        C_Timer.After(
            0.05,
            ProcessInspectQueue
        )

        return
    end

    if UnitIsDeadOrGhost(nextUnit) then
        C_Timer.After(
            0.05,
            ProcessInspectQueue
        )

        return
    end

    if not CanInspect(nextUnit) then

        -- Try again later.
        inspectQueue[nextUnit] = true

        C_Timer.After(
            0.5,
            ProcessInspectQueue
        )

        return
    end

    inspectInProgress = true
    inspectUnit = nextUnit

    NotifyInspect(nextUnit)
end

local function QueueHunterInspect(unit)

    if not UnitExists(unit) then
        return
    end

    if UnitIsUnit(unit, "player") then
        return
    end

    local _, classFile =
        UnitClass(unit)

    if classFile ~= "HUNTER" then
        return
    end

    inspectQueue[unit] = true

    ProcessInspectQueue()
end

-- ============================================================
-- Get inspected player's specialization
-- ============================================================

local function GetInspectedSpecID(unit)

    if not unit then
        return nil
    end

    if not UnitExists(unit) then
        return nil
    end

    local specID =
        GetInspectSpecialization(unit)

    if specID and specID ~= 0 then
        return specID
    end

    return nil
end

-- ============================================================
-- Get group Hunter lust
-- ============================================================

local function GetGroupHunterLustSpellID(unit)

    if not IsUnitValidForLust(unit) then
        return nil, true
    end

    local specID =
        GetInspectedSpecID(unit)

    if not specID then

        QueueHunterInspect(unit)

        return nil, false
    end

    -- --------------------------------------------------------
    -- Marksmanship
    -- --------------------------------------------------------

    if specID == 254 then

        return CONFIG.harriersCrySpellID, true
    end

    -- --------------------------------------------------------
    -- Beast Mastery / Survival
    -- --------------------------------------------------------

    if specID == 253
        or specID == 255 then

        local petUnit =
            GetPetUnitForGroupUnit(unit)

        if not petUnit then
            return nil, true
        end

        if not UnitExists(petUnit) then
            return nil, true
        end

        if UnitIsDeadOrGhost(petUnit) then
            return nil, true
        end

        -- For another Hunter we cannot read arbitrary spellbook
        -- state like our own pet's IsSpellKnown(..., true).
        --
        -- What we CAN reliably confirm:
        --   Hunter spec = BM / Survival
        --   active pet exists
        --   active pet is alive
        --
        -- For Ferocity detection we use inspected pet
        -- specialization if available.
        local petSpecIndex =
            C_SpecializationInfo.GetSpecialization(
                true,
                true
            )

        if petSpecIndex then

            local petSpecID =
                C_SpecializationInfo.GetSpecializationInfo(
                    petSpecIndex,
                    true,
                    true,
                    unit
                )

            -- 1 = Ferocity
            if petSpecID == 1 then
                return CONFIG.primalRageSpellID, true
            end

            return nil, true
        end

        -- Pet specialization isn't available yet.
        return nil, false
    end

    return nil, true
end

-- ============================================================
-- Get lust from any group member
-- ============================================================

local function GetGroupMemberLustSpellID(unit)

    if not IsUnitValidForLust(unit) then
        return nil, true
    end

    local _, classFile =
        UnitClass(unit)

    if not classFile then
        return nil, true
    end

    -- ========================================================
    -- Shaman
    -- ========================================================

    if classFile == "SHAMAN" then

        local faction =
            UnitFactionGroup(unit)

        if faction == "Horde" then
            return CONFIG.bloodlustSpellID, true
        end

        if faction == "Alliance" then
            return CONFIG.heroismSpellID, true
        end

        -- Very unusual fallback.
        local playerFaction =
            UnitFactionGroup("player")

        if playerFaction == "Alliance" then
            return CONFIG.heroismSpellID, true
        end

        if playerFaction == "Horde" then
            return CONFIG.bloodlustSpellID, true
        end

        return nil, true
    end

    -- ========================================================
    -- Mage
    -- ========================================================

    if classFile == "MAGE" then
        return CONFIG.timeWarpSpellID, true
    end

    -- ========================================================
    -- Evoker
    -- ========================================================

    if classFile == "EVOKER" then
        return CONFIG.furyAspectsSpellID, true
    end

    -- ========================================================
    -- Hunter
    -- ========================================================

    if classFile == "HUNTER" then

        return GetGroupHunterLustSpellID(unit)
    end

    return nil, true
end

-- ============================================================
-- Build Heroism announcement
-- ============================================================

local function BuildLustAnnouncement()

    local links = {}

    -- Each GROUP MEMBER gets at most ONE link.
    local addedUnits = {}

    local function AddUnitSpell(unit, spellID)

        if not unit or not spellID then
            return
        end

        if addedUnits[unit] then
            return
        end

        local link =
            GetSpellLinkSafe(spellID)

        if not link then
            return
        end

        addedUnits[unit] = true

        table.insert(
            links,
            link
        )
    end

    -- --------------------------------------------------------
    -- Player
    -- --------------------------------------------------------

    local ownSpellID =
        GetOwnLustSpellID()

    if ownSpellID then

        local link =
            GetSpellLinkSafe(
                ownSpellID
            )

        if link then

            table.insert(
                links,
                link
            )
        end
    end

    -- --------------------------------------------------------
    -- Group members
    -- --------------------------------------------------------

    for _, unit in ipairs(
        GetGroupUnits()
    ) do

        if not UnitIsUnit(
            unit,
            "player"
        ) then

            local spellID, ready =
                GetGroupMemberLustSpellID(
                    unit
                )

            if spellID then

                AddUnitSpell(
                    unit,
                    spellID
                )

            elseif not ready then

                -- Hunter inspection isn't ready yet.
                QueueHunterInspect(unit)
            end
        end
    end

    -- --------------------------------------------------------
    -- Build result
    -- --------------------------------------------------------

    if #links == 0 then
        return "Macros: Press Heroism"
    end

    return "Macros: Press " ..
        table.concat(
            links,
            " "
        )
end

-- ============================================================
-- Build own cast macro
-- ============================================================

local function BuildOwnLustCastMacro()

    local spellID =
        GetOwnLustSpellID()

    if not spellID then
        return nil
    end

    local spellName =
        GetSpellNameSafe(
            spellID
        )

    if not spellName then
        return nil
    end

    return "/cast " ..
        spellName
end

-- ============================================================
-- Update Mark macro
-- ============================================================

local function BuildMarkMacroBody()

    local tank, healer =
        GetPartyRoleUnits()

    local lines = {}

    if tank and UnitExists(tank) then

        table.insert(
            lines,
            "/tm [@" ..
            tank ..
            "] " ..
            CONFIG.tankMarker
        )
    end

    if healer
        and UnitExists(healer)
        and healer ~= tank then

        table.insert(
            lines,
            "/tm [@" ..
            healer ..
            "] " ..
            CONFIG.healerMarker
        )
    end

    if #lines == 0 then
        return "/stopmacro"
    end

    return table.concat(
        lines,
        "\n"
    )
end

local function UpdateMarkMacro()

    if not markButton then
        return
    end

    if InCombatLockdown() then

        needsMarkMacroUpdate = true

        return
    end

    local body =
        BuildMarkMacroBody()

    markButton:SetAttribute(
        "type1",
        "macro"
    )

    markButton:SetAttribute(
        "macrotext1",
        body
    )

    needsMarkMacroUpdate = false
end

-- ============================================================
-- Greeting macro
-- ============================================================

local function UpdateChatMacro()

    if not hiButton then
        return
    end

    if InCombatLockdown() then
        return
    end

    -- Normal
    hiButton:SetAttribute(
        "type1",
        "macro"
    )

    hiButton:SetAttribute(
        "macrotext1",
        "/p " ..
        CONFIG.greetings[1]
    )

    -- Shift
    hiButton:SetAttribute(
        "shift-type1",
        "macro"
    )

    hiButton:SetAttribute(
        "shift-macrotext1",
        "/p " ..
        CONFIG.greetings[2]
    )

    -- Ctrl
    hiButton:SetAttribute(
        "ctrl-type1",
        "macro"
    )

    hiButton:SetAttribute(
        "ctrl-macrotext1",
        "/p " ..
        CONFIG.greetings[3]
    )
end

-- ============================================================
-- Heroism macro
--
-- LMB:
--   If player has lust -> CAST
--   Otherwise -> SAY + PARTY announcement
--
-- SHIFT + LMB:
--   PARTY announcement
--
-- CTRL + LMB:
--   SAY + PARTY announcement
--
-- ALT + LMB:
--   SAY announcement
-- ============================================================

local function UpdateHeroismMacro()

    if not heroButton then
        return
    end

    if InCombatLockdown() then

        needsHeroMacroUpdate = true

        return
    end

    local castMacro =
        BuildOwnLustCastMacro()

    local message =
        BuildLustAnnouncement()

    -- --------------------------------------------------------
    -- Normal click
    -- --------------------------------------------------------

    heroButton:SetAttribute(
        "type1",
        "macro"
    )

    if castMacro then

        heroButton:SetAttribute(
            "macrotext1",
            castMacro
        )

    else

        heroButton:SetAttribute(
            "macrotext1",
            "/say " ..
            message ..
            "\n/p " ..
            message
        )
    end

    -- --------------------------------------------------------
    -- Shift + click
    -- --------------------------------------------------------

    heroButton:SetAttribute(
        "shift-type1",
        "macro"
    )

    heroButton:SetAttribute(
        "shift-macrotext1",
        "/p " ..
        message
    )

    -- --------------------------------------------------------
    -- Ctrl + click
    -- --------------------------------------------------------

    heroButton:SetAttribute(
        "ctrl-type1",
        "macro"
    )

    heroButton:SetAttribute(
        "ctrl-macrotext1",
        "/say " ..
        message ..
        "\n/p " ..
        message
    )

    -- --------------------------------------------------------
    -- Alt + click
    -- --------------------------------------------------------

    heroButton:SetAttribute(
        "alt-type1",
        "macro"
    )

    heroButton:SetAttribute(
        "alt-macrotext1",
        "/say " ..
        message
    )

    needsHeroMacroUpdate = false
end

-- ============================================================
-- Update everything
-- ============================================================

local function UpdateAllMacros()

    if InCombatLockdown() then

        needsMarkMacroUpdate = true
        needsHeroMacroUpdate = true

        return
    end

    UpdateMarkMacro()
    UpdateChatMacro()
    UpdateHeroismMacro()
end

-- ============================================================
-- Button creation
-- ============================================================

local function MakeButton(
    name,
    text,
    texture
)

    local button =
        CreateFrame(
            "Button",
            name,
            mainFrame,
            "SecureActionButtonTemplate"
        )

    button:SetSize(
        CONFIG.buttonWidth,
        CONFIG.buttonHeight
    )

    button:EnableMouse(true)

    button:SetMouseClickEnabled(true)

    button:RegisterForClicks(
        "AnyUp",
        "AnyDown"
    )

    -- --------------------------------------------------------
    -- Background
    -- --------------------------------------------------------

    local bg =
        button:CreateTexture(
            nil,
            "BACKGROUND"
        )

    bg:SetAllPoints()

    bg:SetColorTexture(
        0.05,
        0.05,
        0.07,
        0.92
    )

    -- --------------------------------------------------------
    -- Border
    -- --------------------------------------------------------

    local border =
        button:CreateTexture(
            nil,
            "BORDER"
        )

    border:SetAllPoints()

    border:SetColorTexture(
        0.25,
        0.55,
        0.85,
        0.9
    )

    -- --------------------------------------------------------
    -- Icon
    -- --------------------------------------------------------

    local icon =
        button:CreateTexture(
            nil,
            "ARTWORK"
        )

    icon:SetSize(
        20,
        20
    )

    icon:SetPoint(
        "LEFT",
        7,
        0
    )

    icon:SetTexture(texture)

    -- --------------------------------------------------------
    -- Text
    -- --------------------------------------------------------

    local label =
        button:CreateFontString(
            nil,
            "OVERLAY",
            "GameFontNormal"
        )

    label:SetPoint(
        "LEFT",
        icon,
        "RIGHT",
        7,
        0
    )

    label:SetPoint(
        "RIGHT",
        -6,
        0
    )

    label:SetJustifyH(
        "CENTER"
    )

    label:SetText(
        text
    )

    button._bg = bg
    button._border = border
    button._icon = icon
    button._label = label

    return button
end

-- ============================================================
-- Button hover
-- ============================================================

local function ButtonHoverOn(self)

    self._bg:SetColorTexture(
        0.08,
        0.12,
        0.18,
        0.96
    )

    self._border:SetColorTexture(
        0.45,
        0.75,
        1.0,
        1
    )
end

local function ButtonHoverOff(self)

    self._bg:SetColorTexture(
        0.05,
        0.05,
        0.07,
        0.92
    )

    self._border:SetColorTexture(
        0.25,
        0.55,
        0.85,
        0.9
    )

    GameTooltip:Hide()
end

-- ============================================================
-- Create UI
-- ============================================================

local function CreateUI()

    if mainFrame then
        return
    end

    mainFrame =
        CreateFrame(
            "Frame",
            "PartyButtonsFrame",
            UIParent
        )

    mainFrame:SetSize(
        CONFIG.buttonWidth * 3 +
        CONFIG.spacing * 2,
        CONFIG.buttonHeight
    )

    mainFrame:SetPoint(
        "TOP",
        UIParent,
        "TOP",
        0,
        CONFIG.topOffset
    )

    -- ========================================================
    -- Mark
    -- ========================================================

    markButton =
        MakeButton(
            "PartyButtonsMarkButton",
            "Метки",
            "Interface\\TargetingFrame\\UI-RaidTargetingIcon_6"
        )

    markButton:SetPoint(
        "LEFT",
        mainFrame,
        "LEFT",
        0,
        0
    )

    -- ========================================================
    -- Greeting
    -- ========================================================

    hiButton =
        MakeButton(
            "PartyButtonsHiButton",
            "Сказать Привет",
            "Interface\\Icons\\Achievement_GuildPerk_EverybodysFriend"
        )

    hiButton:SetPoint(
        "LEFT",
        markButton,
        "RIGHT",
        CONFIG.spacing,
        0
    )

    -- ========================================================
    -- Heroism
    -- ========================================================

    heroButton =
        MakeButton(
            "PartyButtonsHeroButton",
            "Героизм",
            CONFIG.bloodlustIcon
        )

    heroButton:SetPoint(
        "LEFT",
        hiButton,
        "RIGHT",
        CONFIG.spacing,
        0
    )

    -- ========================================================
    -- Mark tooltip
    -- ========================================================

    markButton:SetScript(
        "OnEnter",
        function(self)

            ButtonHoverOn(self)

            GameTooltip:SetOwner(
                self,
                "ANCHOR_BOTTOM"
            )

            GameTooltip:AddLine(
                "Метки"
            )

            local tank, healer =
                GetPartyRoleUnits()

            if tank then

                GameTooltip:AddLine(
                    "Танк: " ..
                    UnitName(tank),
                    1,
                    1,
                    1
                )

            else

                GameTooltip:AddLine(
                    "Танк: не найден",
                    1,
                    0.3,
                    0.3
                )
            end

            if healer then

                GameTooltip:AddLine(
                    "Хил: " ..
                    UnitName(healer),
                    0.3,
                    1,
                    0.3
                )

            else

                GameTooltip:AddLine(
                    "Хил: не найден",
                    1,
                    0.3,
                    0.3
                )
            end

            GameTooltip:AddLine(
                "Квадрат → танк",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Треугольник → хил",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Нужно право ставить метки.",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:Show()
        end
    )

    -- ========================================================
    -- Greeting tooltip
    -- ========================================================

    hiButton:SetScript(
        "OnEnter",
        function(self)

            ButtonHoverOn(self)

            GameTooltip:SetOwner(
                self,
                "ANCHOR_BOTTOM"
            )

            GameTooltip:AddLine(
                "Сказать Привет"
            )

            GameTooltip:AddLine(
                "ЛКМ: Hi",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Shift + ЛКМ: yo",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Ctrl + ЛКМ: o/",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:Show()
        end
    )

    -- ========================================================
    -- Heroism tooltip
    -- ========================================================

    heroButton:SetScript(
        "OnEnter",
        function(self)

            ButtonHoverOn(self)

            GameTooltip:SetOwner(
                self,
                "ANCHOR_BOTTOM"
            )

            GameTooltip:AddLine(
                "Героизм / Жажда крови"
            )

            GameTooltip:AddLine(
                "ЛКМ: использовать свой lust",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Shift + ЛКМ: сообщение группе",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Ctrl + ЛКМ: сказать + группа",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:AddLine(
                "Alt + ЛКМ: сказать вслух",
                0.8,
                0.8,
                0.8
            )

            GameTooltip:Show()
        end
    )

    -- ========================================================
    -- Leave
    -- ========================================================

    markButton:SetScript(
        "OnLeave",
        ButtonHoverOff
    )

    hiButton:SetScript(
        "OnLeave",
        ButtonHoverOff
    )

    heroButton:SetScript(
        "OnLeave",
        ButtonHoverOff
    )
end

-- ============================================================
-- Events
-- ============================================================

local eventFrame =
    CreateFrame("Frame")

eventFrame:RegisterEvent(
    "PLAYER_LOGIN"
)

eventFrame:RegisterEvent(
    "GROUP_ROSTER_UPDATE"
)

eventFrame:RegisterEvent(
    "PLAYER_ROLES_ASSIGNED"
)

eventFrame:RegisterEvent(
    "ROLE_CHANGED_INFORM"
)

eventFrame:RegisterEvent(
    "PLAYER_REGEN_ENABLED"
)

eventFrame:RegisterEvent(
    "SPELLS_CHANGED"
)

eventFrame:RegisterEvent(
    "INSPECT_READY"
)

eventFrame:RegisterEvent(
    "PLAYER_SPECIALIZATION_CHANGED"
)

eventFrame:RegisterEvent(
    "UNIT_PET"
)

eventFrame:RegisterEvent(
    "PLAYER_ENTERING_WORLD"
)

eventFrame:RegisterEvent(
    "PET_SPECIALIZATION_CHANGED"
)

eventFrame:SetScript(
    "OnEvent",
    function(_, event, arg1)

        -- ====================================================
        -- Login
        -- ====================================================

        if event == "PLAYER_LOGIN" then

            CreateUI()

            C_Timer.After(
                0.5,
                UpdateAllMacros
            )

            C_Timer.After(
                1.5,
                UpdateAllMacros
            )

            C_Timer.After(
                3.0,
                UpdateHeroismMacro
            )

            Print(
                "Loaded."
            )

            Print(
                "Метки: квадрат на танка + треугольник на хила."
            )

            Print(
                "Привет: ЛКМ / Shift / Ctrl."
            )

            Print(
                "Героизм: ЛКМ = каст своего lust."
            )

            Print(
                "Команды: /pb show | hide | toggle | update | debug"
            )

            return
        end

        -- ====================================================
        -- Inspect result
        -- ====================================================

        if event == "INSPECT_READY" then

            -- Complete current inspection.
            inspectInProgress = false
            inspectUnit = nil

            -- Rebuild immediately.
            if not InCombatLockdown() then

                UpdateHeroismMacro()

            else

                needsHeroMacroUpdate = true
            end

            -- Continue queue.
            C_Timer.After(
                0.15,
                ProcessInspectQueue
            )

            return
        end

        -- ====================================================
        -- Group changes
        -- ====================================================

        if event == "GROUP_ROSTER_UPDATE"
            or event == "PLAYER_ROLES_ASSIGNED"
            or event == "ROLE_CHANGED_INFORM" then

            -- Clear stale inspections.
            inspectQueue = {}

            inspectInProgress = false
            inspectUnit = nil

            if InCombatLockdown() then

                needsMarkMacroUpdate = true
                needsHeroMacroUpdate = true

            else

                C_Timer.After(
                    0.2,
                    function()

                        UpdateMarkMacro()
                        UpdateHeroismMacro()

                        -- Start Hunter inspection.
                        for _, unit in ipairs(
                            GetGroupUnits()
                        ) do

                            local _, classFile =
                                UnitClass(unit)

                            if classFile ==
                                "HUNTER"
                                and not UnitIsUnit(
                                    unit,
                                    "player"
                                ) then

                                QueueHunterInspect(
                                    unit
                                )
                            end
                        end
                    end
                )
            end

            return
        end

        -- ====================================================
        -- Player specialization
        -- ====================================================

        if event ==
            "PLAYER_SPECIALIZATION_CHANGED" then

            if not InCombatLockdown() then

                C_Timer.After(
                    0.2,
                    UpdateHeroismMacro
                )

            else

                needsHeroMacroUpdate = true
            end

            return
        end

        -- ====================================================
        -- Pet changed
        -- ====================================================

        if event == "UNIT_PET" then

            if not InCombatLockdown() then

                C_Timer.After(
                    0.2,
                    UpdateHeroismMacro
                )

            else

                needsHeroMacroUpdate = true
            end

            return
        end

        -- ====================================================
        -- Pet specialization changed
        -- ====================================================

        if event ==
            "PET_SPECIALIZATION_CHANGED" then

            if not InCombatLockdown() then

                C_Timer.After(
                    0.2,
                    UpdateHeroismMacro
                )

            else

                needsHeroMacroUpdate = true
            end

            return
        end

        -- ====================================================
        -- Enter world
        -- ====================================================

        if event ==
            "PLAYER_ENTERING_WORLD" then

            if not InCombatLockdown() then

                C_Timer.After(
                    1.0,
                    UpdateAllMacros
                )
            end

            return
        end

        -- ====================================================
        -- Spell changes
        -- ====================================================

        if event == "SPELLS_CHANGED" then

            if not InCombatLockdown() then

                UpdateHeroismMacro()

            else

                needsHeroMacroUpdate = true
            end

            return
        end

        -- ====================================================
        -- Combat ended
        -- ====================================================

        if event ==
            "PLAYER_REGEN_ENABLED" then

            if needsMarkMacroUpdate
                or needsHeroMacroUpdate then

                UpdateAllMacros()

            else

                UpdateHeroismMacro()
                UpdateChatMacro()
            end

            return
        end
    end
)

-- ============================================================
-- Slash commands
-- ============================================================

SLASH_PARTYBUTTONS1 = "/pb"
SLASH_PARTYBUTTONS2 = "/partybuttons"

SlashCmdList.PARTYBUTTONS = function(msg)

    msg = (msg or "")
        :lower()
        :match("^%s*(.-)%s*$")

    -- ========================================================
    -- SHOW
    -- ========================================================

    if msg == "show" then

        if mainFrame then
            mainFrame:Show()
        end

        return
    end

    -- ========================================================
    -- HIDE
    -- ========================================================

    if msg == "hide" then

        if mainFrame then
            mainFrame:Hide()
        end

        return
    end

    -- ========================================================
    -- TOGGLE
    -- ========================================================

    if msg == "toggle" then

        if mainFrame then

            mainFrame:SetShown(
                not mainFrame:IsShown()
            )
        end

        return
    end

    -- ========================================================
    -- UPDATE
    -- ========================================================

    if msg == "update" then

        if InCombatLockdown() then

            Print(
                "Нельзя обновить secure buttons во время боя."
            )

            return
        end

        UpdateAllMacros()

        Print(
            "PartyButtons updated."
        )

        local tank, healer =
            GetPartyRoleUnits()

        Print(
            "Tank: " ..
            tostring(
                tank
                    and UnitName(tank)
                    or "not found"
            )
        )

        Print(
            "Healer: " ..
            tostring(
                healer
                    and UnitName(healer)
                    or "not found"
            )
        )

        return
    end

    -- ========================================================
    -- DEBUG
    -- ========================================================

    if msg == "debug" then

        Print(
            "===== PartyButtons Debug ====="
        )

        Print(
            "InCombat: " ..
            tostring(
                InCombatLockdown()
            )
        )

        Print(
            "InGroup: " ..
            tostring(
                IsInGroup()
            )
        )

        Print(
            "InRaid: " ..
            tostring(
                IsInRaid()
            )
        )

        local ownLust =
            GetOwnLustSpellID()

        Print(
            "Own lust: " ..
            tostring(
                ownLust
                    and GetSpellNameSafe(
                        ownLust
                    )
                    or "none"
            )
        )

        local tank, healer =
            GetPartyRoleUnits()

        Print(
            "Tank: " ..
            tostring(
                tank
                    and UnitName(tank)
                    or "NOT FOUND"
            )
        )

        Print(
            "Healer: " ..
            tostring(
                healer
                    and UnitName(healer)
                    or "NOT FOUND"
            )
        )

        local message =
            BuildLustAnnouncement()

        Print(
            "Lust message: " ..
            message
        )

        Print(
            "Inspect in progress: " ..
            tostring(
                inspectInProgress
            )
        )

        Print(
            "Inspect unit: " ..
            tostring(
                inspectUnit
                    or "none"
            )
        )

        if heroButton then

            Print(
                "Hero macro:"
            )

            Print(
                tostring(
                    heroButton:GetAttribute(
                        "macrotext1"
                    )
                )
            )
        end

        if markButton then

            Print(
                "Mark macro:"
            )

            Print(
                tostring(
                    markButton:GetAttribute(
                        "macrotext1"
                    )
                )
            )
        end

        return
    end

    -- ========================================================
    -- HELP
    -- ========================================================

    Print(
        "/pb show | hide | toggle | update | debug"
    )
end