Mercurial > wow > askmrrobot

annotate Shopping.lua @ 152:f33e5aaaee9e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Added tag v71 for changeset bdf1b0c98882
author yellowfive
date Thu, 03 Jan 2019 13:36:56 -0800
parents c229c759a125
children 44ce007189e0
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 _frameShop
yellowfive@57 6 local _panelContent
yellowfive@57 7 local _cboPlayers
yellowfive@139 8 local _selectedSetup
yellowfive@81 9
yellowfive@57 10 local _specs = {
yellowfive@57 11 [1] = true,
yellowfive@81 12 [2] = true,
yellowfive@81 13 [3] = true,
yellowfive@81 14 [4] = true,
yellowfive@57 15 }
yellowfive@81 16
yellowfive@57 17 local _isAhOpen = false
yellowfive@57 18
yellowfive@89 19 local function incrementTableItem(tbl, key, inc)
yellowfive@89 20 tbl[key] = tbl[key] and tbl[key] + inc or inc
yellowfive@89 21 end
yellowfive@89 22
yellowfive@57 23 local function onShopFrameClose(widget)
yellowfive@57 24 AceGUI:Release(widget)
yellowfive@57 25 _frameShop = nil
yellowfive@57 26 _cboPlayers = nil
yellowfive@57 27 _panelContent = nil
yellowfive@57 28 end
yellowfive@57 29
yellowfive@57 30 function Amr:HideShopWindow()
yellowfive@57 31 if not _frameShop then return end
yellowfive@57 32 _frameShop:Hide()
yellowfive@57 33 end
yellowfive@57 34
yellowfive@57 35 local function onPlayerChange(widget, eventName, value)
yellowfive@139 36 _selectedSetup = value
yellowfive@57 37 Amr:RefreshShoppingUi()
yellowfive@57 38 end
yellowfive@57 39
yellowfive@57 40 local function onSpecClick(widget)
yellowfive@57 41 local spec = widget:GetUserData("spec")
yellowfive@57 42 _specs[spec] = not _specs[spec]
yellowfive@57 43
yellowfive@57 44 Amr:RefreshShoppingUi()
yellowfive@57 45 end
yellowfive@57 46
yellowfive@57 47 local function onItemClick(widget)
yellowfive@57 48 local name = widget:GetUserData("itemName")
yellowfive@57 49 if name then
yellowfive@57 50 QueryAuctionItems(name)
yellowfive@57 51 end
yellowfive@57 52 end
yellowfive@57 53
yellowfive@124 54
yellowfive@57 55 function Amr:ShowShopWindow()
yellowfive@124 56
yellowfive@57 57 if not _frameShop then
yellowfive@57 58 _frameShop = AceGUI:Create("AmrUiFrame")
yellowfive@57 59 _frameShop:SetStatusTable(Amr.db.profile.shopWindow) -- window position is remembered in db
yellowfive@57 60 _frameShop:SetCallback("OnClose", onShopFrameClose)
yellowfive@57 61 _frameShop:SetLayout("None")
yellowfive@57 62 _frameShop:SetWidth(500)
yellowfive@57 63 _frameShop:SetHeight(500)
yellowfive@57 64 _frameShop:SetBorderColor(Amr.Colors.BorderBlue)
yellowfive@57 65 _frameShop:SetBackgroundColor(Amr.Colors.Bg)
yellowfive@57 66
yellowfive@61 67 if Amr.db.profile.options.uiScale ~= 1 then
yellowfive@61 68 local scale = tonumber(Amr.db.profile.options.uiScale)
yellowfive@61 69 _frameShop:SetScale(scale)
yellowfive@61 70 end
yellowfive@61 71
yellowfive@57 72 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 73 _frameShop:AddChild(lbl)
yellowfive@57 74 lbl:SetWidth(400)
yellowfive@57 75 lbl:SetFont(Amr.CreateFont("Bold", 28, Amr.Colors.White))
yellowfive@57 76 lbl:SetText(L.ShopTitle)
yellowfive@57 77 lbl:SetWordWrap(false)
yellowfive@57 78 lbl:SetJustifyH("CENTER")
yellowfive@57 79 lbl:SetPoint("TOP", _frameShop.content, "TOP", 0, 30)
yellowfive@57 80
yellowfive@57 81 lbl:SetCallback("OnMouseDown", function(widget) _frameShop:StartMove() end)
yellowfive@57 82 lbl:SetCallback("OnMouseUp", function(widget) _frameShop:EndMove() end)
yellowfive@57 83
yellowfive@57 84 -- player picker
yellowfive@57 85 _cboPlayers = AceGUI:Create("AmrUiDropDown")
yellowfive@57 86 _cboPlayers:SetWidth(400)
yellowfive@124 87 _frameShop:AddChild(_cboPlayers)
yellowfive@57 88 _cboPlayers:SetPoint("TOPLEFT", _frameShop.content, "TOPLEFT", 0, -30)
yellowfive@57 89
yellowfive@57 90 _panelContent = AceGUI:Create("AmrUiPanel")
yellowfive@57 91 _panelContent:SetLayout("None")
yellowfive@57 92 _panelContent:SetTransparent()
yellowfive@124 93 _frameShop:AddChild(_panelContent)
yellowfive@139 94 _panelContent:SetPoint("TOPLEFT", _cboPlayers.frame, "BOTTOMLEFT", 0, -10)
yellowfive@57 95 _panelContent:SetPoint("BOTTOMRIGHT", _frameShop.content, "BOTTOMRIGHT")
yellowfive@57 96
yellowfive@57 97 -- update shopping list data
yellowfive@57 98 local player = Amr:ExportCharacter()
yellowfive@57 99 Amr:UpdateShoppingData(player)
yellowfive@57 100
yellowfive@57 101 -- fill player list
yellowfive@57 102 local playerList = {}
yellowfive@139 103 local firstData = nil
yellowfive@139 104 for name, v in pairs(Amr.db.global.Shopping2) do
yellowfive@139 105 for setupName, data in pairs(v.setups) do
yellowfive@139 106 table.insert(playerList, { text = name .. " " .. setupName, value = name .. "@" .. setupName })
yellowfive@139 107 if not firstData then
yellowfive@139 108 firstData = name .. "@" .. setupName
yellowfive@139 109 end
yellowfive@139 110 end
yellowfive@57 111 end
yellowfive@57 112 _cboPlayers:SetItems(playerList)
yellowfive@57 113
yellowfive@57 114 -- set default selected player
yellowfive@139 115 local playerData = Amr.db.global.Shopping2[player.Name .. "-" .. player.Realm]
yellowfive@139 116 if playerData and playerData.setups then
yellowfive@139 117 _selectedSetup = Amr:GetActiveSetupLabel()
yellowfive@139 118 if not _selectedSetup then
yellowfive@139 119 Amr:PickFirstSetupForSpec()
yellowfive@139 120 _selectedSetup = Amr:GetActiveSetupLabel()
yellowfive@139 121 end
yellowfive@139 122 if _selectedSetup and not playerData.setups[_selectedSetup] then
yellowfive@139 123 _selectedSetup = nil
yellowfive@139 124 else
yellowfive@139 125 _selectedSetup = player.Name .. "-" .. player.Realm .. "@" .. _selectedSetup
yellowfive@139 126 end
yellowfive@139 127 end
yellowfive@139 128
yellowfive@139 129 if not _selectedSetup then
yellowfive@139 130 if playerData and playerData.setups then
yellowfive@139 131 for k,v in pairs(playerData.setups) do
yellowfive@139 132 _selectedSetup = player.Name .. "-" .. player.Realm .. "@" .. k
yellowfive@139 133 break
yellowfive@139 134 end
yellowfive@139 135 else
yellowfive@139 136 _selectedSetup = firstData
yellowfive@139 137 end
yellowfive@139 138 end
yellowfive@139 139 _cboPlayers:SelectItem(_selectedSetup)
yellowfive@57 140
yellowfive@57 141 Amr:RefreshShoppingUi()
yellowfive@57 142
yellowfive@57 143 -- set event on dropdown after UI has been initially rendered
yellowfive@57 144 _cboPlayers:SetCallback("OnChange", onPlayerChange)
yellowfive@139 145
yellowfive@139 146 -- set a timer to refresh a bit after opening b/c sometimes some item info isn't available
yellowfive@139 147 Amr.Wait(2, function()
yellowfive@139 148 if _frameShop then
yellowfive@139 149 Amr:RefreshShoppingUi()
yellowfive@139 150 end
yellowfive@139 151 end)
yellowfive@57 152 else
yellowfive@57 153 _frameShop:Show()
yellowfive@57 154 Amr:RefreshShoppingUi()
yellowfive@57 155 end
yellowfive@57 156
yellowfive@57 157 _frameShop:Raise()
yellowfive@57 158 end
yellowfive@57 159
yellowfive@57 160 -- helper to render a section of the shopping list
yellowfive@57 161 local function renderShopSection(list, scroll, header)
yellowfive@57 162 if not list or next(list) == nil then return end
yellowfive@57 163
yellowfive@57 164 local w = 440
yellowfive@57 165
yellowfive@57 166 local panel = AceGUI:Create("AmrUiPanel")
yellowfive@57 167 panel:SetLayout("None")
yellowfive@57 168 panel:SetTransparent()
yellowfive@57 169 panel:SetWidth(w)
yellowfive@57 170 panel:SetHeight(40)
yellowfive@57 171 scroll:AddChild(panel)
yellowfive@57 172
yellowfive@57 173 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 174 panel:AddChild(lbl)
yellowfive@57 175 lbl:SetWidth(w)
yellowfive@57 176 lbl:SetFont(Amr.CreateFont("Regular", 18, Amr.Colors.TextHeaderActive))
yellowfive@57 177 lbl:SetText(header)
yellowfive@57 178 lbl:SetPoint("BOTTOMLEFT", panel.content, "BOTTOMLEFT")
yellowfive@57 179
yellowfive@57 180 for itemId, count in pairs(list) do
yellowfive@57 181 panel = AceGUI:Create("AmrUiPanel")
yellowfive@57 182 panel:SetLayout("None")
yellowfive@57 183 panel:SetTransparent()
yellowfive@57 184 panel:SetWidth(w)
yellowfive@57 185 panel:SetHeight(30)
yellowfive@57 186 scroll:AddChild(panel)
yellowfive@57 187
yellowfive@57 188 lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 189 panel:AddChild(lbl)
yellowfive@89 190 lbl:SetWidth(40)
yellowfive@57 191 lbl:SetWordWrap(false)
yellowfive@57 192 lbl:SetFont(Amr.CreateFont("Bold", 20, Amr.Colors.White))
yellowfive@57 193 lbl:SetText(count .. "x")
yellowfive@57 194 lbl:SetPoint("LEFT", panel.content, "LEFT")
yellowfive@57 195
yellowfive@57 196 local icon = AceGUI:Create("AmrUiIcon")
yellowfive@57 197 icon:SetBorderWidth(1)
yellowfive@57 198 icon:SetIconBorderColor(Amr.Colors.White)
yellowfive@57 199 icon:SetWidth(18)
yellowfive@57 200 icon:SetHeight(18)
yellowfive@124 201 panel:AddChild(icon)
yellowfive@57 202 icon:SetPoint("LEFT", lbl.frame, "RIGHT", 5, 0)
yellowfive@57 203
yellowfive@57 204 local btn = AceGUI:Create("AmrUiTextButton")
yellowfive@57 205 btn:SetWidth(w - 30 - 18 - 15)
yellowfive@57 206 btn:SetJustifyH("LEFT")
yellowfive@57 207 btn:SetWordWrap(false)
yellowfive@57 208 btn:SetFont(Amr.CreateFont("Bold", 14, Amr.Colors.White))
yellowfive@57 209 btn:SetHoverFont(Amr.CreateFont("Bold", 14, Amr.Colors.White))
yellowfive@57 210 btn:SetCallback("OnClick", onItemClick)
yellowfive@57 211 panel:AddChild(btn)
yellowfive@124 212 btn:SetPoint("LEFT", icon.frame, "RIGHT", 5, 0)
yellowfive@57 213
yellowfive@131 214 local item = Item:CreateFromItemID(itemId)
yellowfive@131 215 if item then
yellowfive@131 216 local itemLink = item:GetItemLink()
yellowfive@131 217 if itemLink then
yellowfive@131 218 icon:SetIcon(item:GetItemIcon())
yellowfive@131 219 btn:SetText(itemLink:gsub("%[", ""):gsub("%]", ""))
yellowfive@131 220 btn:SetUserData("itemName", item:GetItemName())
yellowfive@131 221 Amr:SetItemTooltip(btn, itemLink)
yellowfive@131 222 end
yellowfive@131 223 end
yellowfive@57 224 end
yellowfive@57 225
yellowfive@57 226 end
yellowfive@57 227
yellowfive@89 228 -- get the number of a specified gem/enchant/material that the player currently owns
yellowfive@57 229 local function getOwnedCount(itemId)
yellowfive@57 230 local ret = 0
yellowfive@57 231
yellowfive@57 232 local list = Amr.db.char.BagItemsAndCounts
yellowfive@57 233 if list and list[itemId] then
yellowfive@57 234 ret = ret + list[itemId]
yellowfive@57 235 end
yellowfive@57 236
yellowfive@124 237 local bankBags = Amr.db.char.BankItemsAndCounts
yellowfive@124 238 if bankBags then
yellowfive@124 239 for bagId,bagList in pairs(bankBags) do
yellowfive@124 240 if bagList[itemId] then
yellowfive@124 241 ret = ret + bagList[itemId]
yellowfive@124 242 end
yellowfive@124 243 end
yellowfive@57 244 end
yellowfive@57 245
yellowfive@57 246 return ret
yellowfive@57 247 end
yellowfive@57 248
yellowfive@89 249 local function removeOwned(list, owned)
yellowfive@89 250
yellowfive@89 251 for itemId, count in pairs(list) do
yellowfive@89 252 -- load up how many of an item we have
yellowfive@89 253 if not owned.loaded[itemId] then
yellowfive@89 254 owned.counts[itemId] = getOwnedCount(itemId)
yellowfive@89 255 owned.loaded[itemId] = true
yellowfive@89 256 end
yellowfive@89 257
yellowfive@89 258 -- see how many we can remove from the required count
yellowfive@89 259 local used = math.min(owned.counts[itemId], count)
yellowfive@89 260
yellowfive@89 261 -- update owned count so we can't double-use something
yellowfive@89 262 owned.counts[itemId] = owned.counts[itemId] - used;
yellowfive@89 263
yellowfive@89 264 -- reduce the requirement, removing entirely if we have it completely covered
yellowfive@89 265 list[itemId] = list[itemId] - used;
yellowfive@89 266 if list[itemId] == 0 then
yellowfive@89 267 list[itemId] = nil
yellowfive@89 268 end
yellowfive@89 269 end
yellowfive@89 270 end
yellowfive@89 271
yellowfive@89 272 function Amr:RefreshShoppingUi()
yellowfive@89 273
yellowfive@89 274 -- clear out any previous data
yellowfive@89 275 _panelContent:ReleaseChildren()
yellowfive@89 276
yellowfive@139 277 local parts = { strsplit("@", _selectedSetup) }
yellowfive@139 278
yellowfive@139 279 local data = Amr.db.global.Shopping2[parts[1]].setups[parts[2]]
yellowfive@89 280 if not data then
yellowfive@89 281 _panelContent:SetLayout("None")
yellowfive@89 282
yellowfive@89 283 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 284 _panelContent:AddChild(lbl)
yellowfive@89 285 lbl:SetFont(Amr.CreateFont("Italic", 18, Amr.Colors.TextTan))
yellowfive@89 286 lbl:SetText(L.ShopEmpty)
yellowfive@89 287 lbl:SetJustifyH("CENTER")
yellowfive@89 288 lbl:SetPoint("TOP", _panelContent.content, "TOP", 0, -30)
yellowfive@89 289 else
yellowfive@89 290 local allStuff = { gems = {}, enchants = {}, materials = {} }
yellowfive@89 291 local hasStuff = false
yellowfive@89 292 local visited = {}
yellowfive@89 293
yellowfive@139 294 for inventoryId, stuff in pairs(data) do
yellowfive@139 295 hasStuff = true
yellowfive@139 296 if not visited[inventoryId] then
yellowfive@139 297 if stuff.gems then
yellowfive@139 298 for itemId, count in pairs(stuff.gems) do
yellowfive@139 299 incrementTableItem(allStuff.gems, itemId, count)
yellowfive@89 300 end
yellowfive@89 301 end
yellowfive@89 302
yellowfive@139 303 if stuff.enchants then
yellowfive@139 304 for itemId, count in pairs(stuff.enchants) do
yellowfive@139 305 incrementTableItem(allStuff.enchants, itemId, count)
yellowfive@139 306 end
yellowfive@139 307 end
yellowfive@139 308
yellowfive@139 309 if stuff.materials then
yellowfive@139 310 for itemId, count in pairs(stuff.materials) do
yellowfive@139 311 incrementTableItem(allStuff.materials, itemId, count)
yellowfive@139 312 end
yellowfive@139 313 end
yellowfive@139 314
yellowfive@139 315 -- make sure not to count the same physical item twice
yellowfive@139 316 if inventoryId ~= -1 then
yellowfive@139 317 visited[inventoryId] = true
yellowfive@139 318 end
yellowfive@89 319 end
yellowfive@89 320 end
yellowfive@89 321
yellowfive@89 322 if hasStuff then
yellowfive@89 323 -- remove what we already own
yellowfive@89 324 local owned = { counts = {}, loaded = {} }
yellowfive@89 325 removeOwned(allStuff.gems, owned)
yellowfive@89 326 removeOwned(allStuff.enchants, owned)
yellowfive@89 327 removeOwned(allStuff.materials, owned)
yellowfive@89 328
yellowfive@89 329 _panelContent:SetLayout("Fill")
yellowfive@89 330
yellowfive@89 331 local scroll = AceGUI:Create("AmrUiScrollFrame")
yellowfive@89 332 scroll:SetLayout("List")
yellowfive@89 333 _panelContent:AddChild(scroll)
yellowfive@89 334
yellowfive@89 335 renderShopSection(allStuff.gems, scroll, L.ShopHeaderGems)
yellowfive@89 336 renderShopSection(allStuff.enchants, scroll, L.ShopHeaderEnchants)
yellowfive@89 337 renderShopSection(allStuff.materials, scroll, L.ShopHeaderMaterials)
yellowfive@89 338 end
yellowfive@89 339 end
yellowfive@89 340
yellowfive@89 341 end
yellowfive@89 342
yellowfive@124 343 -- compare gear to everything the player owns, and return the minimum gems/enchants/materials needed to optimize,
yellowfive@124 344 -- grouped by inventory ID so that we can combine multiple specs without double-counting
yellowfive@131 345 local function getShoppingData(player, gear)
yellowfive@89 346
yellowfive@89 347 local ret = {}
yellowfive@89 348
yellowfive@89 349 -- used to prevent considering the same item twice
yellowfive@89 350 local usedItems = {}
yellowfive@89 351
yellowfive@89 352 for slotId, optimalItem in pairs(gear) do
yellowfive@124 353 local matchItem = Amr:FindMatchingItem(optimalItem, player, usedItems)
yellowfive@89 354 local inventoryId = optimalItem.inventoryId or -1
yellowfive@89 355
yellowfive@89 356 -- find gem/enchant differences on the best-matching item
yellowfive@89 357
yellowfive@124 358 -- gems
yellowfive@124 359 if not optimalItem.relicBonusIds and (not matchItem or not matchItem.relicBonusIds) then
yellowfive@124 360 for i = 1, 3 do
yellowfive@89 361 local g = optimalItem.gemIds[i]
yellowfive@131 362 local isGemEquipped = g == 0 or (matchItem and matchItem.gemIds and matchItem.gemIds[i] == g)
yellowfive@89 363 if not isGemEquipped then
yellowfive@89 364 if not ret[inventoryId] then
yellowfive@89 365 ret[inventoryId] = { gems = {}, enchants = {}, materials = {} }
yellowfive@89 366 end
yellowfive@89 367 incrementTableItem(ret[inventoryId].gems, g, 1)
yellowfive@89 368 end
yellowfive@89 369 end
yellowfive@89 370 end
yellowfive@89 371
yellowfive@89 372 -- enchant
yellowfive@89 373 if optimalItem.enchantId and optimalItem.enchantId ~= 0 then
yellowfive@89 374 local e = optimalItem.enchantId
yellowfive@89 375 local isEnchantEquipped = matchItem and matchItem.enchantId and matchItem.enchantId == e
yellowfive@89 376
yellowfive@89 377 if not isEnchantEquipped then
yellowfive@124 378 -- enchant info
yellowfive@124 379 local enchInfo = Amr.db.char.ExtraEnchantData[e]
yellowfive@89 380 if enchInfo then
yellowfive@89 381 if not ret[inventoryId] then
yellowfive@89 382 ret[inventoryId] = { gems = {}, enchants = {}, materials = {} }
yellowfive@89 383 end
yellowfive@89 384 incrementTableItem(ret[inventoryId].enchants, enchInfo.itemId, 1)
yellowfive@89 385
yellowfive@89 386 if enchInfo.materials then
yellowfive@89 387 for k, v in pairs(enchInfo.materials) do
yellowfive@89 388 incrementTableItem(ret[inventoryId].materials, k, v)
yellowfive@89 389 end
yellowfive@89 390 end
yellowfive@89 391 end
yellowfive@89 392 end
yellowfive@89 393 end
yellowfive@89 394 end
yellowfive@89 395
yellowfive@89 396 return ret
yellowfive@89 397 end
yellowfive@89 398
yellowfive@139 399 -- look at all gear sets and find stuff that a player needs to acquire to gem/enchant their gear for each setup
yellowfive@57 400 function Amr:UpdateShoppingData(player)
yellowfive@57 401
yellowfive@57 402 local required = {
yellowfive@139 403 setups = {}
yellowfive@57 404 }
yellowfive@139 405
yellowfive@139 406 for i, setup in ipairs(Amr.db.char.GearSetups) do
yellowfive@139 407 local gear = setup.Gear
yellowfive@124 408 if gear then
yellowfive@139 409 required.setups[setup.Label] = getShoppingData(player, gear)
yellowfive@124 410 end
yellowfive@57 411 end
yellowfive@139 412
yellowfive@139 413 Amr.db.global.Shopping2[player.Name .. "-" .. player.Realm] = required
yellowfive@57 414 end
yellowfive@57 415
yellowfive@57 416 Amr:AddEventHandler("AUCTION_HOUSE_SHOW", function()
yellowfive@57 417 _isAhOpen = true
yellowfive@57 418 if Amr.db.profile.options.shopAh then
yellowfive@57 419 Amr:ShowShopWindow()
yellowfive@57 420 end
yellowfive@57 421 end)
yellowfive@57 422
yellowfive@57 423 Amr:AddEventHandler("AUCTION_HOUSE_CLOSED", function()
yellowfive@57 424 _isAhOpen = false
yellowfive@57 425 if Amr.db.profile.options.shopAh then
yellowfive@57 426 Amr:HideShopWindow()
yellowfive@57 427 end
yellowfive@122 428 end)