-- FrameInit.lua
local _;
local unpack, pairs, rawget = unpack, pairs, rawget;
local format = string.format;
local UnitHealthMax, UnitPowerMax = UnitHealthMax, UnitPowerMax;
local UnitInRange = UnitInRange;
local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
local PowerTypeMana = Enum.PowerType.Mana;
local PowerTypeRage = Enum.PowerType.Rage;
local PowerTypeFocus = Enum.PowerType.Focus;
local PowerTypeEnergy = Enum.PowerType.Energy;
local PowerTypeRunic = Enum.PowerType.RunicPower;

local unitEvent = nil;

local M = {};
OmaFrames = M;

-- configurable settings
M.Positions = {"TOPRIGHT", "BOTTOMLEFT"};
M.Width, M.Height = 80, 40;
M.AnchorX, M.AnchorY = 0, -330;
M.IndSize = 14; M.BaseColor = {0, 0, 0};
M.BgColor = {0.7, 0.7, 0.7};
M.HealthColor = {0.3, 0.3, 0.3};
M.ShieldColor = {0, 0.7, 1};
M.ShieldhlColor = {0.5, 0.8, 1};
M.HealpredColor = {0.5, 0.6, 0.5};
M.HealabsorbColor = {0.1, 0.1, 0.1};
M.OverlayColorDispel = {1, 0.5, 0, 0.5};
M.OverlayColorCharm = {0.8, 0, 1, 0.5};
M.OverlayColorAlert = {1, 0, 0, 0.5};
M.PowerColors = {
    [PowerTypeMana] = {0, 0.5, 1},
    [PowerTypeRage] = {1, 0, 0},
    [PowerTypeFocus] = {1, 0.5, 0},
    [PowerTypeEnergy] = {1, 0.8, 0},
    [PowerTypeRunic] = {0.9, 0, 0.1},
};
-- watch to not remove mana entry
setmetatable(M.PowerColors, {__index = function(t) return rawget(t, PowerTypeMana) end});

local positions = M.Positions;
local width, height = M.Width, M.Height;
local indSize = M.IndSize;
local baseColor = M.BaseColor;
local bgColor = M.BgColor;
local healthColor = M.HealthColor;
local shieldColor = M.ShieldColor;
local shieldhlColor = M.ShieldhlColor;
local healpredColor = M.HealpredColor;
local healabsorbColor = M.HealabsorbColor;

local function setupIndicators(frame)
    frame.inds = CreateFrame("Frame", nil, frame);
    frame.inds:SetAllPoints();
    frame.inds:Hide();
    for _, pos in pairs(positions) do
        frame.inds[pos] = frame.inds:CreateTexture(nil, "OVERLAY");
        frame.inds[pos]:SetPoint(pos, frame.inds, pos);
        frame.inds[pos]:SetWidth(indSize);
        frame.inds[pos]:SetHeight(indSize);
        frame.inds[pos]:SetTexture("Interface\\AddOns\\OmaRF\\images\\rhomb");
        frame.inds[pos]:SetVertexColor(1, 0, 0);
        frame.inds[pos]:Hide();
        frame.inds[pos].text = frame.inds:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
        frame.inds[pos].text:SetPoint("BOTTOMRIGHT", frame.inds[pos], "BOTTOMRIGHT");
        frame.inds[pos].text:Hide();
    end
    frame.major = CreateFrame("Frame", nil, frame);
    frame.major:SetPoint("TOPLEFT", frame, "TOPLEFT", 4, -indSize + 4);
    frame.major:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT");
    for i = 1,3 do
        frame.major[i] = frame.major:CreateTexture(nil, "OVERLAY");
        if i == 1 then
            frame.major[i]:SetPoint("TOPLEFT", frame.major, "TOPLEFT");
        else
            frame.major[i]:SetPoint("TOPLEFT", frame.major[i-1], "TOPLEFT");
        end
        frame.major[i]:SetWidth(indSize*2);
        frame.major[i]:SetHeight(indSize*2);
        frame.major[i]:Hide();
        frame.major[i].text = frame.major:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
        frame.major[i].text:SetPoint("BOTTOMRIGHT", frame.major[i], "BOTTOMRIGHT");
        frame.major[i].text:Hide();
        frame.major[i].stack = frame.major:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
        frame.major[i].stack:SetPoint("TOPLEFT", frame.major[i], "TOPLEFT");
        frame.major[i].stack:Hide();
    end
