frame.lua

--[[
    Copyright (c) 2014 by Adam Hellberg.

    Permission is hereby granted, free of charge, to any person obtaining a copy
    of this software and associated documentation files (the "Software"), to deal
    in the Software without restriction, including without limitation the rights
    to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
    copies of the Software, and to permit persons to whom the Software is
    furnished to do so, subject to the following conditions:

    The above copyright notice and this permission notice shall be included in
    all copies or substantial portions of the Software.

    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
    THE SOFTWARE.
]]

local NAME, T = ...

T.BodyguardFrame = {}

local bf = T.BodyguardFrame

local locked = true

local frame
local created = false

local function Create()
    if created then return end

    -- A global name required for dropdown functionality
    frame = CreateFrame("Button", "BodyguardHealthFrame", UIParent, "SecureActionButtonTemplate")

    -- DEBUG
    bf.Frame = frame

    frame:Hide()

    --frame:EnableMouse(false)
    frame:SetMovable(false)

    frame.statusLabel = frame:CreateFontString(nil, "OVERLAY", "GameFontNormal")
    frame.statusLabel:SetWidth(70)
    frame.statusLabel:SetHeight(16)
    frame.statusLabel:SetPoint("TOPRIGHT", frame, "TOPRIGHT", -5, -5)
    frame.statusLabel:SetText("Unknown")
    frame.statusLabel:SetJustifyH("RIGHT")

    frame.nameLabel = frame:CreateFontString(nil, "OVERLAY", "GameFontNormal")
    frame.nameLabel:SetWidth(100)
    frame.nameLabel:SetHeight(16)
    frame.nameLabel:SetPoint("TOPLEFT", frame, "TOPLEFT", 5, -5)
    frame.nameLabel:SetPoint("RIGHT", frame.statusLabel, "LEFT", -5, 0)
    frame.nameLabel:SetText("Bodyguard")
    frame.nameLabel:SetJustifyH("LEFT")

    frame.healthBar = CreateFrame("StatusBar", nil, frame)
    local hb = frame.healthBar
    hb:SetMinMaxValues(0, 1)
    hb:SetValue(1)
    hb:SetPoint("TOP", frame.nameLabel, "BOTTOM", 0, -5)
    hb:SetPoint("BOTTOMLEFT", frame, "BOTTOMLEFT", 5, 5)
    hb:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -5, 5)
    hb:SetStatusBarColor(0, 1, 0)

    frame.healthLabel = hb:CreateFontString(nil, "OVERLAY", "GameFontNormal")
    frame.healthLabel:SetHeight(25)
    frame.healthLabel:SetPoint("TOPLEFT", frame.healthBar, "TOPLEFT")
    frame.healthLabel:SetPoint("BOTTOMRIGHT", frame.healthBar, "BOTTOMRIGHT")
    frame.healthLabel:SetTextColor(1, 1, 1)
    frame.healthLabel:SetText("No HP data")

    created = true

    bf:UpdateSettings()
end

function bf:ResetSettings()
    Create()
    frame:ClearAllPoints()
    frame:SetWidth(200)
    frame:SetHeight(60)
    frame:SetScale(1)
    frame:SetPoint("CENTER")
    self:SaveSettings()
end

