aux-addon-ClassicAPI/components/cache.lua

aux.module 'cache'

MIN_ITEM_ID = 1
MAX_ITEM_ID = 30000

items_schema = {'record', '#', {name='string'}, {quality='number'}, {level='number'}, {class='string'}, {subclass='string'}, {slot='string'}, {max_stack='number'}, {texture='string'}}
merchant_buy_schema = {'record', '#', {unit_price='number'}, {limited='boolean'} }

_G.aux_items = {}
_G.aux_item_ids = {}
_G.aux_auctionable_items = {}
_G.aux_merchant_buy = {}
_G.aux_merchant_sell = {}
_G.aux_characters = {}

function m.LOAD()
	m.scan_wdb()

	aux.control.event_listener('MERCHANT_SHOW', m.on_merchant_show)
	aux.control.event_listener('MERCHANT_CLOSED', m.on_merchant_closed)
	aux.control.event_listener('MERCHANT_UPDATE', m.on_merchant_update)
	aux.control.event_listener('BAG_UPDATE', m.on_bag_update)

	CreateFrame('Frame', nil, MerchantFrame):SetScript('OnUpdate', m.merchant_on_update)

	aux.control.event_listener('NEW_AUCTION_UPDATE', function()
		for _, info in {aux.info.auction_sell_item()} do
			for _, item_id in {aux.cache.item_id(info.name)} do
				_G.aux_merchant_sell[aux.cache.item_id(info.name)] = info.vendor_price / (aux.info.max_item_charges(item_id) or info.count)
			end
		end
	end)
end

do
	local sell_scan_countdown, incomplete_buy_data

	function on_merchant_show()
		m.merchant_sell_scan()
		incomplete_buy_data = not m.merchant_buy_scan()
	end

	function on_merchant_closed()
		sell_scan_countdown = nil
		incomplete_buy_data = false
	end

	function on_merchant_update()
		if incomplete_buy_data then
			incomplete_buy_data = not m.merchant_buy_scan()
		end
	end

	function on_bag_update()
		if MerchantFrame:IsVisible() then
			sell_scan_countdown = 10
		end
	end

	function merchant_on_update()
		if sell_scan_countdown == 0 then
			sell_scan_countdown = nil
			m.merchant_sell_scan()
		elseif sell_scan_countdown then
			sell_scan_countdown = sell_scan_countdown - 1
		end
	end
end

function merchant_loaded()
	for i=1,GetMerchantNumItems() do
		if not GetMerchantItemLink(i) then
			return false
		end
	end
	return true
end

function public.merchant_info(item_id)
	local buy_info
	if _G.aux_merchant_buy[item_id] then
		buy_info = aux.persistence.read(m.merchant_buy_schema, _G.aux_merchant_buy[item_id])
	end

	return _G.aux_merchant_sell[item_id], buy_info and buy_info.unit_price, buy_info and buy_info.limited
end

function public.item_info(item_id)
	local data_string = _G.aux_items[item_id]
	if data_string then
		local cached_data = aux.persistence.read(m.items_schema, data_string)
		return {
			name = cached_data.name,
			itemstring = 'item:'..item_id..':0:0:0',
			quality = cached_data.quality,
			level = cached_data.level,
			class = cached_data.class,
			subclass = cached_data.subclass,
			slot = cached_data.slot,
			max_stack = cached_data.max_stack,
			texture = cached_data.texture,
		}
	end
end

function public.item_id(item_name)
	return _G.aux_item_ids[strlower(item_name)]
end

function merchant_buy_scan()

	local incomplete_data
	for i=1,GetMerchantNumItems() do
		local _, _, price, count, stock = GetMerchantItemInfo(i)
		local link = GetMerchantItemLink(i)
		if link then
			local item_id = aux.info.parse_link(link)
			local new_unit_price, new_limited = price / count, stock >= 0
			if _G.aux_merchant_buy[item_id] then
				local buy_info = aux.persistence.read(m.merchant_buy_schema, _G.aux_merchant_buy[item_id])

				local unit_price
				if buy_info.limited and not new_limited then
					unit_price = new_unit_price
				elseif new_limited and not buy_info.limited then
					unit_price = buy_info.unit_price
				else
					unit_price = min(buy_info.unit_price, new_unit_price)
				end

				_G.aux_merchant_buy[item_id] = aux.persistence.write(m.merchant_buy_schema, {
					unit_price = unit_price,
					limited = buy_info.limited and new_limited,
				})
			else
				_G.aux_merchant_buy[item_id] = aux.persistence.write(m.merchant_buy_schema, {
					unit_price = new_unit_price,
					limited = new_limited,
				})
			end
		else
			incomplete_data = true
		end
	end

	return not incomplete_data
end

function merchant_sell_scan()
	for slot in aux.util.inventory() do
		local item_info = aux.info.container_item(unpack(slot))
		if item_info then
			_G.aux_merchant_sell[item_info.item_id] = item_info.tooltip_money / item_info.aux_quantity
		end
	end
end

function scan_wdb(item_id)
	item_id = item_id or m.MIN_ITEM_ID

	local processed = 0
	while processed <= 100 and item_id <= m.MAX_ITEM_ID do
		local itemstring = 'item:'..item_id
		local name, _, quality, level, class, subclass, max_stack, slot, texture = GetItemInfo(itemstring)
		if name and not _G.aux_item_ids[strlower(name)] then
			_G.aux_item_ids[strlower(name)] = item_id
			_G.aux_items[item_id] = aux.persistence.write(m.items_schema, {
				name = name,
				quality = quality,
				level = level,
				class = class,
				subclass = subclass,
				slot = slot,
				max_stack = max_stack,
				texture = texture,
			})
			local tooltip = aux.info.tooltip(function(tt) tt:SetHyperlink(itemstring) end)
			if aux.info.auctionable(tooltip, quality) then
				tinsert(_G.aux_auctionable_items, strlower(name))
			end
			processed = processed + 1
		end
		item_id = item_id + 1
	end

	if item_id <= m.MAX_ITEM_ID then
		local t0 = GetTime()
		aux.control.thread(aux.control.when, function() return GetTime() - t0 > 0.1 end, m.scan_wdb, item_id)
	else
		sort(_G.aux_auctionable_items, function(a, b) return strlen(a) < strlen(b) or (strlen(a) == strlen(b) and a < b) end)
	end
end

function public.populate_wdb(item_id)
	item_id = item_id or m.MIN_ITEM_ID

	if item_id > m.MAX_ITEM_ID then
		aux.log('Cache populated.')
		return
	end

	if not GetItemInfo('item:'..item_id) then
		aux.log('Fetching item '..item_id..'.')
		AuxTooltip:SetHyperlink('item:'..item_id)
	end

	aux.control.thread(m.populate_wdb, item_id + 1)
end