audiobookshelf/server/utils/toneHelpers.js

const tone = require('node-tone')
const fs = require('../libs/fsExtra')
const Logger = require('../Logger')
const { secondsToTimestamp } = require('./index')

module.exports.writeToneChaptersFile = (chapters, filePath) => {
  var chaptersTxt = ''
  for (const chapter of chapters) {
    chaptersTxt += `${secondsToTimestamp(chapter.start, true, true)} ${chapter.title}\n`
  }
  return fs.writeFile(filePath, chaptersTxt)
}

module.exports.getToneMetadataObject = (libraryItem, chaptersFile) => {
  const coverPath = libraryItem.media.coverPath
  const bookMetadata = libraryItem.media.metadata

  const metadataObject = {
    'Title': bookMetadata.title || '',
    'Album': bookMetadata.title || '',
    'TrackTotal': libraryItem.media.tracks.length
  }
  const additionalFields = []

  if (bookMetadata.subtitle) {
    metadataObject['Subtitle'] = bookMetadata.subtitle
  }
  if (bookMetadata.authorName) {
    metadataObject['Artist'] = bookMetadata.authorName
    metadataObject['AlbumArtist'] = bookMetadata.authorName
  }
  if (bookMetadata.description) {
    metadataObject['Comment'] = bookMetadata.description
    metadataObject['Description'] = bookMetadata.description
  }
  if (bookMetadata.narratorName) {
    metadataObject['Narrator'] = bookMetadata.narratorName
    metadataObject['Composer'] = bookMetadata.narratorName
  }
  if (bookMetadata.firstSeriesName) {
    metadataObject['MovementName'] = bookMetadata.firstSeriesName
  }
  if (bookMetadata.firstSeriesSequence) {
    metadataObject['Movement'] = bookMetadata.firstSeriesSequence
  }
  if (bookMetadata.genres.length) {
    metadataObject['Genre'] = bookMetadata.genres.join('/')
  }
  if (bookMetadata.publisher) {
    metadataObject['Publisher'] = bookMetadata.publisher
  }
  if (bookMetadata.asin) {
    additionalFields.push(`ASIN=${bookMetadata.asin}`)
  }
  if (bookMetadata.isbn) {
    additionalFields.push(`ISBN=${bookMetadata.isbn}`)
  }
  if (coverPath) {
    metadataObject['CoverFile'] = coverPath
  }
  if (parsePublishedYear(bookMetadata.publishedYear)) {
    metadataObject['PublishingDate'] = parsePublishedYear(bookMetadata.publishedYear)
  }
  if (chaptersFile) {
    metadataObject['ChaptersFile'] = chaptersFile
  }

  if (additionalFields.length) {
    metadataObject['AdditionalFields'] = additionalFields
  }

  return metadataObject
}

module.exports.writeToneMetadataJsonFile = (libraryItem, filePath) => {
  const bookMetadata = libraryItem.media.metadata
  const coverPath = libraryItem.media.coverPath
  const chapters = libraryItem.media.chapters

  const metadataObject = {
    'album': bookMetadata.title || '',
    'title': bookMetadata.title || '',
    'trackTotal': libraryItem.media.tracks.length,
    'additionalFields': {}
  }
  if (bookMetadata.subtitle) {
    metadataObject['subtitle'] = bookMetadata.subtitle
  }
  if (bookMetadata.authorName) {
    metadataObject['artist'] = bookMetadata.authorName
    metadataObject['albumArtist'] = bookMetadata.authorName
  }
  if (bookMetadata.description) {
    metadataObject['comment'] = bookMetadata.description
    metadataObject['description'] = bookMetadata.description
  }
  if (bookMetadata.narratorName) {
    metadataObject['narrator'] = bookMetadata.narratorName
    metadataObject['composer'] = bookMetadata.narratorName
  }
  if (bookMetadata.firstSeriesName) {
    metadataObject['movementName'] = bookMetadata.firstSeriesName
  }
  if (bookMetadata.firstSeriesSequence) {
    metadataObject['movement'] = bookMetadata.firstSeriesSequence
  }
  if (bookMetadata.genres.length) {
    metadataObject['genre'] = bookMetadata.genres.join('/')
  }
  if (bookMetadata.publisher) {
    metadataObject['publisher'] = bookMetadata.publisher
  }
  if (bookMetadata.asin) {
    metadataObject.additionalFields['asin'] = bookMetadata.asin
  }
  if (bookMetadata.isbn) {
    metadataObject.additionalFields['isbn'] = bookMetadata.isbn
  }
  if (coverPath) {
    metadataObject['coverFile'] = coverPath
  }
  if (parsePublishedYear(bookMetadata.publishedYear)) {
    metadataObject['publishingDate'] = parsePublishedYear(bookMetadata.publishedYear)
  }
  if (chapters && chapters.length > 0) {
    let metadataChapters = []
    for (const chapter of chapters) {
      metadataChapters.push({
        start: Math.round(chapter.start * 1000),
        length: Math.round((chapter.end - chapter.start) * 1000),
        title: chapter.title,
      })
    }
    metadataObject['chapters'] = metadataChapters
  }

  return fs.writeFile(filePath, JSON.stringify({ meta: metadataObject }))
}

module.exports.tagAudioFile = (filePath, payload) => {
  return tone.tag(filePath, payload).then((data) => {
    Logger.info('Tone results: ', data)
    return true
  }).catch((error) => {
    Logger.error(`[toneHelpers] tagAudioFile: Failed for "${filePath}"`, error)
    return false
  })
}

function parsePublishedYear(publishedYear) {
  if (isNaN(publishedYear) || !publishedYear || Number(publishedYear) <= 0) return null
  return `01/01/${publishedYear}`
}