VendorPriceScan.lua

-- Local references to constants
local addonName = VendorPriceScan.ADDON_NAME
local addonPrefix = VendorPriceScan.ADDON_PREFIX
local addonVersion = VendorPriceScan.ADDON_VERSION
local slashCommands = VendorPriceScan.SLASH_COMMANDS
local colors = VendorPriceScan.COLORS
local strings = VendorPriceScan.STRINGS

-- Initialize saved variables (account-wide)
VendorPriceScan_Data = VendorPriceScan_Data or {}

-- Initialize character-specific saved variables
VendorPriceScan_CharData = VendorPriceScan_CharData or {}

-- Create main frame
local frame = CreateFrame("Frame", "VendorPriceScanFrame")

-- Variable to store the last price captured
local capturedPrice = nil

-- Store original SetTooltipMoney function
local OriginalSetTooltipMoney = SetTooltipMoney

-- Hook SetTooltipMoney to capture vendor prices
if OriginalSetTooltipMoney == SetTooltipMoney then
    SetTooltipMoney = function(tooltip, money, type, prefixText, suffixText)
        -- Store the money value before calling original function
        capturedPrice = money
        -- Call original function and return ALL its values
        return OriginalSetTooltipMoney(tooltip, money, type, prefixText, suffixText)
    end
end

-- Helper function to count items
local function CountItems()
    local count = 0
    for _ in pairs(VendorPriceScan_Data) do
        count = count + 1
    end
    return count
end

-- Event handler
frame:SetScript("OnEvent", function()
    if event == "MERCHANT_SHOW" then
        VendorPriceScan_ScanBags()
    elseif event == "ADDON_LOADED" and arg1 == "VendorPriceScan" then
        -- Initialize saved variables if nil
        if not VendorPriceScan_Data then
            VendorPriceScan_Data = {}
        end
        
        -- Initialize character-specific settings
        if not VendorPriceScan_CharData then
            VendorPriceScan_CharData = {}
        end
        if VendorPriceScan_CharData.showTooltips == nil then
            VendorPriceScan_CharData.showTooltips = true
        end
        if VendorPriceScan_CharData.showTooltipTitle == nil then
            VendorPriceScan_CharData.showTooltipTitle = true
        end
        
        -- Count items
        local count = CountItems()

        -- Single line load message
        DEFAULT_CHAT_FRAME:AddMessage("|c" .. colors.MAGE_BLUE .. addonName .. "|r: |c" .. colors.GOLD .. "v" .. addonVersion .. "|r " .. strings.LOADED_WITH .. " |c" .. colors.COPPER .. count .. "|r " .. strings.ITEMS_LOWERCASE .. " " .. strings.TYPE_FOR_INFO .. " |c" .. colors.GOLD .. slashCommands.PRIMARY .. "|r " .. strings.FOR_INFO)
    end
end)

-- Register events
frame:RegisterEvent("MERCHANT_SHOW")
frame:RegisterEvent("ADDON_LOADED")

-- Create scan tooltip once
local ScanTooltip = CreateFrame("GameTooltip", "VPS_ScanTooltip", UIParent, "GameTooltipTemplate")
ScanTooltip:SetOwner(UIParent, "ANCHOR_NONE")

-- Function to scan all bags
function VendorPriceScan_ScanBags()
    -- Scan all 5 bags (0-4, where 0 is backpack)
    for bag = 0, 4 do
        local numSlots = GetContainerNumSlots(bag)
        
        if numSlots and numSlots > 0 then
            for slot = 1, numSlots do
                local itemLink = GetContainerItemLink(bag, slot)
                if itemLink then
                    -- Extract itemID from link using shared library
                    local itemID = VendorPriceScan_UI.ExtractItemID(itemLink)
                    if itemID then
                        -- Reset captured price
                        capturedPrice = nil
                        
                        -- Trigger tooltip to capture price
                        ScanTooltip:ClearLines()
                        ScanTooltip:SetBagItem(bag, slot)
                        
                        -- Check if we captured a price
                        if capturedPrice and capturedPrice > 0 then
                            local _, qty = GetContainerItemInfo(bag, slot)
                            local pricePerItem = capturedPrice
                            
                            -- If item is stacked, divide by quantity
                            if qty and qty > 1 then
                                pricePerItem = capturedPrice / qty
                            end
                            
                            -- Get item name for saving
                            local itemName = GetItemInfo(itemID)
                            
                            -- Only save if new or price/name has changed
                            local oldData = VendorPriceScan_Data[itemID]
                            local oldPrice = oldData and oldData.vendorPrice or nil
                            local oldName = oldData and oldData.itemName or nil
                            local currentName = itemName or (strings.ITEM_PREFIX .. itemID)
                            
                            if not oldPrice or oldPrice ~= pricePerItem or oldName ~= currentName then
                                -- Save as table with vendorPrice and itemName
                                VendorPriceScan_Data[itemID] = {
                                    vendorPrice = pricePerItem,
                                    itemName = currentName
                                }
                            end
                        else
                            -- No price captured (nil or 0) = item cannot be sold to vendor
                            -- Remove from database if it exists
                            if VendorPriceScan_Data[itemID] then
                                VendorPriceScan_Data[itemID] = nil
                            end
                        end
                    end
                end
            end
        end
    end
