mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2024-12-20 19:06:06 +01:00
113 lines
3.3 KiB
JavaScript
113 lines
3.3 KiB
JavaScript
const { DataTypes, Model } = require('sequelize')
|
|
|
|
const uppercaseFirst = str => `${str[0].toUpperCase()}${str.substr(1)}`
|
|
|
|
/*
|
|
* Polymorphic association: https://sequelize.org/docs/v6/advanced-association-concepts/polymorphic-associations/
|
|
* Feeds can be created from LibraryItem, Collection, Playlist or Series
|
|
*/
|
|
module.exports = (sequelize) => {
|
|
class Feed extends Model {
|
|
getEntity(options) {
|
|
if (!this.entityType) return Promise.resolve(null)
|
|
const mixinMethodName = `get${uppercaseFirst(this.entityType)}`
|
|
return this[mixinMethodName](options)
|
|
}
|
|
}
|
|
|
|
Feed.init({
|
|
id: {
|
|
type: DataTypes.UUID,
|
|
defaultValue: DataTypes.UUIDV4,
|
|
primaryKey: true
|
|
},
|
|
slug: DataTypes.STRING,
|
|
entityType: DataTypes.STRING,
|
|
entityId: DataTypes.UUIDV4,
|
|
entityUpdatedAt: DataTypes.DATE,
|
|
serverAddress: DataTypes.STRING,
|
|
feedURL: DataTypes.STRING,
|
|
imageURL: DataTypes.STRING,
|
|
siteURL: DataTypes.STRING,
|
|
title: DataTypes.STRING,
|
|
description: DataTypes.TEXT,
|
|
author: DataTypes.STRING,
|
|
podcastType: DataTypes.STRING,
|
|
language: DataTypes.STRING,
|
|
ownerName: DataTypes.STRING,
|
|
ownerEmail: DataTypes.STRING,
|
|
explicit: DataTypes.BOOLEAN,
|
|
preventIndexing: DataTypes.BOOLEAN
|
|
}, {
|
|
sequelize,
|
|
modelName: 'Feed'
|
|
})
|
|
|
|
const { User, LibraryItem, Collection, Series, Playlist } = sequelize.models
|
|
|
|
User.hasMany(Feed)
|
|
Feed.belongsTo(User)
|
|
|
|
LibraryItem.hasMany(Feed, {
|
|
foreignKey: 'entityId',
|
|
constraints: false,
|
|
scope: {
|
|
entityType: 'libraryItem'
|
|
}
|
|
})
|
|
Feed.belongsTo(LibraryItem, { foreignKey: 'entityId', constraints: false })
|
|
|
|
Collection.hasMany(Feed, {
|
|
foreignKey: 'entityId',
|
|
constraints: false,
|
|
scope: {
|
|
entityType: 'collection'
|
|
}
|
|
})
|
|
Feed.belongsTo(Collection, { foreignKey: 'entityId', constraints: false })
|
|
|
|
Series.hasMany(Feed, {
|
|
foreignKey: 'entityId',
|
|
constraints: false,
|
|
scope: {
|
|
entityType: 'series'
|
|
}
|
|
})
|
|
Feed.belongsTo(Series, { foreignKey: 'entityId', constraints: false })
|
|
|
|
Playlist.hasMany(Feed, {
|
|
foreignKey: 'entityId',
|
|
constraints: false,
|
|
scope: {
|
|
entityType: 'playlist'
|
|
}
|
|
})
|
|
Feed.belongsTo(Playlist, { foreignKey: 'entityId', constraints: false })
|
|
|
|
Feed.addHook('afterFind', findResult => {
|
|
if (!Array.isArray(findResult)) findResult = [findResult]
|
|
for (const instance of findResult) {
|
|
if (instance.entityType === 'libraryItem' && instance.LibraryItem !== undefined) {
|
|
instance.Entity = instance.LibraryItem
|
|
} else if (instance.mediaItemType === 'collection' && instance.Collection !== undefined) {
|
|
instance.Entity = instance.Collection
|
|
} else if (instance.mediaItemType === 'series' && instance.Series !== undefined) {
|
|
instance.Entity = instance.Series
|
|
} else if (instance.mediaItemType === 'playlist' && instance.Playlist !== undefined) {
|
|
instance.Entity = instance.Playlist
|
|
}
|
|
|
|
// To prevent mistakes:
|
|
delete instance.LibraryItem
|
|
delete instance.dataValues.LibraryItem
|
|
delete instance.Collection
|
|
delete instance.dataValues.Collection
|
|
delete instance.Series
|
|
delete instance.dataValues.Series
|
|
delete instance.Playlist
|
|
delete instance.dataValues.Playlist
|
|
}
|
|
})
|
|
|
|
return Feed
|
|
} |