mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2024-12-20 19:06:06 +01:00
165 lines
5.0 KiB
JavaScript
165 lines
5.0 KiB
JavaScript
|
const { DataTypes, Model } = require('sequelize')
|
||
|
const oldFeed = require('../objects/Feed')
|
||
|
/*
|
||
|
* 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 {
|
||
|
static async getOldFeeds() {
|
||
|
const feeds = await this.findAll({
|
||
|
include: {
|
||
|
model: sequelize.models.feedEpisode
|
||
|
}
|
||
|
})
|
||
|
return feeds.map(f => this.getOldFeed(f))
|
||
|
}
|
||
|
|
||
|
static getOldFeed(feedExpanded) {
|
||
|
const episodes = feedExpanded.feedEpisodes.map((feedEpisode) => feedEpisode.getOldEpisode())
|
||
|
|
||
|
return new oldFeed({
|
||
|
id: feedExpanded.id,
|
||
|
slug: feedExpanded.slug,
|
||
|
userId: feedExpanded.userId,
|
||
|
entityType: feedExpanded.entityType,
|
||
|
entityId: feedExpanded.entityId,
|
||
|
meta: {
|
||
|
title: feedExpanded.title,
|
||
|
description: feedExpanded.description,
|
||
|
author: feedExpanded.author,
|
||
|
imageUrl: feedExpanded.imageURL,
|
||
|
feedUrl: feedExpanded.feedURL,
|
||
|
link: feedExpanded.siteURL,
|
||
|
explicit: feedExpanded.explicit,
|
||
|
type: feedExpanded.podcastType,
|
||
|
language: feedExpanded.language,
|
||
|
preventIndexing: feedExpanded.preventIndexing,
|
||
|
ownerName: feedExpanded.ownerName,
|
||
|
ownerEmail: feedExpanded.ownerEmail
|
||
|
},
|
||
|
serverAddress: feedExpanded.serverAddress,
|
||
|
feedUrl: feedExpanded.feedURL,
|
||
|
episodes,
|
||
|
createdAt: feedExpanded.createdAt.valueOf(),
|
||
|
updatedAt: feedExpanded.updatedAt.valueOf()
|
||
|
})
|
||
|
}
|
||
|
|
||
|
static removeById(feedId) {
|
||
|
return this.destroy({
|
||
|
where: {
|
||
|
id: feedId
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
|
||
|
getEntity(options) {
|
||
|
if (!this.entityType) return Promise.resolve(null)
|
||
|
const mixinMethodName = `get${sequelize.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 (!findResult) return
|
||
|
|
||
|
if (!Array.isArray(findResult)) findResult = [findResult]
|
||
|
for (const instance of findResult) {
|
||
|
if (instance.entityType === 'libraryItem' && instance.libraryItem !== undefined) {
|
||
|
instance.entity = instance.libraryItem
|
||
|
instance.dataValues.entity = instance.dataValues.libraryItem
|
||
|
} else if (instance.entityType === 'collection' && instance.collection !== undefined) {
|
||
|
instance.entity = instance.collection
|
||
|
instance.dataValues.entity = instance.dataValues.collection
|
||
|
} else if (instance.entityType === 'series' && instance.series !== undefined) {
|
||
|
instance.entity = instance.series
|
||
|
instance.dataValues.entity = instance.dataValues.series
|
||
|
} else if (instance.entityType === 'playlist' && instance.playlist !== undefined) {
|
||
|
instance.entity = instance.playlist
|
||
|
instance.dataValues.entity = instance.dataValues.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
|
||
|
}
|