Building Better Addons¶

-- Detect when Edit Mode is entered/exited
EventRegistry:RegisterCallback("EditMode.Enter", function()
    -- Show your addon's Edit Mode controls
end)

EventRegistry:RegisterCallback("EditMode.Exit", function()
    -- Hide Edit Mode controls, apply final positions
end)

-- Register a settings category
local category = Settings.RegisterVerticalLayoutCategory("My Addon")

-- Add controls (11.0.2+ signatures)
local setting = Settings.RegisterAddOnSetting(category,
    "MyAddon_Toggle", "toggle", MyAddon_DB,
    type(false), "Enable Feature", false
)
Settings.CreateCheckbox(category, setting, "Enables the main feature.")

Settings.RegisterAddOnCategory(category)

local addonName, ns = ...

-- tullaCTC's Midnight adaptation:
-- 1. Hook Blizzard's Cooldown widget to intercept SetCooldown calls
-- 2. Use Duration Provider API for secret-safe duration access
-- 3. Apply conditional coloring via Rules API

-- Hook the cooldown widget's countdown FontString
hooksecurefunc(CooldownFrameMixin, "SetCooldown", function(self, start, duration)
    if self:IsForbidden() then return end

    -- Get the built-in countdown text element (12.0+)
    local countdown = self:GetCountdownFontString()
    if not countdown then return end

    -- Restyle the countdown text
    countdown:SetFont("Interface\\AddOns\\" .. addonName .. "\\Fonts\\Main.ttf",
        ns.db.fontSize, "OUTLINE")
    countdown:SetTextColor(ns:GetColorForDuration(duration))
end)

local addonName, ns = ...

-- Wait for the Cooldown Manager to load
local frame = CreateFrame("Frame")
frame:RegisterEvent("ADDON_LOADED")
frame:SetScript("OnEvent", function(self, event, addon)
    if addon == "Blizzard_CooldownViewer" then
        self:UnregisterEvent("ADDON_LOADED")
        ns:SetupCDMExtension()
    end
end)

function ns:SetupCDMExtension()
    -- Create a custom bar anchored below the Cooldown Manager
    local powerBar = CreateFrame("StatusBar", nil, UIParent)
    powerBar:SetSize(200, 8)
    powerBar:SetStatusBarTexture("Interface\\TargetingFrame\\UI-StatusBar")
    powerBar:SetStatusBarColor(0.0, 0.4, 1.0)
    powerBar:SetMinMaxValues(0, 1)

    -- Anchor relative to the CDM frame
    -- CooldownViewerFrame is the parent of the CDM display
    if CooldownViewerFrame then
        powerBar:SetPoint("TOP", CooldownViewerFrame, "BOTTOM", 0, -4)
    end

    -- Background
    local bg = powerBar:CreateTexture(nil, "BACKGROUND")
    bg:SetAllPoints()
    bg:SetColorTexture(0, 0, 0, 0.6)

    -- Update with player power (safe — secondary resources are whitelisted)
    local updater = CreateFrame("Frame")
    updater:RegisterUnitEvent("UNIT_POWER_UPDATE", "player")
    updater:SetScript("OnEvent", function()
        local pct = UnitPowerPercent("player")
        powerBar:SetValue(pct)
    end)
end

local addonName, ns = ...

-- Embed the library
local EME = LibStub("EditModeExpanded-1.0")

-- Create your addon's display frame
local display = CreateFrame("Frame", "MyAddonDisplay", UIParent)
display:SetSize(200, 100)
display:SetPoint("CENTER")

-- Register with Edit Mode Expanded
-- This makes the frame draggable in Edit Mode with per-profile saved positions
EME:RegisterFrame(display, "My Addon Display", ns.db)

local addonName, ns = ...

local myFrame = CreateFrame("Frame", "MyAddonFrame", UIParent,
    "BackdropTemplate")
myFrame:SetSize(150, 40)
myFrame:SetPoint("CENTER")
myFrame:SetBackdrop({
    bgFile = "Interface\\Buttons\\WHITE8X8",
    edgeFile = "Interface\\Buttons\\WHITE8X8",
    edgeSize = 1,
})
myFrame:SetBackdropColor(0, 0, 0, 0.7)
myFrame:SetBackdropBorderColor(0.3, 0.3, 0.3)

-- Make it draggable
myFrame:SetMovable(true)
myFrame:EnableMouse(true)
myFrame:RegisterForDrag("LeftButton")
myFrame:SetScript("OnDragStart", myFrame.StartMoving)
myFrame:SetScript("OnDragStop", function(self)
    self:StopMovingOrSizing()
    -- Save position
    local point, _, relPoint, x, y = self:GetPoint()
    ns.db.position = { point, relPoint, x, y }
end)

-- Show selection glow during Edit Mode
local selectionGlow = myFrame:CreateTexture(nil, "OVERLAY")
selectionGlow:SetAllPoints()
selectionGlow:SetColorTexture(0, 0.5, 1, 0.15)
selectionGlow:Hide()

EventRegistry:RegisterCallback("EditMode.Enter", function()
    selectionGlow:Show()
    myFrame:EnableMouse(true)
end)

EventRegistry:RegisterCallback("EditMode.Exit", function()
    selectionGlow:Hide()
end)

