-- InspectEquip

InspectEquip = LibStub("AceAddon-3.0"):NewAddon("InspectEquip", "AceConsole-3.0", "AceHook-3.0", "AceTimer-3.0", "AceEvent-3.0")
local L = LibStub("AceLocale-3.0"):GetLocale("InspectEquip")
local IE = InspectEquip
local IS = InspectEquip_ItemSources --> ItemSources.lua
local WIN = InspectEquip_InfoWindow --> InfoWindow.xml
local TITLE = InspectEquip_InfoWindowTitle
local AVGIL = InspectEquip_InfoWindowAvgItemLevel
local exMod = nil

local ReforgingInfo = LibStub("LibReforgingInfo-1.0")

local slots = { "HeadSlot", "NeckSlot", "ShoulderSlot", "BackSlot", "ChestSlot",
                "WristSlot", "HandsSlot", "WaistSlot", "LegsSlot", "FeetSlot",
                "Finger0Slot", "Finger1Slot", "Trinket0Slot", "Trinket1Slot", "MainHandSlot",
                "SecondaryHandSlot", "RangedSlot" } -- TabardSlot, ShirtSlot
local noEnchantWarningSlots = {
    ["NeckSlot"] = true, ["WaistSlot"] = true, ["Finger0Slot"] = true, ["Finger1Slot"] = true,
    ["Trinket0Slot"] = true, ["Trinket1Slot"] = true, ["SecondaryHandSlot"] = true, ["RangedSlot"] = true,
}

local lines = {}
local numlines = 0
local curline = 0
local curUnit = nil
local curUnitName = nil
local curUser = nil
local curGUID = nil
local cached = false

local headers = {}
local numheaders = 0

local yoffset = -40
local hooked = false
local autoHidden = false

local origInspectUnit

local tonumber = tonumber
local gmatch = string.gmatch
local tinsert = table.insert
local tsort = table.sort
local band = bit.band
local Examiner = Examiner

--local _,_,_,gameToc = GetBuildInfo()

local tooltipTimer = nil

local dummy_tooltip = CreateFrame("GameTooltip", "dummy_tooltip", UIParent, "GameTooltipTemplate")
dummy_tooltip:SetOwner(WorldFrame, "ANCHOR_NONE")


--------------------------------------------------------------------------------------

InspectEquipConfig = {}
local defaults = {
    tooltips = true,
    showUnknown = true,
    inspectWindow = true,
    charWindow = true,
    checkEnchants = true,
    listItemLevels = true,
    showAvgItemLevel = true,
    ttR = 1.0,
    ttG = 0.75,
    ttB = 0.0,
}

local options = {
    name = "InspectEquip",
    type = "group",
    args = {
        tooltips = {
            order = 1, type = "toggle", width = "full",
            name = L["Add drop information to tooltips"],
            desc = L["Add item drop information to all item tooltips"],
            get = function() return InspectEquipConfig.tooltips end,
            set = function(_,v) InspectEquipConfig.tooltips = v; if v then IE:HookTooltips() end end,
        },
        showunknown = {
            order = 2, type = "toggle", width = "full",
            name = L["Include unknown items in overview"],
            desc = L["Show items that cannot be categorized in a seperate category"],
            get = function() return InspectEquipConfig.showUnknown end,
            set = function(_,v) InspectEquipConfig.showUnknown = v end,
        },
        inspectwindow = {
            order = 3, type = "toggle", width = "full",
            name = L["Attach to inspect window"],
            desc = L["Show the equipment list when inspecting other characters"],
            get = function() return InspectEquipConfig.inspectWindow end,
            set = function(_,v) InspectEquipConfig.inspectWindow = v end,
        },
        charwindow = {
            order = 4, type = "toggle", width = "full",
            name = L["Attach to character window"],
            desc = L["Also show the InspectEquip panel when opening the character window"],
            get = function() return InspectEquipConfig.charWindow end,
            set = function(_,v) InspectEquipConfig.charWindow = v end,
        },
        checkenchants = {
            order = 5, type = "toggle", width = "full",
            name = L["Check for unenchanted items"],
            desc = L["Display a warning for unenchanted items"],
            get = function() return InspectEquipConfig.checkEnchants end,
            set = function(_,v) InspectEquipConfig.checkEnchants = v end,
        },
        listitemlevels = {
            order = 6, type = "toggle", width = "full",
            name = L["Show item level in equipment list"],
            desc = L["Show the item level of each item in the equipment panel"],
            get = function() return InspectEquipConfig.listItemLevels end,
            set = function(_,v) InspectEquipConfig.listItemLevels = v end,
        },
        showavgitemlevel = {
            order = 7, type = "toggle", width = "full",
            name = L["Show average item level in equipment list"],
            desc = L["Show the average item level of all items in the equipment panel"],
            get = function() return InspectEquipConfig.showAvgItemLevel end,
            set = function(_,v) InspectEquipConfig.showAvgItemLevel = v end,
        },
        tooltipcolor = {
            order = 7, type = "color",
            name = L["Tooltip text color"],
            get = function() return InspectEquipConfig.ttR, InspectEquipConfig.ttG, InspectEquipConfig.ttB, 1.0 end,
            set = function(_,r,g,b,a)
                InspectEquipConfig.ttR = r
                InspectEquipConfig.ttG = g
                InspectEquipConfig.ttB = b
            end,
        }
    },
}

