Back to Home

ESO Lua File v100021

ingame/collections/collectionsbook_manager.lua

[◄ back to folders ]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
--
--[[ CollectionsBook Singleton ]]--
--
ZO_COLLECTIONS_SYSTEM_NAME = "collections"
ZO_COLLECTIONS_SEARCH_ROOT = "root"
local CollectionsBook_Singleton = ZO_CallbackObject:Subclass()
function CollectionsBook_Singleton:New(...)
    local collectionsSingleton = ZO_CallbackObject.New(self)
    collectionsSingleton:Initialize(...) -- ZO_CallbackObject does not have an initialize function
    return collectionsSingleton
end
function CollectionsBook_Singleton:Initialize()
    self.ownedHouses = {}
    self.categoriesWithNewCollectibles = {}
    self.searchString = ""
    self.searchResults = {}
    self.searchSpecializationFilters = {}
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTIBLES_SEARCH_RESULTS_READY, function() self:UpdateSearchResults() end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTIBLE_REQUEST_BROWSE_TO, function(eventId, ...) self:BrowseToCollectible(...) end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTIBLE_UPDATED, function(eventId, ...) self:OnCollectibleUpdated(...) end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTION_UPDATED, function(eventId, ...) self:OnCollectionUpdated(...) end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTIBLES_UPDATED, function(eventId, ...) self:OnCollectiblesUpdated(...) end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTIBLE_NOTIFICATION_REMOVED, function(eventId, ...) self:OnCollectionNotificationRemoved(...) end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_COLLECTIBLE_NEW_STATUS_CLEARED, function(eventId, ...) self:OnCollectibleNewStatusRemoved(...) end)
    EVENT_MANAGER:RegisterForEvent("CollectionsBook_Singleton", EVENT_ACTION_UPDATE_COOLDOWNS, function(eventId, ...) self:OnUpdateCooldowns(...) end)
end
function CollectionsBook_Singleton:SetSearchString(searchString)
    self.searchString = searchString or ""
    StartCollectibleSearch(searchString)
end
function CollectionsBook_Singleton:SetSearchCategorySpecializationFilters(...)
    local argCount = select("#", ...)
    if argCount == NonContiguousCount(self.searchSpecializationFilters) then
        local noChange = true
        for i = 1, argCount do
            local categorySpecialization = select(i, ...)
            if not self.searchSpecializationFilters[categorySpecialization] then
                noChange = false
                break
            end
        end
        if noChange then
            return
        end
    end
    ZO_ClearTable(self.searchSpecializationFilters)
    for i = 1, argCount do
        local categorySpecialization = select(i, ...)
        self.searchSpecializationFilters[categorySpecialization] = true
    end
end
function CollectionsBook_Singleton:UpdateSearchResults()
    ZO_ClearTable(self.searchResults)
    local numResults = GetNumCollectiblesSearchResults()
    for i = 1, numResults do
        local categoryIndex, subcategoryIndex, collectibleIndex = GetCollectiblesSearchResult(i)
        local categorySpecialization = GetCollectibleCategorySpecialization(categoryIndex)
        if self.searchSpecializationFilters[categorySpecialization] or NonContiguousCount(self.searchSpecializationFilters) == 0 then
            if not self.searchResults[categoryIndex] then
                self.searchResults[categoryIndex] = {}
            end
            local effectiveSubCategory = subcategoryIndex or ZO_COLLECTIONS_SEARCH_ROOT
            if not self.searchResults[categoryIndex][effectiveSubCategory] then
                self.searchResults[categoryIndex][effectiveSubCategory] = {}
            end
            self.searchResults[categoryIndex][effectiveSubCategory][collectibleIndex] = true
        end
    end
    self:FireCallbacks("UpdateSearchResults")
end
function CollectionsBook_Singleton:GetSearchResults()
    if zo_strlen(self.searchString) > 1 then
        return self.searchResults
    end
    return nil
end
function CollectionsBook_Singleton:BrowseToCollectible(...)
    SYSTEMS:GetObject(ZO_COLLECTIONS_SYSTEM_NAME):BrowseToCollectible(...)
