--[[
T-RestedXP - addon for tracking 100% rested XP in WoW 1.12
T-RestedXP - аддон для отслеживания 100% RESTED XP в WoW 1.12

Addon GitHub link: https://github.com/whtmst/t-restedxp

Author: Mikhail Palagin (Wht Mst)
Website: https://band.link/whtmst

Compatibility:
- Designed for Turtle WoW server
--]]

-- SETTINGS / НАСТРОЙКИ

-- 0% rested XP message alert / Сообщение при 0% rested XP
local noRestedMsg = "=== Fuck I am not RESTED any more ==="

-- 100% rested XP message alert / Сообщение при 100% rested XP
local fullRestedMsg = "=== Hell YE 100% rested ==="

-- Announce channel / Канал анонса ("SELF", "EMOTE", "SAY", "PARTY", "RAID", "GUILD")
local chatChannel = "EMOTE"

-- Max 0% alerts in a row / Максимум оповещений о 0% подряд
local notifyCountZero = 1

-- Interval between 0% alerts (seconds) / Интервал между оповещениями о 0% (сек)
local notifyIntervalZero = 20

-- Show alert in center / Показывать оповещение по центру экрана
local showCenter = true

-- Time to show center message (seconds) / Время показа сообщения по центру (сек)
local centerMessageTime = 3

-- Play sound on alert / Воспроизводить звук при оповещении
local playSound = true

-- Sound for 0% rested XP / Звук для 0%
local soundNameZero = "RaidWarning"

-- Sound for 100% rested XP / Звук для 100%
local soundNameFull = "QUESTCOMPLETED"

-- Max player level / Максимальный уровень игрока
local maxLevel = 60

-- INTERNAL STATE / ВНУТРЕННЕЕ СОСТОЯНИЕ
local lastZeroRestedTime = 0
local zeroRestedCount = 0
local wasFullRested = false
local wasZeroRested = false
local zeroRestedTimerFrame = nil

-- Returns true if player is max level / Возвращает true, если игрок максимального уровня
local function IsPlayerMaxLevel()
    return UnitLevel("player") >= maxLevel
end

-- Returns rested XP percent (0-100), or nil if not available / Возвращает % rested XP (0-100), или nil если нет
local function GetRestedXPPercent()
    local exhaustion = GetXPExhaustion()
    local maxXP = UnitXPMax("player")
    
    -- If no max XP data - return nil / Если нет данных о максимальном XP - возвращаем nil
    if not maxXP or maxXP <= 0 then
        return nil
    end
    
    -- If no exhaustion (rested) - this means 0% / Если нет exhaustion (рестеда) - это означает 0%
    if not exhaustion or exhaustion <= 0 then
        return 0
    end
    
    local percent = (exhaustion / (maxXP * 1.5)) * 100
    return math.min(percent, 100)
end

-- Protection from repeated frame creation / Защита от повторного создания фрейма
if T_RestedXP_MessageFrame then
    T_RestedXP_MessageFrame:Hide()
    T_RestedXP_MessageFrame = nil
end

-- Custom center message frame / Кастомный фрейм для сообщений по центру
local restedXPMessageFrame = CreateFrame("Frame", "T_RestedXP_MessageFrame", UIParent)
restedXPMessageFrame:SetWidth(800)
restedXPMessageFrame:SetHeight(120)
restedXPMessageFrame:SetPoint("CENTER", UIParent, "CENTER", 0, 180)
restedXPMessageFrame:Hide()

local restedXPFontString = restedXPMessageFrame:CreateFontString(nil, "OVERLAY")
local fontPath = "Interface\\AddOns\\T-RestedXP\\fonts\\ARIALN.ttf"
if not restedXPFontString:SetFont(fontPath, 96, "OUTLINE") then
    -- Fallback to default font / Фоллбэк на стандартный шрифт
    restedXPFontString:SetFont("Fonts\\FRIZQT__.TTF", 96, "OUTLINE")
end
restedXPFontString:SetPoint("CENTER", restedXPMessageFrame, "CENTER", 0, 0)
restedXPFontString:SetTextColor(1, 1, 0)

