safe.b68dev.xyz/routes/album.js

100 lines
3.2 KiB
JavaScript

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