wow/wowdb-profiler: Main.lua annotate

annotate Main.lua @ 205:6cac044fd0f9

In the LOOT_OPENED NPC verify function, only set current_action.identifier to the unit_id of the player's target if the identifier is nil.

author	James D. Callahan III <jcallahan@curse.com>
date	Mon, 10 Dec 2012 16:20:06 -0600
parents	83fdf52f0310
children	d401bf9aa80c

rev	line source
jcallahan@0	1 -----------------------------------------------------------------------
jcallahan@0	2 -- Upvalued Lua API.
jcallahan@0	3 -----------------------------------------------------------------------
jcallahan@0	4 local _G = getfenv(0)
jcallahan@0	5
jcallahan@0	6 local pairs = _G.pairs
jcallahan@1	7 local tonumber = _G.tonumber
jcallahan@1	8
jcallahan@1	9 local bit = _G.bit
jcallahan@1	10 local math = _G.math
jcallahan@1	11 local table = _G.table
jcallahan@1	12
jcallahan@78	13 local select = _G.select
jcallahan@78	14
jcallahan@0	15
jcallahan@0	16 -----------------------------------------------------------------------
jcallahan@0	17 -- AddOn namespace.
jcallahan@0	18 -----------------------------------------------------------------------
jcallahan@0	19 local ADDON_NAME, private = ...
jcallahan@0	20
jcallahan@0	21 local LibStub = _G.LibStub
jcallahan@0	22 local WDP = LibStub("AceAddon-3.0"):NewAddon(ADDON_NAME, "AceEvent-3.0", "AceTimer-3.0")
jcallahan@0	23
jcallahan@48	24 local deformat = LibStub("LibDeformat-3.0")
jcallahan@115	25 local LPJ = LibStub("LibPetJournal-2.0")
jcallahan@141	26 local MapData = LibStub("LibMapData-1.0")
jcallahan@48	27
jcallahan@4	28 local DatamineTT = _G.CreateFrame("GameTooltip", "WDPDatamineTT", _G.UIParent, "GameTooltipTemplate")
jcallahan@5	29 DatamineTT:SetOwner(_G.WorldFrame, "ANCHOR_NONE")
jcallahan@5	30
jcallahan@0	31
jcallahan@0	32 -----------------------------------------------------------------------
jcallahan@0	33 -- Local constants.
jcallahan@0	34 -----------------------------------------------------------------------
jcallahan@197	35 local DB_VERSION = 13
jcallahan@151	36 local DEBUGGING = false
jcallahan@156	37 local EVENT_DEBUG = false
jcallahan@92	38
jcallahan@157	39
jcallahan@0	40 local DATABASE_DEFAULTS = {
jcallahan@128	41 char = {},
jcallahan@0	42 global = {
jcallahan@0	43 items = {},
jcallahan@0	44 npcs = {},
jcallahan@0	45 objects = {},
jcallahan@0	46 quests = {},
jcallahan@167	47 spells = {},
jcallahan@17	48 zones = {},
jcallahan@0	49 }
jcallahan@0	50 }
jcallahan@0	51
jcallahan@0	52
jcallahan@1	53 local EVENT_MAPPING = {
jcallahan@90	54 AUCTION_HOUSE_SHOW = true,
jcallahan@90	55 BANKFRAME_OPENED = true,
jcallahan@90	56 BATTLEFIELDS_SHOW = true,
jcallahan@56	57 BLACK_MARKET_ITEM_UPDATE = true,
jcallahan@48	58 CHAT_MSG_LOOT = true,
jcallahan@95	59 CHAT_MSG_MONSTER_SAY = "RecordQuote",
jcallahan@95	60 CHAT_MSG_MONSTER_WHISPER = "RecordQuote",
jcallahan@95	61 CHAT_MSG_MONSTER_YELL = "RecordQuote",
jcallahan@40	62 CHAT_MSG_SYSTEM = true,
jcallahan@23	63 COMBAT_LOG_EVENT_UNFILTERED = true,
jcallahan@18	64 COMBAT_TEXT_UPDATE = true,
jcallahan@140	65 CURSOR_UPDATE = true,
jcallahan@90	66 FORGE_MASTER_OPENED = true,
jcallahan@90	67 GOSSIP_SHOW = true,
jcallahan@93	68 GUILDBANKFRAME_OPENED = true,
jcallahan@42	69 ITEM_TEXT_BEGIN = true,
jcallahan@189	70 ITEM_UPGRADE_MASTER_OPENED = true,
jcallahan@124	71 LOOT_CLOSED = true,
jcallahan@1	72 LOOT_OPENED = true,
jcallahan@89	73 MAIL_SHOW = true,
jcallahan@133	74 MERCHANT_CLOSED = true,
jcallahan@7	75 MERCHANT_SHOW = "UpdateMerchantItems",
jcallahan@61	76 MERCHANT_UPDATE = "UpdateMerchantItems",
jcallahan@25	77 PET_BAR_UPDATE = true,
jcallahan@115	78 PET_JOURNAL_LIST_UPDATE = true,
jcallahan@156	79 PLAYER_REGEN_DISABLED = true,
jcallahan@156	80 PLAYER_REGEN_ENABLED = true,
jcallahan@2	81 PLAYER_TARGET_CHANGED = true,
jcallahan@9	82 QUEST_COMPLETE = true,
jcallahan@9	83 QUEST_DETAIL = true,
jcallahan@9	84 QUEST_LOG_UPDATE = true,
jcallahan@97	85 QUEST_PROGRESS = true,
jcallahan@178	86 SHOW_LOOT_TOAST = true,
jcallahan@88	87 TAXIMAP_OPENED = true,
jcallahan@92	88 TRADE_SKILL_SHOW = true,
jcallahan@167	89 TRAINER_CLOSED = true,
jcallahan@27	90 TRAINER_SHOW = true,
jcallahan@90	91 TRANSMOGRIFY_OPEN = true,
jcallahan@4	92 UNIT_QUEST_LOG_CHANGED = true,
jcallahan@1	93 UNIT_SPELLCAST_FAILED = "HandleSpellFailure",
jcallahan@1	94 UNIT_SPELLCAST_FAILED_QUIET = "HandleSpellFailure",
jcallahan@1	95 UNIT_SPELLCAST_INTERRUPTED = "HandleSpellFailure",
jcallahan@1	96 UNIT_SPELLCAST_SENT = true,
jcallahan@1	97 UNIT_SPELLCAST_SUCCEEDED = true,
jcallahan@90	98 VOID_STORAGE_OPEN = true,
jcallahan@129	99 ZONE_CHANGED = "SetCurrentAreaID",
jcallahan@129	100 ZONE_CHANGED_INDOORS = "SetCurrentAreaID",
jcallahan@129	101 ZONE_CHANGED_NEW_AREA = "SetCurrentAreaID",
jcallahan@0	102 }
jcallahan@0	103
jcallahan@4	104
jcallahan@92	105 local OBJECT_ID_ANVIL = 192628
jcallahan@92	106 local OBJECT_ID_FORGE = 1685
jcallahan@92	107
jcallahan@92	108
jcallahan@1	109 local AF = private.ACTION_TYPE_FLAGS
jcallahan@0	110
jcallahan@4	111
jcallahan@27	112 local PLAYER_CLASS = _G.select(2, _G.UnitClass("player"))
jcallahan@100	113 local PLAYER_FACTION = _G.UnitFactionGroup("player")
jcallahan@34	114 local PLAYER_GUID = _G.UnitGUID("player")
jcallahan@95	115 local PLAYER_NAME = _G.UnitName("player")
jcallahan@39	116 local PLAYER_RACE = _G.select(2, _G.UnitRace("player"))
jcallahan@27	117
jcallahan@95	118
jcallahan@112	119 local CLIENT_LOCALE = _G.GetLocale()
jcallahan@112	120
jcallahan@112	121
jcallahan@112	122 local ALLOWED_LOCALES = {
jcallahan@112	123 enUS = true,
jcallahan@112	124 enGB = true,
jcallahan@112	125 }
jcallahan@112	126
jcallahan@112	127
jcallahan@0	128 -----------------------------------------------------------------------
jcallahan@0	129 -- Local variables.
jcallahan@0	130 -----------------------------------------------------------------------
jcallahan@92	131 local anvil_spell_ids = {}
jcallahan@92	132 local currently_drunk
jcallahan@128	133 local char_db
jcallahan@128	134 local global_db
jcallahan@187	135 local item_process_timer_handle
jcallahan@92	136 local faction_standings = {}
jcallahan@92	137 local forge_spell_ids = {}
jcallahan@95	138 local languages_known = {}
jcallahan@95	139 local name_to_id_map = {}
jcallahan@177	140 local killed_npc_id
jcallahan@2	141 local target_location_timer_handle
jcallahan@86	142 local current_target_id
jcallahan@126	143 local current_area_id
jcallahan@131	144 local current_loot
jcallahan@1	145
jcallahan@121	146 -----------------------------------------------------------------------
jcallahan@121	147 -- Data for our current action. Including possible values as a reference.
jcallahan@121	148 -----------------------------------------------------------------------
jcallahan@122	149 local current_action = {
jcallahan@121	150 identifier = nil,
jcallahan@121	151 loot_label = nil,
jcallahan@121	152 loot_list = nil,
jcallahan@121	153 loot_sources = nil,
jcallahan@121	154 map_level = nil,
jcallahan@121	155 spell_label = nil,
jcallahan@123	156 target_type = nil,
jcallahan@121	157 x = nil,
jcallahan@121	158 y = nil,
jcallahan@121	159 zone_data = nil,
jcallahan@121	160 }
jcallahan@92	161
jcallahan@1	162 -----------------------------------------------------------------------
jcallahan@1	163 -- Helper Functions.
jcallahan@1	164 -----------------------------------------------------------------------
jcallahan@151	165 local function Debug(...)
jcallahan@151	166 if not DEBUGGING then
jcallahan@151	167 return
jcallahan@151	168 end
jcallahan@151	169 _G.print(...)
jcallahan@151	170 end
jcallahan@151	171
jcallahan@151	172
jcallahan@169	173 local TradeSkillExecutePer
jcallahan@169	174 do
jcallahan@169	175 local header_list = {}
jcallahan@169	176
jcallahan@169	177 -- iter_func returns true to indicate that the loop should be broken
jcallahan@169	178 function TradeSkillExecutePer(iter_func)
jcallahan@169	179 if not _G.TradeSkillFrame or not _G.TradeSkillFrame:IsVisible() then
jcallahan@169	180 return
jcallahan@169	181 end
jcallahan@167	182 -- Clear the search box focus so the scan will have correct results.
jcallahan@167	183 local search_box = _G.TradeSkillFrameSearchBox
jcallahan@167	184 search_box:SetText("")
jcallahan@169	185
jcallahan@167	186 _G.TradeSkillSearch_OnTextChanged(search_box)
jcallahan@167	187 search_box:ClearFocus()
jcallahan@167	188 search_box:GetScript("OnEditFocusLost")(search_box)
jcallahan@169	189
jcallahan@169	190 table.wipe(header_list)
jcallahan@169	191
jcallahan@169	192 -- Save the current state of the TradeSkillFrame so it can be restored after we muck with it.
jcallahan@169	193 local have_materials = _G.TradeSkillFrame.filterTbl.hasMaterials
jcallahan@169	194 local have_skillup = _G.TradeSkillFrame.filterTbl.hasSkillUp
jcallahan@169	195
jcallahan@169	196 if have_materials then
jcallahan@169	197 _G.TradeSkillFrame.filterTbl.hasMaterials = false
jcallahan@169	198 _G.TradeSkillOnlyShowMakeable(false)
jcallahan@169	199 end
jcallahan@169	200
jcallahan@169	201 if have_skillup then
jcallahan@169	202 _G.TradeSkillFrame.filterTbl.hasSkillUp = false
jcallahan@169	203 _G.TradeSkillOnlyShowSkillUps(false)
jcallahan@169	204 end
jcallahan@169	205 _G.SetTradeSkillInvSlotFilter(0, 1, 1)
jcallahan@169	206 _G.TradeSkillUpdateFilterBar()
jcallahan@169	207 _G.TradeSkillFrame_Update()
jcallahan@169	208
jcallahan@169	209 -- Expand all headers so we can see all the recipes there are
jcallahan@169	210 for tradeskill_index = 1, _G.GetNumTradeSkills() do
jcallahan@169	211 local name, tradeskill_type, _, is_expanded = _G.GetTradeSkillInfo(tradeskill_index)
jcallahan@169	212
jcallahan@169	213 if tradeskill_type == "header" or tradeskill_type == "subheader" then
jcallahan@169	214 if not is_expanded then
jcallahan@169	215 header_list[name] = true
jcallahan@169	216 _G.ExpandTradeSkillSubClass(tradeskill_index)
jcallahan@169	217 end
jcallahan@169	218 elseif iter_func(name, tradeskill_index) then
jcallahan@169	219 break
jcallahan@169	220 end
jcallahan@169	221 end
jcallahan@169	222
jcallahan@169	223 -- Restore the state of the things we changed.
jcallahan@169	224 for tradeskill_index = 1, _G.GetNumTradeSkills() do
jcallahan@169	225 local name, tradeskill_type, _, is_expanded = _G.GetTradeSkillInfo(tradeskill_index)
jcallahan@169	226
jcallahan@169	227 if header_list[name] then
jcallahan@169	228 _G.CollapseTradeSkillSubClass(tradeskill_index)
jcallahan@169	229 end
jcallahan@169	230 end
jcallahan@169	231 _G.TradeSkillFrame.filterTbl.hasMaterials = have_materials
jcallahan@169	232 _G.TradeSkillOnlyShowMakeable(have_materials)
jcallahan@169	233 _G.TradeSkillFrame.filterTbl.hasSkillUp = have_skillup
jcallahan@169	234 _G.TradeSkillOnlyShowSkillUps(have_skillup)
jcallahan@169	235
jcallahan@169	236 _G.TradeSkillUpdateFilterBar()
jcallahan@169	237 _G.TradeSkillFrame_Update()
jcallahan@167	238 end
jcallahan@169	239 end -- do-block
jcallahan@167	240
jcallahan@167	241
jcallahan@39	242 local ActualCopperCost
jcallahan@39	243 do
jcallahan@39	244 local BARTERING_SPELL_ID = 83964
jcallahan@39	245
jcallahan@39	246 local STANDING_DISCOUNTS = {
jcallahan@39	247 HATED = 0,
jcallahan@39	248 HOSTILE = 0,
jcallahan@39	249 UNFRIENDLY = 0,
jcallahan@39	250 NEUTRAL = 0,
jcallahan@39	251 FRIENDLY = 0.05,
jcallahan@39	252 HONORED = 0.1,
jcallahan@39	253 REVERED = 0.15,
jcallahan@39	254 EXALTED = 0.2,
jcallahan@39	255 }
jcallahan@39	256
jcallahan@39	257
jcallahan@39	258 function ActualCopperCost(copper_cost, rep_standing)
jcallahan@39	259 if not copper_cost or copper_cost == 0 then
jcallahan@39	260 return 0
jcallahan@39	261 end
jcallahan@39	262 local modifier = 1
jcallahan@39	263
jcallahan@39	264 if _G.IsSpellKnown(BARTERING_SPELL_ID) then
jcallahan@39	265 modifier = modifier - 0.1
jcallahan@39	266 end
jcallahan@39	267
jcallahan@39	268 if rep_standing then
jcallahan@39	269 if PLAYER_RACE == "Goblin" then
jcallahan@39	270 modifier = modifier - STANDING_DISCOUNTS["EXALTED"]
jcallahan@39	271 elseif STANDING_DISCOUNTS[rep_standing] then
jcallahan@39	272 modifier = modifier - STANDING_DISCOUNTS[rep_standing]
jcallahan@39	273 end
jcallahan@39	274 end
jcallahan@39	275 return math.floor(copper_cost / modifier)
jcallahan@39	276 end
jcallahan@39	277 end -- do-block
jcallahan@39	278
jcallahan@39	279
jcallahan@153	280 -- Called on a timer
jcallahan@177	281 local function ClearKilledNPC()
jcallahan@177	282 killed_npc_id = nil
jcallahan@177	283 end
jcallahan@177	284
jcallahan@177	285
jcallahan@203	286 local function ClearKilledBossID()
jcallahan@203	287 private.raid_finder_boss_id = nil
jcallahan@203	288 private.world_boss_id = nil
jcallahan@203	289 end
jcallahan@203	290
jcallahan@203	291
jcallahan@177	292 local function IsRaidFinderInstance(instance_type, instance_difficulty)
jcallahan@177	293 return instance_type == "raid" and instance_difficulty == 2 and _G.IsPartyLFG() and _G.IsInLFGDungeon()
jcallahan@153	294 end
jcallahan@153	295
jcallahan@153	296
jcallahan@29	297 local function InstanceDifficultyToken()
jcallahan@29	298 local _, instance_type, instance_difficulty, difficulty_name, _, _, is_dynamic = _G.GetInstanceInfo()
jcallahan@59	299 if not difficulty_name or difficulty_name == "" then
jcallahan@29	300 difficulty_name = "NONE"
jcallahan@29	301 end
jcallahan@59	302
jcallahan@59	303 if not instance_type or instance_type == "" then
jcallahan@59	304 instance_type = "NONE"
jcallahan@59	305 end
jcallahan@137	306
jcallahan@137	307 -- Raid difficulty of 2 is 25-man
jcallahan@177	308 if IsRaidFinderInstance(instance_type, instance_difficulty) then
jcallahan@137	309 difficulty_name = "LOOKING_FOR_RAID"
jcallahan@137	310 end
jcallahan@29	311 return ("%s:%s:%s"):format(instance_type:upper(), difficulty_name:upper():gsub(" ", "_"), _G.tostring(is_dynamic))
jcallahan@29	312 end
jcallahan@29	313
jcallahan@29	314
jcallahan@19	315 local function DBEntry(data_type, unit_id)
jcallahan@19	316 if not data_type or not unit_id then
jcallahan@6	317 return
jcallahan@6	318 end
jcallahan@128	319 local unit = global_db[data_type][unit_id]
jcallahan@6	320
jcallahan@10	321 if not unit then
jcallahan@187	322 unit = {}
jcallahan@187	323 global_db[data_type][unit_id] = unit
jcallahan@6	324 end
jcallahan@10	325 return unit
jcallahan@6	326 end
jcallahan@6	327
jcallahan@6	328
jcallahan@29	329 local function NPCEntry(identifier)
jcallahan@29	330 local npc = DBEntry("npcs", identifier)
jcallahan@29	331
jcallahan@29	332 if not npc then
jcallahan@29	333 return
jcallahan@22	334 end
jcallahan@29	335 local instance_token = InstanceDifficultyToken()
jcallahan@29	336 npc.encounter_data = npc.encounter_data or {}
jcallahan@29	337 npc.encounter_data[instance_token] = npc.encounter_data[instance_token] or {}
jcallahan@30	338 npc.encounter_data[instance_token].stats = npc.encounter_data[instance_token].stats or {}
jcallahan@29	339 return npc
jcallahan@22	340 end
jcallahan@22	341
jcallahan@22	342
jcallahan@1	343 local function CurrentLocationData()
jcallahan@161	344 if _G.GetCurrentMapAreaID() ~= current_area_id then
jcallahan@145	345 return _G.GetRealZoneText(), current_area_id, 0, 0, 0, InstanceDifficultyToken()
jcallahan@145	346 end
jcallahan@1	347 local map_level = _G.GetCurrentMapDungeonLevel() or 0
jcallahan@1	348 local x, y = _G.GetPlayerMapPosition("player")
jcallahan@1	349
jcallahan@1	350 x = x or 0
jcallahan@1	351 y = y or 0
jcallahan@1	352
jcallahan@1	353 if x == 0 and y == 0 then
jcallahan@1	354 for level_index = 1, _G.GetNumDungeonMapLevels() do
jcallahan@1	355 _G.SetDungeonMapLevel(level_index)
jcallahan@1	356 x, y = _G.GetPlayerMapPosition("player")
jcallahan@1	357
jcallahan@1	358 if x and y and (x > 0 or y > 0) then
jcallahan@1	359 _G.SetDungeonMapLevel(map_level)
jcallahan@1	360 map_level = level_index
jcallahan@1	361 break
jcallahan@1	362 end
jcallahan@1	363 end
jcallahan@1	364 end
jcallahan@1	365
jcallahan@1	366 if _G.DungeonUsesTerrainMap() then
jcallahan@1	367 map_level = map_level - 1
jcallahan@1	368 end
jcallahan@31	369 local x = _G.floor(x * 1000)
jcallahan@31	370 local y = _G.floor(y * 1000)
jcallahan@28	371
jcallahan@31	372 if x % 2 ~= 0 then
jcallahan@31	373 x = x + 1
jcallahan@28	374 end
jcallahan@28	375
jcallahan@31	376 if y % 2 ~= 0 then
jcallahan@31	377 y = y + 1
jcallahan@28	378 end
jcallahan@126	379 return _G.GetRealZoneText(), current_area_id, x, y, map_level, InstanceDifficultyToken()
jcallahan@1	380 end
jcallahan@1	381
jcallahan@1	382
jcallahan@1	383 local function ItemLinkToID(item_link)
jcallahan@1	384 if not item_link then
jcallahan@1	385 return
jcallahan@1	386 end
jcallahan@7	387 return tonumber(item_link:match("item:(%d+)"))
jcallahan@1	388 end
jcallahan@0	389
jcallahan@4	390
jcallahan@171	391 local function UnitTypeIsNPC(unit_type)
jcallahan@171	392 return unit_type == private.UNIT_TYPES.NPC or unit_type == private.UNIT_TYPES.VEHICLE
jcallahan@171	393 end
jcallahan@171	394
jcallahan@171	395
jcallahan@34	396 local ParseGUID
jcallahan@4	397 do
jcallahan@4	398 local UNIT_TYPE_BITMASK = 0x007
jcallahan@4	399
jcallahan@34	400 function ParseGUID(guid)
jcallahan@5	401 if not guid then
jcallahan@5	402 return
jcallahan@5	403 end
jcallahan@4	404 local types = private.UNIT_TYPES
jcallahan@4	405 local unit_type = _G.bit.band(tonumber(guid:sub(1, 5)), UNIT_TYPE_BITMASK)
jcallahan@4	406
jcallahan@10	407 if unit_type ~= types.PLAYER and unit_type ~= types.PET then
jcallahan@66	408 return unit_type, tonumber(guid:sub(6, 10), 16)
jcallahan@4	409 end
jcallahan@4	410 return unit_type
jcallahan@4	411 end
jcallahan@4	412 end -- do-block
jcallahan@4	413
jcallahan@4	414
jcallahan@128	415 local function UpdateBlacklistMaps()
jcallahan@128	416 local empty_count = 0
jcallahan@128	417
jcallahan@128	418 for index = 1, _G.MAX_BLACKLIST_BATTLEGROUNDS do
jcallahan@128	419 local map_id = _G.GetBlacklistMap(index)
jcallahan@128	420 char_db.bg_blacklist = char_db.bg_blacklist or {}
jcallahan@128	421 char_db.bg_blacklist[index] = map_id
jcallahan@128	422
jcallahan@128	423 if map_id < 0 then
jcallahan@128	424 empty_count = empty_count + 1
jcallahan@143	425 empty_count = empty_count + 1
jcallahan@128	426 end
jcallahan@128	427 end
jcallahan@128	428
jcallahan@128	429 if empty_count == _G.MAX_BLACKLIST_BATTLEGROUNDS then
jcallahan@128	430 char_db.bg_blacklist = nil
jcallahan@128	431 end
jcallahan@128	432 end
jcallahan@128	433
jcallahan@128	434
jcallahan@141	435 local UpdateDBEntryLocation
jcallahan@141	436 do
jcallahan@141	437 local pi = math.pi
jcallahan@141	438
jcallahan@141	439 -- Fishing node coordinate code based on code in GatherMate2 with permission from Kagaro.
jcallahan@141	440 local function FishingCoordinates(x, y, yard_width, yard_height)
jcallahan@141	441 local facing = _G.GetPlayerFacing()
jcallahan@141	442
jcallahan@141	443 if not facing then
jcallahan@141	444 return x, y
jcallahan@141	445 end
jcallahan@141	446 local rad = facing + pi
jcallahan@141	447 return x + math.sin(rad) * 15 / yard_width, y + math.cos(rad) * 15 / yard_height
jcallahan@10	448 end
jcallahan@10	449
jcallahan@24	450
jcallahan@141	451 function UpdateDBEntryLocation(entry_type, identifier)
jcallahan@141	452 if not identifier then
jcallahan@141	453 return
jcallahan@141	454 end
jcallahan@141	455 local zone_name, area_id, x, y, map_level, difficulty_token = CurrentLocationData()
jcallahan@141	456 local entry = DBEntry(entry_type, identifier)
jcallahan@141	457 entry[difficulty_token] = entry[difficulty_token] or {}
jcallahan@141	458 entry[difficulty_token].locations = entry[difficulty_token].locations or {}
jcallahan@141	459
jcallahan@141	460 local zone_token = ("%s:%d"):format(zone_name, area_id)
jcallahan@141	461 local zone_data = entry[difficulty_token].locations[zone_token]
jcallahan@141	462
jcallahan@141	463 if not zone_data then
jcallahan@141	464 zone_data = {}
jcallahan@141	465 entry[difficulty_token].locations[zone_token] = zone_data
jcallahan@141	466 end
jcallahan@141	467
jcallahan@141	468 -- Special case for Fishing.
jcallahan@141	469 if current_action.spell_label == "FISHING" then
jcallahan@141	470 local yard_width, yard_height = MapData:MapArea(area_id, map_level)
jcallahan@141	471
jcallahan@141	472 if yard_width > 0 and yard_height > 0 then
jcallahan@141	473 x, y = FishingCoordinates(x, y, yard_width, yard_height)
jcallahan@141	474 current_action.x = x
jcallahan@141	475 current_action.y = y
jcallahan@141	476 end
jcallahan@141	477 end
jcallahan@141	478 local location_token = ("%d:%d:%d"):format(map_level, x, y)
jcallahan@141	479
jcallahan@141	480 zone_data[location_token] = zone_data[location_token] or true
jcallahan@141	481 return zone_data
jcallahan@10	482 end
jcallahan@141	483 end -- do-block
jcallahan@10	484
jcallahan@10	485
jcallahan@19	486 local function HandleItemUse(item_link, bag_index, slot_index)
jcallahan@19	487 if not item_link then
jcallahan@19	488 return
jcallahan@19	489 end
jcallahan@19	490 local item_id = ItemLinkToID(item_link)
jcallahan@19	491
jcallahan@19	492 if not bag_index or not slot_index then
jcallahan@19	493 for new_bag_index = 0, _G.NUM_BAG_FRAMES do
jcallahan@19	494 for new_slot_index = 1, _G.GetContainerNumSlots(new_bag_index) do
jcallahan@19	495 if item_id == ItemLinkToID(_G.GetContainerItemLink(new_bag_index, new_slot_index)) then
jcallahan@19	496 bag_index = new_bag_index
jcallahan@19	497 slot_index = new_slot_index
jcallahan@19	498 break
jcallahan@19	499 end
jcallahan@19	500 end
jcallahan@19	501 end
jcallahan@19	502 end
jcallahan@19	503
jcallahan@19	504 if not bag_index or not slot_index then
jcallahan@19	505 return
jcallahan@19	506 end
jcallahan@19	507 local _, _, _, _, _, is_lootable = _G.GetContainerItemInfo(bag_index, slot_index)
jcallahan@19	508
jcallahan@19	509 if not is_lootable then
jcallahan@19	510 return
jcallahan@19	511 end
jcallahan@19	512 DatamineTT:ClearLines()
jcallahan@19	513 DatamineTT:SetBagItem(bag_index, slot_index)
jcallahan@19	514
jcallahan@19	515 for line_index = 1, DatamineTT:NumLines() do
jcallahan@19	516 local current_line = _G["WDPDatamineTTTextLeft" .. line_index]
jcallahan@19	517
jcallahan@19	518 if not current_line then
jcallahan@19	519 break
jcallahan@19	520 end
jcallahan@19	521
jcallahan@19	522 if current_line:GetText() == _G.ITEM_OPENABLE then
jcallahan@122	523 table.wipe(current_action)
jcallahan@123	524 current_action.target_type = AF.ITEM
jcallahan@122	525 current_action.identifier = item_id
jcallahan@122	526 current_action.loot_label = "contains"
jcallahan@19	527 break
jcallahan@19	528 end
jcallahan@19	529 end
jcallahan@19	530 end
jcallahan@19	531
jcallahan@19	532
jcallahan@39	533 local UnitFactionStanding
jcallahan@32	534 local UpdateFactionData
jcallahan@32	535 do
jcallahan@32	536 local MAX_FACTION_INDEX = 1000
jcallahan@20	537
jcallahan@32	538 local STANDING_NAMES = {
jcallahan@32	539 "HATED",
jcallahan@32	540 "HOSTILE",
jcallahan@32	541 "UNFRIENDLY",
jcallahan@32	542 "NEUTRAL",
jcallahan@32	543 "FRIENDLY",
jcallahan@32	544 "HONORED",
jcallahan@32	545 "REVERED",
jcallahan@32	546 "EXALTED",
jcallahan@32	547 }
jcallahan@32	548
jcallahan@39	549
jcallahan@39	550 function UnitFactionStanding(unit)
jcallahan@135	551 local unit_name = _G.UnitName(unit)
jcallahan@39	552 UpdateFactionData()
jcallahan@39	553 DatamineTT:ClearLines()
jcallahan@39	554 DatamineTT:SetUnit(unit)
jcallahan@39	555
jcallahan@39	556 for line_index = 1, DatamineTT:NumLines() do
jcallahan@64	557 local faction_name = _G["WDPDatamineTTTextLeft" .. line_index]:GetText():trim()
jcallahan@39	558
jcallahan@135	559 if faction_name and faction_name ~= unit_name and faction_standings[faction_name] then
jcallahan@39	560 return faction_name, faction_standings[faction_name]
jcallahan@39	561 end
jcallahan@39	562 end
jcallahan@39	563 end
jcallahan@39	564
jcallahan@39	565
jcallahan@32	566 function UpdateFactionData()
jcallahan@32	567 for faction_index = 1, MAX_FACTION_INDEX do
jcallahan@32	568 local faction_name, _, current_standing, _, _, _, _, _, is_header = _G.GetFactionInfo(faction_index)
jcallahan@32	569
jcallahan@86	570 if faction_name then
jcallahan@32	571 faction_standings[faction_name] = STANDING_NAMES[current_standing]
jcallahan@32	572 elseif not faction_name then
jcallahan@32	573 break
jcallahan@32	574 end
jcallahan@20	575 end
jcallahan@20	576 end
jcallahan@32	577 end -- do-block
jcallahan@20	578
jcallahan@48	579
jcallahan@75	580 local GenericLootUpdate
jcallahan@75	581 do
jcallahan@77	582 local function LootTable(entry, loot_type, top_field)
jcallahan@75	583 if top_field then
jcallahan@75	584 entry[top_field] = entry[top_field] or {}
jcallahan@75	585 entry[top_field][loot_type] = entry[top_field][loot_type] or {}
jcallahan@75	586 return entry[top_field][loot_type]
jcallahan@75	587 end
jcallahan@48	588 entry[loot_type] = entry[loot_type] or {}
jcallahan@75	589 return entry[loot_type]
jcallahan@48	590 end
jcallahan@48	591
jcallahan@75	592 function GenericLootUpdate(data_type, top_field)
jcallahan@132	593 local loot_type = current_loot.label
jcallahan@75	594 local loot_count = ("%s_count"):format(loot_type)
jcallahan@77	595 local source_list = {}
jcallahan@75	596
jcallahan@131	597 if current_loot.sources then
jcallahan@131	598 for source_guid, loot_data in pairs(current_loot.sources) do
jcallahan@119	599 local entry, source_id
jcallahan@78	600
jcallahan@131	601 if current_loot.target_type == AF.ITEM then
jcallahan@182	602 -- Debug(("GenericLootUpdate: current_loot.identifier: '%s'"):format(tostring(current_loot.identifier)))
jcallahan@119	603 -- Items return the player as the source, so we need to use the item's ID (disenchant, milling, etc)
jcallahan@131	604 source_id = current_loot.identifier
jcallahan@119	605 entry = DBEntry(data_type, source_id)
jcallahan@131	606 elseif current_loot.target_type == AF.OBJECT then
jcallahan@131	607 source_id = ("%s:%s"):format(current_loot.spell_label, select(2, ParseGUID(source_guid)))
jcallahan@119	608 entry = DBEntry(data_type, source_id)
jcallahan@119	609 else
jcallahan@119	610 source_id = select(2, ParseGUID(source_guid))
jcallahan@119	611 entry = DBEntry(data_type, source_id)
jcallahan@119	612 end
jcallahan@75	613
jcallahan@119	614 if entry then
jcallahan@119	615 local loot_table = LootTable(entry, loot_type, top_field)
jcallahan@77	616
jcallahan@119	617 if not source_list[source_guid] then
jcallahan@119	618 if top_field then
jcallahan@119	619 entry[top_field][loot_count] = (entry[top_field][loot_count] or 0) + 1
jcallahan@119	620 else
jcallahan@119	621 entry[loot_count] = (entry[loot_count] or 0) + 1
jcallahan@119	622 end
jcallahan@119	623 source_list[source_guid] = true
jcallahan@77	624 end
jcallahan@119	625 UpdateDBEntryLocation(data_type, source_id)
jcallahan@75	626
jcallahan@119	627 for item_id, quantity in pairs(loot_data) do
jcallahan@119	628 table.insert(loot_table, ("%d:%d"):format(item_id, quantity))
jcallahan@119	629 end
jcallahan@75	630 end
jcallahan@75	631 end
jcallahan@75	632 end
jcallahan@121	633
jcallahan@121	634 -- This is used for Gas Extractions.
jcallahan@131	635 if #current_loot.list <= 0 then
jcallahan@78	636 return
jcallahan@78	637 end
jcallahan@82	638 local entry
jcallahan@82	639
jcallahan@82	640 -- At this point we only have a name if it's an object.
jcallahan@131	641 if current_loot.target_type == AF.OBJECT then
jcallahan@131	642 entry = DBEntry(data_type, ("%s:%s"):format(current_loot.spell_label, current_loot.object_name))
jcallahan@82	643 else
jcallahan@131	644 entry = DBEntry(data_type, current_loot.identifier)
jcallahan@82	645 end
jcallahan@75	646
jcallahan@75	647 if not entry then
jcallahan@75	648 return
jcallahan@75	649 end
jcallahan@77	650 local loot_table = LootTable(entry, loot_type, top_field)
jcallahan@77	651
jcallahan@131	652 if not source_list[current_loot.identifier] then
jcallahan@77	653 if top_field then
jcallahan@77	654 entry[top_field][loot_count] = (entry[top_field][loot_count] or 0) + 1
jcallahan@77	655 else
jcallahan@77	656 entry[loot_count] = (entry[loot_count] or 0) + 1
jcallahan@77	657 end
jcallahan@77	658 end
jcallahan@75	659
jcallahan@131	660 for index = 1, #current_loot.list do
jcallahan@131	661 table.insert(loot_table, current_loot.list[index])
jcallahan@75	662 end
jcallahan@48	663 end
jcallahan@75	664 end -- do-block
jcallahan@48	665
jcallahan@97	666
jcallahan@97	667 local ReplaceKeywords
jcallahan@97	668 do
jcallahan@97	669 local KEYWORD_SUBSTITUTIONS = {
jcallahan@97	670 class = PLAYER_CLASS,
jcallahan@97	671 name = PLAYER_NAME,
jcallahan@97	672 race = PLAYER_RACE,
jcallahan@97	673 }
jcallahan@97	674
jcallahan@97	675
jcallahan@97	676 function ReplaceKeywords(text)
jcallahan@97	677 if not text or text == "" then
jcallahan@97	678 return ""
jcallahan@97	679 end
jcallahan@97	680
jcallahan@97	681 for category, lookup in pairs(KEYWORD_SUBSTITUTIONS) do
jcallahan@97	682 local category_format = ("<%s>"):format(category)
jcallahan@97	683 text = text:gsub(lookup, category_format):gsub(lookup:lower(), category_format)
jcallahan@97	684 end
jcallahan@97	685 return text
jcallahan@97	686 end
jcallahan@97	687 end -- do-block
jcallahan@97	688
jcallahan@97	689
jcallahan@154	690 -- Contains a dirty hack due to Blizzard's strange handling of Micro Dungeons; GetMapInfo() will not return correct information
jcallahan@154	691 -- unless the WorldMapFrame is shown.
jcallahan@143	692 do
jcallahan@143	693 -- MapFileName = MapAreaID
jcallahan@143	694 local MICRO_DUNGEON_IDS = {
jcallahan@143	695 ShrineofTwoMoons = 903,
jcallahan@143	696 ShrineofSevenStars = 905,
jcallahan@143	697 }
jcallahan@126	698
jcallahan@143	699 function WDP:SetCurrentAreaID(event_name)
jcallahan@156	700 if private.in_combat then
jcallahan@156	701 private.set_area_id = true
jcallahan@156	702 return
jcallahan@156	703 end
jcallahan@155	704 local map_area_id = _G.GetCurrentMapAreaID()
jcallahan@155	705
jcallahan@155	706 if map_area_id == current_area_id then
jcallahan@155	707 return
jcallahan@155	708 end
jcallahan@143	709 local world_map = _G.WorldMapFrame
jcallahan@143	710 local map_visible = world_map:IsVisible()
jcallahan@143	711 local sfx_value = _G.tonumber(_G.GetCVar("Sound_EnableSFX"))
jcallahan@143	712
jcallahan@143	713 if not map_visible then
jcallahan@143	714 _G.SetCVar("Sound_EnableSFX", 0)
jcallahan@143	715 world_map:Show()
jcallahan@143	716 end
jcallahan@143	717 local micro_dungeon_id = MICRO_DUNGEON_IDS[select(5, _G.GetMapInfo())]
jcallahan@143	718
jcallahan@154	719 _G.SetMapToCurrentZone()
jcallahan@154	720
jcallahan@143	721 if micro_dungeon_id then
jcallahan@143	722 current_area_id = micro_dungeon_id
jcallahan@143	723 else
jcallahan@143	724 current_area_id = _G.GetCurrentMapAreaID()
jcallahan@143	725 end
jcallahan@143	726
jcallahan@154	727 if map_visible then
jcallahan@154	728 _G.SetMapByID(map_area_id)
jcallahan@154	729 else
jcallahan@143	730 world_map:Hide()
jcallahan@143	731 _G.SetCVar("Sound_EnableSFX", sfx_value)
jcallahan@143	732 end
jcallahan@143	733 end
jcallahan@154	734 end
jcallahan@126	735
jcallahan@0	736 -----------------------------------------------------------------------
jcallahan@0	737 -- Methods.
jcallahan@0	738 -----------------------------------------------------------------------
jcallahan@0	739 function WDP:OnInitialize()
jcallahan@128	740 local db = LibStub("AceDB-3.0"):New("WoWDBProfilerData", DATABASE_DEFAULTS, "Default")
jcallahan@128	741 global_db = db.global
jcallahan@128	742 char_db = db.char
jcallahan@14	743
jcallahan@14	744 local raw_db = _G["WoWDBProfilerData"]
jcallahan@18	745 local build_num = tonumber(private.build_num)
jcallahan@14	746
jcallahan@136	747 if (raw_db.version and raw_db.version < DB_VERSION) or (raw_db.build_num and raw_db.build_num < build_num) then
jcallahan@74	748 for entry in pairs(DATABASE_DEFAULTS.global) do
jcallahan@128	749 global_db[entry] = {}
jcallahan@74	750 end
jcallahan@74	751 end
jcallahan@35	752 raw_db.build_num = build_num
jcallahan@63	753 raw_db.version = DB_VERSION
jcallahan@0	754 end
jcallahan@0	755
jcallahan@0	756
jcallahan@153	757 function WDP:EventDispatcher(...)
jcallahan@153	758 local event_name = ...
jcallahan@153	759
jcallahan@153	760 if DEBUGGING then
jcallahan@154	761 if event_name == "COMBAT_LOG_EVENT_UNFILTERED" then
jcallahan@154	762 Debug(event_name)
jcallahan@154	763 else
jcallahan@154	764 Debug(...)
jcallahan@153	765 end
jcallahan@153	766 end
jcallahan@153	767 local func = EVENT_MAPPING[event_name]
jcallahan@153	768
jcallahan@153	769 if _G.type(func) == "boolean" then
jcallahan@153	770 self[event_name](self, ...)
jcallahan@153	771 elseif _G.type(func) == "function" then
jcallahan@159	772 self[func](self, ...)
jcallahan@153	773 end
jcallahan@153	774 end
jcallahan@153	775
jcallahan@153	776
jcallahan@0	777 function WDP:OnEnable()
jcallahan@0	778 for event_name, mapping in pairs(EVENT_MAPPING) do
jcallahan@156	779 if EVENT_DEBUG then
jcallahan@153	780 self:RegisterEvent(event_name, "EventDispatcher")
jcallahan@153	781 else
jcallahan@153	782 self:RegisterEvent(event_name, (_G.type(mapping) ~= "boolean") and mapping or nil)
jcallahan@153	783 end
jcallahan@0	784 end
jcallahan@95	785
jcallahan@95	786 for index = 1, _G.GetNumLanguages() do
jcallahan@95	787 languages_known[_G.GetLanguageByIndex(index)] = true
jcallahan@95	788 end
jcallahan@187	789 item_process_timer_handle = self:ScheduleRepeatingTimer("ProcessItems", 30)
jcallahan@31	790 target_location_timer_handle = self:ScheduleRepeatingTimer("UpdateTargetLocation", 0.5)
jcallahan@19	791
jcallahan@19	792 _G.hooksecurefunc("UseContainerItem", function(bag_index, slot_index, target_unit)
jcallahan@19	793 if target_unit then
jcallahan@19	794 return
jcallahan@19	795 end
jcallahan@19	796 HandleItemUse(_G.GetContainerItemLink(bag_index, slot_index), bag_index, slot_index)
jcallahan@19	797 end)
jcallahan@19	798
jcallahan@19	799 _G.hooksecurefunc("UseItemByName", function(identifier, target_unit)
jcallahan@19	800 if target_unit then
jcallahan@19	801 return
jcallahan@19	802 end
jcallahan@19	803 local _, item_link = _G.GetItemInfo(identifier)
jcallahan@19	804 HandleItemUse(item_link)
jcallahan@19	805 end)
jcallahan@129	806 self:SetCurrentAreaID("OnEnable")
jcallahan@126	807
jcallahan@128	808 _G.hooksecurefunc("SetBlacklistMap", UpdateBlacklistMaps)
jcallahan@128	809 _G.hooksecurefunc("ClearBlacklistMap", UpdateBlacklistMaps)
jcallahan@128	810
jcallahan@128	811 UpdateBlacklistMaps()
jcallahan@0	812 end
jcallahan@0	813
jcallahan@0	814
jcallahan@187	815 local function RecordItemData(item_id, item_link, durability)
jcallahan@0	816 if not durability or durability <= 0 then
jcallahan@0	817 return
jcallahan@0	818 end
jcallahan@187	819 local item = DBEntry("items", item_id)
jcallahan@191	820 local item_string = select(3, item_link:find("^\|%x+\|H(.+)\|h%[.+%]"))
jcallahan@0	821
jcallahan@191	822 if item_string then
jcallahan@199	823 local _, _, _, _, _, _, _, suffix_id, unique_id, _, _, upgrade_id = (":"):split(item_string)
jcallahan@201	824 suffix_id = tonumber(suffix_id)
jcallahan@199	825
jcallahan@201	826 if suffix_id and suffix_id ~= 0 then
jcallahan@201	827 item.suffix_id = suffix_id
jcallahan@201	828 item.unique_id = bit.band(unique_id, 0xFFFF)
jcallahan@201	829 end
jcallahan@191	830
jcallahan@191	831 if upgrade_id then
jcallahan@191	832 item.upgrade_id = upgrade_id
jcallahan@191	833 end
jcallahan@0	834 end
jcallahan@187	835 item.durability = durability
jcallahan@0	836 end
jcallahan@0	837
jcallahan@0	838
jcallahan@187	839 function WDP:ProcessItems()
jcallahan@187	840 for slot_index = _G.INVSLOT_FIRST_EQUIPPED, _G.INVSLOT_LAST_EQUIPPED do
jcallahan@1	841 local item_id = _G.GetInventoryItemID("player", slot_index)
jcallahan@0	842
jcallahan@0	843 if item_id and item_id > 0 then
jcallahan@1	844 local _, max_durability = _G.GetInventoryItemDurability(slot_index)
jcallahan@187	845 RecordItemData(item_id, _G.GetInventoryItemLink("player", slot_index), max_durability)
jcallahan@0	846 end
jcallahan@0	847 end
jcallahan@0	848
jcallahan@0	849 for bag_index = 0, _G.NUM_BAG_SLOTS do
jcallahan@0	850 for slot_index = 1, _G.GetContainerNumSlots(bag_index) do
jcallahan@1	851 local item_id = _G.GetContainerItemID(bag_index, slot_index)
jcallahan@0	852
jcallahan@0	853 if item_id and item_id > 0 then
jcallahan@1	854 local _, max_durability = _G.GetContainerItemDurability(bag_index, slot_index)
jcallahan@187	855 RecordItemData(item_id, _G.GetContainerItemLink(bag_index, slot_index), max_durability)
jcallahan@0	856 end
jcallahan@0	857 end
jcallahan@0	858 end
jcallahan@0	859 end
jcallahan@0	860
jcallahan@118	861
jcallahan@118	862 local PlayerTarget
jcallahan@118	863 do
jcallahan@118	864 local GENDER_NAMES = {
jcallahan@118	865 "UNKNOWN",
jcallahan@118	866 "MALE",
jcallahan@118	867 "FEMALE",
jcallahan@118	868 }
jcallahan@118	869
jcallahan@118	870
jcallahan@118	871 local REACTION_NAMES = {
jcallahan@118	872 "HATED",
jcallahan@118	873 "HOSTILE",
jcallahan@118	874 "UNFRIENDLY",
jcallahan@118	875 "NEUTRAL",
jcallahan@118	876 "FRIENDLY",
jcallahan@118	877 "HONORED",
jcallahan@118	878 "REVERED",
jcallahan@118	879 "EXALTED",
jcallahan@118	880 }
jcallahan@118	881
jcallahan@118	882
jcallahan@118	883 local POWER_TYPE_NAMES = {
jcallahan@118	884 ["0"] = "MANA",
jcallahan@118	885 ["1"] = "RAGE",
jcallahan@118	886 ["2"] = "FOCUS",
jcallahan@118	887 ["3"] = "ENERGY",
jcallahan@118	888 ["6"] = "RUNIC_POWER",
jcallahan@118	889 }
jcallahan@118	890
jcallahan@118	891
jcallahan@118	892 function PlayerTarget()
jcallahan@118	893 if not _G.UnitExists("target") or _G.UnitPlayerControlled("target") or currently_drunk then
jcallahan@118	894 current_target_id = nil
jcallahan@118	895 return
jcallahan@118	896 end
jcallahan@118	897 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("target"))
jcallahan@118	898
jcallahan@171	899 if not unit_idnum or not UnitTypeIsNPC(unit_type) then
jcallahan@118	900 return
jcallahan@118	901 end
jcallahan@118	902 current_target_id = unit_idnum
jcallahan@118	903
jcallahan@118	904 local npc = NPCEntry(unit_idnum)
jcallahan@118	905 local _, class_token = _G.UnitClass("target")
jcallahan@118	906 npc.class = class_token
jcallahan@118	907 npc.faction = UnitFactionStanding("target")
jcallahan@118	908 npc.genders = npc.genders or {}
jcallahan@118	909 npc.genders[GENDER_NAMES[_G.UnitSex("target")] or "UNDEFINED"] = true
jcallahan@118	910 npc.is_pvp = _G.UnitIsPVP("target") and true or nil
jcallahan@118	911 npc.reaction = ("%s:%s:%s"):format(_G.UnitLevel("player"), _G.UnitFactionGroup("player"), REACTION_NAMES[_G.UnitReaction("player", "target")])
jcallahan@118	912
jcallahan@118	913 local encounter_data = npc.encounter_data[InstanceDifficultyToken()].stats
jcallahan@118	914 local npc_level = ("level_%d"):format(_G.UnitLevel("target"))
jcallahan@118	915
jcallahan@118	916 if not encounter_data[npc_level] then
jcallahan@118	917 encounter_data[npc_level] = {
jcallahan@118	918 max_health = _G.UnitHealthMax("target"),
jcallahan@118	919 }
jcallahan@118	920
jcallahan@118	921 local max_power = _G.UnitManaMax("target")
jcallahan@118	922
jcallahan@118	923 if max_power > 0 then
jcallahan@118	924 local power_type = _G.UnitPowerType("target")
jcallahan@118	925 encounter_data[npc_level].power = ("%s:%d"):format(POWER_TYPE_NAMES[_G.tostring(power_type)] or power_type, max_power)
jcallahan@118	926 end
jcallahan@118	927 end
jcallahan@118	928 name_to_id_map[_G.UnitName("target")] = unit_idnum
jcallahan@122	929 current_action.identifier = unit_idnum
jcallahan@118	930 return npc, unit_idnum
jcallahan@118	931 end
jcallahan@118	932 end -- do-block
jcallahan@118	933
jcallahan@118	934
jcallahan@113	935 do
jcallahan@113	936 local COORD_MAX = 5
jcallahan@0	937
jcallahan@113	938 function WDP:UpdateTargetLocation()
jcallahan@113	939 if currently_drunk or not _G.UnitExists("target") or _G.UnitPlayerControlled("target") or (_G.UnitIsTapped("target") and not _G.UnitIsDead("target")) then
jcallahan@2	940 return
jcallahan@2	941 end
jcallahan@113	942
jcallahan@113	943 for index = 1, 4 do
jcallahan@113	944 if not _G.CheckInteractDistance("target", index) then
jcallahan@113	945 return
jcallahan@113	946 end
jcallahan@113	947 end
jcallahan@118	948 local npc = PlayerTarget()
jcallahan@113	949
jcallahan@113	950 if not npc then
jcallahan@113	951 return
jcallahan@113	952 end
jcallahan@113	953 local zone_name, area_id, x, y, map_level, difficulty_token = CurrentLocationData()
jcallahan@113	954 local npc_data = npc.encounter_data[difficulty_token].stats[("level_%d"):format(_G.UnitLevel("target"))]
jcallahan@113	955 local zone_token = ("%s:%d"):format(zone_name, area_id)
jcallahan@118	956 npc_data.locations = npc_data.locations or {} -- TODO: Fix this. It is broken. Possibly something to do with the timed updates.
jcallahan@113	957
jcallahan@113	958 local zone_data = npc_data.locations[zone_token]
jcallahan@113	959
jcallahan@113	960 if not zone_data then
jcallahan@113	961 zone_data = {}
jcallahan@113	962 npc_data.locations[zone_token] = zone_data
jcallahan@113	963 end
jcallahan@113	964
jcallahan@113	965 for location_token in pairs(zone_data) do
jcallahan@113	966 local loc_level, loc_x, loc_y = (":"):split(location_token)
jcallahan@113	967 loc_level = tonumber(loc_level)
jcallahan@113	968
jcallahan@113	969 if map_level == loc_level and math.abs(x - loc_x) <= COORD_MAX and math.abs(y - loc_y) <= COORD_MAX then
jcallahan@113	970 return
jcallahan@113	971 end
jcallahan@113	972 end
jcallahan@141	973 zone_data[("%d:%d:%d"):format(map_level, x, y)] = true
jcallahan@2	974 end
jcallahan@113	975 end -- do-block
jcallahan@2	976
jcallahan@118	977
jcallahan@0	978 -----------------------------------------------------------------------
jcallahan@0	979 -- Event handlers.
jcallahan@0	980 -----------------------------------------------------------------------
jcallahan@90	981 function WDP:BLACK_MARKET_ITEM_UPDATE(event_name)
jcallahan@56	982 local num_items = _G.C_BlackMarket.GetNumItems()
jcallahan@56	983
jcallahan@56	984 for index = 1, num_items do
jcallahan@56	985 local name, texture, quantity, item_type, is_usable, level, level_type, seller_name, min_bid, min_increment, current_bid, has_high_bid, num_bids, time_left, item_link, market_id = _G.C_BlackMarket.GetItemInfoByIndex(index);
jcallahan@56	986
jcallahan@56	987 if item_link then
jcallahan@56	988 DBEntry("items", ItemLinkToID(item_link)).black_market = seller_name or "UNKNOWN"
jcallahan@56	989 end
jcallahan@56	990 end
jcallahan@56	991 end
jcallahan@56	992
jcallahan@56	993
jcallahan@178	994 function WDP:SHOW_LOOT_TOAST(event_name, loot_type, item_link, quantity)
jcallahan@178	995 if loot_type ~= "item" then
jcallahan@182	996 Debug(("%s: loot_type is '%s'"):format(event_name, loot_type))
jcallahan@178	997 return
jcallahan@178	998 end
jcallahan@203	999 local npc = NPCEntry(private.raid_finder_boss_id or private.world_boss_id)
jcallahan@179	1000 private.raid_finder_boss_id = nil
jcallahan@203	1001 private.world_boss_id = nil
jcallahan@178	1002
jcallahan@178	1003 if not npc then
jcallahan@182	1004 Debug(("%s: NPC is nil."):format(event_name))
jcallahan@178	1005 return
jcallahan@178	1006 end
jcallahan@178	1007 local item_id = ItemLinkToID(item_link)
jcallahan@178	1008
jcallahan@178	1009 if not item_id then
jcallahan@182	1010 Debug(("%s: ItemID is nil, from item link %s"):format(event_name, item_link))
jcallahan@178	1011 return
jcallahan@178	1012 end
jcallahan@179	1013 local loot_type = "drops"
jcallahan@178	1014 local encounter_data = npc.encounter_data[InstanceDifficultyToken()]
jcallahan@178	1015 encounter_data[loot_type] = encounter_data[loot_type] or {}
jcallahan@178	1016 encounter_data.loot_counts = encounter_data.loot_counts or {}
jcallahan@178	1017 encounter_data.loot_counts[loot_type] = (encounter_data.loot_counts[loot_type] or 0) + 1
jcallahan@178	1018
jcallahan@178	1019 table.insert(encounter_data[loot_type], ("%d:%d"):format(item_id, quantity))
jcallahan@182	1020 Debug(("%s: %sX%d (%d)"):format(event_name, item_link, quantity, item_id))
jcallahan@178	1021 end
jcallahan@178	1022
jcallahan@178	1023
jcallahan@179	1024 do
jcallahan@179	1025 local CHAT_MSG_LOOT_UPDATE_FUNCS = {
jcallahan@179	1026 [AF.NPC] = function(item_id, quantity)
jcallahan@179	1027 Debug(("CHAT_MSG_LOOT: %d (%d)"):format(item_id, quantity))
jcallahan@179	1028 end,
jcallahan@179	1029 [AF.ZONE] = function(item_id, quantity)
jcallahan@179	1030 current_loot = {
jcallahan@179	1031 list = {
jcallahan@179	1032 ("%d:%d"):format(item_id, quantity)
jcallahan@179	1033 },
jcallahan@179	1034 identifier = current_action.identifier,
jcallahan@179	1035 label = current_action.loot_label or "drops",
jcallahan@179	1036 map_level = current_action.map_level,
jcallahan@179	1037 object_name = current_action.object_name,
jcallahan@179	1038 spell_label = current_action.spell_label,
jcallahan@179	1039 target_type = current_action.target_type,
jcallahan@179	1040 x = current_action.x,
jcallahan@179	1041 y = current_action.y,
jcallahan@179	1042 zone_data = current_action.zone_data,
jcallahan@179	1043 }
jcallahan@179	1044 table.wipe(current_action)
jcallahan@179	1045 GenericLootUpdate("zones")
jcallahan@179	1046 end,
jcallahan@179	1047 }
jcallahan@177	1048
jcallahan@177	1049
jcallahan@179	1050 function WDP:CHAT_MSG_LOOT(event_name, message)
jcallahan@179	1051 local category
jcallahan@177	1052
jcallahan@179	1053 if current_action.spell_label ~= "EXTRACT_GAS" then
jcallahan@179	1054 category = AF.ZONE
jcallahan@179	1055 elseif private.raid_finder_boss_id then
jcallahan@179	1056 category = AF.NPC
jcallahan@179	1057 end
jcallahan@179	1058 local update_func = CHAT_MSG_LOOT_UPDATE_FUNCS[category]
jcallahan@177	1059
jcallahan@179	1060 if not category or not update_func then
jcallahan@179	1061 return
jcallahan@179	1062 end
jcallahan@179	1063 local item_link, quantity = deformat(message, _G.LOOT_ITEM_PUSHED_SELF_MULTIPLE)
jcallahan@179	1064
jcallahan@179	1065 if not item_link then
jcallahan@179	1066 quantity, item_link = 1, deformat(message, _G.LOOT_ITEM_PUSHED_SELF)
jcallahan@179	1067 end
jcallahan@179	1068 local item_id = ItemLinkToID(item_link)
jcallahan@179	1069
jcallahan@179	1070 if not item_id then
jcallahan@179	1071 return
jcallahan@179	1072 end
jcallahan@179	1073 update_func(item_id, quantity)
jcallahan@177	1074 end
jcallahan@48	1075 end
jcallahan@48	1076
jcallahan@97	1077 function WDP:RecordQuote(event_name, message, source_name, language_name)
jcallahan@112	1078 if not ALLOWED_LOCALES[CLIENT_LOCALE] or not source_name or not name_to_id_map[source_name] or (language_name ~= "" and not languages_known[language_name]) then
jcallahan@97	1079 return
jcallahan@95	1080 end
jcallahan@97	1081 local npc = NPCEntry(name_to_id_map[source_name])
jcallahan@97	1082 npc.quotes = npc.quotes or {}
jcallahan@97	1083 npc.quotes[event_name] = npc.quotes[event_name] or {}
jcallahan@97	1084 npc.quotes[event_name][ReplaceKeywords(message)] = true
jcallahan@97	1085 end
jcallahan@95	1086
jcallahan@95	1087
jcallahan@95	1088 do
jcallahan@40	1089 local SOBER_MATCH = _G.DRUNK_MESSAGE_ITEM_SELF1:gsub("%%s", ".+")
jcallahan@40	1090
jcallahan@40	1091 local DRUNK_COMPARES = {
jcallahan@40	1092 _G.DRUNK_MESSAGE_SELF2,
jcallahan@40	1093 _G.DRUNK_MESSAGE_SELF3,
jcallahan@40	1094 _G.DRUNK_MESSAGE_SELF4,
jcallahan@40	1095 }
jcallahan@40	1096
jcallahan@40	1097 local DRUNK_MATCHES = {
jcallahan@40	1098 _G.DRUNK_MESSAGE_SELF2:gsub("%%s", ".+"),
jcallahan@40	1099 _G.DRUNK_MESSAGE_SELF3:gsub("%%s", ".+"),
jcallahan@40	1100 _G.DRUNK_MESSAGE_SELF4:gsub("%%s", ".+"),
jcallahan@40	1101 }
jcallahan@40	1102
jcallahan@167	1103 local RECIPE_MATCH = _G.ERR_LEARN_RECIPE_S:gsub("%%s", "(.*)")
jcallahan@167	1104
jcallahan@167	1105
jcallahan@167	1106 local function RecordDiscovery(tradeskill_name, tradeskill_index)
jcallahan@167	1107 if tradeskill_name == private.discovered_recipe_name then
jcallahan@167	1108 DBEntry("spells", tonumber(_G.GetTradeSkillRecipeLink(tradeskill_index):match("^\|c%x%x%x%x%x%x%x%x\|H%w+:(%d+)"))).discovery = ("%d:%d"):format(private.previous_spell_id, private.profession_level)
jcallahan@167	1109
jcallahan@167	1110 private.discovered_recipe_name = nil
jcallahan@167	1111 private.profession_level = nil
jcallahan@167	1112 private.previous_spell_id = nil
jcallahan@169	1113
jcallahan@169	1114 return true
jcallahan@167	1115 end
jcallahan@167	1116 end
jcallahan@167	1117
jcallahan@167	1118
jcallahan@167	1119 local function IterativeRecordDiscovery()
jcallahan@167	1120 TradeSkillExecutePer(RecordDiscovery)
jcallahan@167	1121 end
jcallahan@167	1122
jcallahan@167	1123
jcallahan@92	1124 function WDP:CHAT_MSG_SYSTEM(event_name, message)
jcallahan@167	1125 if not private.trainer_shown then
jcallahan@167	1126 local recipe_name = message:match(RECIPE_MATCH)
jcallahan@167	1127
jcallahan@167	1128 if recipe_name and private.previous_spell_id then
jcallahan@167	1129 local profession_name, prof_level = _G.GetTradeSkillLine()
jcallahan@167	1130
jcallahan@167	1131 if profession_name == _G.UNKNOWN then
jcallahan@167	1132 return
jcallahan@167	1133 end
jcallahan@167	1134 private.discovered_recipe_name = recipe_name
jcallahan@167	1135 private.profession_level = prof_level
jcallahan@167	1136
jcallahan@167	1137 self:ScheduleTimer(IterativeRecordDiscovery, 0.2)
jcallahan@167	1138 end
jcallahan@167	1139 end
jcallahan@167	1140
jcallahan@40	1141 if currently_drunk then
jcallahan@40	1142 if message == _G.DRUNK_MESSAGE_SELF1 or message:match(SOBER_MATCH) then
jcallahan@40	1143 currently_drunk = nil
jcallahan@40	1144 end
jcallahan@40	1145 return
jcallahan@40	1146 end
jcallahan@40	1147
jcallahan@40	1148 for index = 1, #DRUNK_MATCHES do
jcallahan@40	1149 if message == DRUNK_COMPARES[index] or message:match(DRUNK_MATCHES[index]) then
jcallahan@40	1150 currently_drunk = true
jcallahan@40	1151 break
jcallahan@40	1152 end
jcallahan@40	1153 end
jcallahan@40	1154 end
jcallahan@40	1155 end
jcallahan@40	1156
jcallahan@40	1157 -- do-block
jcallahan@40	1158
jcallahan@40	1159 do
jcallahan@23	1160 local FLAGS_NPC = bit.bor(_G.COMBATLOG_OBJECT_TYPE_GUARDIAN, _G.COMBATLOG_OBJECT_CONTROL_NPC)
jcallahan@23	1161 local FLAGS_NPC_CONTROL = bit.bor(_G.COMBATLOG_OBJECT_AFFILIATION_OUTSIDER, _G.COMBATLOG_OBJECT_CONTROL_NPC)
jcallahan@23	1162
jcallahan@23	1163 local function RecordNPCSpell(sub_event, source_guid, source_name, source_flags, dest_guid, dest_name, dest_flags, spell_id, spell_name)
jcallahan@23	1164 if not spell_id then
jcallahan@23	1165 return
jcallahan@23	1166 end
jcallahan@34	1167 local source_type, source_id = ParseGUID(source_guid)
jcallahan@23	1168
jcallahan@171	1169 if not source_id or not UnitTypeIsNPC(source_type) then
jcallahan@23	1170 return
jcallahan@23	1171 end
jcallahan@23	1172
jcallahan@23	1173 if bit.band(FLAGS_NPC_CONTROL, source_flags) == FLAGS_NPC_CONTROL and bit.band(FLAGS_NPC, source_flags) ~= 0 then
jcallahan@29	1174 local encounter_data = NPCEntry(source_id).encounter_data[InstanceDifficultyToken()]
jcallahan@28	1175 encounter_data.spells = encounter_data.spells or {}
jcallahan@28	1176 encounter_data.spells[spell_id] = (encounter_data.spells[spell_id] or 0) + 1
jcallahan@23	1177 end
jcallahan@23	1178 end
jcallahan@23	1179
jcallahan@115	1180 local HEAL_BATTLE_PETS_SPELL_ID = 125801
jcallahan@115	1181
jcallahan@23	1182 local COMBAT_LOG_FUNCS = {
jcallahan@23	1183 SPELL_AURA_APPLIED = RecordNPCSpell,
jcallahan@23	1184 SPELL_CAST_START = RecordNPCSpell,
jcallahan@115	1185 SPELL_CAST_SUCCESS = function(sub_event, source_guid, source_name, source_flags, dest_guid, dest_name, dest_flags, spell_id, spell_name)
jcallahan@115	1186 if spell_id == HEAL_BATTLE_PETS_SPELL_ID then
jcallahan@115	1187 local unit_type, unit_idnum = ParseGUID(source_guid)
jcallahan@115	1188
jcallahan@171	1189 if unit_idnum and UnitTypeIsNPC(unit_type) then
jcallahan@115	1190 NPCEntry(unit_idnum).stable_master = true
jcallahan@115	1191 end
jcallahan@115	1192 end
jcallahan@115	1193 RecordNPCSpell(sub_event, source_guid, source_name, source_flags, dest_guid, dest_name, dest_flags, spell_id, spell_name)
jcallahan@115	1194 end,
jcallahan@65	1195 UNIT_DIED = function(sub_event, source_guid, source_name, source_flags, dest_guid, dest_name, dest_flags, spell_id, spell_name)
jcallahan@65	1196 local unit_type, unit_idnum = ParseGUID(dest_guid)
jcallahan@65	1197
jcallahan@171	1198 if not unit_idnum or not UnitTypeIsNPC(unit_type) then
jcallahan@182	1199 Debug(("%s: %s is not an NPC, or has no ID."):format(sub_event, dest_name))
jcallahan@177	1200 ClearKilledNPC()
jcallahan@98	1201 private.harvesting = nil
jcallahan@65	1202 return
jcallahan@65	1203 end
jcallahan@177	1204
jcallahan@177	1205 if private.RAID_FINDER_BOSS_IDS[unit_idnum] then
jcallahan@184	1206 Debug(("%s: Matching boss %s."):format(sub_event, dest_name))
jcallahan@184	1207 private.raid_finder_boss_id = unit_idnum
jcallahan@203	1208 private.world_boss_id = nil
jcallahan@203	1209 elseif private.WORLD_BOSS_IDS[unit_idnum] then
jcallahan@203	1210 Debug(("%s: Matching world boss %s."):format(sub_event, dest_name))
jcallahan@203	1211 private.raid_finder_boss_id = nil
jcallahan@203	1212 private.world_boss_id = unit_idnum
jcallahan@182	1213 else
jcallahan@203	1214 Debug(("%s: Killed NPC %s (ID: %d) is not in LFG or World boss list."):format(sub_event, dest_name, unit_idnum))
jcallahan@177	1215 end
jcallahan@183	1216
jcallahan@183	1217 if dest_guid ~= _G.UnitGUID("target") then
jcallahan@183	1218 ClearKilledNPC()
jcallahan@183	1219 return
jcallahan@183	1220 end
jcallahan@177	1221 killed_npc_id = unit_idnum
jcallahan@177	1222 WDP:ScheduleTimer(ClearKilledNPC, 0.1)
jcallahan@203	1223 WDP:ScheduleTimer(ClearKilledBossID, 1)
jcallahan@65	1224 end,
jcallahan@23	1225 }
jcallahan@23	1226
jcallahan@23	1227
jcallahan@92	1228 function WDP:COMBAT_LOG_EVENT_UNFILTERED(event_name, time_stamp, sub_event, hide_caster, source_guid, source_name, source_flags, source_raid_flags, dest_guid, dest_name, dest_flags, dest_raid_flags, ...)
jcallahan@23	1229 local combat_log_func = COMBAT_LOG_FUNCS[sub_event]
jcallahan@23	1230
jcallahan@23	1231 if not combat_log_func then
jcallahan@23	1232 return
jcallahan@23	1233 end
jcallahan@23	1234 combat_log_func(sub_event, source_guid, source_name, source_flags, dest_guid, dest_name, dest_flags, ...)
jcallahan@23	1235 end
jcallahan@23	1236
jcallahan@44	1237 local DIPLOMACY_SPELL_ID = 20599
jcallahan@44	1238 local MR_POP_RANK1_SPELL_ID = 78634
jcallahan@44	1239 local MR_POP_RANK2_SPELL_ID = 78635
jcallahan@44	1240
jcallahan@44	1241 local REP_BUFFS = {
jcallahan@44	1242 [_G.GetSpellInfo(30754)] = "CENARION_FAVOR",
jcallahan@44	1243 [_G.GetSpellInfo(24705)] = "GRIM_VISAGE",
jcallahan@44	1244 [_G.GetSpellInfo(32098)] = "HONOR_HOLD_FAVOR",
jcallahan@44	1245 [_G.GetSpellInfo(39913)] = "NAZGRELS_FERVOR",
jcallahan@44	1246 [_G.GetSpellInfo(39953)] = "SONG_OF_BATTLE",
jcallahan@44	1247 [_G.GetSpellInfo(61849)] = "SPIRIT_OF_SHARING",
jcallahan@44	1248 [_G.GetSpellInfo(32096)] = "THRALLMARS_FAVOR",
jcallahan@44	1249 [_G.GetSpellInfo(39911)] = "TROLLBANES_COMMAND",
jcallahan@44	1250 [_G.GetSpellInfo(95987)] = "UNBURDENED",
jcallahan@44	1251 [_G.GetSpellInfo(100951)] = "WOW_ANNIVERSARY",
jcallahan@44	1252 }
jcallahan@44	1253
jcallahan@44	1254
jcallahan@44	1255 local FACTION_NAMES = {
jcallahan@44	1256 CENARION_CIRCLE = _G.GetFactionInfoByID(609),
jcallahan@44	1257 HONOR_HOLD = _G.GetFactionInfoByID(946),
jcallahan@44	1258 THE_SHATAR = _G.GetFactionInfoByID(935),
jcallahan@44	1259 THRALLMAR = _G.GetFactionInfoByID(947),
jcallahan@44	1260 }
jcallahan@44	1261
jcallahan@44	1262
jcallahan@44	1263 local MODIFIERS = {
jcallahan@44	1264 CENARION_FAVOR = {
jcallahan@44	1265 faction = FACTION_NAMES.CENARION_CIRCLE,
jcallahan@44	1266 modifier = 0.25,
jcallahan@44	1267 },
jcallahan@44	1268 GRIM_VISAGE = {
jcallahan@44	1269 modifier = 0.1,
jcallahan@44	1270 },
jcallahan@44	1271 HONOR_HOLD_FAVOR = {
jcallahan@44	1272 faction = FACTION_NAMES.HONOR_HOLD,
jcallahan@44	1273 modifier = 0.25,
jcallahan@44	1274 },
jcallahan@44	1275 NAZGRELS_FERVOR = {
jcallahan@44	1276 faction = FACTION_NAMES.THRALLMAR,
jcallahan@44	1277 modifier = 0.1,
jcallahan@44	1278 },
jcallahan@44	1279 SONG_OF_BATTLE = {
jcallahan@44	1280 faction = FACTION_NAMES.THE_SHATAR,
jcallahan@44	1281 modifier = 0.1,
jcallahan@44	1282 },
jcallahan@44	1283 SPIRIT_OF_SHARING = {
jcallahan@44	1284 modifier = 0.1,
jcallahan@44	1285 },
jcallahan@44	1286 THRALLMARS_FAVOR = {
jcallahan@44	1287 faction = FACTION_NAMES.THRALLMAR,
jcallahan@44	1288 modifier = 0.25,
jcallahan@44	1289 },
jcallahan@44	1290 TROLLBANES_COMMAND = {
jcallahan@44	1291 faction = FACTION_NAMES.HONOR_HOLD,
jcallahan@44	1292 modifier = 0.1,
jcallahan@44	1293 },
jcallahan@44	1294 UNBURDENED = {
jcallahan@44	1295 modifier = 0.1,
jcallahan@44	1296 },
jcallahan@44	1297 WOW_ANNIVERSARY = {
jcallahan@44	1298 modifier = 0.08,
jcallahan@44	1299 }
jcallahan@44	1300 }
jcallahan@44	1301
jcallahan@44	1302
jcallahan@92	1303 function WDP:COMBAT_TEXT_UPDATE(event_name, message_type, faction_name, amount)
jcallahan@177	1304 if message_type ~= "FACTION" or not killed_npc_id then
jcallahan@44	1305 return
jcallahan@44	1306 end
jcallahan@44	1307 UpdateFactionData()
jcallahan@44	1308
jcallahan@46	1309 if not faction_name or not faction_standings[faction_name] then
jcallahan@46	1310 return
jcallahan@46	1311 end
jcallahan@177	1312 local npc = NPCEntry(killed_npc_id)
jcallahan@177	1313 ClearKilledNPC()
jcallahan@46	1314
jcallahan@44	1315 if not npc then
jcallahan@98	1316 private.harvesting = nil
jcallahan@44	1317 return
jcallahan@44	1318 end
jcallahan@98	1319 npc.harvested = private.harvesting
jcallahan@98	1320 private.harvesting = nil
jcallahan@98	1321
jcallahan@44	1322 local modifier = 1
jcallahan@44	1323
jcallahan@44	1324 if _G.IsSpellKnown(DIPLOMACY_SPELL_ID) then
jcallahan@44	1325 modifier = modifier + 0.1
jcallahan@44	1326 end
jcallahan@44	1327
jcallahan@44	1328 if _G.IsSpellKnown(MR_POP_RANK2_SPELL_ID) then
jcallahan@44	1329 modifier = modifier + 0.1
jcallahan@44	1330 elseif _G.IsSpellKnown(MR_POP_RANK1_SPELL_ID) then
jcallahan@44	1331 modifier = modifier + 0.05
jcallahan@44	1332 end
jcallahan@44	1333
jcallahan@44	1334 for buff_name, buff_label in pairs(REP_BUFFS) do
jcallahan@44	1335 if _G.UnitBuff("player", buff_name) then
jcallahan@44	1336 local modded_faction = MODIFIERS[buff_label].faction
jcallahan@44	1337
jcallahan@44	1338 if not modded_faction or faction_name == modded_faction then
jcallahan@44	1339 modifier = modifier + MODIFIERS[buff_label].modifier
jcallahan@44	1340 end
jcallahan@44	1341 end
jcallahan@44	1342 end
jcallahan@65	1343 npc.reputations = npc.reputations or {}
jcallahan@65	1344 npc.reputations[("%s:%s"):format(faction_name, faction_standings[faction_name])] = math.floor(amount / modifier)
jcallahan@32	1345 end
jcallahan@44	1346 end -- do-block
jcallahan@18	1347
jcallahan@18	1348
jcallahan@140	1349 function WDP:CURSOR_UPDATE(event_name)
jcallahan@141	1350 if current_action.fishing_target or _G.Minimap:IsMouseOver() or current_action.spell_label ~= "FISHING" then
jcallahan@140	1351 return
jcallahan@140	1352 end
jcallahan@140	1353 local text = _G["GameTooltipTextLeft1"]:GetText()
jcallahan@140	1354
jcallahan@140	1355 if not text or text == "Fishing Bobber" then
jcallahan@140	1356 text = "NONE"
jcallahan@140	1357 else
jcallahan@140	1358 current_action.fishing_target = true
jcallahan@140	1359 end
jcallahan@140	1360 current_action.identifier = ("%s:%s"):format(current_action.spell_label, text)
jcallahan@140	1361 end
jcallahan@140	1362
jcallahan@140	1363
jcallahan@92	1364 function WDP:ITEM_TEXT_BEGIN(event_name)
jcallahan@42	1365 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("npc"))
jcallahan@42	1366
jcallahan@42	1367 if not unit_idnum or unit_type ~= private.UNIT_TYPES.OBJECT or _G.UnitName("npc") ~= _G.ItemTextGetItem() then
jcallahan@42	1368 return
jcallahan@42	1369 end
jcallahan@42	1370 UpdateDBEntryLocation("objects", unit_idnum)
jcallahan@42	1371 end
jcallahan@42	1372
jcallahan@42	1373
jcallahan@13	1374 do
jcallahan@40	1375 local RE_GOLD = _G.GOLD_AMOUNT:gsub("%%d", "(%%d+)")
jcallahan@40	1376 local RE_SILVER = _G.SILVER_AMOUNT:gsub("%%d", "(%%d+)")
jcallahan@40	1377 local RE_COPPER = _G.COPPER_AMOUNT:gsub("%%d", "(%%d+)")
jcallahan@13	1378
jcallahan@13	1379
jcallahan@13	1380 local function _moneyMatch(money, re)
jcallahan@13	1381 return money:match(re) or 0
jcallahan@1	1382 end
jcallahan@1	1383
jcallahan@0	1384
jcallahan@13	1385 local function _toCopper(money)
jcallahan@13	1386 if not money then
jcallahan@13	1387 return 0
jcallahan@13	1388 end
jcallahan@40	1389 return _moneyMatch(money, RE_GOLD) * 10000 + _moneyMatch(money, RE_SILVER) * 100 + _moneyMatch(money, RE_COPPER)
jcallahan@1	1390 end
jcallahan@1	1391
jcallahan@1	1392
jcallahan@176	1393 local LOOT_OPENED_VERIFY_FUNCS = {
jcallahan@16	1394 [AF.ITEM] = function()
jcallahan@16	1395 local locked_item_id
jcallahan@16	1396
jcallahan@16	1397 for bag_index = 0, _G.NUM_BAG_FRAMES do
jcallahan@16	1398 for slot_index = 1, _G.GetContainerNumSlots(bag_index) do
jcallahan@16	1399 local _, _, is_locked = _G.GetContainerItemInfo(bag_index, slot_index)
jcallahan@16	1400
jcallahan@16	1401 if is_locked then
jcallahan@16	1402 locked_item_id = ItemLinkToID(_G.GetContainerItemLink(bag_index, slot_index))
jcallahan@165	1403 break
jcallahan@16	1404 end
jcallahan@16	1405 end
jcallahan@165	1406
jcallahan@165	1407 if locked_item_id then
jcallahan@165	1408 break
jcallahan@165	1409 end
jcallahan@16	1410 end
jcallahan@16	1411
jcallahan@122	1412 if not locked_item_id or (current_action.identifier and current_action.identifier ~= locked_item_id) then
jcallahan@16	1413 return false
jcallahan@16	1414 end
jcallahan@122	1415 current_action.identifier = locked_item_id
jcallahan@16	1416 return true
jcallahan@16	1417 end,
jcallahan@13	1418 [AF.NPC] = function()
jcallahan@17	1419 if not _G.UnitExists("target") or _G.UnitIsFriend("player", "target") or _G.UnitIsPlayer("target") or _G.UnitPlayerControlled("target") then
jcallahan@15	1420 return false
jcallahan@13	1421 end
jcallahan@205	1422
jcallahan@205	1423 if not current_action.identifier then
jcallahan@205	1424 local unit_type, id_num = ParseGUID(_G.UnitGUID("target"))
jcallahan@205	1425 current_action.identifier = id_num
jcallahan@205	1426 end
jcallahan@13	1427 return true
jcallahan@13	1428 end,
jcallahan@14	1429 [AF.OBJECT] = true,
jcallahan@17	1430 [AF.ZONE] = function()
jcallahan@140	1431 if not _G.IsFishingLoot() then
jcallahan@140	1432 return false
jcallahan@140	1433 end
jcallahan@140	1434 current_action.zone_data = UpdateDBEntryLocation("zones", current_action.identifier)
jcallahan@140	1435 return true
jcallahan@17	1436 end,
jcallahan@13	1437 }
jcallahan@13	1438
jcallahan@13	1439
jcallahan@176	1440 local LOOT_OPENED_UPDATE_FUNCS = {
jcallahan@16	1441 [AF.ITEM] = function()
jcallahan@28	1442 GenericLootUpdate("items")
jcallahan@28	1443 end,
jcallahan@28	1444 [AF.NPC] = function()
jcallahan@75	1445 local difficulty_token = InstanceDifficultyToken()
jcallahan@132	1446 local loot_type = current_loot.label
jcallahan@77	1447 local source_list = {}
jcallahan@75	1448
jcallahan@131	1449 for source_guid, loot_data in pairs(current_loot.sources) do
jcallahan@78	1450 local source_id = select(2, ParseGUID(source_guid))
jcallahan@75	1451 local npc = NPCEntry(source_id)
jcallahan@75	1452
jcallahan@75	1453 if npc then
jcallahan@75	1454 local encounter_data = npc.encounter_data[difficulty_token]
jcallahan@75	1455 encounter_data[loot_type] = encounter_data[loot_type] or {}
jcallahan@75	1456
jcallahan@78	1457 if not source_list[source_guid] then
jcallahan@77	1458 encounter_data.loot_counts = encounter_data.loot_counts or {}
jcallahan@77	1459 encounter_data.loot_counts[loot_type] = (encounter_data.loot_counts[loot_type] or 0) + 1
jcallahan@77	1460 source_list[source_id] = true
jcallahan@77	1461 end
jcallahan@77	1462
jcallahan@75	1463 for item_id, quantity in pairs(loot_data) do
jcallahan@75	1464 table.insert(encounter_data[loot_type], ("%d:%d"):format(item_id, quantity))
jcallahan@75	1465 end
jcallahan@75	1466 end
jcallahan@75	1467 end
jcallahan@75	1468
jcallahan@75	1469 -- TODO: Remove this when GetLootSourceInfo() has values for money
jcallahan@131	1470 if #current_loot.list <= 0 then
jcallahan@79	1471 return
jcallahan@79	1472 end
jcallahan@131	1473 local npc = NPCEntry(current_loot.identifier)
jcallahan@28	1474
jcallahan@28	1475 if not npc then
jcallahan@28	1476 return
jcallahan@28	1477 end
jcallahan@75	1478 local encounter_data = npc.encounter_data[difficulty_token]
jcallahan@29	1479 encounter_data[loot_type] = encounter_data[loot_type] or {}
jcallahan@16	1480
jcallahan@131	1481 if not source_list[current_loot.identifier] then
jcallahan@77	1482 encounter_data.loot_counts = encounter_data.loot_counts or {}
jcallahan@77	1483 encounter_data.loot_counts[loot_type] = (encounter_data.loot_counts[loot_type] or 0) + 1
jcallahan@77	1484 end
jcallahan@77	1485
jcallahan@131	1486 for index = 1, #current_loot.list do
jcallahan@131	1487 table.insert(encounter_data[loot_type], current_loot.list[index])
jcallahan@16	1488 end
jcallahan@16	1489 end,
jcallahan@13	1490 [AF.OBJECT] = function()
jcallahan@28	1491 GenericLootUpdate("objects", InstanceDifficultyToken())
jcallahan@17	1492 end,
jcallahan@17	1493 [AF.ZONE] = function()
jcallahan@141	1494 local location_token = ("%d:%d:%d"):format(current_loot.map_level, current_loot.x, current_loot.y)
jcallahan@41	1495
jcallahan@41	1496 -- This will start life as a boolean true.
jcallahan@131	1497 if _G.type(current_loot.zone_data[location_token]) ~= "table" then
jcallahan@131	1498 current_loot.zone_data[location_token] = {
jcallahan@41	1499 drops = {}
jcallahan@41	1500 }
jcallahan@41	1501 end
jcallahan@132	1502 local loot_count = ("%s_count"):format(current_loot.label)
jcallahan@131	1503 current_loot.zone_data[location_token][loot_count] = (current_loot.zone_data[location_token][loot_count] or 0) + 1
jcallahan@41	1504
jcallahan@131	1505 if current_loot.sources then
jcallahan@131	1506 for source_guid, loot_data in pairs(current_loot.sources) do
jcallahan@131	1507 for item_id, quantity in pairs(loot_data) do
jcallahan@131	1508 table.insert(current_loot.zone_data[location_token].drops, ("%d:%d"):format(item_id, quantity))
jcallahan@131	1509 end
jcallahan@131	1510 end
jcallahan@131	1511 end
jcallahan@131	1512
jcallahan@131	1513 if #current_loot.list <= 0 then
jcallahan@131	1514 return
jcallahan@131	1515 end
jcallahan@131	1516
jcallahan@131	1517 for index = 1, #current_loot.list do
jcallahan@131	1518 table.insert(current_loot.zone_data[location_token].drops, current_loot.loot_list[index])
jcallahan@41	1519 end
jcallahan@13	1520 end,
jcallahan@13	1521 }
jcallahan@13	1522
jcallahan@79	1523 -- Prevent opening the same loot window multiple times from recording data multiple times.
jcallahan@79	1524 local loot_guid_registry = {}
jcallahan@124	1525
jcallahan@124	1526
jcallahan@124	1527 function WDP:LOOT_CLOSED(event_name)
jcallahan@131	1528 current_loot = nil
jcallahan@131	1529 table.wipe(current_action)
jcallahan@124	1530 end
jcallahan@124	1531
jcallahan@13	1532
jcallahan@92	1533 function WDP:LOOT_OPENED(event_name)
jcallahan@132	1534 if current_loot then
jcallahan@18	1535 return
jcallahan@18	1536 end
jcallahan@151	1537
jcallahan@151	1538 if not current_action.target_type then
jcallahan@151	1539 return
jcallahan@151	1540 end
jcallahan@176	1541 local verify_func = LOOT_OPENED_VERIFY_FUNCS[current_action.target_type]
jcallahan@176	1542 local update_func = LOOT_OPENED_UPDATE_FUNCS[current_action.target_type]
jcallahan@13	1543
jcallahan@14	1544 if not verify_func or not update_func then
jcallahan@13	1545 return
jcallahan@13	1546 end
jcallahan@13	1547
jcallahan@14	1548 if _G.type(verify_func) == "function" and not verify_func() then
jcallahan@14	1549 return
jcallahan@14	1550 end
jcallahan@80	1551 local guids_used = {}
jcallahan@132	1552
jcallahan@131	1553 current_loot = {
jcallahan@131	1554 list = {},
jcallahan@131	1555 sources = {},
jcallahan@131	1556 identifier = current_action.identifier,
jcallahan@132	1557 label = current_action.loot_label or "drops",
jcallahan@131	1558 map_level = current_action.map_level,
jcallahan@131	1559 object_name = current_action.object_name,
jcallahan@131	1560 spell_label = current_action.spell_label,
jcallahan@131	1561 target_type = current_action.target_type,
jcallahan@131	1562 x = current_action.x,
jcallahan@131	1563 y = current_action.y,
jcallahan@131	1564 zone_data = current_action.zone_data,
jcallahan@131	1565 }
jcallahan@131	1566 table.wipe(current_action)
jcallahan@13	1567
jcallahan@55	1568 for loot_slot = 1, _G.GetNumLootItems() do
jcallahan@55	1569 local icon_texture, item_text, quantity, quality, locked = _G.GetLootSlotInfo(loot_slot)
jcallahan@55	1570 local slot_type = _G.GetLootSlotType(loot_slot)
jcallahan@13	1571
jcallahan@77	1572 -- TODO: Move LOOT_SLOT_X checks within loop when money is detectable via GetLootSourceInfo
jcallahan@77	1573 if slot_type == _G.LOOT_SLOT_ITEM then
jcallahan@78	1574 local loot_info = {
jcallahan@77	1575 _G.GetLootSourceInfo(loot_slot)
jcallahan@77	1576 }
jcallahan@75	1577
jcallahan@77	1578 -- Odd index is GUID, even is count.
jcallahan@78	1579 for loot_index = 1, #loot_info, 2 do
jcallahan@78	1580 local source_guid = loot_info[loot_index]
jcallahan@77	1581
jcallahan@79	1582 if not loot_guid_registry[source_guid] then
jcallahan@79	1583 local loot_quantity = loot_info[loot_index + 1]
jcallahan@79	1584 local source_type, source_id = ParseGUID(source_guid)
jcallahan@79	1585 -- TODO: Remove debugging
jcallahan@151	1586 local source_key = ("%s:%d"):format(private.UNIT_TYPE_NAMES[source_type + 1], source_id)
jcallahan@151	1587 Debug(("GUID: %s - Type:ID: %s - Amount: %d"):format(loot_info[loot_index], source_key, loot_quantity))
jcallahan@79	1588
jcallahan@79	1589 local item_id = ItemLinkToID(_G.GetLootSlotLink(loot_slot))
jcallahan@131	1590 current_loot.sources[source_guid] = current_loot.sources[source_guid] or {}
jcallahan@131	1591 current_loot.sources[source_guid][item_id] = current_loot.sources[source_guid][item_id] or 0 + loot_quantity
jcallahan@80	1592 guids_used[source_guid] = true
jcallahan@79	1593 end
jcallahan@75	1594 end
jcallahan@82	1595 -- elseif slot_type == _G.LOOT_SLOT_MONEY then
jcallahan@122	1596 -- table.insert(current_action.loot_list, ("money:%d"):format(_toCopper(item_text)))
jcallahan@77	1597 elseif slot_type == _G.LOOT_SLOT_CURRENCY then
jcallahan@131	1598 table.insert(current_loot.list, ("currency:%d:%s"):format(quantity, icon_texture:match("[^\\]+$"):lower()))
jcallahan@13	1599 end
jcallahan@13	1600 end
jcallahan@80	1601
jcallahan@81	1602 for guid in pairs(guids_used) do
jcallahan@80	1603 loot_guid_registry[guid] = true
jcallahan@80	1604 end
jcallahan@13	1605 update_func()
jcallahan@1	1606 end
jcallahan@13	1607 end -- do-block
jcallahan@0	1608
jcallahan@0	1609
jcallahan@89	1610 function WDP:MAIL_SHOW(event_name)
jcallahan@89	1611 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("npc"))
jcallahan@89	1612
jcallahan@89	1613 if not unit_idnum or unit_type ~= private.UNIT_TYPES.OBJECT then
jcallahan@89	1614 return
jcallahan@89	1615 end
jcallahan@89	1616 UpdateDBEntryLocation("objects", unit_idnum)
jcallahan@89	1617 end
jcallahan@89	1618
jcallahan@89	1619
jcallahan@44	1620 do
jcallahan@44	1621 local POINT_MATCH_PATTERNS = {
jcallahan@44	1622 ("^%s$"):format(_G.ITEM_REQ_ARENA_RATING:gsub("%%d", "(%%d+)")), -- May no longer be necessary
jcallahan@44	1623 ("^%s$"):format(_G.ITEM_REQ_ARENA_RATING_3V3:gsub("%%d", "(%%d+)")), -- May no longer be necessary
jcallahan@44	1624 ("^%s$"):format(_G.ITEM_REQ_ARENA_RATING_5V5:gsub("%%d", "(%%d+)")), -- May no longer be necessary
jcallahan@44	1625 ("^%s$"):format(_G.ITEM_REQ_ARENA_RATING_BG:gsub("%%d", "(%%d+)")),
jcallahan@44	1626 ("^%s$"):format(_G.ITEM_REQ_ARENA_RATING_3V3_BG:gsub("%%d", "(%%d+)")),
jcallahan@44	1627 }
jcallahan@5	1628
jcallahan@68	1629 local ITEM_REQ_REPUTATION_MATCH = "Requires (.-) %- (.*)"
jcallahan@87	1630 local ITEM_REQ_QUEST_MATCH1 = "Requires: .*"
jcallahan@87	1631 local ITEM_REQ_QUEST_MATCH2 = "Must have completed: .*"
jcallahan@68	1632
jcallahan@133	1633 local current_merchant
jcallahan@133	1634 local merchant_standing
jcallahan@133	1635
jcallahan@133	1636 function WDP:MERCHANT_CLOSED(event_name)
jcallahan@133	1637 current_merchant = nil
jcallahan@133	1638 merchant_standing = nil
jcallahan@133	1639 end
jcallahan@133	1640
jcallahan@133	1641
jcallahan@89	1642 function WDP:UpdateMerchantItems(event_name)
jcallahan@144	1643 if not current_merchant or event_name == "MERCHANT_SHOW" then
jcallahan@195	1644 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("npc"))
jcallahan@4	1645
jcallahan@171	1646 if not unit_idnum or not UnitTypeIsNPC(unit_type) then
jcallahan@133	1647 return
jcallahan@133	1648 end
jcallahan@195	1649 merchant_standing = select(2, UnitFactionStanding("npc"))
jcallahan@133	1650 current_merchant = NPCEntry(unit_idnum)
jcallahan@133	1651 current_merchant.sells = current_merchant.sells or {}
jcallahan@44	1652 end
jcallahan@55	1653 local current_filters = _G.GetMerchantFilter()
jcallahan@57	1654 _G.SetMerchantFilter(_G.LE_LOOT_FILTER_ALL)
jcallahan@57	1655 _G.MerchantFrame_Update()
jcallahan@57	1656
jcallahan@150	1657 local num_items = _G.GetMerchantNumItems()
jcallahan@150	1658
jcallahan@44	1659 for item_index = 1, num_items do
jcallahan@44	1660 local _, _, copper_price, stack_size, num_available, _, extended_cost = _G.GetMerchantItemInfo(item_index)
jcallahan@44	1661 local item_id = ItemLinkToID(_G.GetMerchantItemLink(item_index))
jcallahan@5	1662
jcallahan@44	1663 if item_id and item_id > 0 then
jcallahan@44	1664 local price_string = ActualCopperCost(copper_price, merchant_standing)
jcallahan@5	1665
jcallahan@68	1666 DatamineTT:ClearLines()
jcallahan@68	1667 DatamineTT:SetMerchantItem(item_index)
jcallahan@68	1668
jcallahan@68	1669 local num_lines = DatamineTT:NumLines()
jcallahan@68	1670
jcallahan@68	1671 for line_index = 1, num_lines do
jcallahan@68	1672 local current_line = _G["WDPDatamineTTTextLeft" .. line_index]
jcallahan@68	1673
jcallahan@68	1674 if not current_line then
jcallahan@68	1675 break
jcallahan@68	1676 end
jcallahan@68	1677 local faction, reputation = current_line:GetText():match(ITEM_REQ_REPUTATION_MATCH)
jcallahan@68	1678
jcallahan@68	1679 if faction or reputation then
jcallahan@68	1680 DBEntry("items", item_id).req_reputation = ("%s:%s"):format(faction:gsub("-", ""), reputation:upper())
jcallahan@68	1681 break
jcallahan@68	1682 end
jcallahan@68	1683 end
jcallahan@68	1684
jcallahan@87	1685 for line_index = 1, num_lines do
jcallahan@87	1686 local current_line = _G["WDPDatamineTTTextLeft" .. line_index]
jcallahan@87	1687
jcallahan@87	1688 if not current_line then
jcallahan@87	1689 break
jcallahan@87	1690 end
jcallahan@87	1691 local line_text = current_line:GetText()
jcallahan@87	1692 local quest_name = line_text:match(ITEM_REQ_QUEST_MATCH1) or line_text:match(ITEM_REQ_QUEST_MATCH2)
jcallahan@87	1693
jcallahan@87	1694 if quest_name then
jcallahan@87	1695 DBEntry("items", item_id).req_quest = ("%s"):format(quest_name:gsub("(.+): ", ""), quest_name)
jcallahan@87	1696 break
jcallahan@87	1697 end
jcallahan@87	1698 end
jcallahan@87	1699
jcallahan@44	1700 if extended_cost then
jcallahan@53	1701 local battleground_rating = 0
jcallahan@53	1702 local personal_rating = 0
jcallahan@53	1703 local required_season_amount
jcallahan@5	1704
jcallahan@68	1705 for line_index = 1, num_lines do
jcallahan@44	1706 local current_line = _G["WDPDatamineTTTextLeft" .. line_index]
jcallahan@5	1707
jcallahan@44	1708 if not current_line then
jcallahan@44	1709 break
jcallahan@44	1710 end
jcallahan@53	1711 required_season_amount = current_line:GetText():match("Requires earning a total of (%d+)\n(.-) for the season.")
jcallahan@5	1712
jcallahan@44	1713 for match_index = 1, #POINT_MATCH_PATTERNS do
jcallahan@44	1714 local match1, match2 = current_line:GetText():match(POINT_MATCH_PATTERNS[match_index])
jcallahan@53	1715 personal_rating = personal_rating + (match1 or 0)
jcallahan@53	1716 battleground_rating = battleground_rating + (match2 or 0)
jcallahan@5	1717
jcallahan@44	1718 if match1 or match2 then
jcallahan@44	1719 break
jcallahan@44	1720 end
jcallahan@44	1721 end
jcallahan@5	1722 end
jcallahan@44	1723 local currency_list = {}
jcallahan@44	1724 local item_count = _G.GetMerchantItemCostInfo(item_index)
jcallahan@50	1725
jcallahan@50	1726 -- Keeping this around in case Blizzard makes the two points diverge at some point.
jcallahan@53	1727 -- price_string = ("%s:%s:%s:%s"):format(price_string, battleground_rating, personal_rating, required_season_amount or 0)
jcallahan@53	1728 price_string = ("%s:%s:%s"):format(price_string, personal_rating, required_season_amount or 0)
jcallahan@5	1729
jcallahan@44	1730 for cost_index = 1, item_count do
jcallahan@44	1731 local icon_texture, amount_required, currency_link = _G.GetMerchantItemCostItem(item_index, cost_index)
jcallahan@44	1732 local currency_id = currency_link and ItemLinkToID(currency_link) or nil
jcallahan@44	1733
jcallahan@44	1734 if (not currency_id or currency_id < 1) and icon_texture then
jcallahan@44	1735 currency_id = icon_texture:match("[^\\]+$"):lower()
jcallahan@44	1736 end
jcallahan@44	1737
jcallahan@44	1738 if currency_id then
jcallahan@44	1739 currency_list[#currency_list + 1] = ("(%s:%s)"):format(amount_required, currency_id)
jcallahan@44	1740 end
jcallahan@44	1741 end
jcallahan@44	1742
jcallahan@44	1743 for currency_index = 1, #currency_list do
jcallahan@44	1744 price_string = ("%s:%s"):format(price_string, currency_list[currency_index])
jcallahan@5	1745 end
jcallahan@5	1746 end
jcallahan@133	1747 current_merchant.sells[item_id] = ("%s:%s:[%s]"):format(num_available, stack_size, price_string)
jcallahan@44	1748 end
jcallahan@44	1749 end
jcallahan@5	1750
jcallahan@44	1751 if _G.CanMerchantRepair() then
jcallahan@133	1752 current_merchant.can_repair = true
jcallahan@5	1753 end
jcallahan@57	1754 _G.SetMerchantFilter(current_filters)
jcallahan@57	1755 _G.MerchantFrame_Update()
jcallahan@4	1756 end
jcallahan@44	1757 end -- do-block
jcallahan@4	1758
jcallahan@89	1759
jcallahan@92	1760 function WDP:PET_BAR_UPDATE(event_name)
jcallahan@122	1761 if current_action.spell_label ~= "MIND_CONTROL" then
jcallahan@25	1762 return
jcallahan@25	1763 end
jcallahan@34	1764 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("pet"))
jcallahan@25	1765
jcallahan@171	1766 if not unit_idnum or not UnitTypeIsNPC(unit_type) then
jcallahan@25	1767 return
jcallahan@25	1768 end
jcallahan@29	1769 NPCEntry(unit_idnum).mind_control = true
jcallahan@122	1770 table.wipe(current_action)
jcallahan@25	1771 end
jcallahan@25	1772
jcallahan@25	1773
jcallahan@115	1774 function WDP:PET_JOURNAL_LIST_UPDATE(event_name)
jcallahan@115	1775 local num_pets = LPJ:NumPets()
jcallahan@115	1776
jcallahan@115	1777 for index, pet_id in LPJ:IteratePetIDs() do
jcallahan@115	1778 local _, _, is_owned, _, level, _, _, name, icon, pet_type, npc_id, _, _, is_wild = _G.C_PetJournal.GetPetInfoByIndex(index)
jcallahan@115	1779
jcallahan@115	1780 if is_owned then
jcallahan@115	1781 local health, max_health, attack, speed, rarity = _G.C_PetJournal.GetPetStats(pet_id)
jcallahan@115	1782
jcallahan@139	1783 if rarity then
jcallahan@139	1784 local rarity_name = _G["BATTLE_PET_BREED_QUALITY" .. rarity]
jcallahan@139	1785 local npc = NPCEntry(npc_id)
jcallahan@139	1786 npc.wild_pet = is_wild or nil
jcallahan@139	1787 npc.battle_pet_data = npc.battle_pet_data or {}
jcallahan@139	1788 npc.battle_pet_data[rarity_name] = npc.battle_pet_data[rarity_name] or {}
jcallahan@139	1789 npc.battle_pet_data[rarity_name]["level_" .. level] = npc.battle_pet_data[rarity_name]["level_" .. level] or {}
jcallahan@139	1790
jcallahan@139	1791 local data = npc.battle_pet_data[rarity_name]["level_" .. level]
jcallahan@139	1792 data.max_health = max_health
jcallahan@139	1793 data.attack = attack
jcallahan@139	1794 data.speed = speed
jcallahan@139	1795 end
jcallahan@115	1796 end
jcallahan@115	1797 end
jcallahan@115	1798 end
jcallahan@115	1799
jcallahan@115	1800
jcallahan@156	1801 function WDP:PLAYER_REGEN_DISABLED(event_name)
jcallahan@156	1802 private.in_combat = true
jcallahan@156	1803 end
jcallahan@156	1804
jcallahan@156	1805
jcallahan@156	1806 function WDP:PLAYER_REGEN_ENABLED(event_name)
jcallahan@156	1807 private.in_combat = nil
jcallahan@156	1808
jcallahan@156	1809 if private.set_area_id then
jcallahan@156	1810 self:SetCurrentAreaID(event_name)
jcallahan@156	1811 private.set_area_id = nil
jcallahan@156	1812 end
jcallahan@156	1813 end
jcallahan@156	1814
jcallahan@156	1815
jcallahan@118	1816 function WDP:PLAYER_TARGET_CHANGED(event_name)
jcallahan@118	1817 if not PlayerTarget() then
jcallahan@118	1818 return
jcallahan@2	1819 end
jcallahan@151	1820 current_action.target_type = AF.NPC
jcallahan@118	1821 self:UpdateTargetLocation()
jcallahan@118	1822 end
jcallahan@2	1823
jcallahan@89	1824
jcallahan@12	1825 do
jcallahan@12	1826 local function UpdateQuestJuncture(point)
jcallahan@12	1827 local unit_name = _G.UnitName("questnpc")
jcallahan@9	1828
jcallahan@12	1829 if not unit_name then
jcallahan@12	1830 return
jcallahan@12	1831 end
jcallahan@34	1832 local unit_type, unit_id = ParseGUID(_G.UnitGUID("questnpc"))
jcallahan@9	1833
jcallahan@12	1834 if unit_type == private.UNIT_TYPES.OBJECT then
jcallahan@38	1835 UpdateDBEntryLocation("objects", unit_id)
jcallahan@12	1836 end
jcallahan@19	1837 local quest = DBEntry("quests", _G.GetQuestID())
jcallahan@12	1838 quest[point] = quest[point] or {}
jcallahan@12	1839 quest[point][("%s:%d"):format(private.UNIT_TYPE_NAMES[unit_type + 1], unit_id)] = true
jcallahan@24	1840
jcallahan@24	1841 return quest
jcallahan@12	1842 end
jcallahan@10	1843
jcallahan@12	1844
jcallahan@92	1845 function WDP:QUEST_COMPLETE(event_name)
jcallahan@97	1846 local quest = UpdateQuestJuncture("end")
jcallahan@97	1847
jcallahan@112	1848 if ALLOWED_LOCALES[CLIENT_LOCALE] then
jcallahan@112	1849 quest.reward_text = ReplaceKeywords(_G.GetRewardText())
jcallahan@112	1850 end
jcallahan@67	1851 -- Make sure the quest NPC isn't erroneously recorded as giving reputation for quests which award it.
jcallahan@177	1852 ClearKilledNPC()
jcallahan@10	1853 end
jcallahan@10	1854
jcallahan@12	1855
jcallahan@92	1856 function WDP:QUEST_DETAIL(event_name)
jcallahan@24	1857 local quest = UpdateQuestJuncture("begin")
jcallahan@24	1858
jcallahan@46	1859 if not quest then
jcallahan@46	1860 return
jcallahan@46	1861 end
jcallahan@24	1862 quest.classes = quest.classes or {}
jcallahan@27	1863 quest.classes[PLAYER_CLASS] = true
jcallahan@24	1864
jcallahan@24	1865 quest.races = quest.races or {}
jcallahan@100	1866 quest.races[(PLAYER_RACE == "Pandaren") and ("%s_%s"):format(PLAYER_RACE, PLAYER_FACTION or "Neutral") or PLAYER_RACE] = true
jcallahan@10	1867 end
jcallahan@12	1868 end -- do-block
jcallahan@9	1869
jcallahan@9	1870
jcallahan@92	1871 function WDP:QUEST_LOG_UPDATE(event_name)
jcallahan@38	1872 local selected_quest = _G.GetQuestLogSelection() -- Save current selection to be restored when we're done.
jcallahan@36	1873 local entry_index, processed_quests = 1, 0
jcallahan@36	1874 local _, num_quests = _G.GetNumQuestLogEntries()
jcallahan@36	1875
jcallahan@36	1876 while processed_quests <= num_quests do
jcallahan@36	1877 local _, _, _, _, is_header, _, _, _, quest_id = _G.GetQuestLogTitle(entry_index)
jcallahan@36	1878
jcallahan@84	1879 if quest_id == 0 then
jcallahan@84	1880 processed_quests = processed_quests + 1
jcallahan@84	1881 elseif not is_header then
jcallahan@36	1882 _G.SelectQuestLogEntry(entry_index);
jcallahan@36	1883
jcallahan@36	1884 local quest = DBEntry("quests", quest_id)
jcallahan@36	1885 quest.timer = _G.GetQuestLogTimeLeft()
jcallahan@37	1886 quest.can_share = _G.GetQuestLogPushable() and true or nil
jcallahan@36	1887 processed_quests = processed_quests + 1
jcallahan@36	1888 end
jcallahan@36	1889 entry_index = entry_index + 1
jcallahan@36	1890 end
jcallahan@36	1891 _G.SelectQuestLogEntry(selected_quest)
jcallahan@4	1892 self:UnregisterEvent("QUEST_LOG_UPDATE")
jcallahan@4	1893 end
jcallahan@4	1894
jcallahan@4	1895
jcallahan@97	1896 function WDP:QUEST_PROGRESS(event_name)
jcallahan@112	1897 if not ALLOWED_LOCALES[CLIENT_LOCALE] then
jcallahan@112	1898 return
jcallahan@112	1899 end
jcallahan@97	1900 DBEntry("quests", _G.GetQuestID()).progress_text = ReplaceKeywords(_G.GetProgressText())
jcallahan@97	1901 end
jcallahan@97	1902
jcallahan@97	1903
jcallahan@92	1904 function WDP:UNIT_QUEST_LOG_CHANGED(event_name, unit_id)
jcallahan@4	1905 if unit_id ~= "player" then
jcallahan@4	1906 return
jcallahan@4	1907 end
jcallahan@4	1908 self:RegisterEvent("QUEST_LOG_UPDATE")
jcallahan@4	1909 end
jcallahan@4	1910
jcallahan@4	1911
jcallahan@92	1912 do
jcallahan@92	1913 local TRADESKILL_TOOLS = {
jcallahan@92	1914 Anvil = anvil_spell_ids,
jcallahan@92	1915 Forge = forge_spell_ids,
jcallahan@92	1916 }
jcallahan@92	1917
jcallahan@92	1918
jcallahan@167	1919 local function RegisterTools(tradeskill_name, tradeskill_index)
jcallahan@167	1920 local spell_id = tonumber(_G.GetTradeSkillRecipeLink(tradeskill_index):match("^\|c%x%x%x%x%x%x%x%x\|H%w+:(%d+)"))
jcallahan@167	1921 local required_tool = _G.GetTradeSkillTools(tradeskill_index)
jcallahan@167	1922
jcallahan@167	1923 if required_tool then
jcallahan@167	1924 for tool_name, registry in pairs(TRADESKILL_TOOLS) do
jcallahan@167	1925 if required_tool:find(tool_name) then
jcallahan@167	1926 registry[spell_id] = true
jcallahan@167	1927 end
jcallahan@167	1928 end
jcallahan@167	1929 end
jcallahan@167	1930 end
jcallahan@167	1931
jcallahan@167	1932
jcallahan@92	1933 function WDP:TRADE_SKILL_SHOW(event_name)
jcallahan@92	1934 local profession_name, prof_level = _G.GetTradeSkillLine()
jcallahan@92	1935
jcallahan@92	1936 if profession_name == _G.UNKNOWN then
jcallahan@92	1937 return
jcallahan@92	1938 end
jcallahan@167	1939 TradeSkillExecutePer(RegisterTools)
jcallahan@92	1940 end
jcallahan@92	1941 end -- do-block
jcallahan@92	1942
jcallahan@92	1943
jcallahan@167	1944 function WDP:TRAINER_CLOSED(event_name)
jcallahan@167	1945 private.trainer_shown = nil
jcallahan@167	1946 end
jcallahan@167	1947
jcallahan@167	1948
jcallahan@92	1949 function WDP:TRAINER_SHOW(event_name)
jcallahan@34	1950 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("target"))
jcallahan@164	1951 local trainer = NPCEntry(unit_idnum)
jcallahan@58	1952
jcallahan@164	1953 if not trainer then
jcallahan@58	1954 return
jcallahan@58	1955 end
jcallahan@164	1956 local trainer_standing = select(2, UnitFactionStanding("target"))
jcallahan@164	1957 trainer.teaches = trainer.teaches or {}
jcallahan@27	1958
jcallahan@167	1959 private.trainer_shown = true
jcallahan@167	1960
jcallahan@27	1961 -- Get the initial trainer filters
jcallahan@27	1962 local available = _G.GetTrainerServiceTypeFilter("available")
jcallahan@27	1963 local unavailable = _G.GetTrainerServiceTypeFilter("unavailable")
jcallahan@27	1964 local used = _G.GetTrainerServiceTypeFilter("used")
jcallahan@27	1965
jcallahan@27	1966 -- Clear the trainer filters
jcallahan@27	1967 _G.SetTrainerServiceTypeFilter("available", 1)
jcallahan@27	1968 _G.SetTrainerServiceTypeFilter("unavailable", 1)
jcallahan@27	1969 _G.SetTrainerServiceTypeFilter("used", 1)
jcallahan@27	1970
jcallahan@27	1971 for index = 1, _G.GetNumTrainerServices(), 1 do
jcallahan@27	1972 local spell_name, rank_name, _, _, required_level = _G.GetTrainerServiceInfo(index)
jcallahan@27	1973
jcallahan@27	1974 if spell_name then
jcallahan@27	1975 DatamineTT:ClearLines()
jcallahan@27	1976 DatamineTT:SetTrainerService(index)
jcallahan@27	1977
jcallahan@27	1978 local _, _, spell_id = DatamineTT:GetSpell()
jcallahan@27	1979
jcallahan@43	1980 if spell_id then
jcallahan@164	1981 local class_professions = trainer.teaches[PLAYER_CLASS]
jcallahan@164	1982
jcallahan@164	1983 if not class_professions then
jcallahan@164	1984 trainer.teaches[PLAYER_CLASS] = {}
jcallahan@164	1985 class_professions = trainer.teaches[PLAYER_CLASS]
jcallahan@164	1986 end
jcallahan@43	1987 local profession, min_skill = _G.GetTrainerServiceSkillReq(index)
jcallahan@43	1988 profession = profession or "General"
jcallahan@43	1989
jcallahan@164	1990 local profession_skills = class_professions[profession]
jcallahan@43	1991
jcallahan@43	1992 if not profession_skills then
jcallahan@43	1993 class_professions[profession] = {}
jcallahan@43	1994 profession_skills = class_professions[profession]
jcallahan@43	1995 end
jcallahan@173	1996 profession_skills[spell_id] = ("%d:%d:%d"):format(required_level, min_skill, _G.GetTrainerServiceCost(index))
jcallahan@27	1997 end
jcallahan@27	1998 end
jcallahan@27	1999 end
jcallahan@27	2000 -- Reset the filters to what they were before
jcallahan@27	2001 _G.SetTrainerServiceTypeFilter("available", available or 0)
jcallahan@27	2002 _G.SetTrainerServiceTypeFilter("unavailable", unavailable or 0)
jcallahan@27	2003 _G.SetTrainerServiceTypeFilter("used", used or 0)
jcallahan@27	2004 end
jcallahan@27	2005
jcallahan@27	2006
jcallahan@1	2007 function WDP:UNIT_SPELLCAST_SENT(event_name, unit_id, spell_name, spell_rank, target_name, spell_line)
jcallahan@1	2008 if private.tracked_line or unit_id ~= "player" then
jcallahan@1	2009 return
jcallahan@1	2010 end
jcallahan@1	2011 local spell_label = private.SPELL_LABELS_BY_NAME[spell_name]
jcallahan@1	2012
jcallahan@1	2013 if not spell_label then
jcallahan@1	2014 return
jcallahan@1	2015 end
jcallahan@150	2016 local item_name, item_link = _G.GameTooltip:GetItem()
jcallahan@150	2017 local unit_name, unit_id = _G.GameTooltip:GetUnit()
jcallahan@1	2018
jcallahan@150	2019 if not unit_name and _G.UnitName("target") == target_name then
jcallahan@150	2020 unit_name = target_name
jcallahan@150	2021 unit_id = "target"
jcallahan@1	2022 end
jcallahan@1	2023 local spell_flags = private.SPELL_FLAGS_BY_LABEL[spell_label]
jcallahan@28	2024 local zone_name, area_id, x, y, map_level, instance_token = CurrentLocationData()
jcallahan@28	2025
jcallahan@205	2026 table.wipe(current_action)
jcallahan@122	2027 current_action.instance_token = instance_token
jcallahan@122	2028 current_action.map_level = map_level
jcallahan@122	2029 current_action.x = x
jcallahan@122	2030 current_action.y = y
jcallahan@122	2031 current_action.zone_data = ("%s:%d"):format(zone_name, area_id)
jcallahan@122	2032 current_action.spell_label = spell_label
jcallahan@105	2033
jcallahan@105	2034 if not private.NON_LOOT_SPELL_LABELS[spell_label] then
jcallahan@122	2035 current_action.loot_label = spell_label:lower()
jcallahan@105	2036 end
jcallahan@1	2037
jcallahan@151	2038 if unit_name and unit_name == target_name and not item_name then
jcallahan@16	2039 if bit.band(spell_flags, AF.NPC) == AF.NPC then
jcallahan@150	2040 if not unit_id or unit_name ~= target_name then
jcallahan@16	2041 return
jcallahan@16	2042 end
jcallahan@123	2043 current_action.target_type = AF.NPC
jcallahan@16	2044 end
jcallahan@16	2045 elseif bit.band(spell_flags, AF.ITEM) == AF.ITEM then
jcallahan@123	2046 current_action.target_type = AF.ITEM
jcallahan@16	2047
jcallahan@150	2048 if item_name and item_name == target_name then
jcallahan@150	2049 current_action.identifier = ItemLinkToID(item_link)
jcallahan@16	2050 elseif target_name and target_name ~= "" then
jcallahan@150	2051 current_action.identifier = ItemLinkToID(select(2, _G.GetItemInfo(target_name)))
jcallahan@16	2052 end
jcallahan@150	2053 elseif not item_name and not unit_name then
jcallahan@1	2054 if bit.band(spell_flags, AF.OBJECT) == AF.OBJECT then
jcallahan@17	2055 if target_name == "" then
jcallahan@17	2056 return
jcallahan@17	2057 end
jcallahan@122	2058 current_action.object_name = target_name
jcallahan@123	2059 current_action.target_type = AF.OBJECT
jcallahan@1	2060 elseif bit.band(spell_flags, AF.ZONE) == AF.ZONE then
jcallahan@123	2061 current_action.target_type = AF.ZONE
jcallahan@1	2062 end
jcallahan@1	2063 end
jcallahan@1	2064 private.tracked_line = spell_line
jcallahan@0	2065 end
jcallahan@0	2066
jcallahan@94	2067
jcallahan@1	2068 function WDP:UNIT_SPELLCAST_SUCCEEDED(event_name, unit_id, spell_name, spell_rank, spell_line, spell_id)
jcallahan@1	2069 if unit_id ~= "player" then
jcallahan@1	2070 return
jcallahan@1	2071 end
jcallahan@1	2072 private.tracked_line = nil
jcallahan@167	2073 private.previous_spell_id = spell_id
jcallahan@85	2074
jcallahan@86	2075 if spell_name:match("^Harvest.+") then
jcallahan@177	2076 killed_npc_id = current_target_id
jcallahan@98	2077 private.harvesting = true
jcallahan@85	2078 end
jcallahan@92	2079
jcallahan@92	2080 if anvil_spell_ids[spell_id] then
jcallahan@92	2081 UpdateDBEntryLocation("objects", OBJECT_ID_ANVIL)
jcallahan@92	2082 elseif forge_spell_ids[spell_id] then
jcallahan@92	2083 UpdateDBEntryLocation("objects", OBJECT_ID_FORGE)
jcallahan@92	2084 end
jcallahan@0	2085 end
jcallahan@0	2086
jcallahan@90	2087
jcallahan@1	2088 function WDP:HandleSpellFailure(event_name, unit_id, spell_name, spell_rank, spell_line, spell_id)
jcallahan@1	2089 if unit_id ~= "player" then
jcallahan@1	2090 return
jcallahan@1	2091 end
jcallahan@0	2092
jcallahan@1	2093 if private.tracked_line == spell_line then
jcallahan@1	2094 private.tracked_line = nil
jcallahan@1	2095 end
jcallahan@147	2096 table.wipe(current_action)
jcallahan@0	2097 end
jcallahan@90	2098
jcallahan@90	2099
jcallahan@90	2100 do
jcallahan@90	2101 local function SetUnitField(field, required_type)
jcallahan@90	2102 local unit_type, unit_idnum = ParseGUID(_G.UnitGUID("npc"))
jcallahan@90	2103
jcallahan@90	2104 if not unit_idnum or (required_type and unit_type ~= required_type) then
jcallahan@90	2105 return
jcallahan@90	2106 end
jcallahan@90	2107
jcallahan@171	2108 if UnitTypeIsNPC(unit_type) then
jcallahan@90	2109 NPCEntry(unit_idnum)[field] = true
jcallahan@90	2110 elseif unit_type == private.UNIT_TYPES.OBJECT then
jcallahan@90	2111 DBEntry("objects", unit_idnum)[field] = true
jcallahan@93	2112 UpdateDBEntryLocation("objects", unit_idnum)
jcallahan@90	2113 end
jcallahan@90	2114 end
jcallahan@90	2115
jcallahan@90	2116
jcallahan@90	2117 function WDP:AUCTION_HOUSE_SHOW(event_name)
jcallahan@90	2118 SetUnitField("auctioneer", private.UNIT_TYPES.NPC)
jcallahan@90	2119 end
jcallahan@90	2120
jcallahan@90	2121
jcallahan@90	2122 function WDP:BANKFRAME_OPENED(event_name)
jcallahan@90	2123 SetUnitField("banker", private.UNIT_TYPES.NPC)
jcallahan@90	2124 end
jcallahan@90	2125
jcallahan@90	2126
jcallahan@90	2127 function WDP:BATTLEFIELDS_SHOW(event_name)
jcallahan@90	2128 SetUnitField("battlemaster", private.UNIT_TYPES.NPC)
jcallahan@90	2129 end
jcallahan@90	2130
jcallahan@90	2131
jcallahan@92	2132 function WDP:FORGE_MASTER_OPENED(event_name)
jcallahan@90	2133 SetUnitField("arcane_reforger", private.UNIT_TYPES.NPC)
jcallahan@90	2134 end
jcallahan@90	2135
jcallahan@90	2136
jcallahan@92	2137 function WDP:GOSSIP_SHOW(event_name)
jcallahan@90	2138 local gossip_options = { _G.GetGossipOptions() }
jcallahan@90	2139
jcallahan@90	2140 for index = 2, #gossip_options, 2 do
jcallahan@90	2141 if gossip_options[index] == "binder" then
jcallahan@90	2142 SetUnitField("innkeeper", private.UNIT_TYPES.NPC)
jcallahan@90	2143 return
jcallahan@90	2144 end
jcallahan@90	2145 end
jcallahan@90	2146 end
jcallahan@90	2147
jcallahan@90	2148
jcallahan@93	2149 function WDP:GUILDBANKFRAME_OPENED(event_name)
jcallahan@93	2150 SetUnitField("guild_bank", private.UNIT_TYPES.OBJECT)
jcallahan@93	2151 end
jcallahan@93	2152
jcallahan@93	2153
jcallahan@189	2154 function WDP:ITEM_UPGRADE_MASTER_OPENED(event_name)
jcallahan@189	2155 SetUnitField("item_upgrade_master", private.UNIT_TYPES.NPC)
jcallahan@189	2156 end
jcallahan@189	2157
jcallahan@189	2158
jcallahan@90	2159 function WDP:TAXIMAP_OPENED(event_name)
jcallahan@90	2160 SetUnitField("flight_master", private.UNIT_TYPES.NPC)
jcallahan@90	2161 end
jcallahan@90	2162
jcallahan@90	2163
jcallahan@90	2164 function WDP:TRANSMOGRIFY_OPEN(event_name)
jcallahan@90	2165 SetUnitField("transmogrifier", private.UNIT_TYPES.NPC)
jcallahan@90	2166 end
jcallahan@90	2167
jcallahan@90	2168
jcallahan@90	2169 function WDP:VOID_STORAGE_OPEN(event_name)
jcallahan@90	2170 SetUnitField("void_storage", private.UNIT_TYPES.NPC)
jcallahan@90	2171 end
jcallahan@90	2172 end -- do-block

Mercurial > wow > wowdb-profiler

annotate Main.lua @ 205:6cac044fd0f9