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

109 lines
3.7 KiB
CoffeeScript

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