local function ShowRestedXPMessage(msg)
    -- Reset previous state / Сброс предыдущего состояния
    restedXPMessageFrame:SetScript("OnUpdate", nil)
    restedXPMessageFrame:Hide()
    
    -- Set new message and state / Установка нового сообщения и состояния
    restedXPFontString:SetText(msg)
    restedXPMessageFrame:SetAlpha(1)
    restedXPMessageFrame:Show()

    -- Save start time / Сохраняем время начала показа
    restedXPMessageFrame.startTime = GetTime()
    restedXPMessageFrame.showDuration = centerMessageTime
    restedXPMessageFrame.fadeDuration = 1

    restedXPMessageFrame:SetScript("OnUpdate", function(frame, elapsed)
        local currentTime = GetTime()
        local timeElapsed = currentTime - restedXPMessageFrame.startTime
        local totalDuration = restedXPMessageFrame.showDuration + restedXPMessageFrame.fadeDuration

        if timeElapsed >= totalDuration then
            -- Time is up, hide frame / Время вышло, скрываем фрейм
            restedXPMessageFrame:Hide()
            restedXPMessageFrame:SetScript("OnUpdate", nil)
            restedXPMessageFrame.startTime = nil
        elseif timeElapsed >= restedXPMessageFrame.showDuration then
            -- Fade phase / Фаза затухания
            local fadeProgress = (timeElapsed - restedXPMessageFrame.showDuration) / restedXPMessageFrame.fadeDuration
            restedXPMessageFrame:SetAlpha(1 - fadeProgress)
        end
    end)
end

-- Send alert to chat and/or center / Отправить оповещение в чат и/или по центру
local function SendRestedAlert(msg, soundName)
    if chatChannel == "SELF" then
        DEFAULT_CHAT_FRAME:AddMessage(msg)
    elseif chatChannel then
        -- Check channel validity / Проверка на валидность канала
        local validChannels = {"EMOTE", "SAY", "PARTY", "RAID", "GUILD", "YELL"}
        local isValidChannel = false
        for _, channel in ipairs(validChannels) do
            if chatChannel == channel then
                isValidChannel = true
                break
            end
        end
        if isValidChannel then
            SendChatMessage(msg, chatChannel)
        else
            DEFAULT_CHAT_FRAME:AddMessage(msg)
        end
    end
    
    if showCenter then
        ShowRestedXPMessage(msg)
    end
    if playSound and soundName then
        PlaySound(soundName)
    end
end

-- Function to stop 0% rested timer / Функция для остановки таймера 0% рестеда
local function StopZeroRestedTimer()
    if zeroRestedTimerFrame then
        zeroRestedTimerFrame:SetScript("OnUpdate", nil)
        zeroRestedTimerFrame = nil
    end
end

-- Function to start 0% rested timer / Функция для запуска таймера 0% рестеда
local function StartZeroRestedTimer()
    -- First stop existing timer / Сначала останавливаем существующий таймер
    StopZeroRestedTimer()
    
    zeroRestedTimerFrame = CreateFrame("Frame")
    zeroRestedTimerFrame:SetScript("OnUpdate", function(self, elapsed)
        local now = GetTime()
        if zeroRestedCount < notifyCountZero and (now - lastZeroRestedTime) >= notifyIntervalZero then
            local percent = GetRestedXPPercent()
            if percent ~= nil and percent <= 0.1 then
                SendRestedAlert(noRestedMsg, soundNameZero)
                lastZeroRestedTime = now
                zeroRestedCount = zeroRestedCount + 1
            end
        end
        
        -- Stop timer if reached notification limit / Останавливаем таймер если достигли лимита оповещений
        if zeroRestedCount >= notifyCountZero then
            -- Use function instead of direct access / Используем функцию вместо прямого обращения
            StopZeroRestedTimer()
        end
    end)
end