end

-- Helper function to create tooltip hooks
local function CreateTooltipHook(originalFunc, getLinkAndQty)
    return function(...)
        -- Call original function and store all return values
        local results = {originalFunc(unpack(arg))}
        
        -- Get itemLink and quantity using the provided function
        local itemLink, qty = getLinkAndQty(unpack(arg))
        
        -- Add vendor price to tooltip if we have the data and tooltips are enabled
        if itemLink and VendorPriceScan_CharData.showTooltips then
            VendorPriceScan_UI.AddVendorPriceToTooltip(arg[1], itemLink, qty, VendorPriceScan_Data)
        end
        
        -- Return all original values
        return unpack(results)
    end
end

-- Hook into GameTooltip.SetBagItem (bags)
GameTooltip.SetBagItem = CreateTooltipHook(
    GameTooltip.SetBagItem,
    function(tooltip, bag, slot)
        local itemLink = GetContainerItemLink(bag, slot)
        local _, qty = GetContainerItemInfo(bag, slot)
        return itemLink, qty
    end
)

-- Hook into GameTooltip.SetAuctionItem (auction house browsing)
GameTooltip.SetAuctionItem = CreateTooltipHook(
    GameTooltip.SetAuctionItem,
    function(tooltip, type, index)
        local itemLink = GetAuctionItemLink(type, index)
        local _, _, qty = GetAuctionItemInfo(type, index)
        return itemLink, qty
    end
)

-- Hook into GameTooltip.SetAuctionSellItem (auction house selling)
GameTooltip.SetAuctionSellItem = CreateTooltipHook(
    GameTooltip.SetAuctionSellItem,
    function(tooltip)
        local itemName, _, qty = GetAuctionSellItemInfo()
        if itemName then
            local itemLink = VendorPriceScan_UI.FindItemLinkByName(itemName)
            return itemLink, qty
        end
        return nil, nil
    end
)

-- Hook into GameTooltip.SetInventoryItem (equipped items and character sheet)
GameTooltip.SetInventoryItem = CreateTooltipHook(
    GameTooltip.SetInventoryItem,
    function(tooltip, unit, slot)
        local itemLink = GetInventoryItemLink(unit, slot)
        return itemLink, 1
    end
)

-- Hook into GameTooltip.SetLootItem (loot window)
GameTooltip.SetLootItem = CreateTooltipHook(
    GameTooltip.SetLootItem,
    function(tooltip, slot)
        local itemLink = GetLootSlotLink(slot)
        local _, _, qty = GetLootSlotInfo(slot)
        return itemLink, qty
    end
)

-- Hook into GameTooltip.SetLootRollItem (loot roll frame - need/greed/pass)
GameTooltip.SetLootRollItem = CreateTooltipHook(
    GameTooltip.SetLootRollItem,
    function(tooltip, rollID)
        local itemLink = GetLootRollItemLink(rollID)
        return itemLink, 1
    end
)

-- Hook into GameTooltip.SetMerchantItem (vendor window)
GameTooltip.SetMerchantItem = CreateTooltipHook(
    GameTooltip.SetMerchantItem,
    function(tooltip, slot)
        local itemLink = GetMerchantItemLink(slot)
        local _, _, _, qty = GetMerchantItemInfo(slot)
        return itemLink, qty
    end
)

