You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.5 KiB
72 lines
2.5 KiB
2 years ago
|
local PetJournalEnhanced = LibStub("AceAddon-3.0"):GetAddon("PetJournalEnhanced")
|
||
|
|
local UniquePets = PetJournalEnhanced:NewModule("UniquePets")
|
||
|
|
local Config = PetJournalEnhanced:GetModule("Config")
|
||
|
|
local LibPetJournal = LibStub("LibPetJournal-2.0")
|
||
|
|
local L = LibStub("AceLocale-3.0"):GetLocale("PetJournalEnhanced")
|
||
|
|
local UNIQUE_PETS = L["Unique Pets"]
|
||
|
|
|
||
|
|
--Callback handler for updating unique pet count
|
||
|
|
function UniquePets:ScanPets()
|
||
|
|
local pets = {}
|
||
|
|
local count = 0
|
||
|
|
for i, petID in LibPetJournal:IteratePetIDs() do
|
||
|
|
local speciesID = C_PetJournal.GetPetInfoByPetID(petID)
|
||
|
|
if speciesID and not pets[speciesID] then
|
||
|
|
count = count + 1
|
||
|
|
pets[speciesID] = speciesID
|
||
|
|
end
|
||
|
|
end
|
||
|
|
UniquePets.frame.uniqueCount:SetText(count)
|
||
|
|
end
|
||
|
|
|
||
|
|
function UniquePets:OnInitialize()
|
||
|
|
self.config = PetJournalEnhanced:GetModule("Config")
|
||
|
|
self.frame = CreateFrame("frame", "PJEUniquePetCount", PetJournal, "InsetFrameTemplate3")
|
||
|
|
|
||
|
|
--Create unique pet count UI elements
|
||
|
|
local frame = self.frame
|
||
|
|
frame:ClearAllPoints()
|
||
|
|
frame:SetPoint("TOPLEFT", PetJournal, 70, -42)
|
||
|
|
frame:SetSize(130, 18)
|
||
|
|
frame.staticText = frame:CreateFontString(nil, "ARTWORK", "GameFontNormalSmall")
|
||
|
|
frame.uniqueCount = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlightSmall")
|
||
|
|
|
||
|
|
frame.staticText:ClearAllPoints()
|
||
|
|
frame.staticText:SetPoint("LEFT", frame, 10, 0)
|
||
|
|
--frame.staticText:SetPoint("RIGHT", frame.uniqueCount, "LEFT", -3, 0)
|
||
|
|
frame.staticText:SetText(UNIQUE_PETS)
|
||
|
|
|
||
|
|
frame.uniqueCount:ClearAllPoints()
|
||
|
|
frame.uniqueCount:SetPoint("RIGHT", frame, -10, 0)
|
||
|
|
frame.uniqueCount:SetText("0")
|
||
|
|
|
||
|
|
PetJournal:HookScript("OnShow", function()
|
||
|
|
self:SetShown(Config.display.uniquePetCount)
|
||
|
|
end )
|
||
|
|
|
||
|
|
--Inital ui state
|
||
|
|
LibPetJournal.RegisterCallback(self, "PetListUpdated", "ScanPets")
|
||
|
|
self:ScanPets()
|
||
|
|
end
|
||
|
|
|
||
|
|
--Shows unique pet count if enabled is not false or nil
|
||
|
|
function UniquePets:SetShown(enabled)
|
||
|
|
if enabled then
|
||
|
|
self.frame:Show()
|
||
|
|
PetJournal.PetCount:SetPoint("TopLeft", 70, -22)
|
||
|
|
PetJournal.PetCount:SetSize(130, 18)
|
||
|
|
else
|
||
|
|
self.frame:Hide()
|
||
|
|
PetJournal.PetCount:SetPoint("TopLeft", 70, -35)
|
||
|
|
PetJournal.PetCount:SetSize(130, 20)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
function UniquePets:GetDisplayUniquePetCount()
|
||
|
|
return self.db.global.display.uniquePetCount
|
||
|
|
end
|
||
|
|
|
||
|
|
function UniquePets:SetDisplayUniquePetCount(enabled)
|
||
|
|
self.db.global.display.uniquePetCount = enabled
|
||
|
|
end
|