ProjectEntityHandler.coffee 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630
  1. Project = require('../../models/Project').Project
  2. settings = require "settings-sharelatex"
  3. Doc = require('../../models/Doc').Doc
  4. Folder = require('../../models/Folder').Folder
  5. File = require('../../models/File').File
  6. FileStoreHandler = require("../FileStore/FileStoreHandler")
  7. Errors = require "../Errors/Errors"
  8. tpdsUpdateSender = require('../ThirdPartyDataStore/TpdsUpdateSender')
  9. projectLocator = require('./ProjectLocator')
  10. path = require "path"
  11. async = require "async"
  12. _ = require('underscore')
  13. logger = require('logger-sharelatex')
  14. docComparitor = require('./DocLinesComparitor')
  15. projectUpdateHandler = require('./ProjectUpdateHandler')
  16. DocstoreManager = require "../Docstore/DocstoreManager"
  17. ProjectGetter = require "./ProjectGetter"
  18. CooldownManager = require '../Cooldown/CooldownManager'
  19. DocumentUpdaterHandler = require('../../Features/DocumentUpdater/DocumentUpdaterHandler')
  20. module.exports = ProjectEntityHandler =
  21. getAllFolders: (project_id, callback) ->
  22. logger.log project_id:project_id, "getting all folders for project"
  23. ProjectGetter.getProjectWithoutDocLines project_id, (err, project) ->
  24. return callback(err) if err?
  25. return callback("no project") if !project?
  26. ProjectEntityHandler.getAllFoldersFromProject project, callback
  27. getAllDocs: (project_id, callback) ->
  28. logger.log project_id:project_id, "getting all docs for project"
  29. # We get the path and name info from the project, and the lines and
  30. # version info from the doc store.
  31. DocstoreManager.getAllDocs project_id, (error, docContentsArray) ->
  32. return callback(error) if error?
  33. # Turn array from docstore into a dictionary based on doc id
  34. docContents = {}
  35. for docContent in docContentsArray
  36. docContents[docContent._id] = docContent
  37. ProjectEntityHandler.getAllFolders project_id, (error, folders = {}) ->
  38. return callback(error) if error?
  39. docs = {}
  40. for folderPath, folder of folders
  41. for doc in (folder.docs or [])
  42. content = docContents[doc._id.toString()]
  43. if content?
  44. docs[path.join(folderPath, doc.name)] = {
  45. _id: doc._id
  46. name: doc.name
  47. lines: content.lines
  48. rev: content.rev
  49. }
  50. logger.log count:_.keys(docs).length, project_id:project_id, "returning docs for project"
  51. callback null, docs
  52. getAllFiles: (project_id, callback) ->
  53. logger.log project_id:project_id, "getting all files for project"
  54. @getAllFolders project_id, (err, folders = {}) ->
  55. return callback(err) if err?
  56. files = {}
  57. for folderPath, folder of folders
  58. for file in (folder.fileRefs or [])
  59. if file?
  60. files[path.join(folderPath, file.name)] = file
  61. callback null, files
  62. getAllFoldersFromProject: (project, callback) ->
  63. folders = {}
  64. processFolder = (basePath, folder) ->
  65. folders[basePath] = folder
  66. for childFolder in (folder.folders or [])
  67. if childFolder.name?
  68. processFolder path.join(basePath, childFolder.name), childFolder
  69. processFolder "/", project.rootFolder[0]
  70. callback null, folders
  71. getAllEntitiesFromProject: (project, callback) ->
  72. logger.log project:project, "getting all files for project"
  73. @getAllFoldersFromProject project, (err, folders = {}) ->
  74. return callback(err) if err?
  75. docs = []
  76. files = []
  77. for folderPath, folder of folders
  78. for doc in (folder.docs or [])
  79. if doc?
  80. docs.push({path: path.join(folderPath, doc.name), doc:doc})
  81. for file in (folder.fileRefs or [])
  82. if file?
  83. files.push({path: path.join(folderPath, file.name), file:file})
  84. callback null, docs, files
  85. getAllDocPathsFromProject: (project, callback) ->
  86. logger.log project:project, "getting all docs for project"
  87. @getAllFoldersFromProject project, (err, folders = {}) ->
  88. return callback(err) if err?
  89. docPath = {}
  90. for folderPath, folder of folders
  91. for doc in (folder.docs or [])
  92. docPath[doc._id] = path.join(folderPath, doc.name)
  93. logger.log count:_.keys(docPath).length, project_id:project._id, "returning docPaths for project"
  94. callback null, docPath
  95. flushProjectToThirdPartyDataStore: (project_id, callback) ->
  96. self = @
  97. logger.log project_id:project_id, "flushing project to tpds"
  98. DocumentUpdaterHandler.flushProjectToMongo project_id, (error) ->
  99. return callback(error) if error?
  100. ProjectGetter.getProject project_id, {name:true}, (error, project) ->
  101. return callback(error) if error?
  102. requests = []
  103. self.getAllDocs project_id, (error, docs) ->
  104. return callback(error) if error?
  105. for docPath, doc of docs
  106. do (docPath, doc) ->
  107. requests.push (cb) ->
  108. tpdsUpdateSender.addDoc {project_id:project_id, doc_id:doc._id, path:docPath, project_name:project.name, rev:doc.rev||0}, cb
  109. self.getAllFiles project_id, (error, files) ->
  110. return callback(error) if error?
  111. for filePath, file of files
  112. do (filePath, file) ->
  113. requests.push (cb) ->
  114. tpdsUpdateSender.addFile {project_id:project_id, file_id:file._id, path:filePath, project_name:project.name, rev:file.rev}, cb
  115. async.series requests, (err) ->
  116. logger.log project_id:project_id, "finished flushing project to tpds"
  117. callback(err)
  118. setRootDoc: (project_id, newRootDocID, callback = (error) ->)->
  119. logger.log project_id: project_id, rootDocId: newRootDocID, "setting root doc"
  120. Project.update {_id:project_id}, {rootDoc_id:newRootDocID}, {}, callback
  121. unsetRootDoc: (project_id, callback = (error) ->) ->
  122. logger.log project_id: project_id, "removing root doc"
  123. Project.update {_id:project_id}, {$unset: {rootDoc_id: true}}, {}, callback
  124. getDoc: (project_id, doc_id, options = {}, callback = (error, lines, rev) ->) ->
  125. if typeof(options) == "function"
  126. callback = options
  127. options = {}
  128. if options["pathname"]
  129. delete options["pathname"]
  130. projectLocator.findElement {project_id: project_id, element_id: doc_id, type: 'doc'}, (error, doc, path) =>
  131. return callback(error) if error?
  132. DocstoreManager.getDoc project_id, doc_id, options, (error, lines, rev, version, ranges) =>
  133. callback(error, lines, rev, version, ranges, path.fileSystem)
  134. else
  135. DocstoreManager.getDoc project_id, doc_id, options, callback
  136. addDoc: (project_or_id, folder_id, docName, docLines, userId, callback = (error, doc, folder_id) ->)=>
  137. ProjectEntityHandler.addDocWithoutUpdatingHistory project_or_id, folder_id, docName, docLines, userId, (error, doc, folder_id, path) ->
  138. return callback(error) if error?
  139. newDocs = [
  140. doc: doc
  141. path: path
  142. docLines: docLines.join('\n')
  143. ]
  144. project_id = project_or_id._id or project_or_id
  145. DocumentUpdaterHandler.updateProjectStructure project_id, userId, {newDocs}, (error) ->
  146. return callback(error) if error?
  147. callback null, doc, folder_id
  148. addDocWithoutUpdatingHistory: (project_or_id, folder_id, docName, docLines, userId, callback = (error, doc, folder_id) ->)=>
  149. # This method should never be called directly, except when importing a project
  150. # from Overleaf. It skips sending updates to the project history, which will break
  151. # the history unless you are making sure it is updated in some other way.
  152. getProject = (cb) ->
  153. if project_or_id._id? # project
  154. return cb(null, project_or_id)
  155. else # id
  156. return ProjectGetter.getProjectWithOnlyFolders project_or_id, cb
  157. getProject (error, project) ->
  158. if err?
  159. logger.err project_id:project_id, err:err, "error getting project for add doc"
  160. return callback(err)
  161. ProjectEntityHandler._addDocWithProject project, folder_id, docName, docLines, userId, callback
  162. _addDocWithProject: (project, folder_id, docName, docLines, userId, callback = (error, doc, folder_id, path) ->)=>
  163. project_id = project._id
  164. logger.log project_id: project_id, folder_id: folder_id, doc_name: docName, "adding doc to project with project"
  165. confirmFolder project, folder_id, (folder_id)=>
  166. doc = new Doc name: docName
  167. # Put doc in docstore first, so that if it errors, we don't have a doc_id in the project
  168. # which hasn't been created in docstore.
  169. DocstoreManager.updateDoc project_id.toString(), doc._id.toString(), docLines, 0, {}, (err, modified, rev) ->
  170. return callback(err) if err?
  171. ProjectEntityHandler._putElement project, folder_id, doc, "doc", (err, result)=>
  172. return callback(err) if err?
  173. tpdsUpdateSender.addDoc {
  174. project_id: project_id,
  175. doc_id: doc?._id
  176. path: result?.path?.fileSystem,
  177. project_name: project.name,
  178. rev: 0
  179. }, (err) ->
  180. return callback(err) if err?
  181. callback(null, doc, folder_id, result?.path?.fileSystem)
  182. restoreDoc: (project_id, doc_id, name, callback = (error, doc, folder_id) ->) ->
  183. # getDoc will return the deleted doc's lines, but we don't actually remove
  184. # the deleted doc, just create a new one from its lines.
  185. ProjectEntityHandler.getDoc project_id, doc_id, include_deleted: true, (error, lines) ->
  186. return callback(error) if error?
  187. ProjectEntityHandler.addDoc project_id, null, name, lines, callback
  188. addFileWithoutUpdatingHistory: (project_id, folder_id, fileName, path, userId, callback = (error, fileRef, folder_id, path, fileStoreUrl) ->)->
  189. ProjectGetter.getProjectWithOnlyFolders project_id, (err, project) ->
  190. if err?
  191. logger.err project_id:project_id, err:err, "error getting project for add file"
  192. return callback(err)
  193. logger.log project_id: project._id, folder_id: folder_id, file_name: fileName, path:path, "adding file"
  194. return callback(err) if err?
  195. confirmFolder project, folder_id, (folder_id)->
  196. fileRef = new File name : fileName
  197. FileStoreHandler.uploadFileFromDisk project._id, fileRef._id, path, (err, fileStoreUrl)->
  198. if err?
  199. logger.err err:err, project_id: project._id, folder_id: folder_id, file_name: fileName, fileRef:fileRef, "error uploading image to s3"
  200. return callback(err)
  201. ProjectEntityHandler._putElement project, folder_id, fileRef, "file", (err, result)=>
  202. if err?
  203. logger.err err:err, project_id: project._id, folder_id: folder_id, file_name: fileName, fileRef:fileRef, "error adding file with project"
  204. return callback(err)
  205. tpdsUpdateSender.addFile {project_id:project._id, file_id:fileRef._id, path:result?.path?.fileSystem, project_name:project.name, rev:fileRef.rev}, (err) ->
  206. return callback(err) if err?
  207. callback(null, fileRef, folder_id, result?.path?.fileSystem, fileStoreUrl)
  208. addFile: (project_id, folder_id, fileName, fsPath, userId, callback = (error, fileRef, folder_id) ->)->
  209. ProjectEntityHandler.addFileWithoutUpdatingHistory project_id, folder_id, fileName, fsPath, userId, (error, fileRef, folder_id, path, fileStoreUrl) ->
  210. newFiles = [
  211. file: fileRef
  212. path: path
  213. url: fileStoreUrl
  214. ]
  215. DocumentUpdaterHandler.updateProjectStructure project_id, userId, {newFiles}, (error) ->
  216. return callback(error) if error?
  217. callback null, fileRef, folder_id
  218. replaceFile: (project_id, file_id, fsPath, userId, callback)->
  219. self = ProjectEntityHandler
  220. FileStoreHandler.uploadFileFromDisk project_id, file_id, fsPath, (err, fileStoreUrl)->
  221. return callback(err) if err?
  222. ProjectGetter.getProject project_id, {rootFolder: true, name:true}, (err, project) ->
  223. return callback(err) if err?
  224. # Note there is a potential race condition here (and elsewhere)
  225. # If the file tree changes between findElement and the Project.update
  226. # then the path to the file element will be out of date. In practice
  227. # this is not a problem so long as we do not do anything longer running
  228. # between them (like waiting for the file to upload.)
  229. projectLocator.findElement {project:project, element_id: file_id, type: 'file'}, (err, fileRef, path)=>
  230. return callback(err) if err?
  231. tpdsUpdateSender.addFile {project_id:project._id, file_id:fileRef._id, path:path.fileSystem, rev:fileRef.rev+1, project_name:project.name}, (err) ->
  232. return callback(err) if err?
  233. conditions = _id:project._id
  234. inc = {}
  235. inc["#{path.mongo}.rev"] = 1
  236. set = {}
  237. set["#{path.mongo}.created"] = new Date()
  238. update =
  239. "$inc": inc
  240. "$set": set
  241. Project.findOneAndUpdate conditions, update, { "new": true}, (err) ->
  242. return callback(err) if err?
  243. newFiles = [
  244. file: fileRef
  245. path: path.fileSystem
  246. url: fileStoreUrl
  247. ]
  248. DocumentUpdaterHandler.updateProjectStructure project_id, userId, {newFiles}, callback
  249. copyFileFromExistingProjectWithProject: (project, folder_id, originalProject_id, origonalFileRef, userId, callback = (error, fileRef, folder_id) ->)->
  250. project_id = project._id
  251. logger.log { project_id, folder_id, originalProject_id, origonalFileRef }, "copying file in s3 with project"
  252. return callback(err) if err?
  253. confirmFolder project, folder_id, (folder_id)=>
  254. if !origonalFileRef?
  255. logger.err { project_id, folder_id, originalProject_id, origonalFileRef }, "file trying to copy is null"
  256. return callback()
  257. fileRef = new File name : origonalFileRef.name
  258. FileStoreHandler.copyFile originalProject_id, origonalFileRef._id, project._id, fileRef._id, (err, fileStoreUrl)->
  259. if err?
  260. logger.err { err, project_id, folder_id, originalProject_id, origonalFileRef }, "error coping file in s3"
  261. return callback(err)
  262. ProjectEntityHandler._putElement project, folder_id, fileRef, "file", (err, result)=>
  263. if err?
  264. logger.err { err, project_id, folder_id }, "error putting element as part of copy"
  265. return callback(err)
  266. tpdsUpdateSender.addFile { project_id, file_id:fileRef._id, path:result?.path?.fileSystem, rev:fileRef.rev, project_name:project.name}, (err) ->
  267. if err?
  268. logger.err { err, project_id, folder_id, originalProject_id, origonalFileRef }, "error sending file to tpds worker"
  269. newFiles = [
  270. file: fileRef
  271. path: result?.path?.fileSystem
  272. url: fileStoreUrl
  273. ]
  274. DocumentUpdaterHandler.updateProjectStructure project_id, userId, {newFiles}, (error) ->
  275. return callback(error) if error?
  276. callback null, fileRef, folder_id
  277. mkdirp: (project_id, path, callback = (err, newlyCreatedFolders, lastFolderInPath)->)->
  278. self = @
  279. folders = path.split('/')
  280. folders = _.select folders, (folder)->
  281. return folder.length != 0
  282. ProjectGetter.getProjectWithOnlyFolders project_id, (err, project)=>
  283. if path == '/'
  284. logger.log project_id: project._id, "mkdir is only trying to make path of / so sending back root folder"
  285. return callback(null, [], project.rootFolder[0])
  286. logger.log project_id: project._id, path:path, folders:folders, "running mkdirp"
  287. builtUpPath = ''
  288. procesFolder = (previousFolders, folderName, callback)=>
  289. previousFolders = previousFolders || []
  290. parentFolder = previousFolders[previousFolders.length-1]
  291. if parentFolder?
  292. parentFolder_id = parentFolder._id
  293. builtUpPath = "#{builtUpPath}/#{folderName}"
  294. projectLocator.findElementByPath project, builtUpPath, (err, foundFolder)=>
  295. if !foundFolder?
  296. logger.log path:path, project_id:project._id, folderName:folderName, "making folder from mkdirp"
  297. @addFolder project_id, parentFolder_id, folderName, (err, newFolder, parentFolder_id)->
  298. return callback(err) if err?
  299. newFolder.parentFolder_id = parentFolder_id
  300. previousFolders.push newFolder
  301. callback null, previousFolders
  302. else
  303. foundFolder.filterOut = true
  304. previousFolders.push foundFolder
  305. callback null, previousFolders
  306. async.reduce folders, [], procesFolder, (err, folders)->
  307. return callback(err) if err?
  308. lastFolder = folders[folders.length-1]
  309. folders = _.select folders, (folder)->
  310. !folder.filterOut
  311. callback(null, folders, lastFolder)
  312. addFolder: (project_id, parentFolder_id, folderName, callback) ->
  313. ProjectGetter.getProjectWithOnlyFolders project_id, (err, project)=>
  314. if err?
  315. logger.err project_id:project_id, err:err, "error getting project for add folder"
  316. return callback(err)
  317. ProjectEntityHandler.addFolderWithProject project, parentFolder_id, folderName, callback
  318. addFolderWithProject: (project, parentFolder_id, folderName, callback = (err, folder, parentFolder_id)->) ->
  319. confirmFolder project, parentFolder_id, (parentFolder_id)=>
  320. folder = new Folder name: folderName
  321. logger.log project: project._id, parentFolder_id:parentFolder_id, folderName:folderName, "adding new folder"
  322. ProjectEntityHandler._putElement project, parentFolder_id, folder, "folder", (err, result)=>
  323. if err?
  324. logger.err err:err, project_id:project._id, "error adding folder to project"
  325. return callback(err)
  326. callback(err, folder, parentFolder_id)
  327. updateDocLines : (project_id, doc_id, lines, version, ranges, callback = (error) ->)->
  328. ProjectGetter.getProjectWithoutDocLines project_id, (err, project)->
  329. return callback(err) if err?
  330. return callback(new Errors.NotFoundError("project not found")) if !project?
  331. logger.log project_id: project_id, doc_id: doc_id, "updating doc lines"
  332. projectLocator.findElement {project:project, element_id:doc_id, type:"docs"}, (err, doc, path)->
  333. if err?
  334. logger.error err: err, doc_id: doc_id, project_id: project_id, lines: lines, "error finding doc while updating doc lines"
  335. return callback err
  336. if !doc?
  337. error = new Errors.NotFoundError("doc not found")
  338. logger.error err: error, doc_id: doc_id, project_id: project_id, lines: lines, "doc not found while updating doc lines"
  339. return callback(error)
  340. logger.log project_id: project_id, doc_id: doc_id, "telling docstore manager to update doc"
  341. DocstoreManager.updateDoc project_id, doc_id, lines, version, ranges, (err, modified, rev) ->
  342. if err?
  343. logger.error err: err, doc_id: doc_id, project_id:project_id, lines: lines, "error sending doc to docstore"
  344. return callback(err)
  345. logger.log project_id: project_id, doc_id: doc_id, modified:modified, "finished updating doc lines"
  346. if modified
  347. # Don't need to block for marking as updated
  348. projectUpdateHandler.markAsUpdated project_id
  349. tpdsUpdateSender.addDoc {project_id:project_id, path:path.fileSystem, doc_id:doc_id, project_name:project.name, rev:rev}, callback
  350. else
  351. callback()
  352. moveEntity: (project_id, entity_id, destFolderId, entityType, userId, callback = (error) ->)->
  353. self = @
  354. logger.log {entityType, entity_id, project_id, destFolderId}, "moving entity"
  355. if !entityType?
  356. logger.err {err: "No entityType set", project_id, entity_id}
  357. return callback("No entityType set")
  358. entityType = entityType.toLowerCase()
  359. ProjectGetter.getProject project_id, {rootFolder:true, name:true}, (err, project)=>
  360. return callback(err) if err?
  361. projectLocator.findElement {project, element_id: entity_id, type: entityType}, (err, entity, entityPath)->
  362. return callback(err) if err?
  363. self._checkValidMove project, entityType, entityPath, destFolderId, (error) ->
  364. return callback(error) if error?
  365. self.getAllEntitiesFromProject project, (error, oldDocs, oldFiles) =>
  366. return callback(error) if error?
  367. self._removeElementFromMongoArray Project, project_id, entityPath.mongo, (err, newProject)->
  368. return callback(err) if err?
  369. self._putElement newProject, destFolderId, entity, entityType, (err, result, newProject)->
  370. return callback(err) if err?
  371. opts =
  372. project_id: project_id
  373. project_name: project.name
  374. startPath: entityPath.fileSystem
  375. endPath: result.path.fileSystem,
  376. rev: entity.rev
  377. tpdsUpdateSender.moveEntity opts
  378. self.getAllEntitiesFromProject newProject, (error, newDocs, newFiles) =>
  379. return callback(error) if error?
  380. DocumentUpdaterHandler.updateProjectStructure project_id, userId, {oldDocs, newDocs, oldFiles, newFiles}, callback
  381. _checkValidMove: (project, entityType, entityPath, destFolderId, callback = (error) ->) ->
  382. return callback() if !entityType.match(/folder/)
  383. projectLocator.findElement { project, element_id: destFolderId, type:"folder"}, (err, destEntity, destFolderPath) ->
  384. return callback(err) if err?
  385. logger.log destFolderPath: destFolderPath.fileSystem, folderPath: entityPath.fileSystem, "checking folder is not moving into child folder"
  386. isNestedFolder = destFolderPath.fileSystem.slice(0, entityPath.fileSystem.length) == entityPath.fileSystem
  387. if isNestedFolder
  388. callback(new Error("destination folder is a child folder of me"))
  389. else
  390. callback()
  391. deleteEntity: (project_id, entity_id, entityType, userId, callback = (error) ->)->
  392. self = @
  393. logger.log entity_id:entity_id, entityType:entityType, project_id:project_id, "deleting project entity"
  394. if !entityType?
  395. logger.err err: "No entityType set", project_id: project_id, entity_id: entity_id
  396. return callback("No entityType set")
  397. entityType = entityType.toLowerCase()
  398. ProjectGetter.getProject project_id, {name:true, rootFolder:true}, (err, project)=>
  399. return callback(error) if error?
  400. projectLocator.findElement {project: project, element_id: entity_id, type: entityType}, (error, entity, path)=>
  401. return callback(error) if error?
  402. ProjectEntityHandler._cleanUpEntity project, entity, entityType, path.fileSystem, userId, (error) ->
  403. return callback(error) if error?
  404. tpdsUpdateSender.deleteEntity project_id:project_id, path:path.fileSystem, project_name:project.name, (error) ->
  405. return callback(error) if error?
  406. self._removeElementFromMongoArray Project, project_id, path.mongo, (error) ->
  407. return callback(error) if error?
  408. callback null
  409. renameEntity: (project_id, entity_id, entityType, newName, userId, callback)->
  410. logger.log(entity_id: entity_id, project_id: project_id, ('renaming '+entityType))
  411. if !entityType?
  412. logger.err err: "No entityType set", project_id: project_id, entity_id: entity_id
  413. return callback("No entityType set")
  414. entityType = entityType.toLowerCase()
  415. ProjectGetter.getProject project_id, {rootFolder:true, name:true}, (error, project)=>
  416. return callback(error) if error?
  417. ProjectEntityHandler.getAllEntitiesFromProject project, (error, oldDocs, oldFiles) =>
  418. return callback(error) if error?
  419. projectLocator.findElement {project:project, element_id:entity_id, type:entityType}, (error, entity, entPath)=>
  420. return callback(error) if error?
  421. endPath = path.join(path.dirname(entPath.fileSystem), newName)
  422. conditions = {_id:project_id}
  423. update = "$set":{}
  424. namePath = entPath.mongo+".name"
  425. update["$set"][namePath] = newName
  426. tpdsUpdateSender.moveEntity({project_id:project_id, startPath:entPath.fileSystem, endPath:endPath, project_name:project.name, rev:entity.rev})
  427. Project.findOneAndUpdate conditions, update, { "new": true}, (error, newProject) ->
  428. return callback(error) if error?
  429. ProjectEntityHandler.getAllEntitiesFromProject newProject, (error, newDocs, newFiles) =>
  430. return callback(error) if error?
  431. DocumentUpdaterHandler.updateProjectStructure project_id, userId, {oldDocs, newDocs, oldFiles, newFiles}, callback
  432. _cleanUpEntity: (project, entity, entityType, path, userId, callback = (error) ->) ->
  433. if(entityType.indexOf("file") != -1)
  434. ProjectEntityHandler._cleanUpFile project, entity, path, userId, callback
  435. else if (entityType.indexOf("doc") != -1)
  436. ProjectEntityHandler._cleanUpDoc project, entity, path, userId, callback
  437. else if (entityType.indexOf("folder") != -1)
  438. ProjectEntityHandler._cleanUpFolder project, entity, path, userId, callback
  439. else
  440. callback()
  441. _cleanUpDoc: (project, doc, path, userId, callback = (error) ->) ->
  442. project_id = project._id.toString()
  443. doc_id = doc._id.toString()
  444. unsetRootDocIfRequired = (callback) =>
  445. if project.rootDoc_id? and project.rootDoc_id.toString() == doc_id
  446. @unsetRootDoc project_id, callback
  447. else
  448. callback()
  449. unsetRootDocIfRequired (error) ->
  450. return callback(error) if error?
  451. DocumentUpdaterHandler.deleteDoc project_id, doc_id, (error) ->
  452. return callback(error) if error?
  453. ProjectEntityHandler._insertDeletedDocReference project._id, doc, (error) ->
  454. return callback(error) if error?
  455. DocstoreManager.deleteDoc project_id, doc_id, (error) ->
  456. return callback(error) if error?
  457. changes = oldDocs: [ {doc, path} ]
  458. DocumentUpdaterHandler.updateProjectStructure project_id, userId, changes, callback
  459. _cleanUpFile: (project, file, path, userId, callback = (error) ->) ->
  460. project_id = project._id.toString()
  461. file_id = file._id.toString()
  462. FileStoreHandler.deleteFile project_id, file_id, (error) ->
  463. return callback(error) if error?
  464. changes = oldFiles: [ {file, path} ]
  465. DocumentUpdaterHandler.updateProjectStructure project_id, userId, changes, callback
  466. _cleanUpFolder: (project, folder, folderPath, userId, callback = (error) ->) ->
  467. jobs = []
  468. for doc in folder.docs
  469. do (doc) ->
  470. docPath = path.join(folderPath, doc.name)
  471. jobs.push (callback) -> ProjectEntityHandler._cleanUpDoc project, doc, docPath, userId, callback
  472. for file in folder.fileRefs
  473. do (file) ->
  474. filePath = path.join(folderPath, file.name)
  475. jobs.push (callback) -> ProjectEntityHandler._cleanUpFile project, file, filePath, userId, callback
  476. for childFolder in folder.folders
  477. do (childFolder) ->
  478. folderPath = path.join(folderPath, childFolder.name)
  479. jobs.push (callback) -> ProjectEntityHandler._cleanUpFolder project, childFolder, folderPath, userId, callback
  480. async.series jobs, callback
  481. _removeElementFromMongoArray : (model, model_id, path, callback = (err, project) ->)->
  482. conditions = {_id:model_id}
  483. update = {"$unset":{}}
  484. update["$unset"][path] = 1
  485. model.update conditions, update, {}, (err)->
  486. pullUpdate = {"$pull":{}}
  487. nonArrayPath = path.slice(0, path.lastIndexOf("."))
  488. pullUpdate["$pull"][nonArrayPath] = null
  489. model.findOneAndUpdate conditions, pullUpdate, {"new": true}, callback
  490. _insertDeletedDocReference: (project_id, doc, callback = (error) ->) ->
  491. Project.update {
  492. _id: project_id
  493. }, {
  494. $push: {
  495. deletedDocs: {
  496. _id: doc._id
  497. name: doc.name
  498. }
  499. }
  500. }, {}, callback
  501. _countElements : (project, callback)->
  502. countFolder = (folder, cb = (err, count)->)->
  503. jobs = _.map folder?.folders, (folder)->
  504. (asyncCb)-> countFolder folder, asyncCb
  505. async.series jobs, (err, subfolderCounts)->
  506. total = 0
  507. if subfolderCounts?.length > 0
  508. total = _.reduce subfolderCounts, (a, b)-> return a + b
  509. if folder?.folders?.length?
  510. total += folder?.folders?.length
  511. if folder?.docs?.length?
  512. total += folder?.docs?.length
  513. if folder?.fileRefs?.length?
  514. total += folder?.fileRefs?.length
  515. cb(null, total)
  516. countFolder project.rootFolder[0], callback
  517. _putElement: (project, folder_id, element, type, callback = (err, path, project)->)->
  518. sanitizeTypeOfElement = (elementType)->
  519. lastChar = elementType.slice -1
  520. if lastChar != "s"
  521. elementType +="s"
  522. if elementType == "files"
  523. elementType = "fileRefs"
  524. return elementType
  525. if !element? or !element._id?
  526. e = new Error("no element passed to be inserted")
  527. logger.err project_id:project._id, folder_id:folder_id, element:element, type:type, "failed trying to insert element as it was null"
  528. return callback(e)
  529. type = sanitizeTypeOfElement type
  530. if path.resolve("/", element.name) isnt "/#{element.name}" or element.name.match("/")
  531. e = new Error("invalid element name")
  532. logger.err project_id:project._id, folder_id:folder_id, element:element, type:type, "failed trying to insert element as name was invalid"
  533. return callback(e)
  534. if !folder_id?
  535. folder_id = project.rootFolder[0]._id
  536. ProjectEntityHandler._countElements project, (err, count)->
  537. if count > settings.maxEntitiesPerProject
  538. logger.warn project_id:project._id, "project too big, stopping insertions"
  539. CooldownManager.putProjectOnCooldown(project._id)
  540. return callback("project_has_to_many_files")
  541. projectLocator.findElement {project:project, element_id:folder_id, type:"folders"}, (err, folder, path)=>
  542. if err?
  543. logger.err err:err, project_id:project._id, folder_id:folder_id, type:type, element:element, "error finding folder for _putElement"
  544. return callback(err)
  545. newPath =
  546. fileSystem: "#{path.fileSystem}/#{element.name}"
  547. mongo: path.mongo
  548. id = element._id+''
  549. element._id = require('mongoose').Types.ObjectId(id)
  550. conditions = _id:project._id
  551. mongopath = "#{path.mongo}.#{type}"
  552. update = "$push":{}
  553. update["$push"][mongopath] = element
  554. logger.log project_id: project._id, element_id: element._id, fileType: type, folder_id: folder_id, mongopath:mongopath, "adding element to project"
  555. Project.findOneAndUpdate conditions, update, {"new": true}, (err, project)->
  556. if err?
  557. logger.err err: err, project_id: project._id, 'error saving in putElement project'
  558. return callback(err)
  559. callback(err, {path:newPath}, project)
  560. confirmFolder = (project, folder_id, callback)->
  561. logger.log folder_id:folder_id, project_id:project._id, "confirming folder in project"
  562. if folder_id+'' == 'undefined'
  563. callback(project.rootFolder[0]._id)
  564. else if folder_id != null
  565. callback folder_id
  566. else
  567. callback(project.rootFolder[0]._id)