groove/audiobookshelf
1
0
Fork 0
mirror of https://github.com/advplyr/audiobookshelf.git synced 2026-07-12 12:21:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Limit collections to return only 2 items when getting list of ALL collections (otherwise, return the entire collection)

Browse source
This commit is contained in:
Brian C. Arnold 2024-05-26 12:18:17 -04:00
parent 36f46072fd
commit 44ff2e7a99
2 changed files with 15 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
server/controllers/LibraryController.js
View file

@ -492,7 +492,7 @@ class LibraryController {
}
// TODO: Create paginated queries
let collections = await Database.collectionModel.getOldCollectionsJsonExpanded(req.user, req.library.id, include)
let collections = await Database.collectionModel.getOldCollectionsJsonExpanded(req.user, req.library.id, include, 2)
payload.total = collections.length

23
server/models/Collection.js
View file

@ -40,9 +40,10 @@ class Collection extends Model {
* @param {[oldUser]} user
* @param {[string]} libraryId
* @param {[string[]]} include
* @param {[number]} limitCollectionSize
* @returns {Promise<object[]>} oldCollection.toJSONExpanded
*/
static async getOldCollectionsJsonExpanded(user, libraryId, include) {
static async getOldCollectionsJsonExpanded(user, libraryId, include, limitCollectionSize) {
let collectionWhere = null
if (libraryId) {
collectionWhere = {
@ -88,7 +89,7 @@ class Collection extends Model {
})
// TODO: Handle user permission restrictions on initial query
return collections.map(c => {
const oldCollection = this.getOldCollection(c)
const oldCollection = this.getOldCollection(c, limitCollectionSize)
// Filter books using user permissions
const books = c.books?.filter(b => {
@ -199,11 +200,15 @@ class Collection extends Model {
/**
* Get old collection from Collection
* @param {Collection} collectionExpanded
* @param {Collection} collectionExpanded
* @param {number} limitCollectionSize
* @returns {oldCollection}
*/
static getOldCollection(collectionExpanded) {
const libraryItemIds = collectionExpanded.books?.map(b => b.libraryItem?.id || null).filter(lid => lid) || []
static getOldCollection(collectionExpanded, limitCollectionSize) {
let libraryItemIds = collectionExpanded.books?.map(b => b.libraryItem?.id || null).filter(lid => lid) || []
if (limitCollectionSize) {
libraryItemIds = libraryItemIds.slice(0,limitCollectionSize)
}
return new oldCollection({
id: collectionExpanded.id,
libraryId: collectionExpanded.libraryId,
@ -239,7 +244,7 @@ class Collection extends Model {
/**
* Get old collection by id
* @param {string} collectionId
* @param {string} collectionId
* @returns {Promise<oldCollection|null>} returns null if not found
*/
static async getOldById(collectionId) {
@ -287,7 +292,7 @@ class Collection extends Model {
/**
* Remove all collections belonging to library
* @param {string} libraryId
* @param {string} libraryId
* @returns {Promise<number>} number of collections destroyed
*/
static async removeAllForLibrary(libraryId) {
@ -316,7 +321,7 @@ class Collection extends Model {
/**
* Initialize model
* @param {import('../Database').sequelize} sequelize
* @param {import('../Database').sequelize} sequelize
*/
static init(sequelize) {
super.init({
@ -339,4 +344,4 @@ class Collection extends Model {
}
}
module.exports = Collection
module.exports = Collection