-- Hook into GameTooltip.SetQuestItem (quest rewards)
GameTooltip.SetQuestItem = CreateTooltipHook(
    GameTooltip.SetQuestItem,
    function(tooltip, type, index)
        local itemLink = GetQuestItemLink(type, index)
        local _, _, qty = GetQuestItemInfo(type, index)
        return itemLink, qty
    end
)

-- Hook into GameTooltip.SetQuestLogItem (quest log rewards)
GameTooltip.SetQuestLogItem = CreateTooltipHook(
    GameTooltip.SetQuestLogItem,
    function(tooltip, type, index)
        local itemLink = GetQuestLogItemLink(type, index)
        return itemLink, 1
    end
)

-- Hook into GameTooltip.SetTradeSkillItem (tradeskill window - crafted item)
GameTooltip.SetTradeSkillItem = CreateTooltipHook(
    GameTooltip.SetTradeSkillItem,
    function(tooltip, skillIndex, reagentIndex)
        if reagentIndex then
            -- Reagent (material)
            local itemLink = GetTradeSkillReagentItemLink(skillIndex, reagentIndex)
            local _, _, qty = GetTradeSkillReagentInfo(skillIndex, reagentIndex)
            return itemLink, qty
        else
            -- Crafted item result
            local itemLink = GetTradeSkillItemLink(skillIndex)
            local minMade, maxMade = GetTradeSkillNumMade(skillIndex)
            return itemLink, minMade or 1
        end
    end
)

-- Hook into GameTooltip.SetCraftItem (crafting window - simpler professions like enchanting)
GameTooltip.SetCraftItem = CreateTooltipHook(
    GameTooltip.SetCraftItem,
    function(tooltip, skillIndex, reagentIndex)
        if reagentIndex then
            -- Reagent (material)
            local itemLink = GetCraftReagentItemLink(skillIndex, reagentIndex)
            local _, _, qty = GetCraftReagentInfo(skillIndex, reagentIndex)
            return itemLink, qty
        else
            -- Crafted item result
            local itemLink = GetCraftItemLink(skillIndex)
            return itemLink, 1
        end
    end
)

-- Slash command
SLASH_VENDORPRICESCAN1 = slashCommands.PRIMARY
SLASH_VENDORPRICESCAN2 = slashCommands.SECONDARY
SlashCmdList["VENDORPRICESCAN"] = function(msg)
    if msg == slashCommands.SUBCOMMANDS.SHOW then
        VendorPriceScan_ShowWindow()
        return
    end
    
    local count = CountItems()
    DEFAULT_CHAT_FRAME:AddMessage("|c" .. colors.MAGE_BLUE .. "[" .. addonPrefix .. "]|r " .. strings.LOADED_ITEMS .. " |c" .. colors.COPPER .. count .. "|r " .. strings.ITEMS_FROM_DATA)
    DEFAULT_CHAT_FRAME:AddMessage("|c" .. colors.MAGE_BLUE .. "[" .. addonPrefix .. "]|r " .. strings.VISIT_VENDOR)
    DEFAULT_CHAT_FRAME:AddMessage("|c" .. colors.MAGE_BLUE .. "[" .. addonPrefix .. "]|r " .. strings.USE_COMMAND .. " |c" .. colors.GOLD .. slashCommands.PRIMARY .. " " .. slashCommands.SUBCOMMANDS.SHOW .. "|r " .. strings.TO_OPEN_WINDOW)
end

-- Show window using shared library
function VendorPriceScan_ShowWindow()
    VendorPriceScan_UI.ShowWindow({
        windowName = "VPS_DatabaseWindow",
        title = addonName .. strings.DATABASE_TITLE,
        data = VendorPriceScan_Data,
        getItemData = function(itemID, rawData)
            return {
                price = rawData.vendorPrice or 0,
                itemName = rawData.itemName
            }
        end,
        formatPrice = function(itemData)
            return VendorPriceScan_UI.FormatMoney(itemData.price)
        end,
        onTooltip = function(tooltip, itemData)
            if itemData.price then
                tooltip:AddLine(strings.VENDOR_PRICE, 1, 1, 1)
                tooltip:AddLine(VendorPriceScan_UI.FormatMoney(itemData.price), 1, 1, 1)
            end
        end,
        sortByPrice = function(a, b)
            return a.price < b.price
        end
    })
end