end
function CollectionsBook_Singleton:RefreshOwnedHouses()
    ZO_ClearTable(self.ownedHouses)
    local numAllHouses = GetTotalCollectiblesByCategoryType(COLLECTIBLE_CATEGORY_TYPE_HOUSE)
    for houseIndex = 1, numAllHouses do
        local collectibleId = GetCollectibleIdFromType(COLLECTIBLE_CATEGORY_TYPE_HOUSE, houseIndex)
        local houseId = GetCollectibleReferenceId(collectibleId)
        local isUnlocked = IsCollectibleUnlocked(collectibleId)
        if isUnlocked then
            self.ownedHouses[collectibleId] = { houseId = houseId }
        end
    end
end
function CollectionsBook_Singleton:RefreshOwnedHouseById(collectibleId, justUnlocked)
    local isUnlocked, _, _, categoryType = select(5, GetCollectibleInfo(collectibleId))
    if categoryType == COLLECTIBLE_CATEGORY_TYPE_HOUSE and isUnlocked then
        local houseId = GetCollectibleReferenceId(collectibleId)
        if justUnlocked then
            self.ownedHouses[collectibleId] = { showPermissionsDialogOnEnter = true }
        end
        self.ownedHouses[collectibleId].houseId = houseId
    end
end
function CollectionsBook_Singleton:OnCollectibleUpdated(collectibleId, justUnlocked)
    self:RefreshOwnedHouseById(collectibleId, justUnlocked)
    self:FireCallbacks("OnCollectibleUpdated", collectibleId, justUnlocked)
end
function CollectionsBook_Singleton:OnCollectionUpdated(...)
    self:FireCallbacks("OnCollectionUpdated", ...)
end
function CollectionsBook_Singleton:OnCollectiblesUpdated(...)
    self:FireCallbacks("OnCollectiblesUpdated", ...)
end
function CollectionsBook_Singleton:OnCollectionNotificationRemoved(...)
    self:FireCallbacks("OnCollectionNotificationRemoved", ...)
end
function CollectionsBook_Singleton:OnCollectibleNewStatusRemoved(collectibleId)
    self:FireCallbacks("OnCollectibleNewStatusRemoved", collectibleId)
end
function CollectionsBook_Singleton:OnUpdateCooldowns(...)
    self:FireCallbacks("OnUpdateCooldowns", ...)
end
function CollectionsBook_Singleton:IsCategoryIndexDLC(categoryIndex)
    return GetCollectibleCategorySpecialization(categoryIndex) == COLLECTIBLE_CATEGORY_SPECIALIZATION_DLC
end
function CollectionsBook_Singleton:GetOwnedHouses()
    return self.ownedHouses
end
function CollectionsBook_Singleton:IsOwnedHouseCollectibleUnlocked(collectibleId)
    return self.ownedHouses[collectibleId] ~= nil
end
function CollectionsBook_Singleton:DoesHousePermissionsDialogNeedToBeShownForCollectible(collectibleId)
    return self.ownedHouses[collectibleId] and self.ownedHouses[collectibleId].showPermissionsDialogOnEnter
end
function CollectionsBook_Singleton:MarkHouseCollectiblePermissionLoadDialogShown(collectibleId)
    if self.ownedHouses[collectibleId] then
        self.ownedHouses[collectibleId].showPermissionsDialogOnEnter = false
    end
end
function CollectionsBook_Singleton:IsCategoryIndexHousing(categoryIndex)
    return GetCollectibleCategorySpecialization(categoryIndex) == COLLECTIBLE_CATEGORY_SPECIALIZATION_HOUSING
end
function CollectionsBook_Singleton.DoesCollectibleListHaveVisibleCollectible(...)
    for i = 1, select("#", ...) do
        local id = select(i, ...)
        if DoesCollectibleHaveVisibleAppearance(id) then
            local isActive = select(7, GetCollectibleInfo(id))
            if isActive and not WouldCollectibleBeHidden(id) then
                return true
            end
        end
    end
    return false
end
function CollectionsBook_Singleton.DoesCollectibleListHaveNewCollectible(...)
    for i = 1, select("#", ...) do
        local id = select(i, ...)
        if IsCollectibleNew(id) then
            return true
        end
    end
    return false
