mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2024-10-06 02:01:58 +02:00
393 lines
16 KiB
JavaScript
393 lines
16 KiB
JavaScript
const Logger = require('../Logger')
|
|
const SocketAuthority = require('../SocketAuthority')
|
|
const Database = require('../Database')
|
|
|
|
const fs = require('../libs/fsExtra')
|
|
|
|
const { getPodcastFeed } = require('../utils/podcastUtils')
|
|
const { removeFile, downloadFile } = require('../utils/fileUtils')
|
|
const { levenshteinDistance } = require('../utils/index')
|
|
const opmlParser = require('../utils/parsers/parseOPML')
|
|
const opmlGenerator = require('../utils/generators/opmlGenerator')
|
|
const prober = require('../utils/prober')
|
|
const ffmpegHelpers = require('../utils/ffmpegHelpers')
|
|
|
|
const LibraryFile = require('../objects/files/LibraryFile')
|
|
const PodcastEpisodeDownload = require('../objects/PodcastEpisodeDownload')
|
|
const PodcastEpisode = require('../objects/entities/PodcastEpisode')
|
|
const AudioFile = require('../objects/files/AudioFile')
|
|
const Task = require("../objects/Task")
|
|
|
|
class PodcastManager {
|
|
constructor(watcher, notificationManager, taskManager) {
|
|
this.watcher = watcher
|
|
this.notificationManager = notificationManager
|
|
this.taskManager = taskManager
|
|
|
|
this.downloadQueue = []
|
|
this.currentDownload = null
|
|
|
|
this.failedCheckMap = {}
|
|
this.MaxFailedEpisodeChecks = 24
|
|
}
|
|
|
|
getEpisodeDownloadsInQueue(libraryItemId) {
|
|
return this.downloadQueue.filter(d => d.libraryItemId === libraryItemId)
|
|
}
|
|
|
|
clearDownloadQueue(libraryItemId = null) {
|
|
if (!this.downloadQueue.length) return
|
|
|
|
if (!libraryItemId) {
|
|
Logger.info(`[PodcastManager] Clearing all downloads in queue (${this.downloadQueue.length})`)
|
|
this.downloadQueue = []
|
|
} else {
|
|
var itemDownloads = this.getEpisodeDownloadsInQueue(libraryItemId)
|
|
Logger.info(`[PodcastManager] Clearing downloads in queue for item "${libraryItemId}" (${itemDownloads.length})`)
|
|
this.downloadQueue = this.downloadQueue.filter(d => d.libraryItemId !== libraryItemId)
|
|
}
|
|
}
|
|
|
|
async downloadPodcastEpisodes(libraryItem, episodesToDownload, isAutoDownload) {
|
|
let index = Math.max(...libraryItem.media.episodes.filter(ep => ep.index == null || isNaN(ep.index)).map(ep => Number(ep.index))) + 1
|
|
for (const ep of episodesToDownload) {
|
|
const newPe = new PodcastEpisode()
|
|
newPe.setData(ep, index++)
|
|
newPe.libraryItemId = libraryItem.id
|
|
newPe.podcastId = libraryItem.media.id
|
|
const newPeDl = new PodcastEpisodeDownload()
|
|
newPeDl.setData(newPe, libraryItem, isAutoDownload, libraryItem.libraryId)
|
|
this.startPodcastEpisodeDownload(newPeDl)
|
|
}
|
|
}
|
|
|
|
async startPodcastEpisodeDownload(podcastEpisodeDownload) {
|
|
SocketAuthority.emitter('episode_download_queue_updated', this.getDownloadQueueDetails())
|
|
if (this.currentDownload) {
|
|
this.downloadQueue.push(podcastEpisodeDownload)
|
|
SocketAuthority.emitter('episode_download_queued', podcastEpisodeDownload.toJSONForClient())
|
|
return
|
|
}
|
|
|
|
const task = new Task()
|
|
const taskDescription = `Downloading episode "${podcastEpisodeDownload.podcastEpisode.title}".`
|
|
const taskData = {
|
|
libraryId: podcastEpisodeDownload.libraryId,
|
|
libraryItemId: podcastEpisodeDownload.libraryItemId,
|
|
}
|
|
task.setData('download-podcast-episode', 'Downloading Episode', taskDescription, false, taskData)
|
|
this.taskManager.addTask(task)
|
|
|
|
SocketAuthority.emitter('episode_download_started', podcastEpisodeDownload.toJSONForClient())
|
|
this.currentDownload = podcastEpisodeDownload
|
|
|
|
// If this file already exists then append the episode id to the filename
|
|
// e.g. "/tagesschau 20 Uhr.mp3" becomes "/tagesschau 20 Uhr (ep_asdfasdf).mp3"
|
|
// this handles podcasts where every title is the same (ref https://github.com/advplyr/audiobookshelf/issues/1802)
|
|
if (await fs.pathExists(this.currentDownload.targetPath)) {
|
|
this.currentDownload.appendEpisodeId = true
|
|
}
|
|
|
|
// Ignores all added files to this dir
|
|
this.watcher.addIgnoreDir(this.currentDownload.libraryItem.path)
|
|
|
|
// Make sure podcast library item folder exists
|
|
if (!(await fs.pathExists(this.currentDownload.libraryItem.path))) {
|
|
Logger.warn(`[PodcastManager] Podcast episode download: Podcast folder no longer exists at "${this.currentDownload.libraryItem.path}" - Creating it`)
|
|
await fs.mkdir(this.currentDownload.libraryItem.path)
|
|
}
|
|
|
|
let success = false
|
|
if (this.currentDownload.urlFileExtension === 'mp3') {
|
|
// Download episode and tag it
|
|
success = await ffmpegHelpers.downloadPodcastEpisode(this.currentDownload).catch((error) => {
|
|
Logger.error(`[PodcastManager] Podcast Episode download failed`, error)
|
|
return false
|
|
})
|
|
} else {
|
|
// Download episode only
|
|
success = await downloadFile(this.currentDownload.url, this.currentDownload.targetPath).then(() => true).catch((error) => {
|
|
Logger.error(`[PodcastManager] Podcast Episode download failed`, error)
|
|
return false
|
|
})
|
|
}
|
|
|
|
if (success) {
|
|
success = await this.scanAddPodcastEpisodeAudioFile()
|
|
if (!success) {
|
|
await fs.remove(this.currentDownload.targetPath)
|
|
this.currentDownload.setFinished(false)
|
|
task.setFailed('Failed to download episode')
|
|
} else {
|
|
Logger.info(`[PodcastManager] Successfully downloaded podcast episode "${this.currentDownload.podcastEpisode.title}"`)
|
|
this.currentDownload.setFinished(true)
|
|
task.setFinished()
|
|
}
|
|
} else {
|
|
task.setFailed('Failed to download episode')
|
|
this.currentDownload.setFinished(false)
|
|
}
|
|
|
|
this.taskManager.taskFinished(task)
|
|
|
|
SocketAuthority.emitter('episode_download_finished', this.currentDownload.toJSONForClient())
|
|
SocketAuthority.emitter('episode_download_queue_updated', this.getDownloadQueueDetails())
|
|
|
|
this.watcher.removeIgnoreDir(this.currentDownload.libraryItem.path)
|
|
this.currentDownload = null
|
|
if (this.downloadQueue.length) {
|
|
this.startPodcastEpisodeDownload(this.downloadQueue.shift())
|
|
}
|
|
}
|
|
|
|
async scanAddPodcastEpisodeAudioFile() {
|
|
const libraryFile = await this.getLibraryFile(this.currentDownload.targetPath, this.currentDownload.targetRelPath)
|
|
|
|
const audioFile = await this.probeAudioFile(libraryFile)
|
|
if (!audioFile) {
|
|
return false
|
|
}
|
|
|
|
const libraryItem = await Database.libraryItemModel.getOldById(this.currentDownload.libraryItem.id)
|
|
if (!libraryItem) {
|
|
Logger.error(`[PodcastManager] Podcast Episode finished but library item was not found ${this.currentDownload.libraryItem.id}`)
|
|
return false
|
|
}
|
|
|
|
const podcastEpisode = this.currentDownload.podcastEpisode
|
|
podcastEpisode.audioFile = audioFile
|
|
|
|
if (audioFile.chapters?.length) {
|
|
podcastEpisode.chapters = audioFile.chapters.map(ch => ({ ...ch }))
|
|
}
|
|
|
|
libraryItem.media.addPodcastEpisode(podcastEpisode)
|
|
if (libraryItem.isInvalid) {
|
|
// First episode added to an empty podcast
|
|
libraryItem.isInvalid = false
|
|
}
|
|
libraryItem.libraryFiles.push(libraryFile)
|
|
|
|
if (this.currentDownload.isAutoDownload) {
|
|
// Check setting maxEpisodesToKeep and remove episode if necessary
|
|
if (libraryItem.media.maxEpisodesToKeep && libraryItem.media.episodesWithPubDate.length > libraryItem.media.maxEpisodesToKeep) {
|
|
Logger.info(`[PodcastManager] # of episodes (${libraryItem.media.episodesWithPubDate.length}) exceeds max episodes to keep (${libraryItem.media.maxEpisodesToKeep})`)
|
|
await this.removeOldestEpisode(libraryItem, podcastEpisode.id)
|
|
}
|
|
}
|
|
|
|
libraryItem.updatedAt = Date.now()
|
|
await Database.updateLibraryItem(libraryItem)
|
|
SocketAuthority.emitter('item_updated', libraryItem.toJSONExpanded())
|
|
const podcastEpisodeExpanded = podcastEpisode.toJSONExpanded()
|
|
podcastEpisodeExpanded.libraryItem = libraryItem.toJSONExpanded()
|
|
SocketAuthority.emitter('episode_added', podcastEpisodeExpanded)
|
|
|
|
if (this.currentDownload.isAutoDownload) { // Notifications only for auto downloaded episodes
|
|
this.notificationManager.onPodcastEpisodeDownloaded(libraryItem, podcastEpisode)
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
async removeOldestEpisode(libraryItem, episodeIdJustDownloaded) {
|
|
var smallestPublishedAt = 0
|
|
var oldestEpisode = null
|
|
libraryItem.media.episodesWithPubDate.filter(ep => ep.id !== episodeIdJustDownloaded).forEach((ep) => {
|
|
if (!smallestPublishedAt || ep.publishedAt < smallestPublishedAt) {
|
|
smallestPublishedAt = ep.publishedAt
|
|
oldestEpisode = ep
|
|
}
|
|
})
|
|
// TODO: Should we check for open playback sessions for this episode?
|
|
// TODO: remove all user progress for this episode
|
|
if (oldestEpisode && oldestEpisode.audioFile) {
|
|
Logger.info(`[PodcastManager] Deleting oldest episode "${oldestEpisode.title}"`)
|
|
const successfullyDeleted = await removeFile(oldestEpisode.audioFile.metadata.path)
|
|
if (successfullyDeleted) {
|
|
libraryItem.media.removeEpisode(oldestEpisode.id)
|
|
libraryItem.removeLibraryFile(oldestEpisode.audioFile.ino)
|
|
return true
|
|
} else {
|
|
Logger.warn(`[PodcastManager] Failed to remove oldest episode "${oldestEpisode.title}"`)
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
async getLibraryFile(path, relPath) {
|
|
var newLibFile = new LibraryFile()
|
|
await newLibFile.setDataFromPath(path, relPath)
|
|
return newLibFile
|
|
}
|
|
|
|
async probeAudioFile(libraryFile) {
|
|
const path = libraryFile.metadata.path
|
|
const mediaProbeData = await prober.probe(path)
|
|
if (mediaProbeData.error) {
|
|
Logger.error(`[PodcastManager] Podcast Episode downloaded but failed to probe "${path}"`, mediaProbeData.error)
|
|
return false
|
|
}
|
|
const newAudioFile = new AudioFile()
|
|
newAudioFile.setDataFromProbe(libraryFile, mediaProbeData)
|
|
newAudioFile.index = 1
|
|
return newAudioFile
|
|
}
|
|
|
|
// Returns false if auto download episodes was disabled (disabled if reaches max failed checks)
|
|
async runEpisodeCheck(libraryItem) {
|
|
const lastEpisodeCheckDate = new Date(libraryItem.media.lastEpisodeCheck || 0)
|
|
const latestEpisodePublishedAt = libraryItem.media.latestEpisodePublished
|
|
Logger.info(`[PodcastManager] runEpisodeCheck: "${libraryItem.media.metadata.title}" | Last check: ${lastEpisodeCheckDate} | ${latestEpisodePublishedAt ? `Latest episode pubDate: ${new Date(latestEpisodePublishedAt)}` : 'No latest episode'}`)
|
|
|
|
// Use latest episode pubDate if exists OR fallback to using lastEpisodeCheckDate
|
|
// lastEpisodeCheckDate will be the current time when adding a new podcast
|
|
const dateToCheckForEpisodesAfter = latestEpisodePublishedAt || lastEpisodeCheckDate
|
|
Logger.debug(`[PodcastManager] runEpisodeCheck: "${libraryItem.media.metadata.title}" checking for episodes after ${new Date(dateToCheckForEpisodesAfter)}`)
|
|
|
|
var newEpisodes = await this.checkPodcastForNewEpisodes(libraryItem, dateToCheckForEpisodesAfter, libraryItem.media.maxNewEpisodesToDownload)
|
|
Logger.debug(`[PodcastManager] runEpisodeCheck: ${newEpisodes ? newEpisodes.length : 'N/A'} episodes found`)
|
|
|
|
if (!newEpisodes) { // Failed
|
|
// Allow up to MaxFailedEpisodeChecks failed attempts before disabling auto download
|
|
if (!this.failedCheckMap[libraryItem.id]) this.failedCheckMap[libraryItem.id] = 0
|
|
this.failedCheckMap[libraryItem.id]++
|
|
if (this.failedCheckMap[libraryItem.id] >= this.MaxFailedEpisodeChecks) {
|
|
Logger.error(`[PodcastManager] runEpisodeCheck ${this.failedCheckMap[libraryItem.id]} failed attempts at checking episodes for "${libraryItem.media.metadata.title}" - disabling auto download`)
|
|
libraryItem.media.autoDownloadEpisodes = false
|
|
delete this.failedCheckMap[libraryItem.id]
|
|
} else {
|
|
Logger.warn(`[PodcastManager] runEpisodeCheck ${this.failedCheckMap[libraryItem.id]} failed attempts at checking episodes for "${libraryItem.media.metadata.title}"`)
|
|
}
|
|
} else if (newEpisodes.length) {
|
|
delete this.failedCheckMap[libraryItem.id]
|
|
Logger.info(`[PodcastManager] Found ${newEpisodes.length} new episodes for podcast "${libraryItem.media.metadata.title}" - starting download`)
|
|
this.downloadPodcastEpisodes(libraryItem, newEpisodes, true)
|
|
} else {
|
|
delete this.failedCheckMap[libraryItem.id]
|
|
Logger.debug(`[PodcastManager] No new episodes for "${libraryItem.media.metadata.title}"`)
|
|
}
|
|
|
|
libraryItem.media.lastEpisodeCheck = Date.now()
|
|
libraryItem.updatedAt = Date.now()
|
|
await Database.updateLibraryItem(libraryItem)
|
|
SocketAuthority.emitter('item_updated', libraryItem.toJSONExpanded())
|
|
return libraryItem.media.autoDownloadEpisodes
|
|
}
|
|
|
|
async checkPodcastForNewEpisodes(podcastLibraryItem, dateToCheckForEpisodesAfter, maxNewEpisodes = 3) {
|
|
if (!podcastLibraryItem.media.metadata.feedUrl) {
|
|
Logger.error(`[PodcastManager] checkPodcastForNewEpisodes no feed url for ${podcastLibraryItem.media.metadata.title} (ID: ${podcastLibraryItem.id})`)
|
|
return false
|
|
}
|
|
var feed = await getPodcastFeed(podcastLibraryItem.media.metadata.feedUrl)
|
|
if (!feed || !feed.episodes) {
|
|
Logger.error(`[PodcastManager] checkPodcastForNewEpisodes invalid feed payload for ${podcastLibraryItem.media.metadata.title} (ID: ${podcastLibraryItem.id})`, feed)
|
|
return false
|
|
}
|
|
|
|
// Filter new and not already has
|
|
var newEpisodes = feed.episodes.filter(ep => ep.publishedAt > dateToCheckForEpisodesAfter && !podcastLibraryItem.media.checkHasEpisodeByFeedUrl(ep.enclosure.url))
|
|
|
|
if (maxNewEpisodes > 0) {
|
|
newEpisodes = newEpisodes.slice(0, maxNewEpisodes)
|
|
}
|
|
|
|
return newEpisodes
|
|
}
|
|
|
|
async checkAndDownloadNewEpisodes(libraryItem, maxEpisodesToDownload) {
|
|
const lastEpisodeCheckDate = new Date(libraryItem.media.lastEpisodeCheck || 0)
|
|
Logger.info(`[PodcastManager] checkAndDownloadNewEpisodes for "${libraryItem.media.metadata.title}" - Last episode check: ${lastEpisodeCheckDate}`)
|
|
var newEpisodes = await this.checkPodcastForNewEpisodes(libraryItem, libraryItem.media.lastEpisodeCheck, maxEpisodesToDownload)
|
|
if (newEpisodes.length) {
|
|
Logger.info(`[PodcastManager] Found ${newEpisodes.length} new episodes for podcast "${libraryItem.media.metadata.title}" - starting download`)
|
|
this.downloadPodcastEpisodes(libraryItem, newEpisodes, false)
|
|
} else {
|
|
Logger.info(`[PodcastManager] No new episodes found for podcast "${libraryItem.media.metadata.title}"`)
|
|
}
|
|
|
|
libraryItem.media.lastEpisodeCheck = Date.now()
|
|
libraryItem.updatedAt = Date.now()
|
|
await Database.updateLibraryItem(libraryItem)
|
|
SocketAuthority.emitter('item_updated', libraryItem.toJSONExpanded())
|
|
|
|
return newEpisodes
|
|
}
|
|
|
|
async findEpisode(rssFeedUrl, searchTitle) {
|
|
const feed = await getPodcastFeed(rssFeedUrl).catch(() => {
|
|
return null
|
|
})
|
|
if (!feed || !feed.episodes) {
|
|
return null
|
|
}
|
|
|
|
const matches = []
|
|
feed.episodes.forEach(ep => {
|
|
if (!ep.title) return
|
|
|
|
const epTitle = ep.title.toLowerCase().trim()
|
|
if (epTitle === searchTitle) {
|
|
matches.push({
|
|
episode: ep,
|
|
levenshtein: 0
|
|
})
|
|
} else {
|
|
const levenshtein = levenshteinDistance(searchTitle, epTitle, true)
|
|
if (levenshtein <= 6 && epTitle.length > levenshtein) {
|
|
matches.push({
|
|
episode: ep,
|
|
levenshtein
|
|
})
|
|
}
|
|
}
|
|
})
|
|
return matches.sort((a, b) => a.levenshtein - b.levenshtein)
|
|
}
|
|
|
|
async getOPMLFeeds(opmlText) {
|
|
var extractedFeeds = opmlParser.parse(opmlText)
|
|
if (!extractedFeeds || !extractedFeeds.length) {
|
|
Logger.error('[PodcastManager] getOPMLFeeds: No RSS feeds found in OPML')
|
|
return {
|
|
error: 'No RSS feeds found in OPML'
|
|
}
|
|
}
|
|
|
|
var rssFeedData = []
|
|
|
|
for (let feed of extractedFeeds) {
|
|
var feedData = await getPodcastFeed(feed.feedUrl, true)
|
|
if (feedData) {
|
|
feedData.metadata.feedUrl = feed.feedUrl
|
|
rssFeedData.push(feedData)
|
|
}
|
|
}
|
|
|
|
return {
|
|
feeds: rssFeedData
|
|
}
|
|
}
|
|
|
|
/**
|
|
* OPML file string for podcasts in a library
|
|
* @param {import('../models/Podcast')[]} podcasts
|
|
* @returns {string} XML string
|
|
*/
|
|
generateOPMLFileText(podcasts) {
|
|
return opmlGenerator.generate(podcasts)
|
|
}
|
|
|
|
getDownloadQueueDetails(libraryId = null) {
|
|
let _currentDownload = this.currentDownload
|
|
if (libraryId && _currentDownload?.libraryId !== libraryId) _currentDownload = null
|
|
|
|
return {
|
|
currentDownload: _currentDownload?.toJSONForClient(),
|
|
queue: this.downloadQueue.filter(item => !libraryId || item.libraryId === libraryId).map(item => item.toJSONForClient())
|
|
}
|
|
}
|
|
}
|
|
module.exports = PodcastManager
|