Mercurial > wow > askmrrobot

annotate Import.lua @ 125:4d175fb190f1 v58

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Updated for 8.0 pre-patch and BfA.
author yellowfive
date Tue, 17 Jul 2018 09:58:51 -0700
parents e31b02b24488
children a0894ffebd15
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 _txtImport
yellowfive@57 6 local _lblError
yellowfive@69 7 local _panelCover
yellowfive@57 8
yellowfive@57 9 local function onImportOkClick(widget)
yellowfive@57 10 local txt = _txtImport:GetText()
yellowfive@57 11 local msg = Amr:ImportCharacter(txt)
yellowfive@57 12 if msg then
yellowfive@57 13 _lblError:SetText(msg)
yellowfive@57 14 _txtImport:SetFocus(true)
yellowfive@57 15 else
yellowfive@57 16 Amr:HideCover()
yellowfive@57 17 Amr:RefreshGearTab()
yellowfive@57 18 end
yellowfive@57 19 end
yellowfive@57 20
yellowfive@57 21 local function onImportCancelClick(widget)
yellowfive@57 22 Amr:HideCover()
yellowfive@57 23 end
yellowfive@57 24
yellowfive@69 25 local function onTextEnterPressed(widget)
yellowfive@69 26 -- hide the overwolf cover when import data is received
yellowfive@69 27 if _panelCover then
yellowfive@69 28 _panelCover:SetVisible(false)
yellowfive@69 29 end
yellowfive@69 30
yellowfive@69 31 -- do an import if the data starts and ends with a dollar sign
yellowfive@69 32 local txt = _txtImport:GetText()
yellowfive@69 33 local txtLen = string.len(txt)
yellowfive@69 34 if txtLen > 2 and string.sub(txt, 1, 1) == '$' then
yellowfive@69 35 onImportOkClick()
yellowfive@69 36 end
yellowfive@69 37
yellowfive@69 38 end
yellowfive@69 39
yellowfive@69 40 local function renderImportWindow(container, fromOverwolf)
yellowfive@57 41
yellowfive@57 42 local panelImport = Amr:RenderCoverChrome(container, 700, 450)
yellowfive@57 43
yellowfive@57 44 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 45 panelImport:AddChild(lbl)
yellowfive@57 46 lbl:SetWidth(600)
yellowfive@57 47 lbl:SetText(L.ImportHeader)
yellowfive@57 48 lbl:SetPoint("TOP", panelImport.content, "TOP", 0, -10)
yellowfive@57 49
yellowfive@57 50 _txtImport = AceGUI:Create("AmrUiTextarea")
yellowfive@57 51 _txtImport:SetWidth(600)
yellowfive@57 52 _txtImport:SetHeight(300)
yellowfive@57 53 _txtImport:SetFont(Amr.CreateFont("Regular", 12, Amr.Colors.Text))
yellowfive@69 54 _txtImport:SetCallback("OnEnterPressed", onTextEnterPressed)
yellowfive@57 55 panelImport:AddChild(_txtImport)
yellowfive@124 56 _txtImport:SetPoint("TOP", lbl.frame, "BOTTOM", 0, -10)
yellowfive@57 57
yellowfive@57 58 local btnImportOk = AceGUI:Create("AmrUiButton")
yellowfive@57 59 btnImportOk:SetText(L.ImportButtonOk)
yellowfive@57 60 btnImportOk:SetBackgroundColor(Amr.Colors.Green)
yellowfive@57 61 btnImportOk:SetFont(Amr.CreateFont("Bold", 16, Amr.Colors.White))
yellowfive@57 62 btnImportOk:SetWidth(120)
yellowfive@57 63 btnImportOk:SetHeight(28)
yellowfive@57 64 btnImportOk:SetCallback("OnClick", onImportOkClick)
yellowfive@57 65 panelImport:AddChild(btnImportOk)
yellowfive@124 66 btnImportOk:SetPoint("TOPLEFT", _txtImport.frame, "BOTTOMLEFT", 0, -10)
yellowfive@57 67
yellowfive@57 68 local btnImportCancel = AceGUI:Create("AmrUiButton")
yellowfive@57 69 btnImportCancel:SetText(L.ImportButtonCancel)
yellowfive@57 70 btnImportCancel:SetBackgroundColor(Amr.Colors.Green)
yellowfive@57 71 btnImportCancel:SetFont(Amr.CreateFont("Bold", 16, Amr.Colors.White))
yellowfive@57 72 btnImportCancel:SetWidth(120)
yellowfive@57 73 btnImportCancel:SetHeight(28)
yellowfive@57 74 btnImportCancel:SetCallback("OnClick", onImportCancelClick)
yellowfive@57 75 panelImport:AddChild(btnImportCancel)
yellowfive@124 76 btnImportCancel:SetPoint("LEFT", btnImportOk.frame, "RIGHT", 20, 0)
yellowfive@57 77
yellowfive@57 78 _lblError = AceGUI:Create("AmrUiLabel")
yellowfive@124 79 panelImport:AddChild(_lblError)
yellowfive@57 80 _lblError:SetWidth(600)
yellowfive@57 81 _lblError:SetFont(Amr.CreateFont("Bold", 14, Amr.Colors.Red))
yellowfive@57 82 _lblError:SetText("")
yellowfive@57 83 _lblError:SetPoint("TOPLEFT", btnImportOk.frame, "BOTTOMLEFT", 0, -20)
yellowfive@57 84
yellowfive@69 85 if fromOverwolf then
yellowfive@69 86 -- show a cover preventing interaction until we receive data from overwolf
yellowfive@69 87 _panelCover = AceGUI:Create("AmrUiPanel")
yellowfive@69 88 _panelCover:SetLayout("None")
yellowfive@69 89 _panelCover:EnableMouse(true)
yellowfive@69 90 _panelCover:SetBackgroundColor(Amr.Colors.Black, 0.75)
yellowfive@124 91 panelImport:AddChild(_panelCover)
yellowfive@69 92 _panelCover:SetPoint("TOPLEFT", panelImport.frame, "TOPLEFT")
yellowfive@69 93 _panelCover:SetPoint("BOTTOMRIGHT", panelImport.frame, "BOTTOMRIGHT")
yellowfive@69 94
yellowfive@69 95 local coverMsg = AceGUI:Create("AmrUiLabel")
yellowfive@124 96 _panelCover:AddChild(coverMsg)
yellowfive@69 97 coverMsg:SetWidth(500)
yellowfive@69 98 coverMsg:SetFont(Amr.CreateFont("Regular", 16, Amr.Colors.TextTan))
yellowfive@69 99 coverMsg:SetJustifyH("MIDDLE")
yellowfive@69 100 coverMsg:SetJustifyV("MIDDLE")
yellowfive@69 101 coverMsg:SetText(L.ImportOverwolfWait)
yellowfive@69 102 coverMsg:SetPoint("CENTER", _panelCover.frame, "CENTER", 0, 20)
yellowfive@69 103
yellowfive@69 104 -- after adding, set cover to sit on top of everything
yellowfive@69 105 _panelCover:SetStrata("FULLSCREEN_DIALOG")
yellowfive@69 106 _panelCover:SetLevel(Amr.FrameLevels.Highest)
yellowfive@69 107 end
yellowfive@57 108 end
yellowfive@57 109
yellowfive@69 110 function Amr:ShowImportWindow(fromOverwolf)
yellowfive@57 111 -- this is shown as a modal dialog
yellowfive@69 112 Amr:ShowCover(function(container)
yellowfive@69 113 renderImportWindow(container, fromOverwolf)
yellowfive@69 114 end)
yellowfive@57 115
yellowfive@57 116 _txtImport:SetText("")
yellowfive@57 117 _txtImport:SetFocus(true)
yellowfive@57 118 end
yellowfive@57 119
yellowfive@57 120 ----------------------------------------------------------------------------
yellowfive@57 121 -- Import Parsing
yellowfive@57 122 ----------------------------------------------------------------------------
yellowfive@57 123
yellowfive@57 124 --
yellowfive@57 125 -- Import a character, returning nil on success, otherwise an error message, import result stored in the db.
yellowfive@57 126 --
yellowfive@57 127 function Amr:ImportCharacter(data, isTest)
yellowfive@57 128
yellowfive@57 129 -- make sure all data is up to date before importing and get a local copy of player's current state
yellowfive@57 130 local currentPlayerData = self:ExportCharacter()
yellowfive@57 131
yellowfive@57 132 if data == nil or string.len(data) == 0 then
yellowfive@57 133 return L.ImportErrorEmpty
yellowfive@57 134 end
yellowfive@57 135
yellowfive@57 136 -- if multiple specs are included in the data, parse each individually, then quit
yellowfive@57 137 local specParts = { strsplit("\n", data) }
yellowfive@57 138 if #specParts > 1 then
yellowfive@57 139 for i = 1, #specParts do
yellowfive@57 140 local err = self:ImportCharacter(specParts[i], isTest)
yellowfive@57 141 if err ~= nil then
yellowfive@57 142 return err
yellowfive@57 143 end
yellowfive@57 144 end
yellowfive@57 145 return
yellowfive@57 146 end
yellowfive@57 147
yellowfive@57 148 local data1 = { strsplit("$", data) }
yellowfive@57 149 if #data1 ~= 3 then
yellowfive@57 150 return L.ImportErrorFormat
yellowfive@57 151 end
yellowfive@57 152
yellowfive@57 153 local parts = { strsplit(";", data1[2]) }
yellowfive@57 154
yellowfive@57 155 -- require a minimum version
yellowfive@57 156 local ver = tonumber(parts[1])
yellowfive@57 157 if ver < Amr.MIN_IMPORT_VERSION then
yellowfive@57 158 return L.ImportErrorVersion
yellowfive@57 159 end
yellowfive@57 160
yellowfive@57 161 -- require name match (don't match realm due to language issues for now)
yellowfive@57 162 if not isTest then
yellowfive@57 163 local region = parts[2]
yellowfive@57 164 local realm = parts[3]
yellowfive@57 165 local name = parts[4]
yellowfive@57 166 if name ~= currentPlayerData.Name then
yellowfive@57 167 local importPlayerName = name .. " (" .. realm .. ")"
yellowfive@57 168 local you = currentPlayerData.Name .. " (" .. currentPlayerData.Realm .. ")"
yellowfive@57 169 return L.ImportErrorChar(importPlayerName, you)
yellowfive@57 170 end
yellowfive@57 171
yellowfive@57 172 -- require race match
yellowfive@57 173 local race = tonumber(parts[6])
yellowfive@57 174 if race ~= Amr.RaceIds[currentPlayerData.Race] then
yellowfive@57 175 return L.ImportErrorRace
yellowfive@57 176 end
yellowfive@57 177
yellowfive@57 178 -- require faction match
yellowfive@57 179 local faction = tonumber(parts[7])
yellowfive@57 180 if faction ~= Amr.FactionIds[currentPlayerData.Faction] then
yellowfive@57 181 return L.ImportErrorFaction
yellowfive@57 182 end
yellowfive@57 183
yellowfive@57 184 -- require level match
yellowfive@57 185 local level = tonumber(parts[8])
yellowfive@57 186 if level ~= currentPlayerData.Level then
yellowfive@57 187 return L.ImportErrorLevel
yellowfive@57 188 end
yellowfive@57 189 end
yellowfive@57 190
yellowfive@57 191 -- if we make it this far, the data is valid, so read item information
yellowfive@124 192 local specSlot = tonumber(parts[11])
yellowfive@57 193
yellowfive@57 194 local importData = {}
yellowfive@57 195 local enchantInfo = {}
yellowfive@57 196
yellowfive@57 197 local prevItemId = 0
yellowfive@57 198 local prevGemId = 0
yellowfive@57 199 local prevEnchantId = 0
yellowfive@57 200 local prevUpgradeId = 0
yellowfive@57 201 local prevBonusId = 0
yellowfive@124 202 local prevLevel = 0
yellowfive@124 203 local prevAzeriteId = 0
yellowfive@57 204 local digits = {
yellowfive@57 205 ["-"] = true,
yellowfive@57 206 ["0"] = true,
yellowfive@57 207 ["1"] = true,
yellowfive@57 208 ["2"] = true,
yellowfive@57 209 ["3"] = true,
yellowfive@57 210 ["4"] = true,
yellowfive@57 211 ["5"] = true,
yellowfive@57 212 ["6"] = true,
yellowfive@57 213 ["7"] = true,
yellowfive@57 214 ["8"] = true,
yellowfive@57 215 ["9"] = true,
yellowfive@57 216 }
yellowfive@124 217 for i = 16, #parts do
yellowfive@57 218 local itemString = parts[i]
yellowfive@57 219 if itemString ~= "" and itemString ~= "_" then
yellowfive@57 220 local tokens = {}
yellowfive@57 221 local bonusIds = {}
yellowfive@124 222 local azerite = {}
yellowfive@57 223 local hasBonuses = false
yellowfive@124 224 local hasAzerites = false
yellowfive@57 225 local token = ""
yellowfive@57 226 local prop = "i"
yellowfive@57 227 local tokenComplete = false
yellowfive@57 228 for j = 1, string.len(itemString) do
yellowfive@57 229 local c = string.sub(itemString, j, j)
yellowfive@57 230 if digits[c] == nil then
yellowfive@57 231 tokenComplete = true
yellowfive@57 232 else
yellowfive@57 233 token = token .. c
yellowfive@57 234 end
yellowfive@57 235
yellowfive@57 236 if tokenComplete or j == string.len(itemString) then
yellowfive@57 237 local val = tonumber(token)
yellowfive@57 238 if prop == "i" then
yellowfive@57 239 val = val + prevItemId
yellowfive@57 240 prevItemId = val
yellowfive@57 241 elseif prop == "u" then
yellowfive@57 242 val = val + prevUpgradeId
yellowfive@57 243 prevUpgradeId = val
yellowfive@81 244 elseif prop == "v" then
yellowfive@81 245 val = val + prevLevel
yellowfive@81 246 prevLevel = val
yellowfive@57 247 elseif prop == "b" then
yellowfive@57 248 val = val + prevBonusId
yellowfive@57 249 prevBonusId = val
yellowfive@57 250 elseif prop == "x" or prop == "y" or prop == "z" then
yellowfive@57 251 val = val + prevGemId
yellowfive@57 252 prevGemId = val
yellowfive@57 253 elseif prop == "e" then
yellowfive@57 254 val = val + prevEnchantId
yellowfive@57 255 prevEnchantId = val
yellowfive@124 256 elseif prop == "a" then
yellowfive@124 257 val = val + prevAzeriteId
yellowfive@124 258 prevAzeriteId = val
yellowfive@57 259 end
yellowfive@57 260
yellowfive@57 261 if prop == "b" then
yellowfive@57 262 table.insert(bonusIds, val)
yellowfive@57 263 hasBonuses = true
yellowfive@124 264 elseif prop == "a" then
yellowfive@124 265 table.insert(azerite, val)
yellowfive@124 266 hasAzerites = true
yellowfive@57 267 else
yellowfive@57 268 tokens[prop] = val
yellowfive@57 269 end
yellowfive@57 270
yellowfive@57 271 token = ""
yellowfive@57 272 tokenComplete = false
yellowfive@57 273
yellowfive@57 274 -- we have moved on to the next token
yellowfive@57 275 prop = c
yellowfive@57 276 end
yellowfive@57 277 end
yellowfive@57 278
yellowfive@57 279 local obj = {}
yellowfive@57 280 importData[tonumber(tokens["s"])] = obj
yellowfive@57 281
yellowfive@57 282 obj.id = tokens["i"]
yellowfive@57 283 obj.suffixId = tokens["f"] or 0
yellowfive@57 284 obj.upgradeId = tokens["u"] or 0
yellowfive@81 285 obj.level = tokens["v"] or 0
yellowfive@57 286 obj.enchantId = tokens["e"] or 0
yellowfive@89 287 obj.inventoryId = tokens["t"] or 0
yellowfive@57 288
yellowfive@57 289 obj.gemIds = {}
yellowfive@57 290 table.insert(obj.gemIds, tokens["x"] or 0)
yellowfive@57 291 table.insert(obj.gemIds, tokens["y"] or 0)
yellowfive@57 292 table.insert(obj.gemIds, tokens["z"] or 0)
yellowfive@57 293 table.insert(obj.gemIds, 0)
yellowfive@57 294
yellowfive@57 295 if hasBonuses then
yellowfive@57 296 obj.bonusIds = bonusIds
yellowfive@57 297 end
yellowfive@57 298
yellowfive@124 299 if hasAzerites then
yellowfive@124 300 obj.azerite = azerite
yellowfive@57 301 end
yellowfive@57 302 end
yellowfive@57 303 end
yellowfive@57 304
yellowfive@57 305 -- now read any extra display information
yellowfive@57 306 parts = { strsplit("@", data1[3]) }
yellowfive@57 307 for i = 1, #parts do
yellowfive@57 308 local infoParts = { strsplit("\\", parts[i]) }
yellowfive@57 309
yellowfive@124 310 if infoParts[1] == "e" then
yellowfive@57 311
yellowfive@57 312 local enchObj = {}
yellowfive@57 313 enchObj.id = tonumber(infoParts[2])
yellowfive@57 314 enchObj.itemId = tonumber(infoParts[3])
yellowfive@57 315 enchObj.spellId = tonumber(infoParts[4])
yellowfive@57 316 enchObj.text = string.gsub(infoParts[5], "_(%a+)_", function(s) return L.StatsShort[s] end)
yellowfive@57 317
yellowfive@57 318 local mats = infoParts[6]
yellowfive@57 319 if string.len(mats) > 0 then
yellowfive@57 320 enchObj.materials = {}
yellowfive@57 321 mats = { strsplit(",", mats) }
yellowfive@57 322 for j = 1, #mats do
yellowfive@57 323 local kv = { strsplit("=", mats[j]) }
yellowfive@57 324 enchObj.materials[tonumber(kv[1])] = tonumber(kv[2])
yellowfive@57 325 end
yellowfive@57 326 end
yellowfive@57 327
yellowfive@124 328 enchantInfo[enchObj.id] = enchObj
yellowfive@57 329 end
yellowfive@57 330 end
yellowfive@57 331
yellowfive@57 332 if isTest then
yellowfive@57 333 print("spec " .. specSlot)
yellowfive@57 334 -- print result for debugging
yellowfive@57 335 for k,v in pairs(importData) do
yellowfive@57 336 local blah = Amr.CreateItemLink(v)
yellowfive@57 337 --print(blah)
yellowfive@57 338 local name, link = GetItemInfo(blah)
yellowfive@57 339 print(link)
yellowfive@57 340 if link == nil then
yellowfive@57 341 print(blah)
yellowfive@57 342 print("bad item: " .. v.id)
yellowfive@57 343 end
yellowfive@124 344 end
yellowfive@57 345 else
yellowfive@57 346 -- we have succeeded, record the result
yellowfive@57 347 Amr.db.char.GearSets[specSlot] = importData
yellowfive@124 348
yellowfive@124 349 for k,v in pairs(enchantInfo) do
yellowfive@124 350 Amr.db.char.ExtraEnchantData[k] = v
yellowfive@124 351 end
yellowfive@57 352
yellowfive@57 353 -- also update shopping list after import
yellowfive@57 354 Amr:UpdateShoppingData(currentPlayerData)
yellowfive@57 355 end
yellowfive@57 356 end