-- Main event handler / Основная обработка событий
local function CheckRestedXP()
    if IsPlayerMaxLevel() then return end

    local percent = GetRestedXPPercent()
    local now = GetTime()

    if percent == nil then
        wasFullRested = false
        wasZeroRested = false
        zeroRestedCount = 0
        StopZeroRestedTimer()
        return
    end

    -- 100% rested XP: only one alert on entering state
    if percent >= 99.9 then
        if not wasFullRested then
            SendRestedAlert(fullRestedMsg, soundNameFull)
            wasFullRested = true
        end
        wasZeroRested = false
        zeroRestedCount = 0
        StopZeroRestedTimer()
        return
    end

    -- 0% rested XP: start timer for repeated notifications / 0% rested XP: запускаем таймер для повторных оповещений
    if percent <= 0.1 then
        if not wasZeroRested then
            -- First notification immediately / Первое оповещение сразу
            SendRestedAlert(noRestedMsg, soundNameZero)
            lastZeroRestedTime = now
            zeroRestedCount = 1
            wasZeroRested = true
            
            -- Start timer for next notifications / Запускаем таймер для следующих оповещений
            StartZeroRestedTimer()
        end
        wasFullRested = false
        return
    end

    -- Reset state if in between
    wasFullRested = false
    wasZeroRested = false
    zeroRestedCount = 0
    StopZeroRestedTimer()
end

-- Event frame setup / Создание фрейма событий
local frame = CreateFrame("Frame")
frame:RegisterEvent("PLAYER_ENTERING_WORLD")
frame:RegisterEvent("PLAYER_XP_UPDATE")
frame:RegisterEvent("UPDATE_EXHAUSTION")
frame:RegisterEvent("PLAYER_LEVEL_UP")
frame:SetScript("OnEvent", function()
    CheckRestedXP()
end)

-- Slash command for manual check / Слэш-команда для ручной проверки
SLASH_TRESTEDXP1 = "/trestedxp"
SlashCmdList["TRESTEDXP"] = function()
    local percent = GetRestedXPPercent()
    local msg
    if percent and percent > 0 then
        msg = string.format("Rested XP: %.1f%% / Отдых: %.1f%%", percent, percent)
        DEFAULT_CHAT_FRAME:AddMessage(msg)
        if showCenter and UIErrorsFrame then
            UIErrorsFrame:AddMessage(msg, 1, 1, 0, 1, 3)
        end
    else
        msg = "NO RESTED XP / НЕТ ОТДЫХА"
        DEFAULT_CHAT_FRAME:AddMessage(msg)
        if showCenter and UIErrorsFrame then
            UIErrorsFrame:AddMessage(msg, 1, 0, 0, 1, 3)
        end
    end
end

-- Slash command for test 0% rested XP / Слэш-команда для теста 0% rested XP
SLASH_TRESTEDXPTESTZERO1 = "/trestedxp-test-0"
SlashCmdList["TRESTEDXPTESTZERO"] = function()
    local oldGetRestedXPPercent = GetRestedXPPercent
    GetRestedXPPercent = function() return 0 end
    CheckRestedXP()
    GetRestedXPPercent = oldGetRestedXPPercent
    DEFAULT_CHAT_FRAME:AddMessage("Test: forced 0% rested XP / Тест: принудительно 0% отдыха")
end

-- Slash command for test 100% rested XP / Слэш-команда для теста 100% rested XP
SLASH_TRESTEDXPTESTFULL1 = "/trestedxp-test-100"
SlashCmdList["TRESTEDXPTESTFULL"] = function()
    local oldGetRestedXPPercent = GetRestedXPPercent
    GetRestedXPPercent = function() return 100 end
    CheckRestedXP()
    GetRestedXPPercent = oldGetRestedXPPercent
    DEFAULT_CHAT_FRAME:AddMessage("Test: forced 100% rested XP / Тест: принудительно 100% отдыха")
end

-- Cleanup on addon unload / Cleanup при выгрузке аддона
local cleanupFrame = CreateFrame("Frame")
cleanupFrame:RegisterEvent("ADDON_LOADED")
cleanupFrame:RegisterEvent("PLAYER_LOGOUT")
cleanupFrame:SetScript("OnEvent", function(self, event, addonName)
    if event == "PLAYER_LOGOUT" or (event == "ADDON_LOADED" and addonName == "T-RestedXP") then
        StopZeroRestedTimer()
        if restedXPMessageFrame then
            restedXPMessageFrame:SetScript("OnUpdate", nil)
            restedXPMessageFrame:Hide()
        end
    end
end)

-- End of file / Конец файла