filesafe/controllers/albumsController.js

const config = require('../config.js')
const db = require('knex')(config.database)
const randomstring = require('randomstring')
const utils = require('./utilsController.js')
const path = require('path')
const fs = require('fs')
const Zip = require('jszip')
const albumsController = {}

albumsController.list = async (req, res, next) => {
  const albumDomain = config.albumDomain || config.domain
  const user = await utils.authorize(req, res)
  if (!user) return

  const fields = ['id', 'name']
  if (req.params.sidebar === undefined) {
    fields.push('timestamp')
    fields.push('identifier')
  }

  const albums = await db.table('albums').select(fields).where({ enabled: 1, userid: user.id })
  if (req.params.sidebar !== undefined) {
    return res.json({ success: true, albums })
  }

  let ids = []
  for (let album of albums) {
    album.date = new Date(album.timestamp * 1000)
    album.date = utils.getPrettyDate(album.date)

    album.identifier = `${albumDomain}/a/${album.identifier}`
    ids.push(album.id)
  }

  const files = await db.table('files').whereIn('albumid', ids).select('albumid')
  const albumsCount = {}

  for (let id of ids) albumsCount[id] = 0
  for (let file of files) albumsCount[file.albumid] += 1
  for (let album of albums) album.files = albumsCount[album.id]

  return res.json({ success: true, albums })
}

albumsController.create = async (req, res, next) => {
  const user = await utils.authorize(req, res)
  if (!user) return

  const name = req.body.name
  if (name === undefined || name === '') {
    return res.json({ success: false, description: 'No album name specified.' })
  }

  const album = await db.table('albums').where({
    name: name,
    enabled: 1,
    userid: user.id
  }).first()

  if (album) {
    return res.json({ success: false, description: 'There\'s already an album with that name.' })
  }

  await db.table('albums').insert({
    name: name,
    enabled: 1,
    userid: user.id,
    identifier: randomstring.generate(8),
    timestamp: Math.floor(Date.now() / 1000),
    editedAt: 0,
    zipGeneratedAt: 0
  })

  return res.json({ success: true })
}

albumsController.delete = async (req, res, next) => {
  const user = await utils.authorize(req, res)
  if (!user) return

  const id = req.body.id
  if (id === undefined || id === '') {
    return res.json({ success: false, description: 'No album specified.' })
  }

  await db.table('albums').where({ id: id, userid: user.id }).update({ enabled: 0 })
  return res.json({ success: true })
}

albumsController.rename = async (req, res, next) => {
  const user = await utils.authorize(req, res)
  if (!user) return

  const id = req.body.id
  if (id === undefined || id === '') {
    return res.json({ success: false, description: 'No album specified.' })
  }

  const name = req.body.name
  if (name === undefined || name === '') {
    return res.json({ success: false, description: 'No name specified.' })
  }

  const album = await db.table('albums').where({ name: name, userid: user.id }).first()
  if (album) {
    return res.json({ success: false, description: 'Name already in use.' })
  }

  await db.table('albums').where({ id: id, userid: user.id }).update({ name: name })
  return res.json({ success: true })
}

albumsController.get = 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 }).first()
  if (!album) return res.json({ success: false, description: 'Album not found.' })

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

  for (let file of files) {
    file.file = `${config.domain}/${file.name}`

    const ext = path.extname(file.name).toLowerCase()
    if ((config.uploads.generateThumbnails.image && utils.imageExtensions.includes(ext)) || (config.uploads.generateThumbnails.video && utils.videoExtensions.includes(ext))) {
      file.thumb = `${config.domain}/thumbs/${file.name.slice(0, -ext.length)}.png`
    }
  }

  return res.json({
    success: true,
    title: title,
    count: files.length,
    files
  })
}

albumsController.generateZip = async (req, res, next) => {
  const identifier = req.params.identifier
  if (identifier === undefined) return res.status(401).json({ success: false, description: 'No identifier provided.' })
  if (!config.uploads.generateZips) return res.status(401).json({ success: false, description: 'Zip generation disabled.' })

  const album = await db.table('albums').where({ identifier, enabled: 1 }).first()
  if (!album) return res.json({ success: false, description: 'Album not found.' })

  if (album.zipGeneratedAt > album.editedAt) {
    const filePath = path.join(config.uploads.folder, 'zips', `${identifier}.zip`)
    const fileName = `${album.name}.zip`
    return res.download(filePath, fileName)
  } else {
    console.log(`Generating zip for album identifier: ${identifier}`)
    const files = await db.table('files').select('name').where('albumid', album.id)
    if (files.length === 0) return res.json({ success: false, description: 'There are no files in the album.' })

    const zipPath = path.join(__dirname, '..', config.uploads.folder, 'zips', `${album.identifier}.zip`)
    let archive = new Zip()

    for (let file of files) {
      try {
        // const exists = fs.statSync(path.join(__dirname, '..', config.uploads.folder, file.name))
        archive.file(file.name, fs.readFileSync(path.join(__dirname, '..', config.uploads.folder, file.name)))
      } catch (err) {
        console.log(err)
      }
    }

    archive
      .generateNodeStream({ type: 'nodebuffer', streamFiles: true })
      .pipe(fs.createWriteStream(zipPath))
      .on('finish', async () => {
        console.log(`Generated zip for album identifier: ${identifier}`)
        await db.table('albums')
          .where('id', album.id)
          .update({ zipGeneratedAt: Math.floor(Date.now() / 1000) })

        const filePath = path.join(config.uploads.folder, 'zips', `${identifier}.zip`)
        const fileName = `${album.name}.zip`
        return res.download(filePath, fileName)
      })
  }
}

module.exports = albumsController