audiobookshelf/server/db/item.db.js

const { Sequelize } = require('sequelize')
const Database = require('../Database')

const getLibraryItemMinified = (libraryItemId) => {
  return Database.models.libraryItem.findByPk(libraryItemId, {
    include: [
      {
        model: Database.models.book,
        attributes: [
          'id', 'title', 'subtitle', 'publishedYear', 'publishedDate', 'publisher', 'description', 'isbn', 'asin', 'language', 'explicit',
          [Sequelize.literal('(SELECT COUNT(*) FROM "audioTracks" WHERE "audioTracks"."mediaItemId" = book.id)'), 'numAudioTracks'],
          [Sequelize.literal('(SELECT COUNT(*) FROM "bookChapters" WHERE "bookChapters"."bookId" = book.id)'), 'numChapters']
        ],
        include: [
          {
            model: Database.models.genre,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'authors',
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'narrators',
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.series,
            attributes: ['id', 'name'],
            through: {
              attributes: ['sequence']
            }
          }
        ]
      },
      {
        model: Database.models.podcast,
        attributes: [
          'id', 'title', 'author', 'releaseDate', 'feedURL', 'imageURL', 'description', 'itunesPageURL', 'itunesId', 'itunesArtistId', 'language', 'podcastType', 'explicit', 'autoDownloadEpisodes',
          [Sequelize.literal('(SELECT COUNT(*) FROM "podcastEpisodes" WHERE "podcastEpisodes"."podcastId" = podcast.id)'), 'numPodcastEpisodes']
        ],
        include: [
          {
            model: Database.models.genre,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
        ]
      }
    ]
  })
}

const getLibraryItemFull = (libraryItemId) => {
  return Database.models.libraryItem.findByPk(libraryItemId, {
    include: [
      {
        model: Database.models.book,
        include: [
          {
            model: Database.models.audioTrack
          },
          {
            model: Database.models.genre,
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'authors',
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'narrators',
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.series,
            through: {
              attributes: ['sequence']
            }
          },
          {
            model: Database.models.bookChapter
          },
          {
            model: Database.models.eBookFile,
            include: 'fileMetadata'
          }
        ]
      },
      {
        model: Database.models.podcast,
        include: [
          {
            model: Database.models.podcastEpisode,
            include: {
              model: Database.models.audioTrack
            }
          },
          {
            model: Database.models.genre,
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            through: {
              attributes: []
            }
          },
        ]
      }
    ]
  })
}

const getLibraryItemExpanded = (libraryItemId) => {
  return Database.models.libraryItem.findByPk(libraryItemId, {
    include: [
      {
        model: Database.models.book,
        include: [
          {
            model: Database.models.fileMetadata,
            as: 'imageFile'
          },
          {
            model: Database.models.audioTrack,
            include: {
              model: Database.models.mediaFile,
              include: [
                'fileMetadata',
                'mediaStreams'
              ]
            }
          },
          {
            model: Database.models.genre,
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'authors',
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'narrators',
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.series,
            through: {
              attributes: ['sequence']
            }
          },
          {
            model: Database.models.bookChapter
          },
          {
            model: Database.models.eBookFile,
            include: 'fileMetadata'
          }
        ]
      },
      {
        model: Database.models.podcast,
        include: [
          {
            model: Database.models.fileMetadata,
            as: 'imageFile'
          },
          {
            model: Database.models.podcastEpisode,
            include: {
              model: Database.models.audioTrack,
              include: {
                model: Database.models.mediaFile,
                include: [
                  'fileMetadata',
                  'mediaStreams'
                ]
              }
            }
          },
          {
            model: Database.models.genre,
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            through: {
              attributes: []
            }
          },
        ]
      },
      {
        model: Database.models.libraryFile,
        include: 'fileMetadata'
      },
      'libraryFolder',
      'library'
    ]
  })
}

const getLibraryItemsForLibrary = async (libraryId) => {
  return Database.models.libraryItem.findAll({
    where: {
      libraryId
    },
    limit: 50,
    order: [
      [Database.models.book, 'title', 'DESC'],
      [Database.models.podcast, 'title', 'DESC']
    ],
    include: [
      {
        model: Database.models.book,
        attributes: [
          'id', 'title', 'subtitle', 'publishedYear', 'publishedDate', 'publisher', 'description', 'isbn', 'asin', 'language', 'explicit',
          [Sequelize.literal('(SELECT COUNT(*) FROM "audioTracks" WHERE "audioTracks"."mediaItemId" = book.id)'), 'numAudioTracks'],
          [Sequelize.literal('(SELECT COUNT(*) FROM "bookChapters" WHERE "bookChapters"."bookId" = book.id)'), 'numChapters']
        ],
        include: [
          {
            model: Database.models.genre,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'authors',
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.person,
            as: 'narrators',
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.series,
            attributes: ['id', 'name'],
            through: {
              attributes: ['sequence']
            }
          }
        ]
      },
      {
        model: Database.models.podcast,
        attributes: [
          'id', 'title', 'author', 'releaseDate', 'feedURL', 'imageURL', 'description', 'itunesPageURL', 'itunesId', 'itunesArtistId', 'language', 'podcastType', 'explicit', 'autoDownloadEpisodes',
          [Sequelize.literal('(SELECT COUNT(*) FROM "podcastEpisodes" WHERE "podcastEpisodes"."podcastId" = podcast.id)'), 'numPodcastEpisodes']
        ],
        include: [
          {
            model: Database.models.genre,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
          {
            model: Database.models.tag,
            attributes: ['id', 'name'],
            through: {
              attributes: []
            }
          },
        ]
      }
    ]
  })
}

module.exports = {
  getLibraryItemMinified,
  getLibraryItemFull,
  getLibraryItemExpanded,
  getLibraryItemsForLibrary
}