ProjectUploadController.js 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. /* eslint-disable
  2. camelcase,
  3. max-len,
  4. no-unused-vars,
  5. */
  6. // TODO: This file was created by bulk-decaffeinate.
  7. // Fix any style issues and re-enable lint.
  8. /*
  9. * decaffeinate suggestions:
  10. * DS102: Remove unnecessary code created because of implicit returns
  11. * DS207: Consider shorter variations of null checks
  12. * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
  13. */
  14. let ProjectUploadController
  15. const logger = require('@overleaf/logger')
  16. const metrics = require('@overleaf/metrics')
  17. const fs = require('fs')
  18. const Path = require('path')
  19. const FileSystemImportManager = require('./FileSystemImportManager')
  20. const ProjectUploadManager = require('./ProjectUploadManager')
  21. const SessionManager = require('../Authentication/SessionManager')
  22. const Settings = require('@overleaf/settings')
  23. const { InvalidZipFileError } = require('./ArchiveErrors')
  24. const multer = require('multer')
  25. const _ = require('lodash')
  26. const upload = multer(
  27. _.defaultsDeep(
  28. {
  29. dest: Settings.path.uploadFolder,
  30. limits: {
  31. fileSize: Settings.maxUploadSize,
  32. },
  33. },
  34. Settings.multerOptions
  35. )
  36. )
  37. module.exports = ProjectUploadController = {
  38. uploadProject(req, res, next) {
  39. const timer = new metrics.Timer('project-upload')
  40. const user_id = SessionManager.getLoggedInUserId(req.session)
  41. const { originalname, path } = req.file
  42. const name = Path.basename(originalname, '.zip')
  43. return ProjectUploadManager.createProjectFromZipArchive(
  44. user_id,
  45. name,
  46. path,
  47. function (error, project) {
  48. fs.unlink(path, function () {})
  49. timer.done()
  50. if (error != null) {
  51. logger.error(
  52. { err: error, filePath: path, fileName: name },
  53. 'error uploading project'
  54. )
  55. if (error instanceof InvalidZipFileError) {
  56. return res.status(422).json({
  57. success: false,
  58. error: req.i18n.translate(error.message),
  59. })
  60. } else {
  61. return res.status(500).json({
  62. success: false,
  63. error: req.i18n.translate('upload_failed'),
  64. })
  65. }
  66. } else {
  67. return res.json({ success: true, project_id: project._id })
  68. }
  69. }
  70. )
  71. },
  72. uploadFile(req, res, next) {
  73. const timer = new metrics.Timer('file-upload')
  74. const name = req.file != null ? req.file.originalname : undefined
  75. const path = req.file != null ? req.file.path : undefined
  76. const project_id = req.params.Project_id
  77. const { folder_id } = req.query
  78. if (name == null || name.length === 0 || name.length > 150) {
  79. logger.err(
  80. { projectId: project_id, fileName: name },
  81. 'bad name when trying to upload file'
  82. )
  83. return res.status(422).json({
  84. success: false,
  85. error: 'invalid_filename',
  86. })
  87. }
  88. const user_id = SessionManager.getLoggedInUserId(req.session)
  89. return FileSystemImportManager.addEntity(
  90. user_id,
  91. project_id,
  92. folder_id,
  93. name,
  94. path,
  95. true,
  96. function (error, entity) {
  97. fs.unlink(path, function () {})
  98. timer.done()
  99. if (error != null) {
  100. logger.error(
  101. {
  102. err: error,
  103. projectId: project_id,
  104. filePath: path,
  105. fileName: name,
  106. folderId: folder_id,
  107. },
  108. 'error uploading file'
  109. )
  110. if (error.name === 'InvalidNameError') {
  111. return res.status(422).json({
  112. success: false,
  113. error: 'invalid_filename',
  114. })
  115. } else if (error.message === 'project_has_too_many_files') {
  116. return res.status(422).json({
  117. success: false,
  118. error: 'project_has_too_many_files',
  119. })
  120. } else {
  121. return res.status(422).json({ success: false })
  122. }
  123. } else {
  124. return res.json({
  125. success: true,
  126. entity_id: entity != null ? entity._id : undefined,
  127. entity_type: entity != null ? entity.type : undefined,
  128. })
  129. }
  130. }
  131. )
  132. },
  133. multerMiddleware(req, res, next) {
  134. if (upload == null) {
  135. return res
  136. .status(500)
  137. .json({ success: false, error: req.i18n.translate('upload_failed') })
  138. }
  139. return upload.single('qqfile')(req, res, function (err) {
  140. if (err instanceof multer.MulterError && err.code === 'LIMIT_FILE_SIZE') {
  141. return res
  142. .status(422)
  143. .json({ success: false, error: req.i18n.translate('file_too_large') })
  144. }
  145. return next(err)
  146. })
  147. },
  148. }