--[[--------------------------------------------------------------------
  Time To Drink  -  Turtle WoW (vanilla 1.12) addon   v1.1
  Watches a chosen healer's mana while you are in a 5-man party
  (disabled in raids) and announces in party chat when they drop
  below a configurable threshold, repeating on an adjustable cooldown.

  Slash commands:
    /ttd            toggle the window
    /ttd status     print current settings + party state
    /ttd test       send a test line to party chat
    /ttd debug      toggle verbose messages
----------------------------------------------------------------------]]

TimeToDrink = {}
local TTD = TimeToDrink

TTD.roster         = {}
TTD.selectedHealer = nil
TTD.threshold      = 40
TTD.cooldown       = 15
TTD.lastAnnounce   = 0
TTD.wasInParty     = false
TTD.pollElapsed    = 0
TTD.debug          = false

local MIN_COOLDOWN  = 10
local POLL_INTERVAL = 0.5

local function msg(text)
    DEFAULT_CHAT_FRAME:AddMessage("|cff33ff99TTD:|r " .. text)
end
local function dbg(text)
    if TTD.debug then DEFAULT_CHAT_FRAME:AddMessage("|cff8888ffTTD debug:|r " .. text) end
end
local function err(text)
    DEFAULT_CHAT_FRAME:AddMessage("|cffff4040TTD error:|r " .. tostring(text))
end

local function TTD_InFivePlayerParty()
    if GetNumRaidMembers() > 0 then return false end
    return GetNumPartyMembers() > 0
end

local function TTD_ClampThreshold(v)
    if not v then return TTD.threshold end
    if v < 1 then v = 1 end
    if v > 100 then v = 100 end
    return math.floor(v)
end

local function TTD_ClampCooldown(v)
    if not v then return TTD.cooldown end
    if v < MIN_COOLDOWN then v = MIN_COOLDOWN end
    return math.floor(v)
end

function TTD_RefreshRoster()
    TTD.roster = {}
    if not TTD_InFivePlayerParty() then return end
    local myname = UnitName("player")
    table.insert(TTD.roster, myname)
    local n = GetNumPartyMembers()
    local i = 1
    while i <= n do
        local nm = UnitName("party" .. i)
        if nm then table.insert(TTD.roster, nm) end
        i = i + 1
    end
    if TTD.selectedHealer then
        local found = false
        local j = 1
        while j <= table.getn(TTD.roster) do
            if TTD.roster[j] == TTD.selectedHealer then found = true end
            j = j + 1
        end
        if not found then TTD.selectedHealer = nil end
    end
end

local function TTD_UnitForName(name)
    if not name then return nil end
    if UnitName("player") == name then return "player" end
    local n = GetNumPartyMembers()
    local i = 1
    while i <= n do
        if UnitName("party" .. i) == name then return "party" .. i end
        i = i + 1
    end
    return nil
end

function TTD_InitHealerDropdown()
    local dd = TimeToDrinkHealerDropdown
    if not dd then return end
    UIDropDownMenu_Initialize(dd, function()
        local i = 1
        while i <= table.getn(TTD.roster) do
            local nm   = TTD.roster[i]
            local info = {}
            info.text  = nm
            info.value = nm
            info.func  = function()
                TTD.selectedHealer = nm
                UIDropDownMenu_SetSelectedValue(dd, nm)
                local t = getglobal("TimeToDrinkHealerDropdownText")
                if t then t:SetText(nm) end
                dbg("healer set to " .. nm)
                if TTD_InFivePlayerParty() then
                    SendChatMessage(nm .. " has been set as the healer in Time To Drink.  The party will be alerted when they drop below " .. TTD.threshold .. "% mana.", "PARTY")
                end
            end
            UIDropDownMenu_AddButton(info)
            i = i + 1
        end
    end)

    local t = getglobal("TimeToDrinkHealerDropdownText")
    if TTD.selectedHealer then
        UIDropDownMenu_SetSelectedValue(dd, TTD.selectedHealer)
        if t then t:SetText(TTD.selectedHealer) end
    else
        if t then t:SetText("Select healer") end
    end
end

