mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2024-11-08 00:54:33 +01:00
304 lines
10 KiB
JavaScript
304 lines
10 KiB
JavaScript
const Path = require('path')
|
|
|
|
const Logger = require('../Logger')
|
|
const SocketAuthority = require('../SocketAuthority')
|
|
const Database = require('../Database')
|
|
|
|
const fs = require('../libs/fsExtra')
|
|
const Feed = require('../objects/Feed')
|
|
const libraryItemsBookFilters = require('../utils/queries/libraryItemsBookFilters')
|
|
|
|
class RssFeedManager {
|
|
constructor() {}
|
|
|
|
async validateFeedEntity(feedObj) {
|
|
if (feedObj.entityType === 'collection') {
|
|
const collection = await Database.collectionModel.getOldById(feedObj.entityId)
|
|
if (!collection) {
|
|
Logger.error(`[RssFeedManager] Removing feed "${feedObj.id}". Collection "${feedObj.entityId}" not found`)
|
|
return false
|
|
}
|
|
} else if (feedObj.entityType === 'libraryItem') {
|
|
const libraryItemExists = await Database.libraryItemModel.checkExistsById(feedObj.entityId)
|
|
if (!libraryItemExists) {
|
|
Logger.error(`[RssFeedManager] Removing feed "${feedObj.id}". Library item "${feedObj.entityId}" not found`)
|
|
return false
|
|
}
|
|
} else if (feedObj.entityType === 'series') {
|
|
const series = await Database.seriesModel.getOldById(feedObj.entityId)
|
|
if (!series) {
|
|
Logger.error(`[RssFeedManager] Removing feed "${feedObj.id}". Series "${feedObj.entityId}" not found`)
|
|
return false
|
|
}
|
|
} else {
|
|
Logger.error(`[RssFeedManager] Removing feed "${feedObj.id}". Invalid entityType "${feedObj.entityType}"`)
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
/**
|
|
* Validate all feeds and remove invalid
|
|
*/
|
|
async init() {
|
|
const feeds = await Database.feedModel.getOldFeeds()
|
|
for (const feed of feeds) {
|
|
// Remove invalid feeds
|
|
if (!(await this.validateFeedEntity(feed))) {
|
|
await Database.removeFeed(feed.id)
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Find open feed for an entity (e.g. collection id, playlist id, library item id)
|
|
* @param {string} entityId
|
|
* @returns {Promise<objects.Feed>} oldFeed
|
|
*/
|
|
findFeedForEntityId(entityId) {
|
|
return Database.feedModel.findOneOld({ entityId })
|
|
}
|
|
|
|
/**
|
|
* Find open feed for a slug
|
|
* @param {string} slug
|
|
* @returns {Promise<objects.Feed>} oldFeed
|
|
*/
|
|
findFeedBySlug(slug) {
|
|
return Database.feedModel.findOneOld({ slug })
|
|
}
|
|
|
|
/**
|
|
* Find open feed for a slug
|
|
* @param {string} slug
|
|
* @returns {Promise<objects.Feed>} oldFeed
|
|
*/
|
|
findFeed(id) {
|
|
return Database.feedModel.findByPkOld(id)
|
|
}
|
|
|
|
async getFeed(req, res) {
|
|
const feed = await this.findFeedBySlug(req.params.slug)
|
|
if (!feed) {
|
|
Logger.warn(`[RssFeedManager] Feed not found ${req.params.slug}`)
|
|
res.sendStatus(404)
|
|
return
|
|
}
|
|
|
|
// Check if feed needs to be updated
|
|
if (feed.entityType === 'libraryItem') {
|
|
const libraryItem = await Database.libraryItemModel.getOldById(feed.entityId)
|
|
|
|
let mostRecentlyUpdatedAt = libraryItem.updatedAt
|
|
if (libraryItem.isPodcast) {
|
|
libraryItem.media.episodes.forEach((episode) => {
|
|
if (episode.updatedAt > mostRecentlyUpdatedAt) mostRecentlyUpdatedAt = episode.updatedAt
|
|
})
|
|
}
|
|
|
|
if (libraryItem && (!feed.entityUpdatedAt || mostRecentlyUpdatedAt > feed.entityUpdatedAt)) {
|
|
Logger.debug(`[RssFeedManager] Updating RSS feed for item ${libraryItem.id} "${libraryItem.media.metadata.title}"`)
|
|
|
|
feed.updateFromItem(libraryItem)
|
|
await Database.updateFeed(feed)
|
|
}
|
|
} else if (feed.entityType === 'collection') {
|
|
const collection = await Database.collectionModel.findByPk(feed.entityId, {
|
|
include: Database.collectionBookModel
|
|
})
|
|
if (collection) {
|
|
const collectionExpanded = await collection.getOldJsonExpanded()
|
|
|
|
// Find most recently updated item in collection
|
|
let mostRecentlyUpdatedAt = collectionExpanded.lastUpdate
|
|
// Check for most recently updated book
|
|
collectionExpanded.books.forEach((libraryItem) => {
|
|
if (libraryItem.media.tracks.length && libraryItem.updatedAt > mostRecentlyUpdatedAt) {
|
|
mostRecentlyUpdatedAt = libraryItem.updatedAt
|
|
}
|
|
})
|
|
// Check for most recently added collection book
|
|
collection.collectionBooks.forEach((collectionBook) => {
|
|
if (collectionBook.createdAt.valueOf() > mostRecentlyUpdatedAt) {
|
|
mostRecentlyUpdatedAt = collectionBook.createdAt.valueOf()
|
|
}
|
|
})
|
|
const hasBooksRemoved = collection.collectionBooks.length < feed.episodes.length
|
|
|
|
if (!feed.entityUpdatedAt || hasBooksRemoved || mostRecentlyUpdatedAt > feed.entityUpdatedAt) {
|
|
Logger.debug(`[RssFeedManager] Updating RSS feed for collection "${collection.name}"`)
|
|
|
|
feed.updateFromCollection(collectionExpanded)
|
|
await Database.updateFeed(feed)
|
|
}
|
|
}
|
|
} else if (feed.entityType === 'series') {
|
|
const series = await Database.seriesModel.getOldById(feed.entityId)
|
|
if (series) {
|
|
const seriesJson = series.toJSON()
|
|
|
|
// Get books in series that have audio tracks
|
|
seriesJson.books = (await libraryItemsBookFilters.getLibraryItemsForSeries(series)).filter((li) => li.media.numTracks)
|
|
|
|
// Find most recently updated item in series
|
|
let mostRecentlyUpdatedAt = seriesJson.updatedAt
|
|
let totalTracks = 0 // Used to detect series items removed
|
|
seriesJson.books.forEach((libraryItem) => {
|
|
totalTracks += libraryItem.media.tracks.length
|
|
if (libraryItem.media.tracks.length && libraryItem.updatedAt > mostRecentlyUpdatedAt) {
|
|
mostRecentlyUpdatedAt = libraryItem.updatedAt
|
|
}
|
|
})
|
|
if (totalTracks !== feed.episodes.length) {
|
|
mostRecentlyUpdatedAt = Date.now()
|
|
}
|
|
|
|
if (!feed.entityUpdatedAt || mostRecentlyUpdatedAt > feed.entityUpdatedAt) {
|
|
Logger.debug(`[RssFeedManager] Updating RSS feed for series "${seriesJson.name}"`)
|
|
|
|
feed.updateFromSeries(seriesJson)
|
|
await Database.updateFeed(feed)
|
|
}
|
|
}
|
|
}
|
|
|
|
const xml = feed.buildXml()
|
|
res.set('Content-Type', 'text/xml')
|
|
res.send(xml)
|
|
}
|
|
|
|
async getFeedItem(req, res) {
|
|
const feed = await this.findFeedBySlug(req.params.slug)
|
|
if (!feed) {
|
|
Logger.debug(`[RssFeedManager] Feed not found ${req.params.slug}`)
|
|
res.sendStatus(404)
|
|
return
|
|
}
|
|
const episodePath = feed.getEpisodePath(req.params.episodeId)
|
|
if (!episodePath) {
|
|
Logger.error(`[RssFeedManager] Feed episode not found ${req.params.episodeId}`)
|
|
res.sendStatus(404)
|
|
return
|
|
}
|
|
res.sendFile(episodePath)
|
|
}
|
|
|
|
async getFeedCover(req, res) {
|
|
const feed = await this.findFeedBySlug(req.params.slug)
|
|
if (!feed) {
|
|
Logger.debug(`[RssFeedManager] Feed not found ${req.params.slug}`)
|
|
res.sendStatus(404)
|
|
return
|
|
}
|
|
|
|
if (!feed.coverPath) {
|
|
res.sendStatus(404)
|
|
return
|
|
}
|
|
|
|
const extname = Path.extname(feed.coverPath).toLowerCase().slice(1)
|
|
res.type(`image/${extname}`)
|
|
const readStream = fs.createReadStream(feed.coverPath)
|
|
readStream.pipe(res)
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param {string} userId
|
|
* @param {*} libraryItem
|
|
* @param {*} options
|
|
* @returns
|
|
*/
|
|
async openFeedForItem(userId, libraryItem, options) {
|
|
const serverAddress = options.serverAddress
|
|
const slug = options.slug
|
|
const preventIndexing = options.metadataDetails?.preventIndexing ?? true
|
|
const ownerName = options.metadataDetails?.ownerName
|
|
const ownerEmail = options.metadataDetails?.ownerEmail
|
|
|
|
const feed = new Feed()
|
|
feed.setFromItem(userId, slug, libraryItem, serverAddress, preventIndexing, ownerName, ownerEmail)
|
|
|
|
Logger.info(`[RssFeedManager] Opened RSS feed "${feed.feedUrl}"`)
|
|
await Database.createFeed(feed)
|
|
SocketAuthority.emitter('rss_feed_open', feed.toJSONMinified())
|
|
return feed
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param {string} userId
|
|
* @param {*} collectionExpanded
|
|
* @param {*} options
|
|
* @returns
|
|
*/
|
|
async openFeedForCollection(userId, collectionExpanded, options) {
|
|
const serverAddress = options.serverAddress
|
|
const slug = options.slug
|
|
const preventIndexing = options.metadataDetails?.preventIndexing ?? true
|
|
const ownerName = options.metadataDetails?.ownerName
|
|
const ownerEmail = options.metadataDetails?.ownerEmail
|
|
|
|
const feed = new Feed()
|
|
feed.setFromCollection(userId, slug, collectionExpanded, serverAddress, preventIndexing, ownerName, ownerEmail)
|
|
|
|
Logger.info(`[RssFeedManager] Opened RSS feed "${feed.feedUrl}"`)
|
|
await Database.createFeed(feed)
|
|
SocketAuthority.emitter('rss_feed_open', feed.toJSONMinified())
|
|
return feed
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param {string} userId
|
|
* @param {*} seriesExpanded
|
|
* @param {*} options
|
|
* @returns
|
|
*/
|
|
async openFeedForSeries(userId, seriesExpanded, options) {
|
|
const serverAddress = options.serverAddress
|
|
const slug = options.slug
|
|
const preventIndexing = options.metadataDetails?.preventIndexing ?? true
|
|
const ownerName = options.metadataDetails?.ownerName
|
|
const ownerEmail = options.metadataDetails?.ownerEmail
|
|
|
|
const feed = new Feed()
|
|
feed.setFromSeries(userId, slug, seriesExpanded, serverAddress, preventIndexing, ownerName, ownerEmail)
|
|
|
|
Logger.info(`[RssFeedManager] Opened RSS feed "${feed.feedUrl}"`)
|
|
await Database.createFeed(feed)
|
|
SocketAuthority.emitter('rss_feed_open', feed.toJSONMinified())
|
|
return feed
|
|
}
|
|
|
|
async handleCloseFeed(feed) {
|
|
if (!feed) return
|
|
await Database.removeFeed(feed.id)
|
|
SocketAuthority.emitter('rss_feed_closed', feed.toJSONMinified())
|
|
Logger.info(`[RssFeedManager] Closed RSS feed "${feed.feedUrl}"`)
|
|
}
|
|
|
|
async closeRssFeed(req, res) {
|
|
const feed = await this.findFeed(req.params.id)
|
|
if (!feed) {
|
|
Logger.error(`[RssFeedManager] RSS feed not found with id "${req.params.id}"`)
|
|
return res.sendStatus(404)
|
|
}
|
|
await this.handleCloseFeed(feed)
|
|
res.sendStatus(200)
|
|
}
|
|
|
|
async closeFeedForEntityId(entityId) {
|
|
const feed = await this.findFeedForEntityId(entityId)
|
|
if (!feed) return
|
|
return this.handleCloseFeed(feed)
|
|
}
|
|
|
|
async getFeeds() {
|
|
const feeds = await Database.models.feed.getOldFeeds()
|
|
Logger.info(`[RssFeedManager] Fetched all feeds`)
|
|
return feeds
|
|
}
|
|
}
|
|
module.exports = RssFeedManager
|