wow/wowdb-profiler: Main.lua annotate

annotate Main.lua @ 146:d49e46c45dcc

Bumped DB version to 9.

author	James D. Callahan III <jcallahan@curse.com>
date	Tue, 02 Oct 2012 14:15:59 -0500
parents	f8a5080a60c6
children	ca490b4a98e8

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

Mercurial > wow > wowdb-profiler

annotate Main.lua @ 146:d49e46c45dcc