overleaf/services/web/app/coffee/Features/Project/ProjectCreationHandler.coffee

logger = require('logger-sharelatex')
async = require("async")
metrics = require('metrics-sharelatex')
Settings = require('settings-sharelatex')
ObjectId = require('mongoose').Types.ObjectId
Project = require('../../models/Project').Project
Folder = require('../../models/Folder').Folder
ProjectEntityUpdateHandler = require('./ProjectEntityUpdateHandler')
ProjectDetailsHandler = require('./ProjectDetailsHandler')
HistoryManager = require('../History/HistoryManager')
User = require('../../models/User').User
fs = require('fs')
Path = require "path"
_ = require "underscore"
AnalyticsManger = require("../Analytics/AnalyticsManager")

module.exports = ProjectCreationHandler =

	createBlankProject : (owner_id, projectName, attributes, callback = (error, project) ->)->
		metrics.inc("project-creation")
		if arguments.length == 3
			callback = attributes
			attributes = null

		ProjectDetailsHandler.validateProjectName projectName, (error) ->
			return callback(error) if error?
			logger.log owner_id:owner_id, projectName:projectName, "creating blank project"
			if attributes?
				ProjectCreationHandler._createBlankProject owner_id, projectName, attributes, (error, project) ->
					return callback(error) if error?
					AnalyticsManger.recordEvent(
						owner_id, 'project-imported', { projectId: project._id, attributes: attributes }
					)
					callback(error, project)
			else
				HistoryManager.initializeProject (error, history) ->
					return callback(error) if error?
					attributes = overleaf: history: id: history?.overleaf_id
					ProjectCreationHandler._createBlankProject owner_id, projectName, attributes, (error, project) ->
						return callback(error) if error?
						AnalyticsManger.recordEvent(
							owner_id, 'project-created', { projectId: project._id }
						)
						callback(error, project)

	_createBlankProject : (owner_id, projectName, attributes, callback = (error, project) ->)->
		rootFolder = new Folder {'name':'rootFolder'}

		attributes.owner_ref = new ObjectId(owner_id)
		attributes.name = projectName
		project = new Project attributes

		Object.assign(project, attributes)

		if Settings.apis?.project_history?.displayHistoryForNewProjects
			project.overleaf.history.display = true
		if Settings.currentImageName?
			# avoid clobbering any imageName already set in attributes (e.g. importedImageName)
			project.imageName ?= Settings.currentImageName
		project.rootFolder[0] = rootFolder
		User.findById owner_id, "ace.spellCheckLanguage", (err, user)->
			if user? # It's possible the owner_id is a UserStub
				project.spellCheckLanguage = user.ace.spellCheckLanguage
			project.save (err)->
				return callback(err) if err?
				callback err, project

	createProjectFromSnippet : (owner_id, projectName, docLines, callback = (error, project) ->)->
		@createBlankProject owner_id, projectName, (error, project)->
			return callback(error) if error?
			ProjectCreationHandler._createRootDoc project, owner_id, docLines, callback

	createBasicProject :  (owner_id, projectName, callback = (error, project) ->)->
		self = @
		@createBlankProject owner_id, projectName, (error, project)->
			return callback(error) if error?
			self._buildTemplate "mainbasic.tex", owner_id, projectName, (error, docLines)->
				return callback(error) if error?
				ProjectCreationHandler._createRootDoc project, owner_id, docLines, callback

	createExampleProject: (owner_id, projectName, callback = (error, project) ->)->
		self = @
		@createBlankProject owner_id, projectName, (error, project)->
			return callback(error) if error?
			async.series [
				(callback) ->
					self._buildTemplate "main.tex", owner_id, projectName, (error, docLines)->
						return callback(error) if error?
						ProjectCreationHandler._createRootDoc project, owner_id, docLines, callback
				(callback) ->
					self._buildTemplate "references.bib", owner_id, projectName, (error, docLines)->
						return callback(error) if error?
						ProjectEntityUpdateHandler.addDoc project._id, project.rootFolder[0]._id, "references.bib", docLines, owner_id, (error, doc)->
							callback(error)
				(callback) ->
					universePath = Path.resolve(__dirname + "/../../../templates/project_files/universe.jpg")
					ProjectEntityUpdateHandler.addFile project._id, project.rootFolder[0]._id, "universe.jpg", universePath, null, owner_id, callback
			], (error) ->
				callback(error, project)

	_createRootDoc: (project, owner_id, docLines, callback = (error, project) ->)->
		ProjectEntityUpdateHandler.addDoc project._id, project.rootFolder[0]._id, "main.tex", docLines, owner_id, (error, doc)->
			if error?
				logger.err err:error, "error adding root doc when creating project"
				return callback(error)
			ProjectEntityUpdateHandler.setRootDoc project._id, doc._id, (error) ->
				callback(error, project)

	_buildTemplate: (template_name, user_id, project_name, callback = (error, output) ->)->
		User.findById user_id, "first_name last_name", (error, user)->
			return callback(error) if error?
			monthNames = [ "January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December" ]

			templatePath = Path.resolve(__dirname + "/../../../templates/project_files/#{template_name}")
			fs.readFile templatePath, (error, template) ->
				return callback(error) if error?
				data =
					project_name: project_name
					user: user
					year: new Date().getUTCFullYear()
					month: monthNames[new Date().getUTCMonth()]
				output = _.template(template.toString(), data)
				callback null, output.split("\n")

metrics.timeAsyncMethod(
	ProjectCreationHandler, 'createBlankProject',
	'mongo.ProjectCreationHandler',
	logger
)