local addonName, ns = ...

-- Defaults for SavedVariables
ns.defaults = {
    classColorHealth = true,
    darkMode = true,
    healthBarTexture = "Interface\\TargetingFrame\\UI-StatusBar",
    scale = 1.0,
}

-- Global compartment click handler
function BetterExample_OnCompartmentClick()
    Settings.OpenToCategory(ns.categoryID)
end

-- Wait for SavedVariables to load
local loader = CreateFrame("Frame")
loader:RegisterEvent("ADDON_LOADED")
loader:SetScript("OnEvent", function(self, event, addon)
    if addon ~= addonName then return end
    self:UnregisterEvent("ADDON_LOADED")

    -- Initialize SavedVariables with defaults
    if not BetterExampleDB then
        BetterExampleDB = {}
    end
    for k, v in pairs(ns.defaults) do
        if BetterExampleDB[k] == nil then
            BetterExampleDB[k] = v
        end
    end
    ns.db = BetterExampleDB

    -- Register settings
    ns:RegisterSettings()

    -- Apply enhancements
    ns:ApplyEnhancements()
end)

local addonName, ns = ...

function ns:RegisterSettings()
    local category = Settings.RegisterVerticalLayoutCategory("BetterExample")
    ns.categoryID = category:GetID()

    -- Class-colored health toggle
    do
        local setting = Settings.RegisterAddOnSetting(category,
            "BetterExample_ClassColor", "classColorHealth", ns.db,
            type(true), "Class-Colored Health Bars", true
        )
        Settings.CreateCheckbox(category, setting,
            "Color health bars by class color on Player and Target frames.")
        setting:SetValueChangedCallback(function()
            ns:RefreshFrames()
        end)
    end

    -- Dark mode toggle
    do
        local setting = Settings.RegisterAddOnSetting(category,
            "BetterExample_DarkMode", "darkMode", ns.db,
            type(true), "Dark Mode", true
        )
        Settings.CreateCheckbox(category, setting,
            "Darken frame backgrounds and borders.")
        setting:SetValueChangedCallback(function()
            ns:RefreshFrames()
        end)
    end

    -- Scale slider
    do
        local setting = Settings.RegisterAddOnSetting(category,
            "BetterExample_Scale", "scale", ns.db,
            type(1.0), "Frame Scale", 1.0
        )
        local options = Settings.CreateSliderOptions(0.5, 2.0, 0.05)
        options:SetLabelFormatter(MinimalSliderWithSteppersMixin.Label.Right)
        Settings.CreateSlider(category, setting, options,
            "Adjust the scale of enhanced frames.")
        setting:SetValueChangedCallback(function()
            ns:RefreshFrames()
        end)
    end

    Settings.RegisterAddOnCategory(category)
end

function ns:ApplyEnhancements()
    local CLASS_COLORS = RAID_CLASS_COLORS

    -- Hook PlayerFrame health bar updates
    hooksecurefunc(PlayerFrame.PlayerFrameContent.PlayerFrameContentMain
        .HealthBarsContainer.HealthBar, "UpdateFillBar", function(self)
        if self:IsForbidden() then return end
        if not ns.db.classColorHealth then return end

        local _, class = UnitClass("player")
        local color = CLASS_COLORS[class]
        if color then
            self:SetStatusBarColor(color.r, color.g, color.b)
        end
    end)

    -- Hook TargetFrame health bar
    hooksecurefunc("TargetFrame_UpdateAuras", function(self)
        if self:IsForbidden() then return end
        if not ns.db.classColorHealth then return end

        local unit = self.unit
        if not unit or not UnitIsPlayer(unit) then return end

        local _, class = UnitClass(unit)
        local color = CLASS_COLORS[class]
        if color and self.HealthBar then
            self.HealthBar:SetStatusBarColor(color.r, color.g, color.b)
        end
    end)

    -- Dark mode: hook frame texture updates
    if ns.db.darkMode then
        ns:ApplyDarkMode()
    end
end

function ns:ApplyDarkMode()
    -- Darken the player frame background
    local frames = { PlayerFrame, TargetFrame, FocusFrame }
    for _, unitFrame in ipairs(frames) do
        if unitFrame and not unitFrame:IsForbidden() then
            -- Add a dark overlay behind the frame
            if not unitFrame.__darkOverlay then
                local overlay = unitFrame:CreateTexture(nil, "BACKGROUND", nil, -8)
                overlay:SetAllPoints()
                overlay:SetColorTexture(0, 0, 0, 0.3)
                unitFrame.__darkOverlay = overlay
            end
        end
    end
end

function ns:RefreshFrames()
    -- Re-apply enhancements when settings change
    -- Combat-safe: visual-only operations
    if PlayerFrame and not PlayerFrame:IsForbidden() then
        if ns.db.darkMode then
            if PlayerFrame.__darkOverlay then
                PlayerFrame.__darkOverlay:Show()
            end
        else
            if PlayerFrame.__darkOverlay then
                PlayerFrame.__darkOverlay:Hide()
            end
        end

        PlayerFrame:SetScale(ns.db.scale)
    end

    if TargetFrame and not TargetFrame:IsForbidden() then
        TargetFrame:SetScale(ns.db.scale)
    end
end