end

local function unitUpdate(self, elapsed)
    -- there's no in/out of range event, have to check each frame
    -- from FrameXML/CompactUnitFrame.lua
    local inRange, checked = UnitInRange(self.displayed);
    if checked and not inRange then
        self:SetAlpha(0.55);
    else
        self:SetAlpha(1);
    end
end

function M.RegisterEvents(frame)
    -- events are taken from FrameXML/CompactUnitFrame.lua
    -- TODO ready check support, raid marker support,
    -- player flags support (/afk, /dnd)
    local displayed = frame.unit ~= frame.displayed and frame.displayed or nil;
    frame:RegisterUnitEvent("UNIT_HEALTH", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_HEALTH_FREQUENT", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_MAXHEALTH", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_POWER", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_MAXPOWER", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_DISPLAYPOWER", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_NAME_UPDATE", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_AURA", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_HEAL_PREDICTION", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_ABSORB_AMOUNT_CHANGED", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_HEAL_ABSORB_AMOUNT_CHANGED", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_THREAT_SITUATION_UPDATE", frame.unit, displayed);
    frame:RegisterUnitEvent("UNIT_CONNECTION", frame.unit, displayed);
    frame:RegisterUnitEvent("INCOMING_RESURRECT_CHANGED", frame.unit, displayed);
end
local registerEvents = M.RegisterEvents;

local function frameShow(frame)
    frame:RegisterEvent("PARTY_MEMBER_ENABLE");
    frame:RegisterEvent("PARTY_MEMBER_DISABLE");
    frame:RegisterEvent("UNIT_ENTERED_VEHICLE");
    frame:RegisterEvent("UNIT_EXITED_VEHICLE");
    frame:RegisterEvent("UNIT_PET");
    frame:RegisterEvent("PLAYER_ROLES_ASSIGNED");
    frame:RegisterEvent("GROUP_ROSTER_UPDATE");
    registerEvents(frame);
    frame:SetScript("OnUpdate", unitUpdate);
    unitEvent(frame, "UPDATE_ALL_BARS", frame.displayed);
end

local function frameHide(frame)
    frame:UnregisterAllEvents();
    frame:SetScript("OnUpdate", nil);
end

local function setupFrame(frame, secure, unit)
    -- create visuals
    secure:SetWidth(width+2);
    frame:SetWidth(width+2);
    frame.base = frame:CreateTexture(nil, "BACKGROUND");
    frame.base:SetAllPoints();
    frame.base:SetColorTexture(1, 1, 1);
    frame.base:SetVertexColor(unpack(baseColor));
    frame.background = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
    frame.background:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
    frame.background:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
    frame.background:SetColorTexture(unpack(bgColor));
    frame.health = frame:CreateTexture(nil, "BORDER");
    frame.health:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
    frame.health:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
    frame.health:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT", 0, 2);
    frame.health:SetVertexColor(unpack(healthColor));
    frame.health:SetWidth(width);
    frame.health.max = UnitHealthMax(unit);
    frame.mana = frame:CreateTexture(nil, "BORDER");
    frame.mana:SetPoint("TOPLEFT", frame.background, "BOTTOMLEFT", 0, 2);
    frame.mana:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT");
    frame.mana:SetColorTexture(1, 1, 1);
    frame.mana:SetWidth(width);
    frame.mana.max = UnitPowerMax(unit);
    frame.shield = frame:CreateTexture(nil, "BORDER");
    frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
    frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
    frame.shield:SetColorTexture(unpack(shieldColor));
    frame.shield:Hide();
    frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
    frame.shieldhl:SetPoint("TOPLEFT", frame.background, "TOPRIGHT", -1, 0);
    frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 1, 0);
    frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
    frame.shieldhl:Hide();
    frame.healpred = frame:CreateTexture(nil, "ARTWORK");
    frame.healpred:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
    frame.healpred:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
    frame.healpred:SetColorTexture(unpack(healpredColor));
    frame.healpred:Hide();
    frame.healabsorb = frame:CreateTexture(nil, "ARTWORK");
    frame.healabsorb:SetPoint("TOPRIGHT", frame.health, "TOPRIGHT");
    frame.healabsorb:SetPoint("BOTTOMRIGHT", frame.health, "BOTTOMRIGHT");
    frame.healabsorb:SetColorTexture(unpack(healabsorbColor));
    frame.healabsorb:Hide();
    frame.role = frame:CreateTexture(nil, "ARTWORK");
    frame.role:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", -2, 2);
    frame.role:SetPoint("TOPLEFT", frame.background, "BOTTOMRIGHT", -14, 14);
    frame.role:SetTexCoord(0.75, 1, 0, 1);
    frame.role:SetTexture("Interface\\LFGFRAME\\LFGROLE");
    frame.role:Hide();
    frame.overlay = frame:CreateTexture(nil, "ARTWORK", nil, 1);
    frame.overlay:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
    frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 0, 2);
    frame.overlay:SetColorTexture(1, 1, 1);
    frame.overlay:Hide();
    frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
    frame.name:SetPoint("CENTER", frame.background, "CENTER", 0, 11);
    frame.text = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
    frame.text:SetFont(STANDARD_TEXT_FONT, 13);
    frame.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
    frame.text:Hide();
    frame.rez = frame:CreateTexture(nil, "OVERLAY");
    frame.rez:SetPoint("TOPLEFT", frame.background, "CENTER", -12, 12);
    frame.rez:SetPoint("BOTTOMRIGHT", frame.background, "CENTER", 12, -12);
    frame.rez:SetTexture("Interface\\RaidFrame\\Raid-Icon-Rez");
    frame.rez:Hide();
    setupIndicators(frame);
    frame:SetScript("OnShow", frameShow);
    frame:SetScript("OnHide", frameHide);
    -- set attributes
    secure:RegisterForClicks("AnyDown");
    secure:SetAttribute("type1", "spell"); -- left click
    secure:SetAttribute("type2", "spell"); -- right click
    secure:SetAttribute("shift-type1", "spell"); -- shift left click
    secure:SetAttribute("shift-type2", "spell"); -- shift right click
    secure:SetAttribute("ctrl-type1", "spell"); -- ctrl left click
    secure:SetAttribute("alt-type2", "spell"); -- alt right click
    secure:SetAttribute("alt-shift-type1", "spell"); -- alt+shift left click
    secure:SetAttribute("alt-shift-type2", "spell"); -- alt+shift right click
    secure:SetAttribute("spell1", "Holy Light");
    secure:SetAttribute("spell2", "Bestow Faith");
    secure:SetAttribute("shift-spell1", "Flash of Light");
    secure:SetAttribute("shift-spell2", "Light of the Martyr");
    secure:SetAttribute("ctrl-spell1", "Cleanse");
    secure:SetAttribute("alt-spell2", "Lay on Hands");
    secure:SetAttribute("alt-shift-spell1", "Beacon of Light");
    secure:SetAttribute("alt-shift-spell2", "Beacon of Faith");
    -- rest give target and menu
    secure:SetAttribute("*type1", "target");
    secure:SetAttribute("*type2", "togglemenu");
