Mercurial > wow > askmrrobot

annotate Export.lua @ 177:44ce007189e0 v83

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix to automatic search on shopping list at auction house.
author yellowfive
date Sun, 26 Jan 2020 12:37:12 -0800
parents 3be9cc6f7d20
children a7f2ba17c39a
rev   line source
yellowfive@57 1 local Amr = LibStub("AceAddon-3.0"):GetAddon("AskMrRobot")
yellowfive@57 2 local L = LibStub("AceLocale-3.0"):GetLocale("AskMrRobot", true)
yellowfive@57 3 local AceGUI = LibStub("AceGUI-3.0")
yellowfive@57 4
yellowfive@57 5 local _lastExport = nil
yellowfive@57 6 local _txt = nil
yellowfive@57 7
yellowfive@57 8 local function createLabel(container, text, width)
yellowfive@57 9 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 10 container:AddChild(lbl)
yellowfive@57 11 lbl:SetWidth(width or 800)
yellowfive@57 12 lbl:SetText(text)
yellowfive@57 13 lbl:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.Text))
yellowfive@57 14 return lbl
yellowfive@57 15 end
yellowfive@57 16
yellowfive@57 17 local function onSplashClose()
yellowfive@57 18 Amr:HideCover()
yellowfive@57 19 Amr.db.char.FirstUse = false
yellowfive@57 20 end
yellowfive@57 21
yellowfive@69 22 local function onTextChanged(widget)
yellowfive@69 23 local val = _txt:GetText()
yellowfive@69 24 if val == "overwolf-bib" then
yellowfive@69 25 -- go to the gear tab, open import window, and show a cover
yellowfive@69 26 Amr:ShowTab("Gear")
yellowfive@69 27 Amr:ShowImportWindow(true)
yellowfive@69 28 end
yellowfive@69 29 end
yellowfive@69 30
yellowfive@57 31 -- render a splash screen with first-time help
yellowfive@57 32 local function renderSplash(container)
yellowfive@57 33 local panel = Amr:RenderCoverChrome(container, 700, 450)
yellowfive@57 34
yellowfive@57 35 local lbl = createLabel(panel, L.ExportSplashTitle, 650)
yellowfive@57 36 lbl:SetJustifyH("CENTER")
yellowfive@57 37 lbl:SetFont(Amr.CreateFont("Bold", 24, Amr.Colors.TextHeaderActive))
yellowfive@57 38 lbl:SetPoint("TOP", panel.content, "TOP", 0, -10)
yellowfive@57 39
yellowfive@57 40 local lbl2 = createLabel(panel, L.ExportSplashSubtitle, 650)
yellowfive@57 41 lbl2:SetJustifyH("CENTER")
yellowfive@57 42 lbl2:SetFont(Amr.CreateFont("Bold", 18, Amr.Colors.TextTan))
yellowfive@57 43 lbl2:SetPoint("TOP", lbl.frame, "BOTTOM", 0, -20)
yellowfive@57 44
yellowfive@57 45 lbl = createLabel(panel, L.ExportSplash1, 650)
yellowfive@57 46 lbl:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.Text))
yellowfive@57 47 lbl:SetPoint("TOPLEFT", lbl2.frame, "BOTTOMLEFT", 0, -70)
yellowfive@57 48
yellowfive@57 49 lbl2 = createLabel(panel, L.ExportSplash2, 650)
yellowfive@57 50 lbl2:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.Text))
yellowfive@57 51 lbl2:SetPoint("TOPLEFT", lbl.frame, "BOTTOMLEFT", 0, -15)
yellowfive@57 52
yellowfive@57 53 local btn = AceGUI:Create("AmrUiButton")
yellowfive@57 54 btn:SetText(L.ExportSplashClose)
yellowfive@57 55 btn:SetBackgroundColor(Amr.Colors.Green)
yellowfive@57 56 btn:SetFont(Amr.CreateFont("Bold", 16, Amr.Colors.White))
yellowfive@57 57 btn:SetWidth(120)
yellowfive@57 58 btn:SetHeight(28)
yellowfive@57 59 btn:SetCallback("OnClick", onSplashClose)
yellowfive@57 60 panel:AddChild(btn)
yellowfive@124 61 btn:SetPoint("BOTTOM", panel.content, "BOTTOM", 0, 20)
yellowfive@57 62 end
yellowfive@57 63
yellowfive@57 64 -- renders the main UI for the Export tab
yellowfive@57 65 function Amr:RenderTabExport(container)
yellowfive@57 66
yellowfive@57 67 local lbl = createLabel(container, L.ExportTitle)
yellowfive@57 68 lbl:SetFont(Amr.CreateFont("Bold", 24, Amr.Colors.TextHeaderActive))
yellowfive@57 69 lbl:SetPoint("TOPLEFT", container.content, "TOPLEFT", 0, -40)
yellowfive@57 70
yellowfive@57 71 local lbl2 = createLabel(container, L.ExportHelp1)
yellowfive@57 72 lbl2:SetPoint("TOPLEFT", lbl.frame, "BOTTOMLEFT", 0, -10)
yellowfive@57 73
yellowfive@57 74 lbl = createLabel(container, L.ExportHelp2)
yellowfive@57 75 lbl:SetPoint("TOPLEFT", lbl2.frame, "BOTTOMLEFT", 0, -10)
yellowfive@57 76
yellowfive@57 77 lbl2 = createLabel(container, L.ExportHelp3)
yellowfive@57 78 lbl2:SetPoint("TOPLEFT", lbl.frame, "BOTTOMLEFT", 0, -10)
yellowfive@57 79
yellowfive@57 80 _txt = AceGUI:Create("AmrUiTextarea")
yellowfive@57 81 _txt:SetWidth(800)
yellowfive@57 82 _txt:SetHeight(300)
yellowfive@57 83 _txt:SetFont(Amr.CreateFont("Regular", 12, Amr.Colors.Text))
yellowfive@69 84 _txt:SetCallback("OnTextChanged", onTextChanged)
yellowfive@57 85 container:AddChild(_txt)
yellowfive@124 86 _txt:SetPoint("TOP", lbl2.frame, "BOTTOM", 0, -20)
yellowfive@57 87
yellowfive@57 88 local data = self:ExportCharacter()
yellowfive@57 89 local txt = Amr.Serializer:SerializePlayerData(data, true)
yellowfive@57 90 _txt:SetText(txt)
yellowfive@57 91 _txt:SetFocus(true)
yellowfive@57 92
yellowfive@57 93 -- update shopping list data
yellowfive@57 94 Amr:UpdateShoppingData(data)
yellowfive@57 95
yellowfive@57 96 -- show help splash if first time a user is using this
yellowfive@57 97 if Amr.db.char.FirstUse then
yellowfive@57 98 Amr:ShowCover(renderSplash)
yellowfive@57 99 AceGUI:ClearFocus()
yellowfive@57 100 end
yellowfive@57 101 end
yellowfive@57 102
yellowfive@57 103 function Amr:ReleaseTabExport()
yellowfive@57 104 end
yellowfive@57 105
yellowfive@57 106 function Amr:GetExportText()
yellowfive@57 107 return _txt:GetText()
yellowfive@57 108 end
yellowfive@57 109
yellowfive@57 110
yellowfive@57 111 -- use some local variables to deal with the fact that a user can close the bank before a scan completes
yellowfive@57 112 local _lastBankBagId = nil
yellowfive@57 113 local _lastBankSlotId = nil
yellowfive@124 114 local _bankOpen = false
yellowfive@57 115
yellowfive@57 116 local function scanBag(bagId, isBank, bagTable, bagItemsWithCount)
yellowfive@57 117 local numSlots = GetContainerNumSlots(bagId)
yellowfive@124 118 local loc = ItemLocation.CreateEmpty()
yellowfive@57 119 for slotId = 1, numSlots do
yellowfive@57 120 local _, itemCount, _, _, _, _, itemLink = GetContainerItemInfo(bagId, slotId)
yellowfive@57 121 if itemLink ~= nil then
yellowfive@57 122 local itemData = Amr.Serializer.ParseItemLink(itemLink)
yellowfive@57 123 if itemData ~= nil then
yellowfive@124 124
yellowfive@124 125 -- see if this is an azerite item and read azerite power ids
yellowfive@124 126 loc:SetBagAndSlot(bagId, slotId)
yellowfive@124 127 if C_AzeriteEmpoweredItem.IsAzeriteEmpoweredItem(loc) then
yellowfive@124 128 local powers = Amr.ReadAzeritePowers(loc)
yellowfive@124 129 if powers then
yellowfive@124 130 itemData.azerite = powers
yellowfive@124 131 end
yellowfive@124 132 end
yellowfive@124 133
yellowfive@124 134 if isBank then
yellowfive@124 135 _lastBankBagId = bagId
yellowfive@124 136 _lastBankSlotId = slotId
yellowfive@124 137 end
yellowfive@57 138
yellowfive@124 139 table.insert(bagTable, itemData)
yellowfive@57 140
yellowfive@57 141 -- all items and counts, used for e.g. shopping list and reagents, etc.
yellowfive@57 142 if bagItemsWithCount then
yellowfive@57 143 if bagItemsWithCount[itemData.id] then
yellowfive@57 144 bagItemsWithCount[itemData.id] = bagItemsWithCount[itemData.id] + itemCount
yellowfive@57 145 else
yellowfive@57 146 bagItemsWithCount[itemData.id] = itemCount
yellowfive@57 147 end
yellowfive@57 148 end
yellowfive@57 149 end
yellowfive@57 150 end
yellowfive@57 151 end
yellowfive@57 152 end
yellowfive@57 153
yellowfive@124 154 -- cache the currently equipped gear for this spec
yellowfive@124 155 local function cacheEquipped()
yellowfive@124 156 local data = Amr.Serializer:GetEquipped()
yellowfive@57 157
yellowfive@124 158 local spec = GetSpecialization()
yellowfive@57 159 Amr.db.char.Equipped[spec] = data.Equipped[spec]
yellowfive@57 160 end
yellowfive@57 161
yellowfive@57 162 local function scanBags()
yellowfive@57 163
yellowfive@57 164 local bagItems = {}
yellowfive@57 165 local itemsAndCounts = {}
yellowfive@57 166
yellowfive@57 167 scanBag(BACKPACK_CONTAINER, false, bagItems, itemsAndCounts) -- backpack
yellowfive@57 168 for bagId = 1, NUM_BAG_SLOTS do
yellowfive@57 169 scanBag(bagId, false, bagItems, itemsAndCounts)
yellowfive@57 170 end
yellowfive@57 171
yellowfive@57 172 Amr.db.char.BagItems = bagItems
yellowfive@57 173 Amr.db.char.BagItemsAndCounts = itemsAndCounts
yellowfive@57 174 end
yellowfive@57 175
yellowfive@57 176 -- scan the player's bank and save the contents, must be at the bank
yellowfive@57 177 local function scanBank()
yellowfive@57 178
yellowfive@57 179 local bankItems = {}
yellowfive@57 180 local itemsAndCounts = {}
yellowfive@127 181
yellowfive@127 182 local bagList = {}
yellowfive@127 183 table.insert(bagList, BANK_CONTAINER)
yellowfive@127 184 table.insert(bagList, REAGENTBANK_CONTAINER)
yellowfive@127 185 for bagId = NUM_BAG_SLOTS + 1, NUM_BAG_SLOTS + NUM_BANKBAGSLOTS do
yellowfive@127 186 table.insert(bagList, bagId)
yellowfive@127 187 end
yellowfive@57 188
yellowfive@127 189 for i,bagId in ipairs(bagList) do
yellowfive@124 190 local bagItems = {}
yellowfive@124 191 local bagItemsAndCounts = {}
yellowfive@124 192 scanBag(bagId, true, bagItems, bagItemsAndCounts)
yellowfive@124 193
yellowfive@124 194 bankItems[bagId] = bagItems
yellowfive@124 195 itemsAndCounts[bagId] = bagItemsAndCounts
yellowfive@57 196 end
yellowfive@57 197
yellowfive@57 198 -- see if the scan completed before the window closed, otherwise we don't overwrite with partial data
yellowfive@124 199 if _bankOpen and _lastBankBagId then
yellowfive@57 200 local itemLink = GetContainerItemLink(_lastBankBagId, _lastBankSlotId)
yellowfive@124 201 if itemLink then --still open
yellowfive@57 202 Amr.db.char.BankItems = bankItems
yellowfive@57 203 Amr.db.char.BankItemsAndCounts = itemsAndCounts
yellowfive@57 204 end
yellowfive@57 205 end
yellowfive@57 206 end
yellowfive@57 207
yellowfive@124 208 local function onBankOpened()
yellowfive@124 209 _bankOpen = true
yellowfive@124 210 scanBank()
yellowfive@124 211 end
yellowfive@124 212
yellowfive@124 213 local function onBankClosed()
yellowfive@124 214 _bankOpen = false
yellowfive@124 215 end
yellowfive@124 216
yellowfive@124 217 -- if a bank bag is updated while the bank is open, re-scan that bag
yellowfive@124 218 local function onBankUpdated(bagID)
yellowfive@124 219 if _bankOpen and (bagID == BANK_CONTAINER or bagID == REAGENTBANK_CONTAINER or (bagID >= NUM_BAG_SLOTS + 1 and bagID <= NUM_BAG_SLOTS + NUM_BANKBAGSLOTS)) then
yellowfive@124 220 local bagItems = {}
yellowfive@124 221 local bagItemsAndCounts = {}
yellowfive@124 222 scanBag(bagID, true, bagItems, bagItemsAndCounts)
yellowfive@124 223
yellowfive@124 224 -- see if the scan completed before the window closed, otherwise we don't overwrite with partial data
yellowfive@124 225 if _bankOpen and _lastBankBagId == bagID then
yellowfive@124 226 local itemLink = GetContainerItemLink(_lastBankBagId, _lastBankSlotId)
yellowfive@124 227 if itemLink then
yellowfive@124 228 Amr.db.char.BankItems[bagID] = bagItems
yellowfive@124 229 Amr.db.char.BankItemsAndCounts[bagID] = bagItemsAndCounts
yellowfive@124 230 end
yellowfive@124 231 end
yellowfive@124 232 end
yellowfive@124 233 end
yellowfive@124 234
yellowfive@124 235 --[[
yellowfive@57 236 -- scan the player's void storage and save the contents, must be at void storage
yellowfive@57 237 local function scanVoid()
yellowfive@57 238
yellowfive@57 239 if IsVoidStorageReady() then
yellowfive@57 240 local voidItems = {}
yellowfive@57 241 local VOID_STORAGE_MAX = 80
yellowfive@57 242 local VOID_STORAGE_PAGES = 2
yellowfive@57 243
yellowfive@57 244 for page = 1,VOID_STORAGE_PAGES do
yellowfive@57 245 for i = 1,VOID_STORAGE_MAX do
yellowfive@57 246 local itemId = GetVoidItemInfo(page, i)
yellowfive@57 247 if itemId then
yellowfive@57 248 local itemLink = GetVoidItemHyperlinkString(((page - 1) * VOID_STORAGE_MAX) + i);
yellowfive@57 249 if itemLink then
yellowfive@57 250 tinsert(voidItems, itemLink)
yellowfive@57 251 end
yellowfive@57 252 end
yellowfive@57 253 end
yellowfive@57 254 end
yellowfive@57 255
yellowfive@57 256 Amr.db.char.VoidItems = voidItems
yellowfive@57 257 end
yellowfive@57 258
yellowfive@57 259 end
yellowfive@124 260 ]]
yellowfive@57 261
yellowfive@165 262 local function scanEssences()
yellowfive@165 263 if not C_AzeriteEssence then return end
yellowfive@165 264
yellowfive@165 265 -- read which essences this player has unlocked
yellowfive@165 266 Amr.db.char.UnlockedEssences = {}
yellowfive@165 267
yellowfive@165 268 local essences = C_AzeriteEssence.GetEssences()
yellowfive@165 269 if essences then
yellowfive@165 270 for i, essence in ipairs(essences) do
yellowfive@165 271 if essence.unlocked then
yellowfive@165 272 table.insert(Amr.db.char.UnlockedEssences, { essence.ID, essence.rank })
yellowfive@165 273 end
yellowfive@165 274 end
yellowfive@165 275 end
yellowfive@165 276
yellowfive@165 277 local specPos = GetSpecialization()
yellowfive@165 278 if not specPos or specPos < 1 or specPos > 4 then return end
yellowfive@165 279
yellowfive@165 280 if not Amr.db.char.Essences then
yellowfive@165 281 Amr.db.char.Essences = {}
yellowfive@165 282 end
yellowfive@165 283
yellowfive@165 284 Amr.db.char.Essences[specPos] = {}
yellowfive@165 285 local active = Amr.db.char.Essences[specPos]
yellowfive@165 286
yellowfive@165 287 local milestones = C_AzeriteEssence.GetMilestones()
yellowfive@165 288 if milestones then
yellowfive@165 289 for i, milestone in ipairs(milestones) do
yellowfive@165 290 -- if no slot, it corresponds to the stamina nodes, skip those
yellowfive@165 291 if milestone.slot ~= nil then
yellowfive@165 292 if milestone.unlocked then
yellowfive@165 293 local essenceId = C_AzeriteEssence.GetMilestoneEssence(milestone.ID)
yellowfive@165 294 if essenceId then
yellowfive@165 295 local essence = C_AzeriteEssence.GetEssenceInfo(essenceId)
yellowfive@165 296 table.insert(active, { milestone.slot, essence.ID, essence.rank })
yellowfive@165 297 end
yellowfive@165 298 end
yellowfive@165 299 end
yellowfive@165 300 end
yellowfive@165 301 end
yellowfive@165 302 end
yellowfive@165 303
yellowfive@81 304 local function scanTalents()
yellowfive@81 305 local specPos = GetSpecialization()
yellowfive@81 306 if not specPos or specPos < 1 or specPos > 4 then return end
yellowfive@81 307
yellowfive@81 308 local talentInfo = {}
yellowfive@81 309 local maxTiers = 7
yellowfive@81 310 for tier = 1, maxTiers do
yellowfive@81 311 for col = 1, 3 do
yellowfive@81 312 local id, name, _, _, _, spellId, _, t, c, selected = GetTalentInfoBySpecialization(specPos, tier, col)
yellowfive@81 313 if selected then
yellowfive@81 314 talentInfo[tier] = col
yellowfive@81 315 end
yellowfive@81 316 end
yellowfive@81 317 end
yellowfive@81 318
yellowfive@81 319 local str = ""
yellowfive@81 320 for i = 1, maxTiers do
yellowfive@81 321 if talentInfo[i] then
yellowfive@81 322 str = str .. talentInfo[i]
yellowfive@81 323 else
yellowfive@81 324 str = str .. '0'
yellowfive@81 325 end
yellowfive@81 326 end
yellowfive@81 327
yellowfive@81 328 Amr.db.char.Talents[specPos] = str
yellowfive@81 329 end
yellowfive@81 330
yellowfive@57 331 -- Returns a data object containing all information about the current player needed for an export:
yellowfive@57 332 -- gear, spec, reputations, bag, bank, and void storage items.
yellowfive@57 333 function Amr:ExportCharacter()
yellowfive@57 334
yellowfive@124 335 -- get all necessary player data
yellowfive@124 336 local data = Amr.Serializer:GetPlayerData()
yellowfive@124 337
yellowfive@124 338 -- cache latest-seen equipped gear for current spec
yellowfive@124 339 local spec = GetSpecialization()
yellowfive@124 340 Amr.db.char.Equipped[spec] = data.Equipped[spec]
yellowfive@124 341
yellowfive@124 342 -- scan current inventory just before export so that it is always fresh
yellowfive@57 343 scanBags()
yellowfive@57 344
yellowfive@81 345 -- scan current spec's talents just before exporting
yellowfive@81 346 scanTalents()
yellowfive@165 347
yellowfive@165 348 -- scan current spec's essences just before exporting
yellowfive@165 349 scanEssences()
yellowfive@81 350
yellowfive@124 351 data.Talents = Amr.db.char.Talents
yellowfive@165 352 data.UnlockedEssences = Amr.db.char.UnlockedEssences
yellowfive@165 353 data.Essences = Amr.db.char.Essences
yellowfive@124 354 data.Equipped = Amr.db.char.Equipped
yellowfive@57 355 data.BagItems = Amr.db.char.BagItems
yellowfive@124 356
yellowfive@124 357 -- flatten bank data (which is stored by bag for more efficient updating)
yellowfive@124 358 data.BankItems = {}
yellowfive@124 359 for k,v in pairs(Amr.db.char.BankItems) do
yellowfive@124 360 for i,v2 in ipairs(v) do
yellowfive@124 361 table.insert(data.BankItems, v2)
yellowfive@124 362 end
yellowfive@124 363 end
yellowfive@124 364
yellowfive@124 365 --data.VoidItems = Amr.db.char.VoidItems
yellowfive@57 366
yellowfive@57 367 return data
yellowfive@57 368 end
yellowfive@57 369
yellowfive@57 370 function Amr:InitializeExport()
yellowfive@57 371 Amr:AddEventHandler("UNIT_INVENTORY_CHANGED", function(unitID)
yellowfive@57 372 if unitID and unitID ~= "player" then return end
yellowfive@124 373 cacheEquipped()
yellowfive@57 374 end)
yellowfive@57 375 end
yellowfive@57 376
yellowfive@124 377 Amr:AddEventHandler("BANKFRAME_OPENED", onBankOpened)
yellowfive@124 378 Amr:AddEventHandler("BANKFRAME_CLOSED", onBankClosed)
yellowfive@124 379 Amr:AddEventHandler("BAG_UPDATE", onBankUpdated)
yellowfive@57 380
yellowfive@124 381 --Amr:AddEventHandler("VOID_STORAGE_OPEN", scanVoid)
yellowfive@124 382 --Amr:AddEventHandler("VOID_STORAGE_CONTENTS_UPDATE", scanVoid)
yellowfive@124 383 --Amr:AddEventHandler("VOID_STORAGE_DEPOSIT_UPDATE", scanVoid)
yellowfive@124 384 --Amr:AddEventHandler("VOID_STORAGE_UPDATE", scanVoid)
yellowfive@81 385
yellowfive@81 386 Amr:AddEventHandler("PLAYER_TALENT_UPDATE", scanTalents)
yellowfive@165 387
yellowfive@165 388 if C_AzeriteEssence then
yellowfive@165 389 Amr:AddEventHandler("AZERITE_ESSENCE_UPDATE", scanEssences)
yellowfive@165 390 end