ProjectEntityHandler.coffee 25 KB

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