Feed and Setting models

2026-03-04 23:19:42 +00:00 · 2023-03-13 17:13:31 -05:00 · 2023-03-13 17:13:31 -05:00 · c7f457da3e
commit c7f457da3e
parent bed3758268
10 changed files with 236 additions and 10 deletions
--- a/server/models/Feed.js
+++ b/server/models/Feed.js
@ -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
+}