mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2024-12-20 19:06:06 +01:00
Feed and Setting models
This commit is contained in:
parent
bed3758268
commit
c7f457da3e
@ -66,6 +66,11 @@ class Database {
|
||||
require('./models/PlaylistMediaItem')(this.sequelize)
|
||||
require('./models/Device')(this.sequelize)
|
||||
require('./models/PlaybackSession')(this.sequelize)
|
||||
require('./models/PlaybackSessionListenTime')(this.sequelize)
|
||||
require('./models/Feed')(this.sequelize)
|
||||
require('./models/FeedEpisode')(this.sequelize)
|
||||
require('./models/Setting')(this.sequelize)
|
||||
require('./models/LibrarySetting')(this.sequelize)
|
||||
|
||||
return this.sequelize.sync()
|
||||
}
|
||||
|
@ -30,6 +30,10 @@ module.exports = (sequelize) => {
|
||||
})
|
||||
|
||||
const { Book, PodcastEpisode, FileMetadata } = sequelize.models
|
||||
|
||||
FileMetadata.hasOne(AudioTrack)
|
||||
AudioTrack.belongsTo(FileMetadata)
|
||||
|
||||
Book.hasMany(AudioTrack, {
|
||||
foreignKey: 'mediaItemId',
|
||||
constraints: false,
|
||||
@ -64,8 +68,5 @@ module.exports = (sequelize) => {
|
||||
}
|
||||
})
|
||||
|
||||
FileMetadata.hasOne(AudioTrack)
|
||||
AudioTrack.belongsTo(FileMetadata)
|
||||
|
||||
return AudioTrack
|
||||
}
|
113
server/models/Feed.js
Normal file
113
server/models/Feed.js
Normal file
@ -0,0 +1,113 @@
|
||||
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
|
||||
}
|
37
server/models/FeedEpisode.js
Normal file
37
server/models/FeedEpisode.js
Normal file
@ -0,0 +1,37 @@
|
||||
const { DataTypes, Model } = require('sequelize')
|
||||
|
||||
module.exports = (sequelize) => {
|
||||
class FeedEpisode extends Model { }
|
||||
|
||||
FeedEpisode.init({
|
||||
id: {
|
||||
type: DataTypes.UUID,
|
||||
defaultValue: DataTypes.UUIDV4,
|
||||
primaryKey: true
|
||||
},
|
||||
title: DataTypes.STRING,
|
||||
author: DataTypes.STRING,
|
||||
description: DataTypes.TEXT,
|
||||
siteURL: DataTypes.STRING,
|
||||
enclosureURL: DataTypes.STRING,
|
||||
enclosureType: DataTypes.STRING,
|
||||
enclosureSize: DataTypes.BIGINT,
|
||||
pubDate: DataTypes.STRING,
|
||||
season: DataTypes.STRING,
|
||||
episode: DataTypes.STRING,
|
||||
episodeType: DataTypes.STRING,
|
||||
duration: DataTypes.INTEGER,
|
||||
filePath: DataTypes.STRING,
|
||||
explicit: DataTypes.BOOLEAN
|
||||
}, {
|
||||
sequelize,
|
||||
modelName: 'FeedEpisode'
|
||||
})
|
||||
|
||||
const { Feed } = sequelize.models
|
||||
|
||||
Feed.hasMany(FeedEpisode)
|
||||
FeedEpisode.belongsTo(Feed)
|
||||
|
||||
return FeedEpisode
|
||||
}
|
@ -19,6 +19,7 @@ module.exports = (sequelize) => {
|
||||
birthtime: DataTypes.DATE(6)
|
||||
}, {
|
||||
sequelize,
|
||||
freezeTableName: true, // sequelize uses datum as singular of data
|
||||
modelName: 'FileMetadata'
|
||||
})
|
||||
|
||||
|
25
server/models/LibrarySetting.js
Normal file
25
server/models/LibrarySetting.js
Normal file
@ -0,0 +1,25 @@
|
||||
const { DataTypes, Model } = require('sequelize')
|
||||
|
||||
module.exports = (sequelize) => {
|
||||
class LibrarySetting extends Model { }
|
||||
|
||||
LibrarySetting.init({
|
||||
id: {
|
||||
type: DataTypes.UUID,
|
||||
defaultValue: DataTypes.UUIDV4,
|
||||
primaryKey: true
|
||||
},
|
||||
key: DataTypes.STRING,
|
||||
value: DataTypes.STRING
|
||||
}, {
|
||||
sequelize,
|
||||
modelName: 'LibrarySetting'
|
||||
})
|
||||
|
||||
const { Library } = sequelize.models
|
||||
|
||||
Library.hasMany(LibrarySetting)
|
||||
LibrarySetting.belongsTo(Library)
|
||||
|
||||
return LibrarySetting
|
||||
}
|
@ -35,6 +35,12 @@ module.exports = (sequelize) => {
|
||||
|
||||
const { Book, PodcastEpisode, User, Device } = sequelize.models
|
||||
|
||||
User.hasMany(PlaybackSession)
|
||||
PlaybackSession.belongsTo(User)
|
||||
|
||||
Device.hasMany(PlaybackSession)
|
||||
PlaybackSession.belongsTo(Device)
|
||||
|
||||
Book.hasMany(PlaybackSession, {
|
||||
foreignKey: 'mediaItemId',
|
||||
constraints: false,
|
||||
@ -69,11 +75,5 @@ module.exports = (sequelize) => {
|
||||
}
|
||||
})
|
||||
|
||||
User.hasMany(PlaybackSession)
|
||||
PlaybackSession.belongsTo(User)
|
||||
|
||||
Device.hasMany(PlaybackSession)
|
||||
PlaybackSession.belongsTo(Device)
|
||||
|
||||
return PlaybackSession
|
||||
}
|
25
server/models/PlaybackSessionListenTime.js
Normal file
25
server/models/PlaybackSessionListenTime.js
Normal file
@ -0,0 +1,25 @@
|
||||
const { DataTypes, Model } = require('sequelize')
|
||||
|
||||
module.exports = (sequelize) => {
|
||||
class PlaybackSessionListenTime extends Model { }
|
||||
|
||||
PlaybackSessionListenTime.init({
|
||||
id: {
|
||||
type: DataTypes.UUID,
|
||||
defaultValue: DataTypes.UUIDV4,
|
||||
primaryKey: true
|
||||
},
|
||||
time: DataTypes.INTEGER,
|
||||
date: DataTypes.STRING
|
||||
}, {
|
||||
sequelize,
|
||||
modelName: 'PlaybackSessionListenTime'
|
||||
})
|
||||
|
||||
const { PlaybackSession } = sequelize.models
|
||||
|
||||
PlaybackSession.hasMany(PlaybackSessionListenTime)
|
||||
PlaybackSessionListenTime.belongsTo(PlaybackSession)
|
||||
|
||||
return PlaybackSessionListenTime
|
||||
}
|
@ -18,7 +18,7 @@ module.exports = (sequelize) => {
|
||||
description: DataTypes.TEXT,
|
||||
pubDate: DataTypes.STRING,
|
||||
enclosureURL: DataTypes.STRING,
|
||||
enclosureLength: DataTypes.BIGINT,
|
||||
enclosureSize: DataTypes.BIGINT,
|
||||
enclosureType: DataTypes.STRING,
|
||||
publishedAt: DataTypes.DATE
|
||||
}, {
|
||||
|
19
server/models/Setting.js
Normal file
19
server/models/Setting.js
Normal file
@ -0,0 +1,19 @@
|
||||
const { DataTypes, Model } = require('sequelize')
|
||||
|
||||
module.exports = (sequelize) => {
|
||||
class Setting extends Model { }
|
||||
|
||||
Setting.init({
|
||||
key: {
|
||||
type: DataTypes.STRING,
|
||||
primaryKey: true
|
||||
},
|
||||
value: DataTypes.STRING,
|
||||
type: DataTypes.INTEGER
|
||||
}, {
|
||||
sequelize,
|
||||
modelName: 'Setting'
|
||||
})
|
||||
|
||||
return Setting
|
||||
}
|
Loading…
Reference in New Issue
Block a user