mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2024-12-20 19:06:06 +01:00
115 lines
3.3 KiB
JavaScript
115 lines
3.3 KiB
JavaScript
|
const { DataTypes, Model } = require('sequelize')
|
||
|
|
||
|
const oldCollection = require('../objects/Collection')
|
||
|
const { areEquivalent } = require('../utils/index')
|
||
|
|
||
|
module.exports = (sequelize) => {
|
||
|
class Collection extends Model {
|
||
|
static async getOldCollections() {
|
||
|
const collections = await this.findAll({
|
||
|
include: {
|
||
|
model: sequelize.models.book,
|
||
|
include: sequelize.models.libraryItem
|
||
|
}
|
||
|
})
|
||
|
return collections.map(c => this.getOldCollection(c))
|
||
|
}
|
||
|
|
||
|
static getOldCollection(collectionExpanded) {
|
||
|
const libraryItemIds = collectionExpanded.books?.map(b => b.libraryItem?.id || null).filter(lid => lid) || []
|
||
|
return new oldCollection({
|
||
|
id: collectionExpanded.id,
|
||
|
libraryId: collectionExpanded.libraryId,
|
||
|
name: collectionExpanded.name,
|
||
|
description: collectionExpanded.description,
|
||
|
books: libraryItemIds,
|
||
|
lastUpdate: collectionExpanded.updatedAt.valueOf(),
|
||
|
createdAt: collectionExpanded.createdAt.valueOf()
|
||
|
})
|
||
|
}
|
||
|
|
||
|
static createFromOld(oldCollection) {
|
||
|
const collection = this.getFromOld(oldCollection)
|
||
|
return this.create(collection)
|
||
|
}
|
||
|
|
||
|
static async fullUpdateFromOld(oldCollection, collectionBooks) {
|
||
|
const existingCollection = await this.findByPk(oldCollection.id, {
|
||
|
include: sequelize.models.collectionBook
|
||
|
})
|
||
|
if (!existingCollection) return false
|
||
|
|
||
|
let hasUpdates = false
|
||
|
const collection = this.getFromOld(oldCollection)
|
||
|
|
||
|
for (const cb of collectionBooks) {
|
||
|
const existingCb = existingCollection.collectionBooks.find(i => i.bookId === cb.bookId)
|
||
|
if (!existingCb) {
|
||
|
await sequelize.models.collectionBook.create(cb)
|
||
|
hasUpdates = true
|
||
|
} else if (existingCb.order != cb.order) {
|
||
|
await existingCb.update({ order: cb.order })
|
||
|
hasUpdates = true
|
||
|
}
|
||
|
}
|
||
|
for (const cb of existingCollection.collectionBooks) {
|
||
|
// collectionBook was removed
|
||
|
if (!collectionBooks.some(i => i.bookId === cb.bookId)) {
|
||
|
await cb.destroy()
|
||
|
hasUpdates = true
|
||
|
}
|
||
|
}
|
||
|
|
||
|
let hasCollectionUpdates = false
|
||
|
for (const key in collection) {
|
||
|
let existingValue = existingCollection[key]
|
||
|
if (existingValue instanceof Date) existingValue = existingValue.valueOf()
|
||
|
if (!areEquivalent(collection[key], existingValue)) {
|
||
|
hasCollectionUpdates = true
|
||
|
}
|
||
|
}
|
||
|
if (hasCollectionUpdates) {
|
||
|
existingCollection.update(collection)
|
||
|
hasUpdates = true
|
||
|
}
|
||
|
return hasUpdates
|
||
|
}
|
||
|
|
||
|
static getFromOld(oldCollection) {
|
||
|
return {
|
||
|
id: oldCollection.id,
|
||
|
name: oldCollection.name,
|
||
|
description: oldCollection.description,
|
||
|
libraryId: oldCollection.libraryId
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static removeById(collectionId) {
|
||
|
return this.destroy({
|
||
|
where: {
|
||
|
id: collectionId
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Collection.init({
|
||
|
id: {
|
||
|
type: DataTypes.UUID,
|
||
|
defaultValue: DataTypes.UUIDV4,
|
||
|
primaryKey: true
|
||
|
},
|
||
|
name: DataTypes.STRING,
|
||
|
description: DataTypes.TEXT
|
||
|
}, {
|
||
|
sequelize,
|
||
|
modelName: 'collection'
|
||
|
})
|
||
|
|
||
|
const { library } = sequelize.models
|
||
|
|
||
|
library.hasMany(Collection)
|
||
|
Collection.belongsTo(library)
|
||
|
|
||
|
return Collection
|
||
|
}
|