overleaf/services/filestore/app/js/FileController.js

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 }
  }
}