Overleaf/services/web/app/coffee/models/Project.coffee

mongoose = require('mongoose')
Settings = require 'settings-sharelatex'
_ = require('underscore')
FolderSchema = require('./Folder.js').FolderSchema
logger = require('logger-sharelatex')
sanitize = require('sanitizer')
concreteObjectId = require('mongoose').Types.ObjectId
Errors  = require "../errors"


Schema = mongoose.Schema
ObjectId = Schema.ObjectId

DeletedDocSchema = new Schema
	name: String

ProjectSchema = new Schema
	name              :   {type:String, default:'new project'}
	lastUpdated       :   {type:Date, default: () -> new Date()}
	lastOpened		  :   {type:Date}
	active			  :   { type: Boolean,  default: true }
	owner_ref         :   {type:ObjectId, ref:'User'}
	collaberator_refs :   [ type:ObjectId, ref:'User' ]
	readOnly_refs     :   [ type:ObjectId, ref:'User' ]
	rootDoc_id        :   {type: ObjectId}
	rootFolder        :   [FolderSchema]
	publicAccesLevel  :   {type: String, default: 'private'}
	compiler		  :   {type:String, default:'pdflatex'}
	spellCheckLanguage :   {type:String, default:'en'}
	deletedByExternalDataSource : {type: Boolean, default: false}
	description : {type:String, default:''}
	archived          : { type: Boolean }
	deletedDocs       : [DeletedDocSchema]
	imageName         : { type: String }

ProjectSchema.statics.getProject = (project_or_id, fields, callback)->
	if project_or_id._id?
		callback null, project_or_id
	else
		try
			concreteObjectId(project_or_id.toString())
		catch e
			return callback(new Errors.NotFoundError(e.message))
		this.findById project_or_id, fields, callback

ProjectSchema.statics.findPopulatedById = (project_id, callback)->
	logger.log project_id:project_id, "findPopulatedById"
	this.find(_id: project_id )
			.populate('collaberator_refs')
			.populate('readOnly_refs')
			.populate('owner_ref')
			.exec (err, projects)->
				if err? 
					logger.err err:err, project_id:project_id, "something went wrong looking for project findPopulatedById"
					callback(err)
				else if !projects? || projects.length == 0
					logger.err project_id:project_id, "something went wrong looking for project findPopulatedById, no project could be found"
					callback "not found"
				else
					logger.log project_id:project_id, "finished findPopulatedById"
					callback(null, projects[0])

ProjectSchema.statics.findAllUsersProjects = (user_id, requiredFields, callback)->
	this.find {owner_ref:user_id}, requiredFields, (err, projects)=>
		this.find {collaberator_refs:user_id}, requiredFields, (err, collabertions)=>
			this.find {readOnly_refs:user_id}, requiredFields, (err, readOnlyProjects)=>
				callback(err, projects, collabertions, readOnlyProjects)

countElements = (project, callback)->
	
	countFolder = (folder, cb)->
		jobs = _.map folder?.folders, (folder)->
			(asyncCb)-> countFolder folder, asyncCb
		async.series jobs, (err, results)->
			total = _.reduce results, (a, b)-> return a+b
			total += folder?.docs?.length
			total += folder?.fileRefs?.length
			cb(null, subTotal)

	countFolder project.rootFolder[0], callback

getIndexOf = (searchEntity, id)->
	length = searchEntity.length
	count = 0
	while(count < length)
		if searchEntity[count]._id+"" == id+""
			return count
		count++



applyToAllFilesRecursivly = ProjectSchema.statics.applyToAllFilesRecursivly = (folder, fun)->
	_.each folder.fileRefs, (file)->
		fun(file)
	_.each folder.folders, (folder)->
		applyToAllFilesRecursivly(folder, fun)


ProjectSchema.methods.getSafeProjectName = ->
	safeProjectName = this.name.replace(new RegExp("\\W", "g"), '_')
	return sanitize.escape(safeProjectName)

conn = mongoose.createConnection(Settings.mongo.url, server: poolSize: Settings.mongo.poolSize || 10)

Project = conn.model('Project', ProjectSchema)

mongoose.model 'Project', ProjectSchema
exports.Project = Project
exports.ProjectSchema = ProjectSchema