Mercurial > wow > askmrrobot

annotate Core.lua @ 112:57c6cac5143c v52

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