ChoreTracker = LibStub('AceAddon-3.0'):NewAddon('ChoreTracker', 'AceConsole-3.0', 'AceEvent-3.0') local core = ChoreTracker local LQT, LDB, LDBIcon -- Localization local L = LibStub('AceLocale-3.0'):GetLocale('ChoreTracker') -- Get localized instances local Z = LibStub('LibBabble-Zone-3.0'):GetLookupTable() -- local CURRENT_MAX_LEVEL = 85 local MAX_VALOR_POINTS = 1000 local defaults = { global = {}, profile = { minimap = { hide = false, }, sortType = 1, sortDirection = 1, currentOnTop = false, showServer = false, showTotalVp = true, instances = { [Z['Dragon Soul']] = { abbreviation = 'DS', enable = true, removed = false, }, [Z['Baradin Hold']] = { abbreviation = 'BH', enable = true, removed = false, }, [Z['Firelands']] = { abbreviation = 'FL', enable = true, removed = false, }, [Z['The Bastion of Twilight']] = { abbreviation = 'BoT', enable = false, removed = false, }, [Z['Blackwing Descent']] = { abbreviation = 'BWD', enable = false, removed = false, }, [Z['Throne of the Four Winds']] = { abbreviation = '4W', enable = false, removed = false, }, }, lfrs = {} }, } -- Setup LFR Defaults local RFDungeonCount = GetNumRFDungeons() for i = 1, RFDungeonCount do id, instanceName = GetRFDungeonInfo(i) defaults.profile.lfrs[instanceName] = { enable = true, removed = false, } defaults.profile.lfrs[instanceName].abbreviation = string.sub(instanceName, 0, 1) end local options = { name = 'ChoreTracker', type = 'group', args = { general = { name = L['Settings'], type = 'group', order = 1, args = { minimap = { name = L['Hide Minimap Icon'], desc = L['Removes the icon from your minimap.'], type = 'toggle', order = 1, get = function(info) return core.db.profile.minimap.hide end, set = function(info, value) core.db.profile.minimap.hide = value LDBIcon[value and 'Hide' or 'Show'](LDBIcon, 'ChoreTracker') end, }, verticalHeader = { name = L['Vertical Sorting'], type = 'header', order = 2, }, currentOnTop = { name = L['Current Character On Top'], desc = L['Place the character you are currently logged in as on the top of the list.'], type = 'toggle', width = 'full', order = 3, get = function(info) return core.db.profile.currentOnTop end, set = function(info, value) core.db.profile.currentOnTop = value end, }, sortType = { name = L['Sort Field'], desc = L['Field to sort the tooltip by.'], type = 'select', order = 5, values = { L['Character'], L['Valor Points'], L['Class'] }, get = function(info) return core.db.profile.sortType end, set = function(info, value) core.db.profile.sortType = value end, }, sortingDirection = { name = L['Sorting Direction'], desc = L['Which direction to sort.'], type = 'select', order = 6, values = { L['Ascending'], L['Descending'] }, get = function(info) return core.db.profile.sortDirection end, set = function(info, value) core.db.profile.sortDirection = value end, }, otherHeader = { name = '', type = 'header', order = 20, }, showServer = { name = L['Show Server'], desc = L['Show the server abbreviation next to a character name in the list.'], type = 'toggle', width = 'full', order = 21, get = function(info) return core.db.profile.showServer end, set = function(info, value) core.db.profile.showServer = value end, }, showTotalVp = { name = L['Show Total VP'], desc = L['Show the total valor points for all characters in the tooltip.'], type = 'toggle', width = 'full', order = 22, get = function(info) return core.db.profile.showTotalVp end, set = function(info, value) core.db.profile.showTotalVp = value end, }, }, }, instances = { name = L['Instances'], type = 'group', order = 20, args = { }, }, }, } function core:OnInitialize() self.db = LibStub('AceDB-3.0'):New('ChoreTrackerDB', defaults, 'Default') self.character = { name = UnitName('player'), level = UnitLevel('player'), class = UnitClass('player'), realm = GetRealmName(), } self.character.class = self.character.class:lower():gsub("%s*(.-)%s*", "%1") if self.db.global[self.character.realm] == nil then self.db.global[self.character.realm] = {} end if self.db.global[self.character.realm][self.character.name] == nil and self.character.level == CURRENT_MAX_LEVEL then self.db.global[self.character.realm][self.character.name] = {} self.db.global[self.character.realm][self.character.name].class = self.character.class self.db.global[self.character.realm][self.character.name].valorPoints = { total = 0, points = 0, resetTime = 0, } self.db.global[self.character.realm][self.character.name].lockouts = {} end if self.db.global[self.character.realm][self.character.name] ~= nil then if self.db.global[self.character.realm][self.character.name].lfrs == nil and self.character.level == CURRENT_MAX_LEVEL then self.db.global[self.character.realm][self.character.name].lfrs = {} end end -- Add LFR stuff to profile if it isn't there already core:LFRProfileUpdate() end function core:OnEnable() LQT = LibStub('LibQTip-1.0') self.instanceInfoTime = false self.vpResetTime = false -- Setup font strings for later. (RAID_CLASS_COLORS always indexed in English?) self.fontObjects = { } for class, color in pairs(RAID_CLASS_COLORS) do class = class:lower() self.fontObjects[class] = CreateFont('ClassFont' .. class) self.fontObjects[class]:CopyFontObject(GameTooltipText) self.fontObjects[class]:SetTextColor(color.r, color.g, color.b) end self.fontObjects['green'] = CreateFont('FlagFontGreen') self.fontObjects['green']:CopyFontObject(GameTooltipText) self.fontObjects['green']:SetTextColor(0, 255, 0) self.fontObjects['red'] = CreateFont('FlagFontRed') self.fontObjects['red']:CopyFontObject(GameTooltipText) self.fontObjects['red']:SetTextColor(255, 0, 0) -- Setup instance stuff for options core:DrawInstanceOptions() -- Add options to Interface Panel LibStub('AceConfigRegistry-3.0'):RegisterOptionsTable('ChoreTracker', options) local ACD = LibStub('AceConfigDialog-3.0') ACD:AddToBlizOptions('ChoreTracker', 'ChoreTracker') options.args.profile = LibStub('AceDBOptions-3.0'):GetOptionsTable(self.db) -- Setup LDB LDB = LibStub('LibDataBroker-1.1'):NewDataObject('ChoreTracker', { type = 'data source', text = 'ChoreTracker', icon = 'Interface\\AddOns\\ChoreTracker\\icon', OnClick = function(self, button) if button == 'RightButton' then if LibStub("AceConfigDialog-3.0").OpenFrames['ChoreTracker'] then LibStub('AceConfigDialog-3.0'):Close('ChoreTracker') else LibStub('AceConfigDialog-3.0'):Open('ChoreTracker') end else -- Cycle through our sort options if core.db.profile.sortType == 1 then core.db.profile.sortType = 2 core:DrawTooltip() elseif core.db.profile.sortType == 2 then core.db.profile.sortType = 3 core:DrawTooltip() else core.db.profile.sortType = 1 core:DrawTooltip() end end end, OnEnter = function(self) core:DrawTooltip() core.tooltip:SmartAnchorTo(self) core.tooltip:Show() end, OnLeave = function(self) LQT:Release(core.tooltip) core.tooltip = nil end, }) -- Deal with minimap LDBIcon = LibStub('LibDBIcon-1.0') LDBIcon:Register('ChoreTracker', LDB, self.db.profile.minimap) if self.db.profile.minimap.hide then LDBIcon:Hide('ChoreTracker') else LDBIcon:Show('ChoreTracker') end -- Register events if self.character.level == CURRENT_MAX_LEVEL then self:RegisterEvent('PLAYER_ENTERING_WORLD') self:RegisterEvent('CALENDAR_UPDATE_EVENT_LIST') self:RegisterEvent('LFG_UPDATE_RANDOM_INFO') self:RegisterEvent('LFG_LOCK_INFO_RECEIVED') self:RegisterEvent('UPDATE_INSTANCE_INFO') self:RegisterEvent('CURRENCY_DISPLAY_UPDATE') self:RegisterEvent('CHAT_MSG_CURRENCY') self:RegisterEvent('INSTANCE_ENCOUNTER_ENGAGE_UNIT') end -- Get calendar events information OpenCalendar() -- Reset data if necessary core:ResetRaidLockouts() core:ResetValorPoints() core:ResetLFRLockouts() end --[[ EVENTS ]]-- function core:PLAYER_ENTERING_WORLD() self.vpResetTime = core:FindLockout(Z['Baradin Hold']) end function core:CALENDAR_UPDATE_EVENT_LIST() self.vpResetTime = core:FindLockout(Z['Baradin Hold']) end function core:UPDATE_INSTANCE_INFO() self.instanceInfoTime = time() core:UpdateRaidLockouts() end function core:LFG_UPDATE_RANDOM_INFO() core:UpdateValorPoints() end function core:LFG_LOCK_INFO_RECEIVED() core:UpdateLFRLockouts() end function core:CURRENCY_DISPLAY_UPDATE() core:UpdateValorPoints() end function core:CHAT_MSG_CURRENCY() RequestRaidInfo() RequestLFDPlayerLockInfo() end -- Might only fire for encounters with a boss frame. function core:INSTANCE_ENCOUNTER_ENGAGE_UNIT() RequestRaidInfo() RequestLFDPlayerLockInfo() end --[[ FUNCTIONS ]]-- function core:UpdateValorPoints() local _, amount, _, earnedThisWeek = GetCurrencyInfo(396) if self.db.global[self.character.realm][self.character.name].valorPoints == nil then self.db.global[self.character.realm][self.character.name].valorPoints = {} end self.db.global[self.character.realm][self.character.name].valorPoints.points = earnedThisWeek self.db.global[self.character.realm][self.character.name].valorPoints.total = amount if self.vpResetTime ~= false then self.db.global[self.character.realm][self.character.name].valorPoints.resetTime = self.vpResetTime end end function core:ResetValorPoints() for realm, realmTable in pairs(self.db.global) do for name in pairs(realmTable) do if self.db.global[realm][name].valorPoints.resetTime < time() then self.db.global[realm][name].valorPoints.points = 0 self.db.global[realm][name].valorPoints.resetTime = 0 end end end end function core:UpdateRaidLockouts() local savedInstances = GetNumSavedInstances() for i = 1, savedInstances do local instanceName, _, instanceReset, _, _, _, _, _, _, _, _, defeatedBosses = GetSavedInstanceInfo(i) if self.db.profile.instances[instanceName] ~= nil then if instanceReset > 0 then self.db.global[self.character.realm][self.character.name].lockouts[instanceName] = {} self.db.global[self.character.realm][self.character.name].lockouts[instanceName].defeatedBosses = defeatedBosses self.db.global[self.character.realm][self.character.name].lockouts[instanceName].resetTime = self.instanceInfoTime + instanceReset end end end end function core:ResetRaidLockouts() for realm,realmTable in pairs(self.db.global) do for name in pairs(realmTable) do for instance,instanceTable in pairs(self.db.global[realm][name].lockouts) do if instanceTable.resetTime < time() then self.db.global[realm][name].lockouts[instance] = nil end end end end end function core:UpdateLFRLockouts() local RFDungeonCount = GetNumRFDungeons() for i = 1, RFDungeonCount do id, instanceName = GetRFDungeonInfo(i) _, defeatedBosses = GetLFGDungeonNumEncounters(id) if self.db.profile.lfrs[instanceName] ~= nil then self.db.global[self.character.realm][self.character.name].lfrs[instanceName] = {} self.db.global[self.character.realm][self.character.name].lfrs[instanceName].defeatedBosses = defeatedBosses self.db.global[self.character.realm][self.character.name].lfrs[instanceName].resetTime = self.vpResetTime end end end function core:ResetLFRLockouts() for realm,realmTable in pairs(self.db.global) do for name in pairs(realmTable) do for instance,instanceTable in pairs(self.db.global[realm][name].lfrs) do if instanceTable.resetTime < time() then self.db.global[realm][name].lfrs[instance] = nil end end end end end function core:DrawInstanceOptions() -- Redraw our instance options everytime they are updated. options.args.instances.args = { instance = { name = L['Add instance to track.'], desc = L['Enter an instance on a lockout that you would like ChoreTracker to track.'], type = 'input', order = 1, set = function(info, value) if core:FindLockout(value) then self.db.profile.instances[value] = { } self.db.profile.instances[value].abbreviation = string.sub(value,0,1) self.db.profile.instances[value].enable = true self.db.profile.instances[value].removed = false core:DrawInstanceOptions() else print('Invalid instance') end end, }, lfrsHeader = { name = L['Looking for Raid Instances'], type = 'header', order = 100, }, instancesHeader = { name = L['Instances'], type = 'header', order = 500, }, } local i = 1 for instance, abbreviation in pairs(self.db.profile.instances) do if self.db.profile.instances[instance].removed == false then options.args.instances.args[instance .. 'Enable'] = { type = 'toggle', name = instance, order = 500 + (5 * i) + 0, get = function(info) return self.db.profile.instances[instance].enable end, set = function(info, value) self.db.profile.instances[instance].enable = value core:DrawInstanceOptions() end, } options.args.instances.args[instance .. 'Abbreviation'] = { type = 'input', name = '', order = 500 + (5 * i) + 1, width = 'half', get = function(info) return self.db.profile.instances[instance].abbreviation end, set = function(info, value) self.db.profile.instances[instance].abbreviation = value end, } options.args.instances.args[instance .. 'Remove'] = { type = 'execute', name = L['Remove'], order = 500 + (5 * i) + 2, width = 'half', confirm = true, func = function() self.db.profile.instances[instance].removed = true core:DrawInstanceOptions() end, } options.args.instances.args[instance .. 'Spacer'] = { type = 'description', name = '', order = 500 + (5 * i) + 3, } i = i + 1 end end i = 1 for instance, abbreviation in pairs(self.db.profile.lfrs) do if self.db.profile.lfrs[instance].removed == false then options.args.instances.args[instance .. 'Enable'] = { type = 'toggle', name = instance, order = 100 + (5 * i) + 0, get = function(info) return self.db.profile.lfrs[instance].enable end, set = function(info, value) self.db.profile.lfrs[instance].enable = value core:DrawInstanceOptions() end, } options.args.instances.args[instance .. 'Abbreviation'] = { type = 'input', name = '', order = 100 + (5 * i) + 1, width = 'half', get = function(info) return self.db.profile.lfrs[instance].abbreviation end, set = function(info, value) self.db.profile.lfrs[instance].abbreviation = value end, } options.args.instances.args[instance .. 'Remove'] = { type = 'execute', name = L['Remove'], order = 100 + (5 * i) + 2, width = 'half', confirm = true, func = function() self.db.profile.lfrs[instance].removed = true core:DrawInstanceOptions() end, } options.args.instances.args[instance .. 'Spacer'] = { type = 'description', name = '', order = 100 + (5 * i) + 3, } i = i + 1 end end end function core:FindLockout(instance) -- We need to have access to the instance lockouts on the calendar. local currentCalendarSetting = GetCVar('calendarShowResets') SetCVar('calendarShowResets', 1) -- Figure out what time the server resets daily information local questReset = GetQuestResetTime() local resetTime = date('*t', time() + questReset) -- Figure out reset day using next BH lockout local _, month, day, year = CalendarGetDate() local monthOffset = 0 local resetDate = false while resetDate == false do local todaysEvents = CalendarGetNumDayEvents(monthOffset, day) for i = 1,todaysEvents do if todaysEvents == 0 then break end local title,hour,minute = CalendarGetDayEvent(monthOffset, day, i) if title == instance then resetDate = { year = year, month = month + monthOffset, day = day } end end day = day + 1 if day > 31 then if monthOffset == 1 then break end day = 1 monthOffset = 1 end end -- Reset the calendar to the original settings SetCVar('calendarShowResets', currentCalendarSetting) -- And combine for the reset timestamp if(resetDate ~= false) then resetDate.hour = resetTime.hour resetDate.min = resetTime.min resetDate.sec = resetTime.sec return time(resetDate) else return false end end function core:DrawTooltip() -- UpdateChores before we show the tooltip to make sure we have the most recent data if self.character.level == CURRENT_MAX_LEVEL then -- Should not update without being 100% sure our raid info is correct core:UpdateValorPoints() core:UpdateRaidLockouts() core:UpdateLFRLockouts() end if self.tooltip then self.tooltip:ClearAllPoints() self.tooltip:Clear() self.tooltip = nil end local columnCount = 2 for instance in pairs(self.db.profile.instances) do if self.db.profile.instances[instance].enable == true and self.db.profile.instances[instance].removed == false then columnCount = columnCount + 1 end end for instance in pairs(self.db.profile.lfrs) do if self.db.profile.lfrs[instance].enable == true and self.db.profile.lfrs[instance].removed == false then columnCount = columnCount + 1 end end self.tooltip = LQT:Acquire('ChoreTrackerTooltip', columnCount, 'LEFT', 'CENTER', 'RIGHT') -- Populate a table with the information we want for our tooltip local tooltipTable = {} local currentTable = {} for realm in pairs(self.db.global) do for name in pairs(self.db.global[realm]) do local valorPoints = self.db.global[realm][name].valorPoints.points local class = self.db.global[realm][name].class local totalVp = self.db.global[realm][name].valorPoints.total if valorPoints == nil then valorPoints = 0 end local characterTable = { name = name, realm = realm, class = class, valorPoints = valorPoints, totalVp = totalVp } for instance in pairs(self.db.profile.instances) do local defeatedBosses if self.db.global[realm][name].lockouts[instance] ~= nil then defeatedBosses = self.db.global[realm][name].lockouts[instance].defeatedBosses else defeatedBosses = 0 end characterTable[instance] = defeatedBosses end for instance in pairs(self.db.profile.lfrs) do local defeatedBosses if self.db.global[realm][name].lfrs[instance] ~= nil then defeatedBosses = self.db.global[realm][name].lfrs[instance].defeatedBosses else defeatedBosses = 0 end characterTable[instance] = defeatedBosses end if name == UnitName('player') and self.db.profile.currentOnTop == true then currentTable = characterTable else table.insert(tooltipTable, characterTable) end end end -- Sort table according to options. local sortTooltip = function(a, b) local aValue, bValue if self.db.profile.sortType == 1 then aValue = a.name:lower() bValue = b.name:lower() elseif self.db.profile.sortType == 2 then aValue = a.valorPoints bValue = b.valorPoints elseif self.db.profile.sortType == 3 then aValue = a.class bValue = b.class end if self.db.profile.sortDirection == 1 then return aValue < bValue else return aValue > bValue end end table.sort(tooltipTable, sortTooltip ) -- Toss the current character on top if it is set that way if self.db.profile.currentOnTop == true then table.insert(tooltipTable, 1, currentTable) end -- Create a table for the header; vpPos to decide where to place Valor Points column -- Draw tooltip table then looped through. -- Draw the tooltip self.tooltip:AddHeader('') self.tooltip:SetScale(1) local valorPointColumn = self.tooltip:AddColumn('LEFT') self.tooltip:SetCell(1, 1, '') self.tooltip:SetCell(1, 2, 'VP') -- Build and sort our headers local headerTable = { } --headerTable['Valor Points'] = { abbreviation = 'VP', enable = true, removed = false, } for instance, instanceInfo in pairs(self.db.profile.instances) do if self.db.profile.instances[instance].enable == true and self.db.profile.instances[instance].removed == false then table.insert(headerTable,instanceInfo) end end for instance, instanceInfo in pairs(self.db.profile.lfrs) do if self.db.profile.lfrs[instance].enable == true and self.db.profile.lfrs[instance].removed == false then table.insert(headerTable,instanceInfo) end end local nextColumn = 3 for instance,instanceInfo in pairs(headerTable) do self.tooltip:SetCell(1, nextColumn, instanceInfo.abbreviation, nil, 'CENTER') nextColumn = nextColumn + 1 end if self.db.profile.showTotalVp == true then self.tooltip:SetCell(1, nextColumn, 'Total VP') nextColumn = nextColumn + 1 end for _,information in pairs(tooltipTable) do if self.db.profile.showServer then information.name = information.name .. '-' .. string.sub(information.realm,0,3) end local characterLine = self.tooltip:AddLine('') self.tooltip:SetCell(characterLine, 1, information.name, self.fontObjects[information.class], 'LEFT') local valorPointColor if information.valorPoints == MAX_VALOR_POINTS then valorPointColor = self.fontObjects['red'] else valorPointColor = self.fontObjects['green'] end self.tooltip:SetCell(characterLine, 2, information.valorPoints, valorPointColor, 'RIGHT') local nextColumn = 3 for instance, abbreviation in pairs(self.db.profile.instances) do if self.db.profile.instances[instance].enable == true and self.db.profile.instances[instance].removed == false then local instanceColor if information[instance] == 0 then instanceColor = self.fontObjects['green'] else instanceColor = self.fontObjects['red'] end self.tooltip:SetCell(characterLine, nextColumn, information[instance], instanceColor, 'RIGHT') nextColumn = nextColumn + 1 end end for instance, abbreviation in pairs(self.db.profile.lfrs) do if self.db.profile.lfrs[instance].enable == true and self.db.profile.lfrs[instance].removed == false then local instanceColor if information[instance] == 0 then instanceColor = self.fontObjects['green'] else instanceColor = self.fontObjects['red'] end self.tooltip:SetCell(characterLine, nextColumn, information[instance], instanceColor, 'RIGHT') nextColumn = nextColumn + 1 end end if self.db.profile.showTotalVp then self.tooltip:SetCell(characterLine, nextColumn, information.totalVp, self.fontObjects['green'], 'RIGHT') end end end --[[ PROFILE UPDATES ]]-- function core:LFRProfileUpdate() for realm,realmTable in pairs(self.db.global) do for name in pairs(realmTable) do if self.db.global[realm][name].lfrs == nil then self.db.global[realm][name].lfrs = {} end for instance in pairs(self.db.profile.lfrs) do if self.db.global[realm][name].lfrs[instance] == nil then self.db.global[realm][name].lfrs[instance] = {} self.db.global[realm][name].lfrs[instance].defeatedBosses = 0 self.db.global[realm][name].lfrs[instance].resetTime = 0 end end end end end