end
function CollectionsBook_Singleton.GetCategoryCollectibleIds(categoryIndex, subCategoryIndex, index, ...)
    if index >= 1 then
        local id = GetCollectibleId(categoryIndex, subCategoryIndex, index)
        index = index - 1
        return CollectionsBook_Singleton.GetCategoryCollectibleIds(categoryIndex, subCategoryIndex, index, id, ...)
    end
    return ...
end
function CollectionsBook_Singleton:ClearCachedNewStatusForAllCollectibles()
    ZO_ClearTable(self.categoriesWithNewCollectibles)
end
function CollectionsBook_Singleton:ClearCachedNewStatusForCollectible(collectibleId)
    local categoryIndex, subcategoryIndex, collectibleIndex = GetCategoryInfoFromCollectibleId(collectibleId)
    if categoryIndex then
        if subcategoryIndex then
            local categoryNewData = self.categoriesWithNewCollectibles[categoryIndex]
            if categoryNewData and categoryNewData.subcategories then
                categoryNewData.subcategories[subcategoryIndex] = nil
                categoryNewData.categoryOrSubcategoriesHaveAnyNew = nil
            end
        else
            self.categoriesWithNewCollectibles[categoryIndex] = nil
        end
    end
end
function CollectionsBook_Singleton:DoesCategoryHaveAnyNewCollectibles(categoryIndex, subcategoryIndex)
    if subcategoryIndex == nil then
        local categoryHasAnyNew
        local numSubCategories, numCollectibles, unlockedCollectibles = select(2, GetCollectibleCategoryInfo(categoryIndex))
        local categoryNewData = self.categoriesWithNewCollectibles[categoryIndex]
        if categoryNewData then
            if categoryNewData.categoryOrSubcategoriesHaveAnyNew ~= nil then
                return categoryNewData.categoryOrSubcategoriesHaveAnyNew
            elseif categoryNewData.hasAnyNew ~= nil then
                categoryHasAnyNew = categoryNewData.hasAnyNew
            else
                categoryHasAnyNew = CollectionsBook_Singleton.DoesCollectibleListHaveNewCollectible(CollectionsBook_Singleton.GetCategoryCollectibleIds(categoryIndex, subcategoryIndex, numCollectibles))
                categoryNewData.hasAnyNew = categoryHasAnyNew
            end
        else
            categoryHasAnyNew = CollectionsBook_Singleton.DoesCollectibleListHaveNewCollectible(CollectionsBook_Singleton.GetCategoryCollectibleIds(categoryIndex, subcategoryIndex, numCollectibles))
            categoryNewData = { hasAnyNew = categoryHasAnyNew }
            self.categoriesWithNewCollectibles[categoryIndex] = categoryNewData
        end
        local categoryOrSubcategoriesHaveAnyNew = false
        if categoryHasAnyNew then
            categoryOrSubcategoriesHaveAnyNew = true
        else
            for i = 1, numSubCategories do
                if self:DoesCategoryHaveAnyNewCollectibles(categoryIndex, i) then
                    categoryOrSubcategoriesHaveAnyNew = true
                    break
                end
            end
        end
        
        categoryNewData.categoryOrSubcategoriesHaveAnyNew = categoryOrSubcategoriesHaveAnyNew
        return categoryOrSubcategoriesHaveAnyNew
    else
        local categoryNewData = self.categoriesWithNewCollectibles[categoryIndex]
        if categoryNewData == nil then
            local NO_SUBCATEGORY = nil
            local numCollectibles = select(2, GetCollectibleSubCategoryInfo(categoryIndex, NO_SUBCATEGORY))
            categoryHasAnyNew = CollectionsBook_Singleton.DoesCollectibleListHaveNewCollectible(CollectionsBook_Singleton.GetCategoryCollectibleIds(categoryIndex, NO_SUBCATEGORY, numCollectibles))
            categoryNewData = { hasAnyNew = categoryHasAnyNew }
            self.categoriesWithNewCollectibles[categoryIndex] = categoryNewData
        end
        if categoryNewData.subcategories == nil then
            categoryNewData.subcategories = {}
        end
        local subcategoryHasAnyNew = categoryNewData.subcategories[subcategoryIndex]
        if subcategoryHasAnyNew == nil then
            local numCollectibles = select(2, GetCollectibleSubCategoryInfo(categoryIndex, subcategoryIndex))
            subcategoryHasAnyNew = CollectionsBook_Singleton.DoesCollectibleListHaveNewCollectible(CollectionsBook_Singleton.GetCategoryCollectibleIds(categoryIndex, subcategoryIndex, numCollectibles))
            categoryNewData.subcategories[subcategoryIndex] = subcategoryHasAnyNew
        end
        return subcategoryHasAnyNew
    end
    return false
