local T = ShaguTweaks.T

local module = ShaguTweaks:register({
    title = T["Improved Roll Frames"],
    description = T["Smaller roll frames with roll tracking."],
    expansions = { ["vanilla"] = true },
    category = T["Loot"],
    enabled = nil,
  })

module.enable = function(self)
    local _G = ShaguTweaks.GetGlobalEnv()
    local font_default, font_size = "Fonts\\skurri.TTF", 15
    local ROLL_FRAME_COUNT = 4

    ShaguTweaks.roll = CreateFrame("Frame", "STLootRoll", UIParent)
    ShaguTweaks.roll.frames = {}

    -- squash vanilla item placeholders
    local LOOT_ROLL_GREED = string.gsub(LOOT_ROLL_GREED, "%%s|Hitem:%%d:%%d:%%d:%%d|h%[%%s%]|h%%s", "%%s")
    local LOOT_ROLL_NEED = string.gsub(LOOT_ROLL_NEED, "%%s|Hitem:%%d:%%d:%%d:%%d|h%[%%s%]|h%%s", "%%s")
    local LOOT_ROLL_PASSED = string.gsub(LOOT_ROLL_PASSED, "%%s|Hitem:%%d:%%d:%%d:%%d|h%[%%s%]|h%%s", "%%s")

    -- try to detect the everyone string
    local _, _, everyone, _ = strfind(LOOT_ROLL_ALL_PASSED, LOOT_ROLL_PASSED)
    ShaguTweaks.roll.blacklist = {}
    if YOU then ShaguTweaks.roll.blacklist[YOU] = true end
    if everyone then ShaguTweaks.roll.blacklist[everyone] = true end

    ShaguTweaks.roll.cache = {}

    ShaguTweaks.roll.scan = CreateFrame("Frame", "STLootRollMonitor", UIParent)
    ShaguTweaks.roll.scan:RegisterEvent("CHAT_MSG_LOOT")
    ShaguTweaks.roll.scan:SetScript("OnEvent", function()
    local player, item = ShaguTweaks.cmatch(arg1, LOOT_ROLL_GREED)
    if player and item then
        ShaguTweaks.roll:AddCache(item, player, "GREED")
        return
    end

    local player, item = ShaguTweaks.cmatch(arg1, LOOT_ROLL_NEED)
    if player and item then
        ShaguTweaks.roll:AddCache(item, player, "NEED")
        return
    end

    local player, item = ShaguTweaks.cmatch(arg1, LOOT_ROLL_PASSED)
    if player and item then
        ShaguTweaks.roll:AddCache(item, player, "PASS")
        return
    end
    end)

    function ShaguTweaks.roll:AddCache(hyperlink, name, roll)
    -- skip invalid names
    if not hyperlink or not name then return end
    if ShaguTweaks.roll.blacklist[name] then return end

    local _, _, itemLink = string.find(hyperlink, "(item:%d+:%d+:%d+:%d+)")
    if not itemLink then return end
    local itemName = GetItemInfo(itemLink)
    if not itemName then return end

    local now = GetTime()
    local itemCache = ShaguTweaks.roll.cache[itemName]

    -- delete obsolete tables
    if itemCache and itemCache["TIMESTAMP"] < now - 60 then
        ShaguTweaks.roll.cache[itemName] = nil
        itemCache = nil
    end

    -- initialize itemtable
    if not itemCache then
        itemCache = { ["GREED"] = {}, ["NEED"] = {}, ["PASS"] = {}, ["TIMESTAMP"] = now }
        ShaguTweaks.roll.cache[itemName] = itemCache
    end

    -- ignore already listed names
    for _, existing in pairs(itemCache[roll]) do
        if name == existing then return end
    end

    table.insert(itemCache[roll], name)

    local count_greed, count_need, count_pass

    for id=1,ROLL_FRAME_COUNT do
        if ShaguTweaks.roll.frames[id]:IsVisible() and ShaguTweaks.roll.frames[id].itemname == itemName then
        if not count_greed then
            count_greed = table.getn(itemCache["GREED"])
            count_need  = table.getn(itemCache["NEED"])
            count_pass  = table.getn(itemCache["PASS"])
        end
        ShaguTweaks.roll.frames[id].greed.count:SetText(count_greed > 0 and count_greed or "")
        ShaguTweaks.roll.frames[id].need.count:SetText(count_need > 0 and count_need or "")
        ShaguTweaks.roll.frames[id].pass.count:SetText(count_pass > 0 and count_pass or "")
        end
    end
    end

    function ShaguTweaks.roll:CreateLootRoll(id)
    local size = 22
    -- local rawborder, border = GetBorderSize()
    local border = 4
    local esize = 22
    local f = CreateFrame("Frame", "STLootRollFrame" .. id, UIParent)

    local function CreateBackdrop(f,b,a)
        if not f then return end
        f.backdrop = CreateFrame("Frame", nil, f)
        f.backdrop:SetPoint("TOPLEFT", f, "TOPLEFT", -b, b)
        f.backdrop:SetPoint("BOTTOMRIGHT", f, "BOTTOMRIGHT", b, -b)
        f.backdrop:SetBackdrop({
            bgFile = "Interface/Tooltips/UI-Tooltip-Background",
            edgeFile = "Interface/Tooltips/UI-Tooltip-Border",
            edgeSize = 12,
            insets = { left = 4, right = 4, top = 4, bottom = 4 },
        })

        f.backdrop:SetBackdropColor(0, 0, 0, a)
        f.backdrop:SetBackdropBorderColor(1, 1, 1, a)
    end

    CreateBackdrop(f, border, .1)
    -- CreateBackdrop(f, nil, nil, .8)
    -- CreateBackdropShadow(f)
    f.backdrop:SetFrameStrata("BACKGROUND")
    f.hasItem = 1

    f:SetWidth(385)
    f:SetHeight(size)

    f.icon = CreateFrame("Button", "STLootRollFrame" .. id .. "Icon", f)
    CreateBackdrop(f.icon, border, .1)
    f.icon:SetPoint("LEFT", f, "LEFT", -30, 0)
    f.icon:SetWidth(esize*1.2)
    f.icon:SetHeight(esize*1.2)

    f.icon.tex = f.icon:CreateTexture("OVERLAY")
    f.icon.tex:SetTexCoord(.08, .92, .08, .92)
    f.icon.tex:SetAllPoints(f.icon)

    f.icon:SetScript("OnEnter", function()
        GameTooltip:SetOwner(this, "ANCHOR_RIGHT")
        GameTooltip:SetLootRollItem(this:GetParent().rollID)
        CursorUpdate()
    end)

    f.icon:SetScript("OnLeave", function()
        GameTooltip:Hide()
    end)

    f.icon:SetScript("OnClick", function()
        if IsControlKeyDown() then
        DressUpItemLink(GetLootRollItemLink(this:GetParent().rollID))
        elseif IsShiftKeyDown() then
        if ChatEdit_InsertLink then
            ChatEdit_InsertLink(GetLootRollItemLink(this:GetParent().rollID))
        elseif ChatFrameEditBox:IsVisible() then
            ChatFrameEditBox:Insert(GetLootRollItemLink(this:GetParent().rollID))
        end
        end
    end)

    f.need = CreateFrame("Button", "STLootRollFrame" .. id .. "Need", f)
    f.need:SetPoint("LEFT", f.icon, "RIGHT", border*3, -1)
    f.need:SetWidth(esize)
    f.need:SetHeight(esize)
    f.need:SetNormalTexture("Interface\\Buttons\\UI-GroupLoot-Dice-Up")
    f.need:SetHighlightTexture("Interface\\Buttons\\UI-GroupLoot-Dice-Highlight")

    f.need.count = f.need:CreateFontString("NEED")
    f.need.count:SetPoint("CENTER", f.need, "CENTER", 0, 0)
    f.need.count:SetJustifyH("CENTER")
    f.need.count:SetFont(font_default, font_size, "OUTLINE")

    f.need:SetScript("OnClick", function()
        RollOnLoot(this:GetParent().rollID, 1)
    end)
    f.need:SetScript("OnEnter", function()
        GameTooltip:SetOwner(this, "ANCHOR_RIGHT")
        GameTooltip:SetText("|cff33ffcc" .. NEED)
        if f.itemname and ShaguTweaks.roll.cache[f.itemname] then
        for _, player in pairs(ShaguTweaks.roll.cache[f.itemname]["NEED"]) do
            GameTooltip:AddLine(player)
        end
        end
        GameTooltip:Show()
    end)
    f.need:SetScript("OnLeave", function()
        GameTooltip:Hide()
    end)

    f.greed = CreateFrame("Button", "STLootRollFrame" .. id .. "Greed", f)
    f.greed:SetPoint("LEFT", f.icon, "RIGHT", border*5+esize, -2)
    f.greed:SetWidth(esize)
    f.greed:SetHeight(esize)
    f.greed:SetNormalTexture("Interface\\Buttons\\UI-GroupLoot-Coin-Up")
    f.greed:SetHighlightTexture("Interface\\Buttons\\UI-GroupLoot-Coin-Highlight")

    f.greed.count = f.greed:CreateFontString("GREED")
    f.greed.count:SetPoint("CENTER", f.greed, "CENTER", 0, 1)
    f.greed.count:SetJustifyH("CENTER")
    f.greed.count:SetFont(font_default, font_size, "OUTLINE")

    f.greed:SetScript("OnClick", function()
        RollOnLoot(this:GetParent().rollID, 2)
    end)
    f.greed:SetScript("OnEnter", function()
        GameTooltip:SetOwner(this, "ANCHOR_RIGHT")
        GameTooltip:SetText("|cff33ffcc" .. GREED)
        if f.itemname and ShaguTweaks.roll.cache[f.itemname] then
        for _, player in pairs(ShaguTweaks.roll.cache[f.itemname]["GREED"]) do
            GameTooltip:AddLine(player)
        end
        end
        GameTooltip:Show()
    end)
    f.greed:SetScript("OnLeave", function()
        GameTooltip:Hide()
    end)

    f.pass = CreateFrame("Button", "STLootRollFrame" .. id .. "Pass", f)
    f.pass:SetPoint("LEFT", f.icon, "RIGHT", border*7+esize*2, 0)
    f.pass:SetWidth(esize)
    f.pass:SetHeight(esize)
    f.pass:SetNormalTexture("Interface\\Buttons\\UI-GroupLoot-Pass-Up")
    f.pass:SetHighlightTexture("Interface\\Buttons\\UI-GroupLoot-Coin-Highlight")

    f.pass.count = f.pass:CreateFontString("PASS")
    f.pass.count:SetPoint("CENTER", f.pass, "CENTER", 0, -1)
    f.pass.count:SetJustifyH("CENTER")
    f.pass.count:SetFont(font_default, font_size, "OUTLINE")

    f.pass:SetScript("OnClick", function()
        RollOnLoot(this:GetParent().rollID, 0)
    end)
    f.pass:SetScript("OnEnter", function()
        GameTooltip:SetOwner(this, "ANCHOR_RIGHT")
        GameTooltip:SetText("|cff33ffcc" .. PASS)
        if f.itemname and ShaguTweaks.roll.cache[f.itemname] then
        for _, player in pairs(ShaguTweaks.roll.cache[f.itemname]["PASS"]) do
            GameTooltip:AddLine(player)
        end
        end
        GameTooltip:Show()
    end)
    f.pass:SetScript("OnLeave", function()
        GameTooltip:Hide()
    end)

    f.boe = CreateFrame("Frame", "STLootRollFrame" .. id .. "BOE", f)
    f.boe:SetPoint("LEFT", f.icon, "RIGHT", border*9+esize*3, 0)
    f.boe:SetWidth(esize*2)
    f.boe:SetHeight(esize)
    f.boe.text = f.boe:CreateFontString("BOE")
    f.boe.text:SetAllPoints(f.boe)
    f.boe.text:SetJustifyH("LEFT")
    f.boe.text:SetFont(font_default, font_size, "OUTLINE")

    f.name = CreateFrame("Frame", "STLootRollFrame" .. id .. "Name", f)
    f.name:SetPoint("LEFT", f.icon, "RIGHT", border*11+esize*4, 0)
    f.name:SetPoint("RIGHT", f, "RIGHT", border*2, 0)
    f.name:SetHeight(esize)
    f.name.text = f.name:CreateFontString("NAME")
    f.name.text:SetAllPoints(f.name)
    f.name.text:SetJustifyH("LEFT")
    f.name.text:SetFont(font_default, font_size, "OUTLINE")

    f.time = CreateFrame("Frame", "STLootRollFrame" .. id .. "Time", f)
    f.time:SetPoint("TOPLEFT", f, "TOPLEFT", 0, 0)
    f.time:SetPoint("BOTTOMRIGHT", f, "BOTTOMRIGHT", 0, 0)
    f.time:SetFrameStrata("LOW")
    f.time.bar = CreateFrame("StatusBar", "STLootRollFrame" .. id .. "TimeBar", f.time)
    f.time.bar:SetAllPoints(f.time)
    -- f.time.bar:SetStatusBarTexture(pfUI.media["img:bar"])
    f.time.bar:SetStatusBarTexture("Interface\\TargetingFrame\\UI-StatusBar")
    f.time.bar:SetMinMaxValues(0, 100)
    -- local r, g, b, a = strsplit(",", C.appearance.border.color)
    local r, g, b, a = 255/255, 210/255, 0/255, 1
    -- local r, g, b, a = 1, 1, 1, 1
    f.time.bar:SetStatusBarColor(r, g, b)
    f.time.bar:SetValue(20)
    f.time.bar.rollFrame = f
    f.time.bar:SetScript("OnUpdate", function()
        local frame = this.rollFrame
        if not frame.rollID then return end
        local left = GetLootRollTimeLeft(frame.rollID)
        if left < 0 or left > frame.rollTime then left = 0 end
        this:SetValue(left)
    end)

    return f
    end

    ShaguTweaks.roll:RegisterEvent("CANCEL_LOOT_ROLL")
    ShaguTweaks.roll:SetScript("OnEvent", function()
    for i=1,ROLL_FRAME_COUNT do
        if ShaguTweaks.roll.frames[i].rollID == arg1 then
        ShaguTweaks.roll.frames[i]:Hide()
        ShaguTweaks.roll.frames[i].rollID = nil
        return
        end
    end
    end)

    function _G.GroupLootFrame_OpenNewFrame(id, rollTime)
    local visible, available
    for i=1,ROLL_FRAME_COUNT do
        local isVisible = ShaguTweaks.roll.frames[i]:IsVisible()
        visible = visible or isVisible
        available = available or not isVisible and i
    end

    -- clear cache if possible
    if not visible then ShaguTweaks.roll.cache = {} end

    -- setup the first available roll frame
    if available then
        ShaguTweaks.roll.frames[available].rollID = id
        ShaguTweaks.roll.frames[available].rollTime = rollTime
        ShaguTweaks.roll:UpdateLootRoll(available)
    end
    end

    function ShaguTweaks.roll:UpdateLootRoll(id)
    local frame = ShaguTweaks.roll.frames[id]
    local texture, name, count, quality, bop = GetLootRollItemInfo(frame.rollID)
    local color = ITEM_QUALITY_COLORS[quality]

    frame.itemname = name

    local itemCache = ShaguTweaks.roll.cache[name]
    local count_greed = itemCache and table.getn(itemCache["GREED"]) or 0
    local count_need  = itemCache and table.getn(itemCache["NEED"]) or 0
    local count_pass  = itemCache and table.getn(itemCache["PASS"]) or 0

    frame.greed.count:SetText(count_greed > 0 and count_greed or "")
    frame.need.count:SetText(count_need > 0 and count_need or "")
    frame.pass.count:SetText(count_pass > 0 and count_pass or "")

    frame.name.text:SetText(name)
    frame.name.text:SetTextColor(color.r, color.g, color.b, 1)
    frame.icon.tex:SetTexture(texture)
    frame.backdrop:SetBackdropBorderColor(color.r, color.g, color.b)
    frame.time.bar:SetMinMaxValues(0, frame.rollTime)

    -- if C.loot.raritytimer == "1" then
        frame.time.bar:SetStatusBarColor(color.r, color.g, color.b, .5)
    -- end

    if bop then
        -- frame.boe.text:SetText(T["BoP"])
        frame.boe.text:SetText("BoP")
        frame.boe.text:SetTextColor(1,.3,.3,1)
    else
        -- frame.boe.text:SetText(T["BoE"])
        frame.boe.text:SetText("BoE")
        frame.boe.text:SetTextColor(.3,1,.3,1)
    end

    frame:Show()
    end

    for i=1,ROLL_FRAME_COUNT do
    if not ShaguTweaks.roll.frames[i] then
        ShaguTweaks.roll.frames[i] = ShaguTweaks.roll:CreateLootRoll(i)
        -- ShaguTweaks.roll.frames[i]:SetPoint("CENTER", 0, -i*35)
        ShaguTweaks.roll.frames[i]:SetPoint("CENTER", 15, i*35)
        -- UpdateMovable(ShaguTweaks.roll.frames[i])
        ShaguTweaks.roll.frames[i]:Hide()
    end
    end
end