FileWriter = require('../../infrastructure/FileWriter') AuthorizationManager = require('../Authorization/AuthorizationManager') ProjectLocator = require('../Project/ProjectLocator') ProjectGetter = require('../Project/ProjectGetter') DocstoreManager = require('../Docstore/DocstoreManager') FileStoreHandler = require('../FileStore/FileStoreHandler') FileWriter = require('../../infrastructure/FileWriter') _ = require "underscore" Settings = require 'settings-sharelatex' AccessDeniedError = (message) -> error = new Error(message) error.name = 'AccessDenied' error.__proto__ = AccessDeniedError.prototype return error AccessDeniedError.prototype.__proto__ = Error.prototype BadEntityTypeError = (message) -> error = new Error(message) error.name = 'BadEntityType' error.__proto__ = BadEntityTypeError.prototype return error BadEntityTypeError.prototype.__proto__ = Error.prototype BadDataError = (message) -> error = new Error(message) error.name = 'BadData' error.__proto__ = BadDataError.prototype return error BadDataError.prototype.__proto__ = Error.prototype ProjectNotFoundError = (message) -> error = new Error(message) error.name = 'ProjectNotFound' error.__proto__ = ProjectNotFoundError.prototype return error ProjectNotFoundError.prototype.__proto__ = Error.prototype SourceFileNotFoundError = (message) -> error = new Error(message) error.name = 'BadData' error.__proto__ = SourceFileNotFoundError.prototype return error SourceFileNotFoundError.prototype.__proto__ = Error.prototype module.exports = ProjectFileAgent = sanitizeData: (data) -> return _.pick( data, 'source_project_id', 'source_entity_path' ) _validate: (data) -> return ( data.source_project_id? && data.source_entity_path? ) decorateLinkedFileData: (data, callback = (err, newData) ->) -> callback = _.once(callback) { source_project_id } = data return callback(new BadDataError()) if !source_project_id? ProjectGetter.getProject source_project_id, (err, project) -> return callback(err) if err? return callback(new ProjectNotFoundError()) if !project? callback(err, _.extend(data, {source_project_display_name: project.name})) checkAuth: (project_id, data, current_user_id, callback = (error, allowed)->) -> callback = _.once(callback) if !ProjectFileAgent._validate(data) return callback(new BadDataError()) {source_project_id, source_entity_path} = data AuthorizationManager.canUserReadProject current_user_id, source_project_id, null, (err, canRead) -> return callback(err) if err? callback(null, canRead) writeIncomingFileToDisk: (project_id, data, current_user_id, callback = (error, fsPath) ->) -> callback = _.once(callback) if !ProjectFileAgent._validate(data) return callback(new BadDataError()) {source_project_id, source_entity_path} = data ProjectLocator.findElementByPath { project_id: source_project_id, path: source_entity_path }, (err, entity, type) -> if err? if err.toString().match(/^not found.*/) err = new SourceFileNotFoundError() return callback(err) ProjectFileAgent._writeEntityToDisk source_project_id, entity._id, type, callback _writeEntityToDisk: (project_id, entity_id, type, callback=(err, location)->) -> callback = _.once(callback) if type == 'doc' DocstoreManager.getDoc project_id, entity_id, (err, lines) -> return callback(err) if err? FileWriter.writeLinesToDisk entity_id, lines, callback else if type == 'file' FileStoreHandler.getFileStream project_id, entity_id, null, (err, fileStream) -> return callback(err) if err? FileWriter.writeStreamToDisk entity_id, fileStream, callback else callback(new BadEntityTypeError()) handleError: (error, req, res, next) -> if error instanceof AccessDeniedError res.status(403).send("You do not have access to this project") else if error instanceof BadDataError res.status(400).send("The submitted data is not valid") else if error instanceof BadEntityTypeError res.status(400).send("The file is the wrong type") else if error instanceof SourceFileNotFoundError res.status(404).send("Source file not found") else if error instanceof ProjectNotFoundError res.status(404).send("Project not found") else next(error) next()