function bf:UpdateSettings()
    if T.InCombat then return end

    Create()

    local settings = T.DB.profile.FrameSettings

    frame:ClearAllPoints()
    frame:SetWidth(settings.Width or 200)
    frame:SetHeight(settings.Height or 60)
    if settings.RelPoint then
        frame:SetPoint(settings.Point or "CENTER", nil, settings.RelPoint, settings.Offset.X or 0, settings.Offset.Y or 0)
    else
        frame:SetPoint(settings.Point or "CENTER", settings.Offset.X or 0, settings.Offset.Y or 0)
    end

    if settings.Scale <= 0 then
        settings.Scale = 1
    end

    frame:SetScale(settings.Scale)

    local lsm = T.LSM

    local backdrop = {
        bgFile = lsm:Fetch(lsm.MediaType.BACKGROUND, settings.Backdrop.Background),
        edgeFile = lsm:Fetch(lsm.MediaType.BORDER, settings.Backdrop.Border),
        tile = settings.Backdrop.Tile,
        edgeSize = settings.Backdrop.BorderSize,
        tileSize = settings.Backdrop.TileSize,
        insets = {
            left = settings.Backdrop.Insets.Left,
            right = settings.Backdrop.Insets.Right,
            top = settings.Backdrop.Insets.Top,
            bottom = settings.Backdrop.Insets.Bottom
        }
    }

    frame:SetBackdrop(backdrop)
    local bdcolor = settings.Backdrop.Color
    frame:SetBackdropColor(bdcolor.R, bdcolor.G, bdcolor.B, bdcolor.A)
    local bdbcolor = settings.Backdrop.BorderColor
    frame:SetBackdropBorderColor(bdbcolor.R, bdbcolor.G, bdbcolor.B, bdbcolor.A)

    frame.healthBar:SetStatusBarTexture(lsm:Fetch(lsm.MediaType.STATUSBAR, settings.Texture), "ARTWORK")

    local fontFlags = settings.FontFlags
    if fontFlags == "NONE" then fontFlags = nil end

    frame.healthLabel:SetFont(lsm:Fetch(lsm.MediaType.FONT, settings.Font), settings.FontSize, fontFlags)
    frame.healthLabel:SetTextColor(settings.FontColor.R, settings.FontColor.G, settings.FontColor.B, settings.FontColor.A)

    frame:SetAlpha(settings.Opacity / 100)

    self:SaveSettings()
end

function bf:SaveSettings()
    Create()

    local settings = T.DB.profile.FrameSettings

    settings.Width = frame:GetWidth()
    settings.Height = frame:GetHeight()
    settings.Scale = frame:GetScale()
    local point, _, relPoint, x, y = frame:GetPoint(1)
    settings.Point = point
    settings.RelPoint = relPoint
    settings.Offset.X = x
    settings.Offset.Y = y
    --settings.Texture = hb:GetStatusBarTexture():GetTexture()
end

function bf:UpdateName(name)
    Create()
    frame.nameLabel:SetText(name)
    if not T.DB.profile.FrameSettings.ClickThrough and not T.InCombat then
        frame:SetAttribute("macrotext1", "/targetexact " .. name)
    end
end

function bf:UpdateStatus(status)
    Create()
    local text = "Unknown"
    for label, id in pairs(T.LBG.Status) do
        if id == status then text = label break end
    end
    frame.statusLabel:SetText(text)
end

local function round(num)
    return math.floor(num + 0.5)
end

local at_warn_threshold = false
local health_warnings = {5, 10, 20, 30}
local health_warns = {}

local shown_by_unlock = false

local BreakUpLargeNumbers = BreakUpLargeNumbers

local suffixes = {"k", "M"}

-- Upvalue since this is used from combat events
local format = string.format
local gsub = string.gsub

local function shorten(number)
    local suffix = ""
    local suffixIndex = 0

    while number > 1000 and suffixIndex < #suffixes do
        number = number / 1000
        suffixIndex = suffixIndex + 1
        suffix = suffixes[suffixIndex]
    end

    -- The gsub will strip trailing zeroes and decimal points when needed
    -- 100.00 becomes 100, but 100.02 stays the same
    return gsub(format("%.2f", number), "%.?0+$", "") .. suffix
end

local text_formatters = {
    NONE = function() return "" end,
    PERCENTAGE = function(health, maxHealth, pct) return format("%d%%", pct) end,
    SHORT = function(health) return shorten(health) end,
    LONG = function(health)
        return BreakUpLargeNumbers(health)
    end,
    MIXED = function(health, maxHealth, pct)
        return format("%s (%d%%)", shorten(health), pct)
    end
}

