restructuring
This commit is contained in:
Manuel Simon Hirsig
@@ -0,0 +1,241 @@
|
||||
local private , public = {}, {}
|
||||
Aux.scan = public
|
||||
|
||||
local PAGE_SIZE = 50
|
||||
|
||||
local state
|
||||
local threads = {}
|
||||
|
||||
function private.last_page(total_count)
|
||||
local total_pages = math.ceil(total_count / PAGE_SIZE)
|
||||
local last_page = max(total_pages - 1, 0)
|
||||
local last_page_limit = Aux.util.safe_index(private.current_query().blizzard_query, 'last_page')
|
||||
return last_page_limit and min(last_page_limit, last_page) or last_page
|
||||
end
|
||||
|
||||
function private.current_query()
|
||||
return private.current_thread().params.queries[private.current_thread().query_index]
|
||||
end
|
||||
|
||||
function private.current_thread()
|
||||
for _, thread in threads do
|
||||
if thread.id == Aux.control.thread_id then
|
||||
return thread
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function public.start(params)
|
||||
if threads[params.type] then
|
||||
public.abort(threads[params.type].id)
|
||||
end
|
||||
|
||||
local thread_id = Aux.control.new_thread(function()
|
||||
return private.wait_for_callback(private.current_thread().params.on_scan_start, private.scan)
|
||||
end)
|
||||
threads[params.type] = {
|
||||
id = thread_id,
|
||||
params = params,
|
||||
}
|
||||
return thread_id
|
||||
end
|
||||
|
||||
function public.abort(scan_id)
|
||||
local aborted_threads = {}
|
||||
for t, thread in pairs(threads) do
|
||||
if not scan_id or thread.id == scan_id then
|
||||
Aux.control.kill_thread(thread.id)
|
||||
threads[t] = nil
|
||||
tinsert(aborted_threads, thread)
|
||||
end
|
||||
end
|
||||
|
||||
for _, thread in ipairs(aborted_threads) do
|
||||
if thread.params.on_abort then
|
||||
thread.params.on_abort()
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function private.wait_for_results(k)
|
||||
if private.current_thread().params.type == 'bidder' then
|
||||
return Aux.control.wait_until(function() return Aux.bids_loaded end, k)
|
||||
elseif private.current_thread().params.type == 'owner' then
|
||||
return private.wait_for_owner_results(k)
|
||||
elseif private.current_thread().params.type == 'list' then
|
||||
return private.wait_for_list_results(k)
|
||||
end
|
||||
end
|
||||
|
||||
function private.wait_for_owner_results(k)
|
||||
local updated
|
||||
if private.current_thread().page == Aux.current_owner_page then
|
||||
updated = true
|
||||
else
|
||||
Aux.control.on_next_event('AUCTION_OWNED_LIST_UPDATE', function()
|
||||
updated = true
|
||||
end)
|
||||
end
|
||||
|
||||
Aux.control.wait_until(function() return updated end, k)
|
||||
end
|
||||
|
||||
function private.wait_for_list_results(k)
|
||||
local updated, last_update
|
||||
local listener = Aux.control.event_listener('AUCTION_ITEM_LIST_UPDATE', function()
|
||||
last_update = GetTime()
|
||||
updated = true
|
||||
end)
|
||||
listener:start()
|
||||
Aux.control.wait_until(function()
|
||||
-- order important, owner_data_complete must be called after an update to request missing data
|
||||
local ok = updated and private.owner_data_complete() or last_update and GetTime() - last_update > 5
|
||||
updated = false
|
||||
return ok
|
||||
end, function()
|
||||
listener:stop()
|
||||
return k()
|
||||
end)
|
||||
end
|
||||
|
||||
function private.owner_data_complete()
|
||||
if private.current_thread().params.ignore_owner or aux_ignore_owner then
|
||||
return true
|
||||
end
|
||||
local count = GetNumAuctionItems(private.current_thread().params.type)
|
||||
for i=1,count do
|
||||
local auction_info = Aux.info.auction(i, private.current_thread().params.type)
|
||||
if auction_info and not auction_info.owner then
|
||||
return false
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
function private.wait_for_callback(...)
|
||||
local ok = true
|
||||
|
||||
local f = tremove(arg, 1)
|
||||
local k = tremove(arg)
|
||||
|
||||
if f then
|
||||
tinsert(arg, {
|
||||
suspend = function() ok = false end,
|
||||
resume = function() ok = true end,
|
||||
})
|
||||
f(unpack(arg))
|
||||
end
|
||||
|
||||
if ok then
|
||||
return k()
|
||||
else
|
||||
return Aux.control.wait_until(function() return ok end, k)
|
||||
end
|
||||
end
|
||||
|
||||
function private.scan()
|
||||
private.current_thread().query_index = private.current_thread().query_index and private.current_thread().query_index + 1 or 1
|
||||
if private.current_query() then
|
||||
if private.current_query().blizzard_query then
|
||||
private.current_thread().page = private.current_query().blizzard_query.first_page or 0
|
||||
else
|
||||
private.current_thread().page = nil
|
||||
end
|
||||
private.wait_for_callback(private.current_thread().params.on_start_query, private.current_thread().query_index, private.process_query)
|
||||
else
|
||||
local on_complete = private.current_thread().params.on_complete
|
||||
threads[private.current_thread().params.type] = nil
|
||||
if on_complete then
|
||||
return on_complete()
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function private.process_query()
|
||||
if private.current_query().blizzard_query then
|
||||
return private.submit_query(private.scan_page)
|
||||
else
|
||||
return private.scan_page()
|
||||
end
|
||||
end
|
||||
|
||||
function private.scan_page()
|
||||
private.scan_auctions(function()
|
||||
|
||||
private.wait_for_callback(private.current_thread().params.on_page_scanned, function()
|
||||
|
||||
if private.current_query().blizzard_query and private.current_thread().page < private.last_page(private.current_thread().total_auctions) then
|
||||
private.current_thread().page = private.current_thread().page + 1
|
||||
return private.process_query()
|
||||
else
|
||||
return private.scan()
|
||||
end
|
||||
|
||||
end)
|
||||
end)
|
||||
end
|
||||
|
||||
function private.scan_auctions(k)
|
||||
return private.scan_auctions_helper(1, k)
|
||||
end
|
||||
|
||||
function private.scan_auctions_helper(i, k)
|
||||
local recurse = function()
|
||||
if i >= (private.current_thread().page_auctions or GetNumAuctionItems(private.current_thread().params.type)) then
|
||||
return k()
|
||||
else
|
||||
return private.scan_auctions_helper(i + 1, k)
|
||||
end
|
||||
end
|
||||
|
||||
local auction_info = Aux.info.auction(i, private.current_thread().params.type)
|
||||
if auction_info and (auction_info.owner or private.current_thread().params.ignore_owner or aux_ignore_owner) then
|
||||
auction_info.index = i
|
||||
auction_info.page = private.current_thread().page
|
||||
auction_info.blizzard_query = private.current_query().blizzard_query
|
||||
auction_info.query_type = private.current_thread().params.type
|
||||
|
||||
Aux.history.process_auction(auction_info)
|
||||
|
||||
if not private.current_query().validator or private.current_query().validator(auction_info) then
|
||||
return private.wait_for_callback(private.current_thread().params.on_auction, auction_info, recurse)
|
||||
end
|
||||
end
|
||||
|
||||
return recurse()
|
||||
end
|
||||
|
||||
function private.submit_query(k)
|
||||
Aux.control.wait_until(function() return private.current_thread().params.type ~= 'list' or CanSendAuctionQuery() end, function()
|
||||
|
||||
if private.current_thread().params.on_submit_query then
|
||||
private.current_thread().params.on_submit_query()
|
||||
end
|
||||
if private.current_thread().params.type == 'bidder' then
|
||||
GetBidderAuctionItems(private.current_thread().page)
|
||||
elseif private.current_thread().params.type == 'owner' then
|
||||
GetOwnerAuctionItems(private.current_thread().page)
|
||||
else
|
||||
QueryAuctionItems(
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'name'),
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'min_level'),
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'max_level'),
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'slot'),
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'class'),
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'subclass'),
|
||||
private.current_thread().page,
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'usable'),
|
||||
Aux.util.safe_index(private.current_query().blizzard_query, 'quality')
|
||||
)
|
||||
end
|
||||
private.wait_for_results(function()
|
||||
private.current_thread().page_auctions, private.current_thread().total_auctions = GetNumAuctionItems(private.current_thread().params.type)
|
||||
private.wait_for_callback(
|
||||
private.current_thread().params.on_page_loaded,
|
||||
private.current_thread().page - (private.current_query().blizzard_query.first_page or 0) + 1,
|
||||
max(1, private.last_page(private.current_thread().total_auctions) - (private.current_query().blizzard_query.first_page or 0) + 1),
|
||||
k
|
||||
)
|
||||
end)
|
||||
end)
|
||||
end
|
||||
Reference in New Issue
Block a user