initial commit

LibTSM/Service/AltTracking.lua

@@ -0,0 +1,74 @@
+-- ------------------------------------------------------------------------------ --
+--                                TradeSkillMaster                                --
+--                          https://tradeskillmaster.com                          --
+--    All Rights Reserved - Detailed license information included with addon.     --
+-- ------------------------------------------------------------------------------ --
+
+local _, TSM = ...
+local AltTracking = TSM.Init("Service.AltTracking")
+local Database = TSM.Include("Util.Database")
+local TempTable = TSM.Include("Util.TempTable")
+local Vararg = TSM.Include("Util.Vararg")
+local Settings = TSM.Include("Service.Settings")
+local private = {
+	quantityDB = nil,
+}
+
+
+
+-- ============================================================================
+-- Module Loading
+-- ============================================================================
+
+AltTracking:OnSettingsLoad(function()
+	private.quantityDB = Database.NewSchema("INVENTORY_ALT_QUANTITY")
+		:AddUniqueStringField("itemString")
+		:AddNumberField("quantity")
+		:Commit()
+
+	local altItemQuantity = TempTable.Acquire()
+	for _, factionrealm, character, syncScopeKey in Settings.ConnectedFactionrealmAltCharacterIterator() do
+		for _, key in Vararg.Iterator("bagQuantity", "bankQuantity", "reagentBankQuantity", "auctionQuantity", "mailQuantity") do
+			for itemString, quantity in pairs(Settings.Get("sync", syncScopeKey, "internalData", key)) do
+				altItemQuantity[itemString] = (altItemQuantity[itemString] or 0) + quantity
+			end
+		end
+		local pendingMailLookup = Settings.Get("factionrealm", factionrealm, "internalData", "pendingMail")
+		local pendingMail = pendingMailLookup[character]
+		local isValid = true
+		if pendingMail then
+			for itemString, quantity in pairs(pendingMail) do
+				if type(quantity) ~= "number" then
+					isValid = false
+					break
+				end
+				altItemQuantity[itemString] = (altItemQuantity[itemString] or 0) + quantity
+			end
+		end
+		if not isValid then
+			pendingMailLookup[character] = nil
+		end
+	end
+	private.quantityDB:BulkInsertStart()
+	for itemString, quantity in pairs(altItemQuantity) do
+		private.quantityDB:BulkInsertNewRow(itemString, quantity)
+	end
+	private.quantityDB:BulkInsertEnd()
+	TempTable.Release(altItemQuantity)
+end)
+
+
+
+-- ============================================================================
+-- Module Functions
+-- ============================================================================
+
+function AltTracking.BaseItemIterator()
+	return private.quantityDB:NewQuery()
+		:Select("itemString")
+		:IteratorAndRelease()
+end
+
+function AltTracking.GetQuantityByBaseItemString(baseItemString)
+	return private.quantityDB:GetUniqueRowField("itemString", baseItemString, "quantity") or 0
+end