[Fixes #44, Fixes #43] Resolve TMog source correctly after purchase to hide it

Author: plusmouse

Source/PurchaseWatch/Frame.lua

@@ -118,7 +118,7 @@ end
 function CollectionatorPurchaseWatchFrameMixin:ProcessTMogDetails(itemLink)
   local _, source = C_TransmogCollection.GetItemInfo(itemLink)
   if source == nil then
-    _, source = C_TransmogCollection.GetItemInfo(C_Item.GetItemInfoInstant(itemLink))
+    source = Collectionator.Utilities.RecoverTMogSource(C_Item.GetItemInfoInstant(itemLink))
   end
   if source ~= nil then
     local sourceInfo = C_TransmogCollection.GetSourceInfo(source)

Source/Replicate/Scanner/TMog.lua

@@ -20,7 +20,7 @@ end
 function CollectionatorReplicateTMogScannerFrameMixin:GetItem(index, link, scanInfo)
   local _, source = C_TransmogCollection.GetItemInfo(link)
   if source == nil then
-    _, source = C_TransmogCollection.GetItemInfo(C_Item.GetItemInfoInstant(link))
+    source = Collectionator.Utilities.RecoverTMogSource(C_Item.GetItemInfoInstant(link))
   end
 
   if source ~= nil then

Source/Summary/Scanner/TMog.lua

@@ -1,16 +1,5 @@
 CollectionatorSummaryTMogScannerFrameMixin = CreateFromMixins(CollectionatorSummaryScannerFrameMixin)
 
-local modelScene
-modelScene = CreateFrame("ModelScene", nil, UIParent, "ModelSceneMixinTemplate")
-modelScene:TransitionToModelSceneID(596, CAMERA_TRANSITION_TYPE_IMMEDIATE, CAMERA_MODIFICATION_TYPE_DISCARD, true)
-modelScene:Hide()
-local frame = CreateFrame("Frame")
-frame:SetScript("OnUpdate", function()
-  if(modelScene:GetPlayerActor():SetModelByUnit("player")) then
-    frame:SetScript("OnUpdate", nil)
-  end
-end)
-
 function CollectionatorSummaryTMogScannerFrameMixin:OnLoad()
   CollectionatorSummaryScannerFrameMixin.OnLoad(self)
 
@@ -34,29 +23,7 @@ function CollectionatorSummaryTMogScannerFrameMixin:GetItem(index, itemKeyInfo,
   if itemKeyInfo.appearanceLink ~= nil then
     source = tonumber(itemKeyInfo.appearanceLink:match("transmogappearance:(%d+)"))
   else
-    source = select(2, C_TransmogCollection.GetItemInfo(scanInfo.itemKey.itemID))
-  end
-
-  if not source and C_Item.IsDressableItemByID(scanInfo.itemKey.itemID) then
-    local pa = modelScene:GetPlayerActor()
-    local invType = select(4, C_Item.GetItemInfoInstant(scanInfo.itemKey.itemID))
-    local mainhandOverride = invType == "INVTYPE_WEAPON" or invType == "INVTYPE_RANGEDRIGHT"
-    local slot = Collectionator.Constants.SlotMap[invType]
-    if slot then
-      local link = select(2, C_Item.GetItemInfo(scanInfo.itemKey.itemID))
-      local result
-      if mainhandOverride then
-        result = pa:TryOn(link, "MAINHANDSLOT")
-      else
-        result = pa:TryOn(link)
-      end
-      if result == Enum.ItemTryOnReason.Success then
-        local info = pa:GetItemTransmogInfo(slot)
-        if info then
-          source = info.appearanceID
-        end
-      end
-    end
+    source = Collectionator.Utilities.RecoverTMogSource(scanInfo.itemKey.itemID)
   end
 
   if source ~= nil and source > 0 and select(2, C_TransmogCollection.AccountCanCollectSource(source)) then

Source/Utilities/Manifest.xml

@@ -6,4 +6,5 @@
   <Script file="GroupedByID.lua" />
   <Script file="GetPrice.lua" />
   <Script file="GetRealmAndFaction.lua" />
+  <Script file="RecoverTMogSource.lua" />
 </Ui>

Source/Utilities/RecoverTMogSource.lua

@@ -0,0 +1,38 @@
+local modelScene
+modelScene = CreateFrame("ModelScene", nil, UIParent, "ModelSceneMixinTemplate")
+modelScene:TransitionToModelSceneID(596, CAMERA_TRANSITION_TYPE_IMMEDIATE, CAMERA_MODIFICATION_TYPE_DISCARD, true)
+modelScene:Hide()
+local frame = CreateFrame("Frame")
+frame:SetScript("OnUpdate", function()
+  if(modelScene:GetPlayerActor():SetModelByUnit("player")) then
+    frame:SetScript("OnUpdate", nil)
+  end
+end)
+
+function Collectionator.Utilities.RecoverTMogSource(itemID)
+  local possible = select(2, C_TransmogCollection.GetItemInfo(itemID))
+  if possible then
+    return possible
+  end
+  if C_Item.IsDressableItemByID(itemID) then
+    local pa = modelScene:GetPlayerActor()
+    local invType = select(4, C_Item.GetItemInfoInstant(itemID))
+    local mainhandOverride = invType == "INVTYPE_WEAPON" or invType == "INVTYPE_RANGEDRIGHT"
+    local slot = Collectionator.Constants.SlotMap[invType]
+    if slot then
+      local link = select(2, C_Item.GetItemInfo(itemID))
+      local result
+      if mainhandOverride then
+        result = pa:TryOn(link, "MAINHANDSLOT")
+      else
+        result = pa:TryOn(link)
+      end
+      if result == Enum.ItemTryOnReason.Success then
+        local info = pa:GetItemTransmogInfo(slot)
+        if info then
+          return info.appearanceID
+        end
+      end
+    end
+  end
+end