function bf:UpdateHealthBar(health, maxHealth)
    health = health or T.DB.char.Health or T.LBG:GetHealth()
    maxHealth = maxHealth or T.DB.char.MaxHealth or T.LBG:GetMaxHealth()

    Create()

    local red = 1
    local green = 1
    local blue = 0

    local ratio = (maxHealth > 0) and (health / maxHealth) or 0

    if T.DB.profile.FrameSettings.HealthBasedColor then
        if ratio > 0.5 then
            red = 2 - ratio * 2
        elseif ratio < 0.5 then
            green = ratio * 2
        end
    else
        red = T.DB.profile.FrameSettings.CustomColor.R
        green = T.DB.profile.FrameSettings.CustomColor.G
        blue = T.DB.profile.FrameSettings.CustomColor.B
    end

    local hb = frame.healthBar

    hb:SetStatusBarColor(red, green, blue)

    hb:SetMinMaxValues(0, maxHealth)
    hb:SetValue(health)

    local percentage = round(ratio * 100)

    local style = T.DB.profile.FrameSettings.HealthTextStyle

    local text = text_formatters[style](health, maxHealth, percentage)

    frame.healthLabel:SetText(text)

    if T.LBG:GetStatus() == T.LBG.Status.Inactive then return end

    if percentage <= 30 then
        at_warn_threshold = true
        for i = 1, #health_warnings do
            local threshold = health_warnings[i]
            if percentage <= threshold then
                if not health_warns[i] then
                    if T.DB.profile.EnableWarn then
                        PlaySoundFile(T.LSM.Fetch(T.LSM.MediaType.SOUND, T.DB.WarnSound), "Master")
                        RaidNotice_AddMessage(RaidWarningFrame, ("%s @ %d%%!"):format(T.LBG:GetName(), percentage), ChatTypeInfo["RAID_WARNING"])
                    end
                    health_warns[i] = true
                end
                break
            end
        end
    elseif at_warn_threshold then
        at_warn_threshold = false
        for i = 1, #health_warnings do
            health_warns[i] = false
        end
    end
end

function bf:SetMenu(enabled)
    Create()
    if enabled then
        frame:EnableMouse(true)
        frame:SetAttribute("type1", "macro")
        local name = T.LBG:GetName()
        if name then
            frame:SetAttribute("macrotext1", "/targetexact " .. name)
        end
        frame:SetScript("OnMouseUp", function(self, button)
            if button ~= "RightButton" or T.InCombat then return end
            T.Dropdown:Show(frame)
        end)
    else
        frame:SetAttribute("type1", nil)
        frame:SetScript("OnMouseUp", nil)
        frame:EnableMouse(false)
    end
    T.DB.profile.FrameSettings.ClickThrough = not enabled
end

function bf:EnableMenu()
    self:SetMenu(true)
end

function bf:DisableMenu()
    self:SetMenu(false)
end

function bf:Show(force)
    if T.InCombat then
        T:Log("Tried to show frame in combat, queueing action.", true)
        T:QueueShow()
        return
    end
    if not force and (self:IsShowing() or not T.DB.profile.Enabled) then return end
    if not force then shown_by_unlock = false end
    Create()
    self:UpdateSettings()
    frame:Show()
end

function bf:Hide()
    if T.InCombat then
        T:Log("Tried to hide frame in combat, queueing action.", true)
        T:QueueHide()
        return
    end
    if not self:IsShowing() then return end
    Create()
    frame:Hide()
    shown_by_unlock = false
end

function bf:IsShowing()
    return created and frame:IsVisible()
end

function bf:IsLocked()
    return locked
end

function bf:Unlock()
    Create()
    frame:SetMovable(true)

    frame:SetScript("OnMouseDown", function(f) f:StartMoving() end)
    frame:SetScript("OnMouseUp", function(f)
        f:StopMovingOrSizing()
        bf:SaveSettings()
    end)

    if not self:IsShowing() then
        self:Show(true)
        shown_by_unlock = true
    end

    locked = false
end

function bf:Lock()
    Create()
    frame:SetMovable(false)

    frame:SetScript("OnMouseDown", nil)
    frame:SetScript("OnMouseUp", nil)

    self:SaveSettings()

    locked = true

    if not T.DB.profile.FrameSettings.ClickThrough then
        self:EnableMenu()
    else
        frame:EnableMouse(false)
    end

    if not T.DB.profile.Enabled or not T.DB.char.HasBodyguard or shown_by_unlock then
        self:Hide()
    end

    shown_by_unlock = false
end

function bf:EnterCombat()
    if T.DB.profile.FrameSettings.UseCombatOpacity then
        frame:SetAlpha(T.DB.profile.FrameSettings.CombatOpacity / 100)
    end
end