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/PlaylistMediaItem')(this.sequelize)
|
||||||
require('./models/Device')(this.sequelize)
|
require('./models/Device')(this.sequelize)
|
||||||
require('./models/PlaybackSession')(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()
|
return this.sequelize.sync()
|
||||||
}
|
}
|
||||||
|
@ -30,6 +30,10 @@ module.exports = (sequelize) => {
|
|||||||
})
|
})
|
||||||
|
|
||||||
const { Book, PodcastEpisode, FileMetadata } = sequelize.models
|
const { Book, PodcastEpisode, FileMetadata } = sequelize.models
|
||||||
|
|
||||||
|
FileMetadata.hasOne(AudioTrack)
|
||||||
|
AudioTrack.belongsTo(FileMetadata)
|
||||||
|
|
||||||
Book.hasMany(AudioTrack, {
|
Book.hasMany(AudioTrack, {
|
||||||
foreignKey: 'mediaItemId',
|
foreignKey: 'mediaItemId',
|
||||||
constraints: false,
|
constraints: false,
|
||||||
@ -64,8 +68,5 @@ module.exports = (sequelize) => {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
FileMetadata.hasOne(AudioTrack)
|
|
||||||
AudioTrack.belongsTo(FileMetadata)
|
|
||||||
|
|
||||||
return AudioTrack
|
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)
|
birthtime: DataTypes.DATE(6)
|
||||||
}, {
|
}, {
|
||||||
sequelize,
|
sequelize,
|
||||||
|
freezeTableName: true, // sequelize uses datum as singular of data
|
||||||
modelName: 'FileMetadata'
|
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
|
const { Book, PodcastEpisode, User, Device } = sequelize.models
|
||||||
|
|
||||||
|
User.hasMany(PlaybackSession)
|
||||||
|
PlaybackSession.belongsTo(User)
|
||||||
|
|
||||||
|
Device.hasMany(PlaybackSession)
|
||||||
|
PlaybackSession.belongsTo(Device)
|
||||||
|
|
||||||
Book.hasMany(PlaybackSession, {
|
Book.hasMany(PlaybackSession, {
|
||||||
foreignKey: 'mediaItemId',
|
foreignKey: 'mediaItemId',
|
||||||
constraints: false,
|
constraints: false,
|
||||||
@ -69,11 +75,5 @@ module.exports = (sequelize) => {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
User.hasMany(PlaybackSession)
|
|
||||||
PlaybackSession.belongsTo(User)
|
|
||||||
|
|
||||||
Device.hasMany(PlaybackSession)
|
|
||||||
PlaybackSession.belongsTo(Device)
|
|
||||||
|
|
||||||
return PlaybackSession
|
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,
|
description: DataTypes.TEXT,
|
||||||
pubDate: DataTypes.STRING,
|
pubDate: DataTypes.STRING,
|
||||||
enclosureURL: DataTypes.STRING,
|
enclosureURL: DataTypes.STRING,
|
||||||
enclosureLength: DataTypes.BIGINT,
|
enclosureSize: DataTypes.BIGINT,
|
||||||
enclosureType: DataTypes.STRING,
|
enclosureType: DataTypes.STRING,
|
||||||
publishedAt: DataTypes.DATE
|
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