local function TTD_CreateUI()
    if TimeToDrinkUI then return end

    local f = CreateFrame("Frame", "TimeToDrinkUI", UIParent)
    f:SetWidth(260)
    f:SetHeight(180)
    f:SetPoint("CENTER", UIParent, "CENTER", 0, 120)
    f:SetBackdrop({
        bgFile   = "Interface\\DialogFrame\\UI-DialogBox-Background",
        edgeFile = "Interface\\DialogFrame\\UI-DialogBox-Border",
        tile = true, tileSize = 32, edgeSize = 32,
        insets = { left = 11, right = 12, top = 12, bottom = 11 },
    })
    f:SetMovable(true)
    f:EnableMouse(true)
    f:RegisterForDrag("LeftButton")
    f:SetScript("OnDragStart", function() this:StartMoving() end)
    f:SetScript("OnDragStop", function() this:StopMovingOrSizing() end)
    f:Hide()

    local title = f:CreateFontString(nil, "OVERLAY", "GameFontNormal")
    title:SetPoint("TOP", f, "TOP", 0, -16)
    title:SetText("Time To Drink")

    local close = CreateFrame("Button", nil, f, "UIPanelCloseButton")
    close:SetPoint("TOPRIGHT", f, "TOPRIGHT", -6, -6)

    local tl = f:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall")
    tl:SetPoint("TOPLEFT", f, "TOPLEFT", 22, -46)
    tl:SetText("Mana threshold (1-100%)")

    local te = CreateFrame("EditBox", "TimeToDrinkThresholdEdit", f, "InputBoxTemplate")
    te:SetWidth(50); te:SetHeight(20)
    te:SetPoint("TOPLEFT", tl, "BOTTOMLEFT", 4, -4)
    te:SetAutoFocus(false)
    te:SetMaxLetters(3)
    te:SetText("" .. TTD.threshold)
    te:SetScript("OnTextChanged", function()
        local s = this:GetText()
        local clean = string.gsub(s, "[^0-9]", "")
        if clean ~= s then this:SetText(clean) end
    end)
    local function commitThreshold()
        TTD.threshold = TTD_ClampThreshold(tonumber(this:GetText()))
        this:SetText("" .. TTD.threshold)
        this:ClearFocus()
    end
    te:SetScript("OnEnterPressed", commitThreshold)
    te:SetScript("OnEditFocusLost", commitThreshold)

    local cl = f:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall")
    cl:SetPoint("TOPLEFT", tl, "TOPLEFT", 120, 0)
    cl:SetText("Cooldown (min 10s)")

    local ce = CreateFrame("EditBox", "TimeToDrinkCooldownEdit", f, "InputBoxTemplate")
    ce:SetWidth(50); ce:SetHeight(20)
    ce:SetPoint("TOPLEFT", cl, "BOTTOMLEFT", 4, -4)
    ce:SetAutoFocus(false)
    ce:SetMaxLetters(4)
    ce:SetText("" .. TTD.cooldown)
    ce:SetScript("OnTextChanged", function()
        local s = this:GetText()
        local clean = string.gsub(s, "[^0-9]", "")
        if clean ~= s then this:SetText(clean) end
    end)
    local function commitCooldown()
        TTD.cooldown = TTD_ClampCooldown(tonumber(this:GetText()))
        this:SetText("" .. TTD.cooldown)
        this:ClearFocus()
    end
    ce:SetScript("OnEnterPressed", commitCooldown)
    ce:SetScript("OnEditFocusLost", commitCooldown)

    local hl = f:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall")
    hl:SetPoint("TOPLEFT", f, "TOPLEFT", 22, -104)
    hl:SetText("Healer")

    local dd = CreateFrame("Frame", "TimeToDrinkHealerDropdown", f, "UIDropDownMenuTemplate")
    dd:SetPoint("TOPLEFT", hl, "BOTTOMLEFT", -16, -4)

    local status = f:CreateFontString("TimeToDrinkStatus", "OVERLAY", "GameFontDisableSmall")
    status:SetPoint("BOTTOM", f, "BOTTOM", 0, 18)
    status:SetText("Type numbers, pick a healer, press Enter.")
end

function TTD_ShowUI()
    TTD_CreateUI()
    TTD_RefreshRoster()
    TTD_InitHealerDropdown()
    if TimeToDrinkThresholdEdit then TimeToDrinkThresholdEdit:SetText("" .. TTD.threshold) end
    if TimeToDrinkCooldownEdit  then TimeToDrinkCooldownEdit:SetText("" .. TTD.cooldown) end
    TimeToDrinkUI:Show()
