overleaf/services/filestore/app/js/FileController.js
Simon Detheridge ec80052d9a Decaf cleanup of FileController
Remove deprecated res.send
2020-01-07 09:26:23 +00:00

169 lines
4.3 KiB
JavaScript

const PersistorManager = require('./PersistorManager')
const logger = require('logger-sharelatex')
const FileHandler = require('./FileHandler')
const metrics = require('metrics-sharelatex')
const parseRange = require('range-parser')
const Errors = require('./Errors')
const { pipeline } = require('stream')
const maxSizeInBytes = 1024 * 1024 * 1024 // 1GB
module.exports = {
getFile,
getFileHead,
insertFile,
copyFile,
deleteFile,
directorySize
}
function getFile(req, res) {
const { key, bucket } = req
const { format, style } = req.query
const options = {
key,
bucket,
format,
style
}
metrics.inc('getFile')
logger.log({ key, bucket, format, style }, 'receiving request to get file')
if (req.headers.range) {
const range = _getRange(req.headers.range)
if (range) {
options.start = range.start
options.end = range.end
logger.log(
{ start: range.start, end: range.end },
'getting range of bytes from file'
)
}
}
FileHandler.getFile(bucket, key, options, function(err, fileStream) {
if (err) {
if (err instanceof Errors.NotFoundError) {
res.sendStatus(404)
} else {
logger.err({ err, key, bucket, format, style }, 'problem getting file')
res.sendStatus(500)
}
return
}
if (req.query.cacheWarm) {
logger.log(
{ key, bucket, format, style },
'request is only for cache warm so not sending stream'
)
return res.sendStatus(200)
}
logger.log({ key, bucket, format, style }, 'sending file to response')
pipeline(fileStream, res)
})
}
function getFileHead(req, res) {
const { key, bucket } = req
metrics.inc('getFileSize')
logger.log({ key, bucket }, 'receiving request to get file metadata')
FileHandler.getFileSize(bucket, key, function(err, fileSize) {
if (err) {
if (err instanceof Errors.NotFoundError) {
res.sendStatus(404)
} else {
res.sendStatus(500)
}
return
}
res.set('Content-Length', fileSize)
res.status(200).end()
})
}
function insertFile(req, res) {
metrics.inc('insertFile')
const { key, bucket } = req
logger.log({ key, bucket }, 'receiving request to insert file')
FileHandler.insertFile(bucket, key, req, function(err) {
if (err) {
logger.log({ err, key, bucket }, 'error inserting file')
res.sendStatus(500)
} else {
res.sendStatus(200)
}
})
}
function copyFile(req, res) {
metrics.inc('copyFile')
const { key, bucket } = req
const oldProjectId = req.body.source.project_id
const oldFileId = req.body.source.file_id
logger.log(
{ key, bucket, oldProject_id: oldProjectId, oldFile_id: oldFileId },
'receiving request to copy file'
)
PersistorManager.copyFile(
bucket,
`${oldProjectId}/${oldFileId}`,
key,
function(err) {
if (err) {
if (err instanceof Errors.NotFoundError) {
res.sendStatus(404)
} else {
logger.log(
{ err, oldProject_id: oldProjectId, oldFile_id: oldFileId },
'something went wrong copying file'
)
res.sendStatus(500)
}
return
}
res.sendStatus(200)
}
)
}
function deleteFile(req, res) {
metrics.inc('deleteFile')
const { key, bucket } = req
logger.log({ key, bucket }, 'receiving request to delete file')
return FileHandler.deleteFile(bucket, key, function(err) {
if (err != null) {
logger.log({ err, key, bucket }, 'something went wrong deleting file')
return res.sendStatus(500)
} else {
return res.sendStatus(204)
}
})
}
function directorySize(req, res) {
metrics.inc('projectSize')
const { project_id: projectId, bucket } = req
logger.log({ projectId, bucket }, 'receiving request to project size')
FileHandler.getDirectorySize(bucket, projectId, function(err, size) {
if (err) {
logger.log({ err, projectId, bucket }, 'error inserting file')
return res.sendStatus(500)
}
res.json({ 'total bytes': size })
})
}
function _getRange(header) {
const parsed = parseRange(maxSizeInBytes, header)
if (parsed === -1 || parsed === -2 || parsed.type !== 'bytes') {
return null
} else {
const range = parsed[0]
return { start: range.start, end: range.end }
}
}