const Path = require('path') const AudioFile = require('../objects/AudioFile') const prober = require('../utils/prober') const Logger = require('../Logger') const { LogLevel } = require('../utils/constants') class AudioFileScanner { constructor() { } getTrackNumberFromMeta(scanData) { return !isNaN(scanData.trackNumber) && scanData.trackNumber !== null ? Math.trunc(Number(scanData.trackNumber)) : null } getTrackNumberFromFilename(bookScanData, filename) { const { title, author, series, publishYear } = bookScanData var partbasename = Path.basename(filename, Path.extname(filename)) // Remove title, author, series, and publishYear from filename if there if (title) partbasename = partbasename.replace(title, '') if (author) partbasename = partbasename.replace(author, '') if (series) partbasename = partbasename.replace(series, '') if (publishYear) partbasename = partbasename.replace(publishYear) // Remove eg. "disc 1" from path partbasename = partbasename.replace(/\bdisc \d\d?\b/i, '') // Remove "cd01" or "cd 01" from path partbasename = partbasename.replace(/\bcd ?\d\d?\b/i, '') var numbersinpath = partbasename.match(/\d{1,4}/g) if (!numbersinpath) return null var number = numbersinpath.length ? parseInt(numbersinpath[0]) : null return number } getCdNumberFromFilename(bookScanData, filename) { const { title, author, series, publishYear } = bookScanData var partbasename = Path.basename(filename, Path.extname(filename)) // Remove title, author, series, and publishYear from filename if there if (title) partbasename = partbasename.replace(title, '') if (author) partbasename = partbasename.replace(author, '') if (series) partbasename = partbasename.replace(series, '') if (publishYear) partbasename = partbasename.replace(publishYear) var cdNumber = null var cdmatch = partbasename.match(/\b(disc|cd) ?(\d\d?)\b/i) if (cdmatch && cdmatch.length > 2 && cdmatch[2]) { if (!isNaN(cdmatch[2])) { cdNumber = Number(cdmatch[2]) } } return cdNumber } getAverageScanDurationMs(results) { if (!results.length) return 0 var total = 0 for (let i = 0; i < results.length; i++) total += results[i].elapsed return Math.floor(total / results.length) } async scan(audioFileData, bookScanData, verbose = false) { var probeStart = Date.now() // Logger.debug(`[AudioFileScanner] Start Probe ${audioFileData.fullPath}`) var probeData = await prober.probe2(audioFileData.fullPath, verbose) if (probeData.error) { Logger.error(`[AudioFileScanner] ${probeData.error} : "${audioFileData.fullPath}"`) return null } // Logger.debug(`[AudioFileScanner] Finished Probe ${audioFileData.fullPath} elapsed ${msToTimestamp(Date.now() - probeStart, true)}`) var audioFile = new AudioFile() audioFileData.trackNumFromMeta = this.getTrackNumberFromMeta(probeData) audioFileData.trackNumFromFilename = this.getTrackNumberFromFilename(bookScanData, audioFileData.filename) audioFileData.cdNumFromFilename = this.getCdNumberFromFilename(bookScanData, audioFileData.filename) audioFile.setDataFromProbe(audioFileData, probeData) if (audioFile.embeddedCoverArt) { } return { audioFile, elapsed: Date.now() - probeStart } } // Returns array of { AudioFile, elapsed, averageScanDuration } from audio file scan objects async executeAudioFileScans(audioFileDataArray, bookScanData) { var proms = [] for (let i = 0; i < audioFileDataArray.length; i++) { proms.push(this.scan(audioFileDataArray[i], bookScanData)) } var scanStart = Date.now() var results = await Promise.all(proms).then((scanResults) => scanResults.filter(sr => sr)) return { audioFiles: results.map(r => r.audioFile), elapsed: Date.now() - scanStart, averageScanDuration: this.getAverageScanDurationMs(results) } } async scanAudioFiles(audioFileDataArray, bookScanData, audiobook, preferAudioMetadata, libraryScan = null) { var hasUpdated = false var audioScanResult = await this.executeAudioFileScans(audioFileDataArray, bookScanData) if (audioScanResult.audioFiles.length) { if (libraryScan) { libraryScan.addLog(LogLevel.DEBUG, `Book "${bookScanData.path}" Audio file scan took ${audioScanResult.elapsed}ms for ${audioScanResult.audioFiles.length} with average time of ${audioScanResult.averageScanDuration}ms`) } var totalAudioFilesToInclude = audiobook.audioFilesToInclude.filter(af => !audioScanResult.audioFiles.find(_af => _af.ino === af.ino)).length + audioScanResult.audioFiles.length // validate & add/update audio files to audiobook for (let i = 0; i < audioScanResult.audioFiles.length; i++) { var newAF = audioScanResult.audioFiles[i] var existingAF = audiobook.getAudioFileByIno(newAF.ino) var trackIndex = null if (totalAudioFilesToInclude === 1) { // Single track audiobooks trackIndex = 1 } else if (existingAF && existingAF.manuallyVerified) { // manually verified audio files use existing index trackIndex = existingAF.index } else { trackIndex = newAF.validateTrackIndex() } if (trackIndex !== null) { if (audiobook.checkHasTrackNum(trackIndex, newAF.ino)) { newAF.setDuplicateTrackNumber(trackIndex) } else { newAF.index = trackIndex } } if (existingAF) { if (audiobook.updateAudioFile(newAF)) { // console.log('update dauido file') hasUpdated = true } } else { audiobook.addAudioFile(newAF) // console.log('added auido file') hasUpdated = true } } if (hasUpdated) { audiobook.rebuildTracks() } // Set book details from audio file ID3 tags, optional prefer if (audiobook.setDetailsFromFileMetadata(preferAudioMetadata)) { hasUpdated = true } } return hasUpdated } } module.exports = new AudioFileScanner()