end

function TTD_ToggleUI()
    TTD_CreateUI()
    if TimeToDrinkUI:IsShown() then
        TimeToDrinkUI:Hide()
    else
        TTD_ShowUI()
    end
end

local function TTD_CheckHealer()
    if not TTD_InFivePlayerParty() then return end
    if not TTD.selectedHealer then return end
    local unit = TTD_UnitForName(TTD.selectedHealer)
    if not unit or not UnitExists(unit) then return end
    if UnitIsDeadOrGhost(unit) then return end
    if UnitPowerType(unit) ~= 0 then return end
    local maxMana = UnitManaMax(unit)
    if not maxMana or maxMana == 0 then return end
    local pct = (UnitMana(unit) / maxMana) * 100
    if pct < TTD.threshold then
        if (GetTime() - TTD.lastAnnounce) >= TTD.cooldown then
            SendChatMessage("The healer has less than " .. TTD.threshold .. "% mana", "PARTY")
            TTD.lastAnnounce = GetTime()
            dbg("announced (" .. math.floor(pct) .. "% now)")
        end
    end
end

local driver = CreateFrame("Frame", "TimeToDrinkDriver", UIParent)
driver:RegisterEvent("VARIABLES_LOADED")
driver:RegisterEvent("PARTY_MEMBERS_CHANGED")
driver:RegisterEvent("RAID_ROSTER_UPDATE")
driver:RegisterEvent("PLAYER_LOGOUT")

driver:SetScript("OnEvent", function()
    if event == "VARIABLES_LOADED" then
        if type(TimeToDrinkDB) ~= "table" then TimeToDrinkDB = {} end
        if TimeToDrinkDB.threshold then TTD.threshold = TTD_ClampThreshold(TimeToDrinkDB.threshold) end
        if TimeToDrinkDB.cooldown  then TTD.cooldown  = TTD_ClampCooldown(TimeToDrinkDB.cooldown) end
        if TimeToDrinkDB.healer    then TTD.selectedHealer = TimeToDrinkDB.healer end

    elseif event == "PLAYER_LOGOUT" then
        if type(TimeToDrinkDB) ~= "table" then TimeToDrinkDB = {} end
        TimeToDrinkDB.threshold = TTD.threshold
        TimeToDrinkDB.cooldown  = TTD.cooldown
        TimeToDrinkDB.healer    = TTD.selectedHealer

    else
        local inParty = TTD_InFivePlayerParty()
        TTD_RefreshRoster()
        if TimeToDrinkUI and TimeToDrinkUI:IsShown() then TTD_InitHealerDropdown() end
        if inParty and not TTD.wasInParty then
            local ok, e = pcall(TTD_ShowUI)
            if not ok then err(e) end
        end
        TTD.wasInParty = inParty
    end
end)
driver:Show()

driver:SetScript("OnUpdate", function()
    TTD.pollElapsed = TTD.pollElapsed + arg1
    if TTD.pollElapsed < POLL_INTERVAL then return end
    TTD.pollElapsed = 0
    TTD_CheckHealer()
end)

SLASH_TIMETODRINK1 = "/ttd"
SLASH_TIMETODRINK2 = "/timetodrink"
SlashCmdList["TIMETODRINK"] = function(arg)
    local cmd = string.lower(arg or "")

    if cmd == "status" then
        msg("threshold=" .. TTD.threshold .. "%  cooldown=" .. TTD.cooldown .. "s  healer=" ..
            (TTD.selectedHealer or "none") .. "  inParty=" .. tostring(TTD_InFivePlayerParty()) ..
            "  raid=" .. tostring(GetNumRaidMembers() > 0))
        return
    end

    if cmd == "test" then
        if TTD_InFivePlayerParty() then
            SendChatMessage("Time To Drink test - if you see this, chat works.", "PARTY")
            msg("sent test line to party chat.")
        else
            msg("not in a 5-man party, so nothing was sent to party chat.")
        end
        return
    end

    if cmd == "debug" then
        TTD.debug = not TTD.debug
        msg("debug " .. (TTD.debug and "ON" or "OFF"))
        return
    end

    local ok, e = pcall(TTD_ToggleUI)
    if not ok then err(e) end
end

msg("loaded v1.3. /ttd to open, /ttd status, /ttd test. (5-man parties only)")