filesafe/routes/album.js

const routes = require('express').Router()
const path = require('path')
const paths = require('./../controllers/pathsController')
const utils = require('./../controllers/utilsController')
const config = require('./../config')
const db = require('knex')(config.database)

routes.get('/a/:identifier', async (req, res, next) => {
  const identifier = req.params.identifier
  if (identifier === undefined)
    return res.status(401).json({
      success: false,
      description: 'No identifier provided.'
    })

  const album = await db.table('albums')
    .where({
      identifier,
      enabled: 1
    })
    .select('id', 'name', 'identifier', 'editedAt', 'download', 'public', 'description')
    .first()

  if (!album)
    return res.status(404).sendFile(path.join(paths.errorRoot, config.errorPages[404]))
  else if (album.public === 0)
    return res.status(403).json({
      success: false,
      description: 'This album is not available for public.'
    })

  const nojs = req.query.nojs !== undefined

  // Cache ID - we initialize a separate cache for No-JS version
  const cacheid = nojs ? `${album.id}-nojs` : album.id

  if (!utils.albumsCache[cacheid])
    utils.albumsCache[cacheid] = {
      cache: null,
      generating: false,
      // Cache will actually be deleted after the album has been updated,
      // so storing this timestamp may be redundant, but just in case.
      generatedAt: 0
    }

  if (!utils.albumsCache[cacheid].cache && utils.albumsCache[cacheid].generating)
    return res.json({
      success: false,
      description: 'This album is still generating its public page.'
    })
  else if ((album.editedAt < utils.albumsCache[cacheid].generatedAt) || utils.albumsCache[cacheid].generating)
    return res.send(utils.albumsCache[cacheid].cache)

  // Use current timestamp to make sure cache is invalidated
  // when an album is edited during this generation process.
  utils.albumsCache[cacheid].generating = true
  utils.albumsCache[cacheid].generatedAt = Math.floor(Date.now() / 1000)

  const files = await db.table('files')
    .select('name', 'size')
    .where('albumid', album.id)
    .orderBy('id', 'DESC')

  album.thumb = ''
  album.totalSize = 0

  for (const file of files) {
    album.totalSize += parseInt(file.size)

    file.extname = path.extname(file.name)
    if (utils.mayGenerateThumb(file.extname)) {
      file.thumb = `thumbs/${file.name.slice(0, -file.extname.length)}.png`
      // If thumbnail for album is still not set, set it to current file's full URL.
      // A potential improvement would be to let the user set a specific image as an album cover.
      if (!album.thumb) album.thumb = file.name
    }
  }

  album.description = album.description
    ? album.description.replace(/\n/g, '<br>')
    : null

  album.downloadLink = album.download === 0
    ? null
    : `api/album/zip/${album.identifier}?v=${album.editedAt}`

  album.url = `a/${album.identifier}`

  return res.render('album', { config, album, files, nojs }, (error, html) => {
    utils.albumsCache[cacheid].cache = error ? null : html
    utils.albumsCache[cacheid].generating = false

    // Express should already send error to the next handler
    if (error) return
    return res.send(utils.albumsCache[cacheid].cache)
  })
})

module.exports = routes