const { Request, Response, NextFunction } = require('express') const Logger = require('../Logger') const Database = require('../Database') /** * @typedef RequestUserObject * @property {import('../models/User')} user * * @typedef {Request & RequestUserObject} RequestWithUser */ class ToolsController { constructor() {} /** * POST: /api/tools/item/:id/encode-m4b * Start an audiobook merge to m4b task * * @this import('../routers/ApiRouter') * * @param {RequestWithUser} req * @param {Response} res */ async encodeM4b(req, res) { if (req.libraryItem.isMissing || req.libraryItem.isInvalid) { Logger.error(`[MiscController] encodeM4b: library item not found or invalid ${req.params.id}`) return res.status(404).send('Audiobook not found') } if (req.libraryItem.mediaType !== 'book') { Logger.error(`[MiscController] encodeM4b: Invalid library item ${req.params.id}: not a book`) return res.status(400).send('Invalid library item: not a book') } if (req.libraryItem.media.tracks.length <= 0) { Logger.error(`[MiscController] encodeM4b: Invalid audiobook ${req.params.id}: no audio tracks`) return res.status(400).send('Invalid audiobook: no audio tracks') } if (this.abMergeManager.getPendingTaskByLibraryItemId(req.libraryItem.id)) { Logger.error(`[MiscController] encodeM4b: Audiobook ${req.params.id} is already processing`) return res.status(400).send('Audiobook is already processing') } const options = req.query || {} this.abMergeManager.startAudiobookMerge(req.user.id, req.libraryItem, options) res.sendStatus(200) } /** * DELETE: /api/tools/item/:id/encode-m4b * Cancel a running m4b merge task * * @this import('../routers/ApiRouter') * * @param {RequestWithUser} req * @param {Response} res */ async cancelM4bEncode(req, res) { const workerTask = this.abMergeManager.getPendingTaskByLibraryItemId(req.params.id) if (!workerTask) return res.sendStatus(404) this.abMergeManager.cancelEncode(workerTask.task) res.sendStatus(200) } /** * POST: /api/tools/item/:id/embed-metadata * Start audiobook embed task * * @this import('../routers/ApiRouter') * * @param {RequestWithUser} req * @param {Response} res */ async embedAudioFileMetadata(req, res) { if (req.libraryItem.isMissing || !req.libraryItem.hasAudioFiles || !req.libraryItem.isBook) { Logger.error(`[ToolsController] Invalid library item`) return res.sendStatus(400) } if (this.audioMetadataManager.getIsLibraryItemQueuedOrProcessing(req.libraryItem.id)) { Logger.error(`[ToolsController] Library item (${req.libraryItem.id}) is already in queue or processing`) return res.status(400).send('Library item is already in queue or processing') } const options = { forceEmbedChapters: req.query.forceEmbedChapters === '1', backup: req.query.backup === '1' } this.audioMetadataManager.updateMetadataForItem(req.user.id, req.libraryItem, options) res.sendStatus(200) } /** * POST: /api/tools/batch/embed-metadata * Start batch audiobook embed task * * @this import('../routers/ApiRouter') * * @param {RequestWithUser} req * @param {Response} res */ async batchEmbedMetadata(req, res) { const libraryItemIds = req.body.libraryItemIds || [] if (!libraryItemIds.length) { return res.status(400).send('Invalid request payload') } const libraryItems = [] for (const libraryItemId of libraryItemIds) { const libraryItem = await Database.libraryItemModel.getOldById(libraryItemId) if (!libraryItem) { Logger.error(`[ToolsController] Batch embed metadata library item (${libraryItemId}) not found`) return res.sendStatus(404) } // Check user can access this library item if (!req.user.checkCanAccessLibraryItem(libraryItem)) { Logger.error(`[ToolsController] Batch embed metadata library item (${libraryItemId}) not accessible to user "${req.user.username}"`) return res.sendStatus(403) } if (libraryItem.isMissing || !libraryItem.hasAudioFiles || !libraryItem.isBook) { Logger.error(`[ToolsController] Batch embed invalid library item (${libraryItemId})`) return res.sendStatus(400) } if (this.audioMetadataManager.getIsLibraryItemQueuedOrProcessing(libraryItemId)) { Logger.error(`[ToolsController] Batch embed library item (${libraryItemId}) is already in queue or processing`) return res.status(400).send('Library item is already in queue or processing') } libraryItems.push(libraryItem) } const options = { forceEmbedChapters: req.query.forceEmbedChapters === '1', backup: req.query.backup === '1' } this.audioMetadataManager.handleBatchEmbed(req.user.id, libraryItems, options) res.sendStatus(200) } /** * * @param {RequestWithUser} req * @param {Response} res * @param {NextFunction} next */ async middleware(req, res, next) { if (!req.user.isAdminOrUp) { Logger.error(`[LibraryItemController] Non-root user "${req.user.username}" attempted to access tools route`) return res.sendStatus(403) } if (req.params.id) { const item = await Database.libraryItemModel.getOldById(req.params.id) if (!item?.media) return res.sendStatus(404) // Check user can access this library item if (!req.user.checkCanAccessLibraryItem(item)) { return res.sendStatus(403) } req.libraryItem = item } next() } } module.exports = new ToolsController()