LibStub("AceConfig-3.0"):RegisterOptionsTable("InspectEquip", options)
LibStub("AceConfigDialog-3.0"):AddToBlizOptions("InspectEquip")

--------------------------------------------------------------------------------------

function IE:OnInitialize()
    setmetatable(InspectEquipConfig, {__index = defaults})

    self:SetParent(Examiner or InspectFrame)
    WIN:Hide()
    TITLE:SetText("InspectEquip")

    if Examiner and Examiner.CreateModule then
        exMod = Examiner:CreateModule("InspectEquip")
        exMod.OnCacheLoaded = function(s, entry, unit)
            if InspectEquipConfig.inspectWindow then
                IE:Inspect("cache", entry)
            end
        end
        exMod.OnClearInspect = function(s) WIN:Hide() end
        exMod.OnInspect = function(s, unit)
            if InspectEquipConfig.inspectWindow then
                IE:SetParent(Examiner); IE:Inspect(unit)
            end
        end
    end

    self:RegisterEvent("PLAYER_ENTERING_WORLD")
    self:RegisterEvent("ADDON_LOADED")
end

function IE:OnEnable()
    origInspectUnit = origInspectUnit or InspectUnit
    InspectUnit = function(...) IE:InspectUnit(...) end
    self:SecureHookScript(PaperDollFrame, "OnShow", "PaperDollFrame_OnShow")
    self:SecureHookScript(PaperDollFrame, "OnHide", "PaperDollFrame_OnHide")
    if GearManagerDialog then -- 4.0
        self:SecureHookScript(GearManagerDialog, "OnShow", "GearManagerDialog_OnShow")
        self:SecureHookScript(GearManagerDialog, "OnHide", "GearManagerDialog_OnHide")
    end
    if OutfitterFrame then
        self:SecureHookScript(OutfitterFrame, "OnShow", "GearManagerDialog_OnShow")
        self:SecureHookScript(OutfitterFrame, "OnHide", "GearManagerDialog_OnHide")
    end
    self:RegisterEvent("UNIT_INVENTORY_CHANGED")
    self:RegisterEvent("INSPECT_READY")
end

function IE:OnDisable()
    InspectUnit = origInspectUnit
    if hooked then
        hooked = false
        self:Unhook("InspectFrame_UnitChanged")
    end
    self:UnhookAll()
    self:UnregisterEvent("UNIT_INVENTORY_CHANGED")
    self:UnregisterEvent("INSPECT_READY")
    self:CancelAllTimers()
    WIN:Hide()
end

local entered = false

function IE:PLAYER_ENTERING_WORLD()
    entered = true
    self:ScheduleTooltipHook()
    self:UnregisterEvent("PLAYER_ENTERING_WORLD")
end

function IE:ADDON_LOADED(e, name)
    if entered then
        self:ScheduleTooltipHook()
    end
end

-- Ugly hack, but some addons override the OnTooltipSetItem handler on
-- ItemRefTooltip, breaking IE. Using this timer, IE hopefully hooks after them.
function IE:ScheduleTooltipHook()
    if InspectEquipConfig.tooltips then
        if tooltipTimer then
            self:CancelTimer(tooltipTimer, true)
        end
        tooltipTimer = self:ScheduleTimer('HookTooltips', 3)
    end
end

function IE:SetParent(frame)
    WIN:SetParent(frame)
    WIN:ClearAllPoints()
    if not (frame == Examiner) then
        WIN:SetPoint("TOPLEFT", frame, "TOPRIGHT", 5, 0)
    else
        WIN:SetPoint("TOPLEFT", frame, "TOPRIGHT", -25, -13)
    end
