--[[
	Author: Dennis Werner Garske (DWG) / brian / Mewtiny
	License: MIT License
]]
local _G = _G or getfenv(0)
local CleveRoids = _G.CleveRoids or {}

local Extension = CleveRoids.RegisterExtension("CT_UnitFrames")
Extension.RegisterEvent("ADDON_LOADED", "OnLoad")

function Extension.SetHook(widget)
    local hookedOnEnter = widget:GetScript("OnEnter")
    local hookedOnLeave = widget:GetScript("OnLeave")

    widget:SetScript("OnEnter", function()
        hookedOnEnter()
        CleveRoids.SetMouseoverFrom("ctuf", "targettarget")
    end)

    widget:SetScript("OnLeave", function()
        hookedOnLeave()
        CleveRoids.ClearMouseoverFrom("ctuf")
        CleveRoids.ClearMouseoverFrom("native")
    end)
end

function Extension.OnLoad()
    if arg1 ~= "CT_UnitFrames" or not CT_AssistFrame then
        return
    end
    CleveRoids.Print("CT_UnitFrames module loaded.")

    Extension.SetHook(CT_AssistFrame)
    Extension.SetHook(CT_AssistFrameHealthBar)
    Extension.SetHook(CT_AssistFrameManaBar)
    Extension.SetHook(CT_AssistFrame_Drag)
end

_G["CleveRoids"] = CleveRoids