Mercurial > wow > askmrrobot

annotate Core.lua @ 163:f27a7c64b21f v76

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tweaks to item matching in junk list features.
author yellowfive
date Mon, 06 May 2019 18:05:16 -0700
parents 35612aee8e15
children 3be9cc6f7d20
rev   line source
yellowfive@57 1 AskMrRobot = LibStub("AceAddon-3.0"):NewAddon("AskMrRobot", "AceEvent-3.0", "AceComm-3.0", "AceConsole-3.0", "AceSerializer-3.0")
yellowfive@57 2 local Amr = AskMrRobot
yellowfive@57 3 Amr.Serializer = LibStub("AskMrRobot-Serializer")
yellowfive@57 4
yellowfive@57 5 Amr.ADDON_NAME = "AskMrRobot"
yellowfive@57 6
yellowfive@57 7 -- types of inter-addon messages that we receive, used to parcel them out to the proper handlers
yellowfive@57 8 Amr.MessageTypes = {
yellowfive@57 9 Version = "_V",
yellowfive@124 10 VersionRequest = "_VR"
yellowfive@57 11 }
yellowfive@57 12
yellowfive@57 13 local L = LibStub("AceLocale-3.0"):GetLocale("AskMrRobot", true)
yellowfive@57 14 local AceGUI = LibStub("AceGUI-3.0")
yellowfive@57 15
yellowfive@57 16 -- minimap icon and LDB support
yellowfive@57 17 local _amrLDB = LibStub("LibDataBroker-1.1"):NewDataObject(Amr.ADDON_NAME, {
yellowfive@57 18 type = "launcher",
yellowfive@57 19 text = "Ask Mr. Robot",
yellowfive@57 20 icon = "Interface\\AddOns\\" .. Amr.ADDON_NAME .. "\\Media\\icon",
yellowfive@57 21 OnClick = function(self, button, down)
yellowfive@57 22 if button == "LeftButton" then
yellowfive@124 23 --if IsControlKeyDown() then
yellowfive@124 24 -- Amr:Wipe()
yellowfive@124 25 --else
yellowfive@57 26 Amr:Toggle()
yellowfive@124 27 --end
yellowfive@57 28 elseif button == "RightButton" then
yellowfive@57 29 Amr:EquipGearSet()
yellowfive@57 30 end
yellowfive@57 31 end,
yellowfive@57 32 OnTooltipShow = function(tt)
yellowfive@57 33 tt:AddLine("Ask Mr. Robot", 1, 1, 1);
yellowfive@57 34 tt:AddLine(" ");
yellowfive@57 35 tt:AddLine(L.MinimapTooltip)
yellowfive@57 36 end
yellowfive@57 37 })
yellowfive@57 38 local _icon = LibStub("LibDBIcon-1.0")
yellowfive@57 39
yellowfive@57 40
yellowfive@57 41 -- initialize the database
yellowfive@57 42 local function initializeDb()
yellowfive@57 43
yellowfive@57 44 local defaults = {
yellowfive@139 45 char = {
yellowfive@139 46 LastVersion = 0, -- used to clean out old stuff
yellowfive@129 47 FirstUse = true, -- true if this is first time use, gets cleared after seeing the export help splash window
yellowfive@129 48 Talents = {}, -- for each spec, selected talents
yellowfive@129 49 Equipped = {}, -- for each spec, slot id to item info
yellowfive@129 50 BagItems = {}, -- list of item info for bags
yellowfive@129 51 BankItems = {}, -- list of item info for bank
yellowfive@129 52 BagItemsAndCounts = {}, -- used mainly for the shopping list
yellowfive@129 53 BankItemsAndCounts = {}, -- used mainly for the shopping list
yellowfive@139 54 GearSetups = {}, -- imported gear sets
yellowfive@161 55 JunkData = {}, -- imported data about items that can be vendored/scrapped/disenchanted
yellowfive@129 56 ExtraEnchantData = {}, -- enchant id to enchant display information and material information
yellowfive@129 57 Logging = { -- character logging settings
yellowfive@129 58 Enabled = false, -- whether logging is currently on or not
yellowfive@129 59 LastZone = nil, -- last zone the player was in
yellowfive@129 60 LastDiff = nil, -- last difficulty for the last zone the player was in
yellowfive@129 61 LastWipe = nil -- last time a wipe was called by this player
yellowfive@129 62 }
yellowfive@129 63 },
yellowfive@57 64 profile = {
yellowfive@57 65 minimap = { -- minimap hide/show and position settings
yellowfive@57 66 hide = false
yellowfive@57 67 },
yellowfive@57 68 window = {}, -- main window position settings
yellowfive@57 69 shopWindow = {}, -- shopping list window position settings
yellowfive@161 70 junkWindow = {}, -- junk list window position settings
yellowfive@57 71 options = {
yellowfive@57 72 autoGear = false, -- auto-equip saved gear sets when changing specs
yellowfive@161 73 junkVendor = false, -- auto-show junk list at vendor/scrapper
yellowfive@61 74 shopAh = false, -- auto-show shopping list at AH
yellowfive@91 75 disableEm = false, -- disable auto-creation of equipment manager sets
yellowfive@61 76 uiScale = 1 -- custom scale for AMR UI
yellowfive@57 77 },
yellowfive@57 78 Logging = { -- global logging settings
yellowfive@57 79 Auto = {} -- for each instanceId, for each difficultyId, true if auto-logging enabled
yellowfive@57 80 }
yellowfive@57 81 },
yellowfive@57 82 global = {
yellowfive@57 83 Region = nil, -- region that this user is in, all characters on the same account should be the same region
yellowfive@139 84 Shopping2 = {}, -- shopping list data stored globally for access on any character
yellowfive@57 85 Logging = { -- a lot of log data is stored globally for simplicity, can only be raiding with one character at a time
yellowfive@57 86 Wipes = {}, -- times that a wipe was called
yellowfive@57 87 PlayerData = {}, -- player data gathered at fight start
yellowfive@57 88 PlayerExtras = {} -- player extra data like auras, gathered at fight start
yellowfive@57 89 }
yellowfive@57 90 }
yellowfive@57 91 }
yellowfive@57 92
yellowfive@124 93 Amr.db = LibStub("AceDB-3.0"):New("AskMrRobotDb4", defaults)
yellowfive@110 94
yellowfive@110 95 -- set defaults for auto logging; if a new zone is added and some other stuff was turned on, turn on the new zone too
yellowfive@110 96 local hasSomeLogging = false
yellowfive@110 97 local addedLogging = {}
yellowfive@57 98 for i, instanceId in ipairs(Amr.InstanceIdsOrdered) do
yellowfive@110 99 local byDiff = Amr.db.profile.Logging.Auto[instanceId]
yellowfive@57 100 if not byDiff then
yellowfive@57 101 byDiff = {}
yellowfive@110 102 Amr.db.profile.Logging.Auto[instanceId] = byDiff
yellowfive@110 103 addedLogging[instanceId] = byDiff
yellowfive@57 104 end
yellowfive@57 105
yellowfive@57 106 for k, difficultyId in pairs(Amr.Difficulties) do
yellowfive@110 107 if not byDiff[difficultyId] then
yellowfive@57 108 byDiff[difficultyId] = false
yellowfive@110 109 else
yellowfive@110 110 hasSomeLogging = true
yellowfive@57 111 end
yellowfive@57 112 end
yellowfive@137 113 end
yellowfive@137 114
yellowfive@137 115 for k,v in pairs(Amr.db.profile.Logging.Auto) do
yellowfive@137 116 if not Amr.IsSupportedInstanceId(k) then
yellowfive@137 117 Amr.db.profile.Logging.Auto[k] = nil
yellowfive@137 118 end
yellowfive@57 119 end
yellowfive@57 120
yellowfive@110 121 if hasSomeLogging then
yellowfive@110 122 for instanceId, byDiff in pairs(addedLogging) do
yellowfive@110 123 for k, difficultyId in pairs(Amr.Difficulties) do
yellowfive@110 124 byDiff[difficultyId] = true
yellowfive@110 125 end
yellowfive@110 126 end
yellowfive@110 127 end
yellowfive@110 128
yellowfive@139 129 -- upgrade old gear set info to new format
yellowfive@139 130 if Amr.db.char.GearSets then
yellowfive@139 131 Amr.db.char.GearSets = nil
yellowfive@139 132 end
yellowfive@139 133
yellowfive@139 134 if not Amr.db.char.GearSetups then
yellowfive@139 135 Amr.db.char.GearSetups = {}
yellowfive@139 136 end
yellowfive@139 137
yellowfive@139 138 if Amr.db.global.Shopping then
yellowfive@139 139 Amr.db.global.Shopping = nil
yellowfive@139 140 end
yellowfive@57 141
yellowfive@57 142 Amr.db.RegisterCallback(Amr, "OnProfileChanged", "RefreshConfig")
yellowfive@57 143 Amr.db.RegisterCallback(Amr, "OnProfileCopied", "RefreshConfig")
yellowfive@57 144 Amr.db.RegisterCallback(Amr, "OnProfileReset", "RefreshConfig")
yellowfive@57 145 end
yellowfive@57 146
yellowfive@57 147 function Amr:OnInitialize()
yellowfive@57 148
yellowfive@57 149 initializeDb()
yellowfive@57 150
yellowfive@57 151 Amr:RegisterChatCommand("amr", "SlashCommand")
yellowfive@57 152
yellowfive@57 153 _icon:Register(Amr.ADDON_NAME, _amrLDB, self.db.profile.minimap)
yellowfive@57 154
yellowfive@57 155 -- listen for inter-addon communication
yellowfive@139 156 self:RegisterComm(Amr.ChatPrefix, "OnCommReceived")
yellowfive@57 157 end
yellowfive@57 158
yellowfive@57 159 local _enteredWorld = false
yellowfive@57 160 local _pendingInit = false
yellowfive@57 161
yellowfive@139 162 -- upgrade some stuff from old to new formats
yellowfive@139 163 local function upgradeFromOld()
yellowfive@139 164
yellowfive@139 165 local currentVersion = tonumber(GetAddOnMetadata(Amr.ADDON_NAME, "Version"))
yellowfive@139 166 if Amr.db.char.LastVersion < 65 then
yellowfive@143 167
yellowfive@143 168 if not Amr.db.profile.options.disableEm then
yellowfive@143 169 for i = 1,GetNumSpecializations() do
yellowfive@143 170 local _, specName = GetSpecializationInfo(i)
yellowfive@143 171 if specName then
yellowfive@143 172 local setid = C_EquipmentSet.GetEquipmentSetID("AMR " .. specName)
yellowfive@143 173 if setid then
yellowfive@143 174 C_EquipmentSet.DeleteEquipmentSet(setid)
yellowfive@143 175 end
yellowfive@139 176 end
yellowfive@139 177 end
yellowfive@139 178 end
yellowfive@143 179
yellowfive@139 180 end
yellowfive@139 181 Amr.db.char.LastVersion = currentVersion
yellowfive@139 182
yellowfive@139 183 end
yellowfive@139 184
yellowfive@133 185 local function finishInitialize()
yellowfive@57 186
yellowfive@57 187 -- record region, the only thing that we still can't get from the log file
yellowfive@57 188 Amr.db.global.Region = Amr.RegionNames[GetCurrentRegion()]
yellowfive@57 189
yellowfive@57 190 -- make sure that some initialization is deferred until after PLAYER_ENTERING_WORLD event so that data we need is available;
yellowfive@57 191 -- also delay this initialization for a few extra seconds to deal with some event spam that is otherwise hard to identify and ignore when a player logs in
yellowfive@57 192 Amr.Wait(5, function()
yellowfive@145 193 --Amr:InitializeVersions()
yellowfive@57 194 Amr:InitializeGear()
yellowfive@57 195 Amr:InitializeExport()
yellowfive@57 196 Amr:InitializeCombatLog()
yellowfive@139 197
yellowfive@139 198 upgradeFromOld()
yellowfive@57 199 end)
yellowfive@57 200 end
yellowfive@57 201
yellowfive@133 202 local function onPlayerEnteringWorld()
yellowfive@57 203
yellowfive@57 204 _enteredWorld = true
yellowfive@57 205
yellowfive@57 206 if _pendingInit then
yellowfive@57 207 finishInitialize()
yellowfive@57 208 _pendingInit = false
yellowfive@57 209 end
yellowfive@57 210 end
yellowfive@57 211
yellowfive@57 212 function Amr:OnEnable()
yellowfive@57 213
yellowfive@81 214 --[[
yellowfive@57 215 -- listen for changes to the snapshot enable state, and always make sure it is enabled if using the core AskMrRobot addon
yellowfive@57 216 self:RegisterMessage("AMR_SNAPSHOT_STATE_CHANGED", function(eventName, isEnabled)
yellowfive@57 217 if not isEnabled then
yellowfive@57 218 -- immediately re-enable on any attempt to disable
yellowfive@57 219 Amr.Serializer:EnableSnapshots()
yellowfive@57 220 end
yellowfive@57 221 end)
yellowfive@57 222 self.Serializer:EnableSnapshots()
yellowfive@81 223 ]]
yellowfive@57 224
yellowfive@57 225 -- update based on current configuration whenever enabled
yellowfive@57 226 self:RefreshConfig()
yellowfive@57 227
yellowfive@57 228 -- if we have fully entered the world, do initialization; otherwise wait for PLAYER_ENTERING_WORLD to continue
yellowfive@57 229 if not _enteredWorld then
yellowfive@57 230 _pendingInit = true
yellowfive@57 231 else
yellowfive@57 232 _pendingInit = false
yellowfive@57 233 finishInitialize()
yellowfive@57 234 end
yellowfive@57 235 end
yellowfive@57 236
yellowfive@57 237 function Amr:OnDisable()
yellowfive@57 238 -- disabling is not supported
yellowfive@57 239 end
yellowfive@57 240
yellowfive@57 241
yellowfive@57 242 ----------------------------------------------------------------------------------------
yellowfive@57 243 -- Slash Commands
yellowfive@57 244 ----------------------------------------------------------------------------------------
yellowfive@57 245 local _slashMethods = {
yellowfive@57 246 hide = "Hide",
yellowfive@57 247 show = "Show",
yellowfive@57 248 toggle = "Toggle",
yellowfive@124 249 equip = "EquipGearSet",
yellowfive@57 250 version = "PrintVersions",
yellowfive@161 251 junk = "ShowJunkWindow",
yellowfive@124 252 --wipe = "Wipe",
yellowfive@124 253 --undowipe = "UndoWipe",
yellowfive@61 254 reset = "Reset",
yellowfive@57 255 test = "Test"
yellowfive@57 256 }
yellowfive@57 257
yellowfive@57 258 function Amr:SlashCommand(input)
yellowfive@57 259 input = string.lower(input)
yellowfive@57 260 local parts = {}
yellowfive@57 261 for w in input:gmatch("%S+") do
yellowfive@57 262 table.insert(parts, w)
yellowfive@57 263 end
yellowfive@57 264
yellowfive@57 265 if #parts == 0 then return end
yellowfive@57 266
yellowfive@57 267 local func = _slashMethods[parts[1]]
yellowfive@57 268 if not func then return end
yellowfive@57 269
yellowfive@57 270 local funcArgs = {}
yellowfive@57 271 for i = 2, #parts do
yellowfive@57 272 table.insert(funcArgs, parts[i])
yellowfive@57 273 end
yellowfive@57 274
yellowfive@57 275 Amr[func](Amr, unpack(funcArgs))
yellowfive@57 276 end
yellowfive@57 277
yellowfive@57 278
yellowfive@57 279 ----------------------------------------------------------------------------------------
yellowfive@57 280 -- Configuration
yellowfive@57 281 ----------------------------------------------------------------------------------------
yellowfive@57 282
yellowfive@57 283 -- refresh all state based on the current values of configuration options
yellowfive@57 284 function Amr:RefreshConfig()
yellowfive@57 285
yellowfive@57 286 self:UpdateMinimap()
yellowfive@57 287 self:RefreshOptionsUi()
yellowfive@57 288 self:RefreshLogUi()
yellowfive@57 289 end
yellowfive@57 290
yellowfive@57 291 function Amr:UpdateMinimap()
yellowfive@57 292
yellowfive@57 293 if self.db.profile.minimap.hide or not Amr:IsEnabled() then
yellowfive@57 294 _icon:Hide(Amr.ADDON_NAME)
yellowfive@57 295 else
yellowfive@57 296 -- change icon color if logging
yellowfive@57 297 if Amr:IsLogging() then
yellowfive@57 298 _amrLDB.icon = 'Interface\\AddOns\\AskMrRobot\\Media\\icon_green'
yellowfive@57 299 else
yellowfive@57 300 _amrLDB.icon = 'Interface\\AddOns\\AskMrRobot\\Media\\icon'
yellowfive@57 301 end
yellowfive@57 302
yellowfive@57 303 _icon:Show(Amr.ADDON_NAME)
yellowfive@57 304 end
yellowfive@57 305 end
yellowfive@57 306
yellowfive@57 307
yellowfive@57 308 ----------------------------------------------------------------------------------------
yellowfive@57 309 -- Version Checking
yellowfive@57 310 ----------------------------------------------------------------------------------------
yellowfive@57 311
yellowfive@57 312 -- version of addon being run by each person in the player's raid or group
yellowfive@57 313 Amr.GroupVersions = {}
yellowfive@57 314
yellowfive@57 315 local function toGroupVersionKey(realm, name)
yellowfive@57 316 realm = string.gsub(realm, "%s+", "")
yellowfive@57 317 return name .. "-" .. realm
yellowfive@57 318 end
yellowfive@57 319
yellowfive@57 320 -- prune out version information for players no longer in the current raid group
yellowfive@57 321 local function pruneVersionInfo()
yellowfive@57 322
yellowfive@57 323 local newVersions = {}
yellowfive@57 324 local units = Amr:GetGroupUnitIdentifiers()
yellowfive@57 325
yellowfive@57 326 for i, unitId in ipairs(units) do
yellowfive@57 327 local realm, name = Amr:GetRealmAndName(unitId)
yellowfive@57 328 if realm then
yellowfive@57 329 local key = toGroupVersionKey(realm, name)
yellowfive@57 330 newVersions[key] = Amr.GroupVersions[key]
yellowfive@57 331 end
yellowfive@57 332 end
yellowfive@57 333
yellowfive@57 334 Amr.GroupVersions = newVersions
yellowfive@57 335 end
yellowfive@57 336
yellowfive@57 337 -- send version information to other people in the same raid group
yellowfive@57 338 local function sendVersionInfo()
yellowfive@57 339
yellowfive@57 340 local realm = GetRealmName()
yellowfive@57 341 local name = UnitName("player")
yellowfive@57 342 local ver = GetAddOnMetadata(Amr.ADDON_NAME, "Version")
yellowfive@57 343
yellowfive@57 344 local msg = string.format("%s\n%s\n%s\n%s", Amr.MessageTypes.Version, realm, name, ver)
yellowfive@57 345 Amr:SendAmrCommMessage(msg)
yellowfive@57 346 end
yellowfive@57 347
yellowfive@57 348 local function onVersionInfoReceived(message)
yellowfive@57 349
yellowfive@57 350 -- message will be of format: realm\nname\nversion
yellowfive@57 351 local parts = {}
yellowfive@57 352 for part in string.gmatch(message, "([^\n]+)") do
yellowfive@57 353 table.insert(parts, part)
yellowfive@57 354 end
yellowfive@57 355
yellowfive@57 356 local key = toGroupVersionKey(parts[2], parts[3])
yellowfive@57 357 local ver = parts[4]
yellowfive@57 358
yellowfive@57 359 Amr.GroupVersions[key] = tonumber(ver)
yellowfive@57 360
yellowfive@57 361 -- make sure that versions are properly pruned in case this message arrived late and the player has since been removed from the group
yellowfive@57 362 pruneVersionInfo()
yellowfive@57 363 end
yellowfive@57 364
yellowfive@57 365 -- get the addon version another person in the player's raid/group is running, or 0 if they are not running the addon
yellowfive@57 366 function Amr:GetAddonVersion(realm, name)
yellowfive@57 367 local ver = Amr.GroupVersions[toGroupVersionKey(realm, name)]
yellowfive@57 368 return ver or 0
yellowfive@57 369 end
yellowfive@57 370
yellowfive@57 371 function Amr:PrintVersions()
yellowfive@57 372
yellowfive@57 373 if not IsInGroup() and not IsInRaid() then
yellowfive@57 374 self:Print(L.VersionChatNotGrouped)
yellowfive@57 375 return
yellowfive@57 376 end
yellowfive@57 377
yellowfive@57 378 local units = self:GetGroupUnitIdentifiers()
yellowfive@57 379
yellowfive@57 380 local msg = {}
yellowfive@57 381 table.insert(msg, L.VersionChatTitle)
yellowfive@57 382
yellowfive@57 383 for i, unitId in ipairs(units) do
yellowfive@57 384 local realm, name = self:GetRealmAndName(unitId)
yellowfive@57 385 if realm then
yellowfive@57 386 local key = toGroupVersionKey(realm, name)
yellowfive@57 387 local ver = Amr.GroupVersions[key]
yellowfive@57 388 if not ver then
yellowfive@57 389 table.insert(msg, key .. " |cFFFF0000" .. L.VersionChatNotInstalled .. "|r")
yellowfive@57 390 else
yellowfive@57 391 table.insert(msg, key .. " v" .. ver)
yellowfive@57 392 end
yellowfive@57 393 end
yellowfive@57 394 end
yellowfive@57 395
yellowfive@57 396 msg = table.concat(msg, "\n")
yellowfive@57 397 print(msg)
yellowfive@57 398 end
yellowfive@57 399
yellowfive@57 400 function Amr:InitializeVersions()
yellowfive@57 401 Amr:AddEventHandler("GROUP_ROSTER_UPDATE", pruneVersionInfo)
yellowfive@57 402 Amr:AddEventHandler("GROUP_ROSTER_UPDATE", sendVersionInfo)
yellowfive@57 403
yellowfive@57 404 -- request version information from anyone in my group upon initialization
yellowfive@57 405 if IsInGroup() or IsInRaid() then
yellowfive@57 406 Amr:SendAmrCommMessage(Amr.MessageTypes.VersionRequest)
yellowfive@57 407 end
yellowfive@57 408 end
yellowfive@57 409
yellowfive@57 410
yellowfive@57 411 ----------------------------------------------------------------------------------------
yellowfive@57 412 -- Generic Helpers
yellowfive@57 413 ----------------------------------------------------------------------------------------
yellowfive@57 414
yellowfive@57 415 local _waitTable = {}
yellowfive@57 416 local _waitFrame = nil
yellowfive@57 417
yellowfive@57 418 -- execute the specified function after the specified delay (in seconds)
yellowfive@57 419 function Amr.Wait(delay, func, ...)
yellowfive@57 420 if not _waitFrame then
yellowfive@57 421 _waitFrame = CreateFrame("Frame", "AmrWaitFrame", UIParent)
yellowfive@57 422 _waitFrame:SetScript("OnUpdate", function (self, elapse)
yellowfive@57 423 local count = #_waitTable
yellowfive@57 424 local i = 1
yellowfive@57 425 while(i <= count) do
yellowfive@57 426 local waitRecord = table.remove(_waitTable, i)
yellowfive@57 427 local d = table.remove(waitRecord, 1)
yellowfive@57 428 local f = table.remove(waitRecord, 1)
yellowfive@57 429 local p = table.remove(waitRecord, 1)
yellowfive@57 430 if d > elapse then
yellowfive@57 431 table.insert(_waitTable, i, { d-elapse, f, p })
yellowfive@57 432 i = i + 1
yellowfive@57 433 else
yellowfive@57 434 count = count - 1
yellowfive@57 435 f(unpack(p))
yellowfive@57 436 end
yellowfive@57 437 end
yellowfive@57 438 end)
yellowfive@57 439 end
yellowfive@57 440 table.insert(_waitTable, { delay, func, {...} })
yellowfive@57 441 return true
yellowfive@57 442 end
yellowfive@57 443
yellowfive@57 444 -- helper to iterate over a table in order by its keys
yellowfive@57 445 function Amr.spairs(t, order)
yellowfive@57 446 -- collect the keys
yellowfive@57 447 local keys = {}
yellowfive@57 448 for k in pairs(t) do keys[#keys+1] = k end
yellowfive@57 449
yellowfive@57 450 -- if order function given, sort by it by passing the table and keys a, b,
yellowfive@57 451 -- otherwise just sort the keys
yellowfive@57 452 if order then
yellowfive@57 453 table.sort(keys, function(a,b) return order(t, a, b) end)
yellowfive@57 454 else
yellowfive@57 455 table.sort(keys)
yellowfive@57 456 end
yellowfive@57 457
yellowfive@57 458 -- return the iterator function
yellowfive@57 459 local i = 0
yellowfive@57 460 return function()
yellowfive@57 461 i = i + 1
yellowfive@57 462 if keys[i] then
yellowfive@57 463 return keys[i], t[keys[i]]
yellowfive@57 464 end
yellowfive@57 465 end
yellowfive@57 466 end
yellowfive@57 467
yellowfive@57 468 function Amr.StartsWith(str, prefix)
yellowfive@57 469 if string.len(str) < string.len(prefix) then return false end
yellowfive@57 470 return string.sub(str, 1, string.len(prefix)) == prefix
yellowfive@57 471 end
yellowfive@57 472
yellowfive@124 473 function Amr.IsEmpty(table)
yellowfive@124 474 return next(table) == nil
yellowfive@124 475 end
yellowfive@124 476
yellowfive@124 477 function Amr.Contains(table, value)
yellowfive@124 478 if not table then return false end
yellowfive@124 479 for k,v in pairs(table) do
yellowfive@124 480 if v == value then
yellowfive@124 481 return true
yellowfive@124 482 end
yellowfive@124 483 end
yellowfive@124 484 return false
yellowfive@124 485 end
yellowfive@124 486
yellowfive@57 487 -- helper to get the unit identifiers (e.g. to pass to GetUnitName) for all members of the player's current group/raid
yellowfive@57 488 function Amr:GetGroupUnitIdentifiers()
yellowfive@57 489
yellowfive@57 490 local units = {}
yellowfive@57 491 if IsInRaid() then
yellowfive@57 492 for i = 1,40 do
yellowfive@57 493 table.insert(units, "raid" .. i)
yellowfive@57 494 end
yellowfive@57 495 elseif IsInGroup() then
yellowfive@57 496 table.insert(units, "player")
yellowfive@57 497 for i = 1,4 do
yellowfive@57 498 table.insert(units, "party" .. i)
yellowfive@57 499 end
yellowfive@57 500 else
yellowfive@57 501 table.insert(units, "player")
yellowfive@57 502 end
yellowfive@57 503
yellowfive@57 504 return units
yellowfive@57 505 end
yellowfive@57 506
yellowfive@57 507 -- helper to get the realm and name from a unitId (e.g. "player" or "raid1")
yellowfive@57 508 function Amr:GetRealmAndName(unitId)
yellowfive@57 509
yellowfive@57 510 local name = GetUnitName(unitId, true)
yellowfive@57 511 if not name then return end
yellowfive@57 512
yellowfive@57 513 local realm = GetRealmName()
yellowfive@57 514 local splitPos = string.find(name, "-")
yellowfive@57 515 if splitPos ~= nil then
yellowfive@57 516 realm = string.sub(name, splitPos + 1)
yellowfive@57 517 name = string.sub(name, 1, splitPos - 1)
yellowfive@57 518 end
yellowfive@57 519
yellowfive@57 520 return realm, name
yellowfive@57 521 end
yellowfive@57 522
yellowfive@57 523 -- find the unitid of a player given the name and realm... this comes from the server so the realm will be in english...
yellowfive@57 524 -- TODO: more robust handling of players with same name but different realms in the same group on non-english clients
yellowfive@57 525 function Amr:GetUnitId(unitRealm, unitName)
yellowfive@57 526
yellowfive@57 527 local nameMatches = {}
yellowfive@57 528
yellowfive@57 529 local units = Amr:GetGroupUnitIdentifiers()
yellowfive@57 530 for i, unitId in ipairs(units) do
yellowfive@57 531 local realm, name = Amr:GetRealmAndName(unitId)
yellowfive@57 532 if realm then
yellowfive@57 533 -- remove spaces to ensure proper matches
yellowfive@57 534 realm = string.gsub(realm, "%s+", "")
yellowfive@57 535 unitRealm = string.gsub(unitRealm, "%s+", "")
yellowfive@57 536
yellowfive@57 537 if unitRealm == realm and unitName == name then return unitId end
yellowfive@57 538 if unitName == name then
yellowfive@57 539 table.insert(nameMatches, unitId)
yellowfive@57 540 end
yellowfive@57 541 end
yellowfive@57 542 end
yellowfive@57 543
yellowfive@57 544 -- only one player with same name, must be the player of interest
yellowfive@57 545 if #nameMatches == 1 then return nameMatches[1] end
yellowfive@57 546
yellowfive@57 547 -- could not find or ambiguous
yellowfive@57 548 return nil
yellowfive@57 549 end
yellowfive@57 550
yellowfive@133 551 --[[
yellowfive@57 552 -- search the tooltip for txt, returns true if it is encountered on any line
yellowfive@57 553 function Amr:IsTextInTooltip(tt, txt)
yellowfive@57 554 local regions = { tt:GetRegions() }
yellowfive@57 555 for i, region in ipairs(regions) do
yellowfive@57 556 if region and region:GetObjectType() == "FontString" then
yellowfive@57 557 if region:GetText() == txt then
yellowfive@57 558 return true
yellowfive@57 559 end
yellowfive@57 560 end
yellowfive@57 561 end
yellowfive@57 562 return false
yellowfive@57 563 end
yellowfive@133 564 ]]
yellowfive@57 565
yellowfive@124 566 -- helper to determine if we can equip an item (it is soulbound)
yellowfive@59 567 function Amr:CanEquip(bagId, slotId)
yellowfive@124 568 local item = Item:CreateFromBagAndSlot(bagId, slotId)
yellowfive@124 569 if item then
yellowfive@124 570 local loc = item:GetItemLocation()
yellowfive@124 571 return C_Item.IsBound(loc)
yellowfive@124 572 else
yellowfive@124 573 -- for now just return true if we can't find the item... will get an error trying to equip if it isn't bound
yellowfive@124 574 return true
yellowfive@124 575 end
yellowfive@124 576
yellowfive@124 577 --local tt = Amr.GetItemTooltip(bagId, slotId)
yellowfive@124 578 --if self:IsTextInTooltip(tt, ITEM_SOULBOUND) then return true end
yellowfive@124 579 --if self:IsTextInTooltip(tt, ITEM_BNETACCOUNTBOUND) then return true end
yellowfive@124 580 --if self:IsTextInTooltip(tt, ITEM_ACCOUNTBOUND) then return true end
yellowfive@57 581 end
yellowfive@57 582
yellowfive@57 583 -- helper to determine if an item has a unique constraint
yellowfive@133 584 --[[
yellowfive@57 585 function Amr:IsUnique(bagId, slotId)
yellowfive@81 586 local tt = Amr.GetItemTooltip(bagId, slotId)
yellowfive@57 587 if self:IsTextInTooltip(tt, ITEM_UNIQUE_EQUIPPABLE) then return true end
yellowfive@57 588 if self:IsTextInTooltip(tt, ITEM_UNIQUE) then return true end
yellowfive@57 589 return false
yellowfive@57 590 end
yellowfive@133 591 ]]
yellowfive@57 592
yellowfive@57 593
yellowfive@57 594 ----------------------------------------------------------------------------------------
yellowfive@57 595 -- Inter-Addon Communication
yellowfive@57 596 ----------------------------------------------------------------------------------------
yellowfive@57 597 function Amr:SendAmrCommMessage(message, channel)
yellowfive@57 598 -- prepend version to all messages
yellowfive@57 599 local v = GetAddOnMetadata(Amr.ADDON_NAME, "Version")
yellowfive@57 600 message = v .. "\r" .. message
yellowfive@57 601
yellowfive@79 602 Amr:SendCommMessage(Amr.ChatPrefix, message, channel or (IsInGroup(LE_PARTY_CATEGORY_INSTANCE) and "INSTANCE_CHAT" or "RAID"))
yellowfive@57 603 end
yellowfive@57 604
yellowfive@57 605 function Amr:OnCommReceived(prefix, message, distribution, sender)
yellowfive@57 606
yellowfive@57 607 local parts = {}
yellowfive@57 608 for part in string.gmatch(message, "([^\r]+)") do
yellowfive@57 609 table.insert(parts, part)
yellowfive@57 610 end
yellowfive@57 611
yellowfive@57 612 local ver = parts[1]
yellowfive@57 613 if ver then ver = tonumber(ver) end
yellowfive@57 614 if ver then
yellowfive@57 615 -- newest versions of the addon start all messages with a version number
yellowfive@57 616 message = parts[2]
yellowfive@57 617 end
yellowfive@57 618
yellowfive@57 619 -- we always allow version checks, even from old versions of the addon that aren't otherwise compatible
yellowfive@57 620 if Amr.StartsWith(message, Amr.MessageTypes.Version) then
yellowfive@57 621 -- version checking between group members
yellowfive@57 622 if Amr.StartsWith(message, Amr.MessageTypes.VersionRequest) then
yellowfive@57 623 sendVersionInfo()
yellowfive@57 624 else
yellowfive@57 625 onVersionInfoReceived(message)
yellowfive@57 626 end
yellowfive@57 627
yellowfive@57 628 return
yellowfive@57 629 end
yellowfive@57 630
yellowfive@57 631 -- any other kind of message is ignored if the version is too old
yellowfive@57 632 if not ver or ver < Amr.MIN_ADDON_VERSION then return end
yellowfive@57 633
yellowfive@124 634 --[[
yellowfive@57 635 if Amr.StartsWith(message, Amr.MessageTypes.Team) then
yellowfive@57 636 -- if fully initialized, process team optimizer messages
yellowfive@57 637 if Amr["ProcessTeamMessage"] then
yellowfive@57 638 Amr:ProcessTeamMessage(message)
yellowfive@57 639 end
yellowfive@57 640 else
yellowfive@57 641 -- if we are fully loaded, process a player snapshot when it is received (combat logging)
yellowfive@57 642 if Amr["ProcessPlayerSnapshot"] then
yellowfive@57 643 self:ProcessPlayerSnapshot(message)
yellowfive@57 644 end
yellowfive@57 645 end
yellowfive@124 646 ]]
yellowfive@57 647 end
yellowfive@57 648
yellowfive@57 649
yellowfive@57 650 ----------------------------------------------------------------------------------------
yellowfive@57 651 -- Events
yellowfive@57 652 ----------------------------------------------------------------------------------------
yellowfive@57 653 local _eventHandlers = {}
yellowfive@57 654
yellowfive@57 655 local function handleEvent(eventName, ...)
yellowfive@57 656 local list = _eventHandlers[eventName]
yellowfive@57 657 if list then
yellowfive@57 658 --print(eventName .. " handled")
yellowfive@57 659 for i, handler in ipairs(list) do
yellowfive@57 660 if type(handler) == "function" then
yellowfive@57 661 handler(select(1, ...))
yellowfive@57 662 else
yellowfive@57 663 Amr[handler](Amr, select(1, ...))
yellowfive@57 664 end
yellowfive@57 665 end
yellowfive@57 666 end
yellowfive@57 667 end
yellowfive@57 668
yellowfive@57 669 -- WoW and Ace seem to work on a "one handler" kind of approach to events (as far as I can tell from the sparse documentation of both).
yellowfive@57 670 -- This is a simple wrapper to allow adding multiple handlers to the same event, thus allowing better encapsulation of code from file to file.
yellowfive@57 671 function Amr:AddEventHandler(eventName, methodOrName)
yellowfive@57 672 local list = _eventHandlers[eventName]
yellowfive@57 673 if not list then
yellowfive@57 674 list = {}
yellowfive@57 675 _eventHandlers[eventName] = list
yellowfive@57 676 Amr:RegisterEvent(eventName, handleEvent)
yellowfive@57 677 end
yellowfive@57 678 table.insert(list, methodOrName)
yellowfive@57 679 end
yellowfive@57 680
yellowfive@57 681 Amr:AddEventHandler("PLAYER_ENTERING_WORLD", onPlayerEnteringWorld)
yellowfive@57 682
yellowfive@57 683
yellowfive@57 684 ----------------------------------------------------------------------------------------
yellowfive@57 685 -- Debugging
yellowfive@57 686 ----------------------------------------------------------------------------------------
yellowfive@124 687 function Amr:dump(o)
yellowfive@124 688 if type(o) == 'table' then
yellowfive@124 689 local s = '{ '
yellowfive@124 690 for k,v in pairs(o) do
yellowfive@124 691 if type(k) ~= 'number' then k = '"'..k..'"' end
yellowfive@124 692 s = s .. '['..k..'] = ' .. Amr:dump(v) .. ','
yellowfive@124 693 end
yellowfive@124 694 return s .. '} '
yellowfive@124 695 else
yellowfive@124 696 return tostring(o)
yellowfive@124 697 end
yellowfive@124 698 end
yellowfive@124 699
yellowfive@69 700 function Amr:Test()
yellowfive@124 701
yellowfive@69 702 end