TradeSkillMaster/TradeSkillMaster_AuctionDB/Modules/GUI.lua

-- ------------------------------------------------------------------------------ --
--                           TradeSkillMaster_AuctionDB                           --
--           http://www.curse.com/addons/wow/tradeskillmaster_auctiondb           --
--                                                                                --
--             A TradeSkillMaster Addon (http://tradeskillmaster.com)             --
--    All Rights Reserved* - Detailed license information included with addon.    --
-- ------------------------------------------------------------------------------ --

-- load the parent file (TSM) into a local variable and register this file as a module
local TSM = select(2, ...)
local GUI = TSM:NewModule("GUI")
local L = LibStub("AceLocale-3.0"):GetLocale("TradeSkillMaster_AuctionDB") -- loads the localization table
	
local private = {}

function GUI:Show(frame)
	private.statusBar = private.statusBar or private:CreateStatusBar(frame.content)
	private.statusBar:Show()
	GUI:UpdateStatus("", 0, 0)
	
	private.startScanContent = private.startScanContent or private:CreateStartScanContent(frame)
	private.startScanContent:Show()
end

function GUI:Hide()
	private.statusBar:Hide()
	private.startScanContent:Hide()
	
	TSM.Scan:DoneScanning()
	TSMAPI.AuctionScan:StopScan()
end

function GUI:UpdateStatus(text, major, minor)
	if text then
		private.statusBar:SetStatusText(text)
	end
	if major or minor then
		private.statusBar:UpdateStatus(major, minor)
	end
end

function private:CreateStatusBar(parent)
	local frame = TSMAPI.GUI:CreateStatusBar(parent, "TSMAuctionDBStatusBar")
	TSMAPI.GUI:CreateHorizontalLine(frame, -30, parent)
	
	return frame
end

function private:CreateStartScanContent(parent)
	local frame = CreateFrame("Frame", nil, parent)
	frame:SetAllPoints(parent)
	frame:Hide()

	-- Don't create or handle the GetAll button if player has disabled GetAll.
	-- NOTE: This GUI creation is only done once per game reload, so this choice
	-- won't change until the user does a UI "/reload" or logs out of the game.
	local includeGetAll = not TSM.db.profile.disableGetAll

	local function UpdateGetAllButton()
		if not frame.startGetAllButton then
			return  -- Do nothing if the GetAll feature is disabled.
		end

		if TSM.Scan.isScanning then
			frame:Disable()
		elseif not select(2, CanSendAuctionQuery()) then
			-- Server says that GetAll isn't ready. Check our stored cooldown value.
			local previous = TSM.db.profile.lastGetAll or time()
			if previous > (time() - 15*60) then  -- 15 minute enforced cooldown between GetAll scans...
				local diff = previous + 15*60 - time()
				local diffMin = math.floor(diff/60)
				local diffSec = diff - diffMin*60
				frame.getAllStatusText:SetText("|cff990000"..format(L["Ready in %s min and %s sec"], diffMin, diffSec))
			else
				frame.getAllStatusText:SetText("|cff990000"..L["Not Ready"])
			end
			frame:Enable()
			frame.startGetAllButton:Disable()
		else
			-- Server says that GetAll is ready.
			frame:Enable()
			frame.getAllStatusText:SetText("|cff009900"..L["Ready"])
			frame.startGetAllButton:Enable()
		end
	end

	if includeGetAll then
		frame:SetScript("OnShow", function(self)
			TSMAPI:CreateTimeDelay("auctionDBGetAllStatus", 0, UpdateGetAllButton, 0.2)
		end)

		frame:SetScript("OnHide", function(self)
			TSMAPI:CancelFrame("auctionDBGetAllStatus")
		end)
	end

	frame.Enable = function(self)
		if self.startGetAllButton then self.startGetAllButton:Enable() end
		self.startFullScanButton:Enable()
		self.startGroupScanButton:Enable()
		self.startBagScanButton:Enable()
		self.startBankScanButton:Enable()
	end

	frame.Disable = function(self)
		if self.startGetAllButton then self.startGetAllButton:Disable() end
		self.startFullScanButton:Disable()
		self.startGroupScanButton:Disable()
		self.startBagScanButton:Disable()
		self.startBankScanButton:Disable()
	end

	-- Top row: Auto updater.
	local text = TSMAPI.GUI:CreateLabel(frame)
	text:SetFont(TSMAPI.Design:GetContentFont(), 24)
	text:SetPoint("TOP", 0, -24)
	text:SetHeight(24)
	text:SetJustifyH("CENTER")
	text:SetJustifyV("CENTER")
	text:SetText(TSMAPI.Design:GetInlineColor("link").."TSM_AuctionDB")
	local ag = text:CreateAnimationGroup()
	local a1 = ag:CreateAnimation("Alpha")
	a1:SetChange(-.5)
	a1:SetDuration(.5)
	ag:SetLooping("BOUNCE")
	ag:Play()

	local content = CreateFrame("Frame", nil, frame)
	content:SetAllPoints(parent.content)
	TSMAPI.Design:SetFrameBackdropColor(content)

	-- Group tree.
	local container = CreateFrame("Frame", nil, content)
	container:SetPoint("TOPLEFT", 5, -35)
	container:SetPoint("BOTTOMRIGHT", -205, 5)
	TSMAPI.Design:SetFrameColor(container)
	frame.groupTree = TSMAPI:CreateGroupTree(container, nil, "AuctionDB")

	local bar = TSMAPI.GUI:CreateVerticalLine(content, 0)
	bar:ClearAllPoints()
	bar:SetPoint("TOPRIGHT", -200, -30)
	bar:SetPoint("BOTTOMRIGHT", -200, 0)

	local buttonFrame = CreateFrame("Frame", nil, content)
	buttonFrame:SetPoint("TOPLEFT", content, "TOPRIGHT", -200, 0)
	buttonFrame:SetPoint("BOTTOMRIGHT")

	-- Row: GetAll Scan.
	-- NOTE: We hide this button if the player has disabled GetAll scans.
	local yOffset = -50
	if includeGetAll then
		local btn = TSMAPI.GUI:CreateButton(buttonFrame, 18)
		btn:SetPoint("TOPLEFT", 6, yOffset)
		btn:SetPoint("TOPRIGHT", -6, yOffset)
		btn:SetHeight(22)
		btn:SetScript("OnClick", TSM.Scan.StartGetAllScan)
		btn:SetText(L["Run GetAll Scan"])
		btn.tooltip = L["A GetAll scan is the fastest in-game method for scanning every item on the auction house. However, there are many possible bugs on Blizzard's end with it including the chance for it to disconnect you from the game. Also, it has a 15 minute cooldown. You can disable the GetAll button via TSM's AuctionDB options if this feature doesn't work well on your server."]
		frame.startGetAllButton = btn

		local text = TSMAPI.GUI:CreateLabel(buttonFrame)
		text:SetPoint("TOPLEFT", btn, "BOTTOMLEFT", 0, -3)
		text:SetPoint("TOPRIGHT", btn, "BOTTOMRIGHT", 0, -3)
		text:SetHeight(16)
		text:SetJustifyH("CENTER")
		text:SetJustifyV("CENTER")
		frame.getAllStatusText = text

		yOffset = yOffset - 50

		TSMAPI.GUI:CreateHorizontalLine(buttonFrame, yOffset)

		yOffset = yOffset - 20
	end

	-- Row: Full Scan.
	local btn = TSMAPI.GUI:CreateButton(buttonFrame, 18)
	btn:SetPoint("TOPLEFT", 6, yOffset)
	btn:SetPoint("TOPRIGHT", -6, yOffset)
	btn:SetHeight(22)
	btn:SetScript("OnClick", TSM.Scan.StartFullScan)
	btn:SetText(L["Run Full Scan"])
	btn.tooltip = L["A full auction house scan will scan every item on the auction house but is far slower than a GetAll scan. Expect this scan to take several minutes or longer."]
	frame.startFullScanButton = btn

	yOffset = yOffset - 40

	TSMAPI.GUI:CreateHorizontalLine(buttonFrame, yOffset)

	yOffset = yOffset - 20

	-- Row: Group Scan.
	local btn = TSMAPI.GUI:CreateButton(buttonFrame, 18)
	btn:SetPoint("TOPLEFT", 6, yOffset)
	btn:SetPoint("TOPRIGHT", -6, yOffset)
	btn:SetHeight(22)
	btn:SetScript("OnClick", GUI.StartGroupScan)
	btn:SetText(L["Scan Selected Groups"])
	btn.tooltip = L["This will do a slow auction house scan of every item in the selected groups and update their AuctionDB prices. This may take several minutes."]
	frame.startGroupScanButton = btn

	yOffset = yOffset - 40

	TSMAPI.GUI:CreateHorizontalLine(buttonFrame, yOffset)

	yOffset = yOffset - 20

	-- Row: Bag Scan.
	local btn = TSMAPI.GUI:CreateButton(buttonFrame, 18)
	btn:SetPoint("TOPLEFT", 6, yOffset)
	btn:SetPoint("TOPRIGHT", -6, yOffset)
	btn:SetHeight(22)
	btn:SetScript("OnClick", GUI.StartBagScan)
	btn:SetText(L["Scan Bags"])
	btn.tooltip = L["Scan bag items."]
	frame.startBagScanButton = btn

	yOffset = yOffset - 30

	-- Row: Bank Scan.
	local btn = TSMAPI.GUI:CreateButton(buttonFrame, 18)
	btn:SetPoint("TOPLEFT", 6, yOffset)
	btn:SetPoint("TOPRIGHT", -6, yOffset)
	btn:SetHeight(22)
	btn:SetScript("OnClick", GUI.StartBankScan)
	btn:SetText(L["Scan Bank"])
	btn.tooltip = L["Scan bank items."]
	frame.startBankScanButton = btn

	return frame
end

function GUI:StartGroupScan()
	local items = {}
	for groupName, data in pairs(private.startScanContent.groupTree:GetSelectedGroupInfo()) do
		groupName = TSMAPI:FormatGroupPath(groupName, true)
		for itemString in pairs(data.items) do
			tinsert(items, itemString)
		end
	end
	TSM.Scan:StartGroupScan(items)
end

function private:GetItemTrackerItems(key)
	local data = TSMAPI:ModuleAPI("ItemTracker", key)
	if not data then
		TSM:Print(L["ItemTracker data unavailable."])
		return
	end

	local specificItems = {}
	local baseCandidates = {}
	local hasSpecific = {}

	for itemString in pairs(data) do
		if type(itemString) == "string" then
			local baseItemString = TSMAPI:GetBaseItemString(itemString) or itemString
			if itemString ~= baseItemString then
				specificItems[itemString] = true
				hasSpecific[baseItemString] = true
			else
				baseCandidates[baseItemString] = true
			end
		end
	end

	for baseItemString in pairs(baseCandidates) do
		if not hasSpecific[baseItemString] then
			specificItems[baseItemString] = true
		end
	end

	local items = {}
	for itemString in pairs(specificItems) do
		tinsert(items, itemString)
	end
	return items
end

function GUI:StartBagScan()
	local items = private:GetItemTrackerItems("playerbags")
	if not items then
		return
	end
	if #items == 0 then
		TSM:Print(L["No bag items found."])
		return
	end

	TSM.Scan:StartGroupScan(items)
end

function GUI:StartBankScan()
	local items = private:GetItemTrackerItems("playerbank")
	if not items then
		return
	end
	if #items == 0 then
		TSM:Print(L["No bank items found."])
		return
	end

	TSM.Scan:StartGroupScan(items)
end