end
do
    local ANY_SUBCATEGORY = nil
    function CollectionsBook_Singleton.HasAnyNewCollectibles()
        if getCollectiblesFunction == nil then
            getCollectiblesFunction = CollectionsBook_Singleton.GetCategoryCollectibleIds
        end
        local numCategories = GetNumCollectibleCategories()
        for categoryIndex = 1, numCategories do
            if COLLECTIONS_BOOK_SINGLETON:DoesCategoryHaveAnyNewCollectibles(categoryIndex, ANY_SUBCATEGORY) then
                return true
            end
        end
        return false
    end
end
do
    local function ShouldAddCollectibleToList(topLevelCategoryIndex, subCategoryIndex, entryIndex, ...)
        local collectibleId = GetCollectibleId(topLevelCategoryIndex, subCategoryIndex, entryIndex)
        if not IsCollectibleUnlocked(collectibleId) then
            return false
        end
        local collectibleCategoryType = GetCollectibleCategoryType(collectibleId)
        for i = 1, select("#", ...) do
            local filterFunction = select(i, ...)
            if not filterFunction(collectibleCategoryType) then
                return false
            end
        end
        return true
    end
    function CollectionsBook_Singleton.GetCollectibleIdsFromTopLevelCategory(topLevelCategoryIndex, ...) --... are filter functions that takes COLLECTIBLE_CATEGORY_TYPE as an argument
        local foundCollectibleIds = {}
        local _, numSubCategories, numCollectibles = GetCollectibleCategoryInfo(topLevelCategoryIndex)
        for entryIndex = 1, numCollectibles do
            if ShouldAddCollectibleToList(topLevelCategoryIndex, nil, entryIndex, ...) then
                local collectibleId = GetCollectibleId(topLevelCategoryIndex, nil, entryIndex)
                table.insert(foundCollectibleIds, collectibleId)
            end
        end
        for subCategoryIndex = 1, numSubCategories do
            local _, subCategoryEntries = GetCollectibleSubCategoryInfo(topLevelCategoryIndex, subCategoryIndex)
            for entryIndex = 1, subCategoryEntries do
                if ShouldAddCollectibleToList(topLevelCategoryIndex, subCategoryIndex, entryIndex, ...) then
                    local collectibleId = GetCollectibleId(topLevelCategoryIndex, subCategoryIndex, entryIndex)
                    table.insert(foundCollectibleIds, collectibleId)
                end
            end
        end
        return foundCollectibleIds
    end
end
function ZO_GetCollectibleCategoryAndName(collectibleId)
    local categoryIndex, subcategoryIndex = GetCategoryInfoFromCollectibleId(collectibleId)
    local categoryName
    if subcategoryIndex then
        categoryName = GetCollectibleSubCategoryInfo(categoryIndex, subcategoryIndex)
    else
        categoryName = GetCollectibleCategoryInfo(categoryIndex)
    end
    local collectibleName = GetCollectibleName(collectibleId)
    return categoryName, collectibleName
end
    if IsConsoleUI() then
        ZO_Dialogs_ShowGamepadDialog("CHAPTER_UPGRADE_STORE_CONSOLE")
    else
        ZO_Dialogs_ShowPlatformDialog("CHAPTER_UPGRADE_STORE")
    end
end
COLLECTIONS_BOOK_SINGLETON = CollectionsBook_Singleton:New()