end

function IE:NewLine()
    local row = CreateFrame("Frame", nil, WIN)
    row:SetHeight(12)
    row:SetWidth(200)
    row:SetPoint("TOPLEFT", WIN, "TOPLEFT", 15, yoffset)

    local txt = row:CreateFontString(nil, "ARTWORK")
    txt:SetJustifyH("LEFT")
    txt:SetFontObject(GameFontHighlightSmall)
    txt:SetPoint("TOPLEFT", row, "TOPLEFT", 0, 0)

    row.text = txt
    yoffset = yoffset - 15
    numlines = numlines + 1
    lines[numlines] = row

    row:EnableMouse(true)
    row:SetScript("OnEnter", IE.Line_OnEnter)
    row:SetScript("OnLeave", IE.Line_OnLeave)
    row:SetScript("OnMouseDown", IE.Line_OnClick)
end

function IE:ResetDisplay()
    for i = 1, numlines do
        lines[i].text:SetText("")
        lines[i]:Hide()
    end
    curline = 0
end

function IE:AddLine(text, link, item)
    curline = curline + 1
    if curline > numlines then
        self:NewLine()
    end
    local line = lines[curline]
    line.link = link
    line.item = item
    line.text:SetText(text)
    line:SetWidth(line.text:GetStringWidth())
    line:SetFrameLevel(WIN:GetFrameLevel() + 1)
    line:Show()
end

function IE:FullUnitName(name, realm)
    if realm and realm ~= "" then
        return name .. "-" .. realm
    else
        return name
    end
end

function IE:GetExaminerCache(unit)
    local name, realm = UnitName(unit)
    return Examiner_Cache and Examiner_Cache[self:FullUnitName(name, realm)]
end

function IE:InspectUnit(unit, ...)
    origInspectUnit(unit, ...)

    if InspectEquipConfig.inspectWindow then
        self:SetParent(Examiner or InspectFrame)
        WIN:Hide()
        if not hooked and InspectFrame_UnitChanged then
            hooked = true
            self:SecureHook("InspectFrame_UnitChanged")
        end

        self:Inspect(unit)
    end
end

function IE:InspectFrame_UnitChanged()
    if InspectFrame.unit and InspectEquipConfig.inspectWindow then
        self:InspectUnit(InspectFrame.unit)
    else
        WIN:Hide()
    end
end

function IE:PaperDollFrame_OnShow()
    if InspectEquipConfig.charWindow then
        IE:SetParent(CharacterFrame)
        IE:Inspect("player")
    end
end

function IE:PaperDollFrame_OnHide()
    if WIN:GetParent() == CharacterFrame then
        WIN:Hide()
        autoHidden = false
    end
end

function IE:GearManagerDialog_OnShow()
    if WIN:GetParent() == CharacterFrame and WIN:IsShown() then
        WIN:Hide()
        autoHidden = true
    end
end

function IE:GearManagerDialog_OnHide()
    if autoHidden and WIN:GetParent() == CharacterFrame then
        WIN:Show()
        autoHidden = false
    end
end

function IE:UNIT_INVENTORY_CHANGED(event, unit)
    if (unit == "player") and (WIN:IsVisible() or autoHidden) and (WIN:GetParent() == CharacterFrame) then
        IE:Inspect("player")
    elseif (unit == curUnit) and (UnitName(unit) == curUnitName) and (WIN:IsVisible()) then
        IE:Inspect(curUnit)
    end
end