end

-- vehicle toggle from Shadowed Unit Frames
local vehicletoggle = [=[
    local unit = self:GetAttribute("unit");
    if unit and newstate == "vehicle" and not UnitTargetsVehicleInRaidUI(unit) then
        print(unit, "in vehicle, still target player");
        self:SetAttribute("toggleForVehicle", false);
    else
        self:SetAttribute("toggleForVehicle", true);
    end
]=]

-- TODO reorganizing needed for eventhandler stuff
function M.InitializeParty(parent, party, eventhandler)
    unitEvent = eventhandler;
    local secure = CreateFrame("Button", "OmaPartySecure0", parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");
    local frame = CreateFrame("Frame", "OmaParty0", parent);
    local unit = "player";
    secure:SetAttribute("unit", unit);
    secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
    secure:SetHeight(height+2);
    frame.unit = unit;
    frame.vehicle = "vehicle";
    frame.displayed = unit;
    frame:SetAttribute("unit", unit);
    frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
    frame:SetHeight(height+2);
    frame:SetScript("OnEvent", eventhandler);
    frame:Hide();
    setupFrame(frame, secure, unit);
    RegisterStateDriver(frame, "visibility", "[@player,group:raid] hide; show");
    RegisterStateDriver(secure, "visibility", "[@player,group:raid] hide; show");
    RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
    secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
    party[0] = {secure=secure, frame=frame};
    for i = 1,4 do
        local secure = CreateFrame("Button", "OmaPartySecure"..i, parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");
        local frame = CreateFrame("Frame", "OmaParty"..i, parent);
        local unit = "party"..i;
        secure:SetAttribute("unit", unit);
        secure:SetPoint("TOPLEFT", party[i-1].secure, "TOPRIGHT");
        secure:SetPoint("BOTTOMLEFT", party[i-1].secure, "BOTTOMRIGHT");
        frame.unit = unit;
        frame.vehicle = unit.."pet";
        frame.displayed = unit;
        frame:SetAttribute("unit", unit);
        frame:SetPoint("TOPLEFT", party[i-1].frame, "TOPRIGHT");
        frame:SetPoint("BOTTOMLEFT", party[i-1].frame, "BOTTOMRIGHT");
        frame:SetScript("OnEvent", unitEvent);
        frame:Hide();
        setupFrame(frame, secure, unit);
        local visible = format("[@%s,exists,group:raid] hide; [@%s,exists] show; hide", unit, unit);
        local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
        RegisterStateDriver(frame, "visibility", visible);
        RegisterStateDriver(secure, "visibility", visible);
        RegisterStateDriver(secure, "vehicleui", vehicle);
        secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
        party[i] = {secure=secure, frame=frame};
    end
end

function M.InitializeRaid(parent, raid, eventhandler)
    local secure = CreateFrame("Button", "OmaRaidSecure1", parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");
    local frame = CreateFrame("Frame", "OmaRaid1", parent);
    local unit = "raid1";
    secure:SetAttribute("unit", unit);
    secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
    secure:SetHeight(height+2);
    frame.unit = unit;
    frame.vehicle = unit.."pet";
    frame.displayed = unit;
    frame:SetAttribute("unit", unit);
    frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
    frame:SetHeight(height+2);
    frame:SetScript("OnEvent", eventhandler);
    frame:Hide();
    setupFrame(frame, secure, unit);
    RegisterUnitWatch(frame);
    RegisterUnitWatch(secure);
    local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
    RegisterStateDriver(secure, "vehicleui", vehicle);
    secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
    raid[1] = {secure=secure, frame=frame};
    for y = 1,7 do
        local i = y*5+1;
        local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");
        local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
        local unit = "raid"..i;
        secure:SetAttribute("unit", unit);
        secure:SetPoint("TOPLEFT", raid[i-5].secure, "BOTTOMLEFT");
        secure:SetHeight(height+2);
        frame.unit = unit;
        frame.vehicle = unit.."pet";
        frame.displayed = unit;
        frame:SetAttribute("unit", unit);
        frame:SetPoint("TOPLEFT", raid[i-5].frame, "BOTTOMLEFT");
        frame:SetHeight(height+2);
        frame:SetScript("OnEvent", unitEvent);
        frame:Hide();
        setupFrame(frame, secure, unit);
        RegisterUnitWatch(frame);
        RegisterUnitWatch(secure);
        local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
        RegisterStateDriver(secure, "vehicleui", vehicle);
        secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
        raid[i] = {secure=secure, frame=frame};
    end
    for y = 0,7 do
        for x = 1,4 do
            local i = y*5+x+1;
            local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");
            local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
            local unit = "raid"..i;
            secure:SetAttribute("unit", unit);
            secure:SetPoint("TOPLEFT", raid[i-1].secure, "TOPRIGHT");
            secure:SetPoint("BOTTOMLEFT", raid[i-1].secure, "BOTTOMRIGHT");
            frame.unit = unit;
            frame.vehicle = unit.."pet";
            frame.displayed = unit;
            frame:SetAttribute("unit", unit);
            frame:SetPoint("TOPLEFT", raid[i-1].frame, "TOPRIGHT");
            frame:SetPoint("BOTTOMLEFT", raid[i-1].frame, "BOTTOMRIGHT");
            frame:SetScript("OnEvent", unitEvent);
            frame:Hide();
            setupFrame(frame, secure, unit);
            RegisterUnitWatch(frame);
            RegisterUnitWatch(secure);
            local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
            RegisterStateDriver(secure, "vehicleui", vehicle);
            secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
            raid[i] = {secure=secure, frame=frame};
        end
    end
end