function IE:INSPECT_READY(event, guid)
    if (guid == curGUID) and (UnitName(curUnit) == curUnitName) and (WIN:IsVisible()) then
        IE:Inspect(curUnit)
    else
        -- probably mouseover changed... nothing we can do apparently :(
    -- (inspect by unit name / guid does not work)
    end
end

function IE:Inspect(unit, entry)
    local unitName, unitRealm
    cached = (unit == "cache")

    if (cached and (not entry)) or (not self:IsEnabled()) then
        WIN:Hide()
        return
    end

    local cacheItems = cached and entry.Items or nil

    if cached then
        unitName, unitRealm = entry.name, entry.realm
        curGUID = nil
    else
        if (not unit or not UnitExists(unit)) then
            unit = "player"
        end
        unitName, unitRealm = UnitName(unit)
        curGUID = UnitGUID(unit)

        if not CanInspect(unit) then
            entry = self:GetExaminerCache(unit)
            if entry then
                cached = true
                cacheItems = entry.Items
            end
        else
            --ClearInspectPlayer()
            --NotifyInspect(unit)
        end
    end
    if unitRealm == "" then unitRealm = nil end
    curUnit = unit
    curUnitName = unitName
    curUser = self:FullUnitName(unitName, unitRealm)
    TITLE:SetText("InspectEquip: " .. curUser .. (cached and " (Cache)" or ""))

    self:ResetDisplay()

    local items = { cats = {}, items = {} }
    local itemsFound = false
    local getItem
    if cached then
        getItem = function(slot)
            local istr = cacheItems[slot]
            if istr then
                local itemId = tonumber(istr:match("item:(%d+)"))
                return select(2, GetItemInfo(istr)) or ("[" .. itemId .. "]")
            else
                return nil
            end
        end
    else
        getItem = function(slot) return GetInventoryItemLink(unit, GetInventorySlotInfo(slot)) end
    end

    local calciv = InspectEquipConfig.showAvgItemLevel
    local iLevelSum, iCount = 0,0

    for _,slot in pairs(slots) do
        local itemLink = getItem(slot)
        if itemLink then
            -- get source
            local source = self:GetItemSourceCategories(itemLink)
            if (not source) and InspectEquipConfig.showUnknown then
                local _,_,rar = GetItemInfo(itemLink)
                if rar >= 2 then
                    source = {L["Unknown"]}
                end
            end

            if source then
                local enchantId = tonumber(itemLink:match("Hitem:%d+:(%d+):"))
                local reforgeId = ReforgingInfo:GetReforgeID(itemLink)
                local reforgeText = nil
                if reforgeId then
                    local refid1,refid2 = ReforgingInfo:GetReforgedStatShortNames(reforgeId)
                    reforgeText = " |cff00cc00[" .. string.sub(refid1, 1, 2) .. ">" .. string.sub(refid2, 1, 2) .. "]|r"
                end
                itemsFound = true

                -- find category
                local cat = items
                local entry
                for _, entry in pairs(source) do
                    if cat.cats[entry] == nil then
                        cat.cats[entry] = { count = 0, cats = {}, items = {} }
                    end
                    cat = cat.cats[entry]
                    cat.count = cat.count + 1
                end

                -- add item to category
                cat.hasItems = true
                cat.items[cat.count] = {link = itemLink, enchant = enchantId, reforged = reforgeText, slot = slot}
            end

            -- calculate avg ilvl
            if calciv then
                local _,_,rar,lvl = GetItemInfo(itemLink)
                if lvl then
                    iLevelSum = iLevelSum + lvl
                    iCount = iCount + 1
                end
            end

        end
    end

    if itemsFound then
        self:AddCategory(items, "")
        if calciv and iCount > 0 then
            local avgLvl = iLevelSum / iCount
            AVGIL:SetText(L["Avg. Item Level"] .. ": " .. string.format("%.2f", avgLvl))
            AVGIL:Show()
        else
            AVGIL:Hide()
        end
        self:FixWindowSize()
        if WIN:GetParent() == CharacterFrame and ((GearManagerDialog and GearManagerDialog:IsVisible()) or (OutfitterFrame and OutfitterFrame:IsVisible())) then
            autoHidden = true
        else
            WIN:Show()
        end
    else
        WIN:Hide()
    end
end

function IE:AddCategory(cat, prefix)
    -- add items
    if #cat.items > 0 then
        self:AddItems(cat.items, prefix .. "  ")
    end

    -- sort subcategories by item count
    local t = {}
    for name, subcat in pairs(cat.cats) do
        tinsert(t, {name = name, subcat = subcat})
    end
    tsort(t, function(a,b) return a.subcat.count > b.subcat.count end)

    -- add subcategories
    for i = 1, #t do
        local name = t[i].name
        local subcat = t[i].subcat
        self:AddLine(prefix .. name .. " (" .. subcat.count .. ")")
        self:AddCategory(subcat, prefix .. "  ")
    end
end

function IE:AddItems(tab, padding)
    for _, item in pairs(tab) do
        local suffix = ""
        local prefix = padding
        if InspectEquipConfig.listItemLevels then
            local _,_,_,ilvl = GetItemInfo(item.link)
            if ilvl then
                prefix = padding .. "|cffaaaaaa[" .. ilvl .. "]|r "
            end
        end
        if InspectEquipConfig.checkEnchants and (item.enchant == 0) and (not noEnchantWarningSlots[item.slot]) then
            suffix = "|cffff0000*|r"
        end
        -- set tooltip used for gem+bonus scanning
        dummy_tooltip:ClearLines()
        dummy_tooltip:SetHyperlink(item.link)
        -- if it does have a socket bonus, check for empty gems
        for i = 1, 4 do
            local gemtexture = _G["dummy_tooltipTexture"..i]
            -- if the gem is empty, add to label text
            if gemtexture and gemtexture:IsVisible() and string.find(gemtexture:GetTexture(), "Empty") then
                suffix = suffix .. " |cff00ff00*|r"
                break
            end
        end
        -- and iterate over the lines
        for i = 1, dummy_tooltip:NumLines() do
            -- check the text for socket bonus
            local line = _G["dummy_tooltipTextLeft" .. i]
            if line and line:GetText() and string.find(line:GetText(), "Socket Bonus: ") then -- this won't be localised!
                -- if the socket bonus isn't the right colour (it not gemmed for socket bonus) add text
                if line:GetTextColor() ~= 0 then
                    suffix = suffix .. " |cffffa500*|r"
                    break
                end
            end
        end
    if item.reforged then
        suffix = suffix .. item.reforged
    end
        self:AddLine(prefix .. item.link .. suffix, item.link, item)
    end
end

function IE:GetItemData(item)
    local id
    if type(item) == "number" then
        id = item
    else -- item string/link
        id = tonumber(item:match("item:(%d+)"))
    end

    if id then
        return IS.Items[id]
    else
        return nil
    end
end

function IE:GetItemSourceCategories(itemLink)
    local data = IE:GetItemData(itemLink)
    if data then
        for entry in gmatch(data, "[^;]+") do
            local next_field = gmatch(entry, "[^_]+")
            local cat = next_field()

            if cat == "r" or cat == "d" then
                -- raid/dungeon
                local zone = IS.Zones[tonumber(next_field())]
                local mode = next_field()
                local zoneType
                if cat == "r" then zoneType = L["Raid"] else zoneType = L["Instances"] end
                return {zoneType, zone}
            elseif cat == "v" or cat == "g" then
                -- vendor
                local mainCat
                if cat == "v" then mainCat = L["Vendor"] else mainCat = L["Guild Vendor"] end
                local typ = next_field()
                while typ do
                    if typ == "c" then
                        -- currency
                        local currency = tonumber(next_field())
                        next_field()
                        local curName = GetCurrencyInfo(currency)
                        return {mainCat, curName}
                    elseif typ == "i" then
                        -- item
                        next_field()
                    elseif typ == "m" then
                        -- money
                        next_field()
                    end
                    typ = next_field()
                end
                return {mainCat}
            elseif cat == "f" then
                -- reputation rewards
                return {L["Reputation rewards"]}
            elseif cat == "m" then
                -- darkmoon cards
                return {L["Darkmoon Faire"]}
            elseif cat == "w" then
                -- world drops
                return {L["World drops"]}
            elseif cat == "c" then
                -- crafted
                return {L["Crafted"]}
            end

        end
    end
    return nil
end

function IE:FixWindowSize()
    local maxwidth = TITLE:GetStringWidth()
    for i = 1, numlines do
        local width = lines[i].text:GetStringWidth()
        if maxwidth < width then maxwidth = width end
    end
    local height = (curline * 15) + 55
    if InspectEquipConfig.showAvgItemLevel then
        height = height + 15
    end
    WIN:SetWidth(maxwidth + 40)
    WIN:SetHeight(height)
end

function IE.Line_OnEnter(row)
    if row.link then
        GameTooltip:SetOwner(WIN, "ANCHOR_RIGHT")
        if (not cached) and (UnitName(curUnit) == curUnitName) then
            row.link = GetInventoryItemLink(curUnit, GetInventorySlotInfo(row.item.slot)) or row.link
        end
        GameTooltip:SetHyperlink(row.link)
        if row.item and InspectEquipConfig.checkEnchants and (row.item.enchant == 0) and (not noEnchantWarningSlots[row.item.slot]) then
            GameTooltip:AddLine(" ")
            GameTooltip:AddLine("|cffff0000" .. L["Item is not enchanted"] .. "|r")
        end
        GameTooltip:Show()
    end
end

function IE.Line_OnLeave(row)
    GameTooltip:Hide()
end

function IE.Line_OnClick(row, button)
    if row.link then
        if IsControlKeyDown() then
            DressUpItemLink(row.link)
        elseif IsShiftKeyDown() then
            ChatEdit_InsertLink(row.link)
        end
    end
end