request = require("./request") _ = require("underscore") settings = require("settings-sharelatex") {db, ObjectId} = require("../../../../app/js/infrastructure/mongojs") UserModel = require("../../../../app/js/models/User").User AuthenticationManager = require("../../../../app/js/Features/Authentication/AuthenticationManager") count = 0 class User constructor: (options = {}) -> @email = "acceptance-test-#{count}@example.com" @password = "acceptance-test-#{count}-password" count++ @jar = request.jar() @request = request.defaults({ jar: @jar }) login: (callback = (error) ->) -> @ensureUserExists (error) => return callback(error) if error? @getCsrfToken (error) => return callback(error) if error? @request.post { url: "/login" json: { @email, @password } }, callback ensureUserExists: (callback = (error) ->) -> filter = {@email} options = {upsert: true, new: true, setDefaultsOnInsert: true} UserModel.findOneAndUpdate filter, {}, options, (error, user) => return callback(error) if error? AuthenticationManager.setUserPassword user._id, @password, (error) => return callback(error) if error? @id = user?._id?.toString() @_id = user?._id?.toString() @first_name = user?.first_name @referal_id = user?.referal_id callback(null, @password) logout: (callback = (error) ->) -> @getCsrfToken (error) => return callback(error) if error? @request.get { url: "/logout" json: email: @email password: @password }, (error, response, body) => return callback(error) if error? db.users.findOne {email: @email}, (error, user) => return callback(error) if error? @id = user?._id?.toString() @_id = user?._id?.toString() callback() ensure_admin: (callback = (error) ->) -> db.users.update {_id: ObjectId(@id)}, { $set: { isAdmin: true }}, callback upgradeFeatures: (callback = (error) -> ) -> features = { collaborators: -1 # Infinite versioning: true dropbox:true compileTimeout: 60 compileGroup:"priority" templates: true references: true trackChanges: true trackChangesVisible: true } db.users.update {_id: ObjectId(@id)}, { $set: { features: features }}, callback downgradeFeatures: (callback = (error) -> ) -> features = { collaborators: 1 versioning: false dropbox:false compileTimeout: 60 compileGroup:"standard" templates: false references: false trackChanges: false trackChangesVisible: false } db.users.update {_id: ObjectId(@id)}, { $set: { features: features }}, callback defaultFeatures: (callback = (error) -> ) -> features = settings.defaultFeatures db.users.update {_id: ObjectId(@id)}, { $set: { features: features }}, callback full_delete_user: (email, callback = (error) ->) -> db.users.findOne {email: email}, (error, user) => if !user? return callback() user_id = user._id db.projects.remove owner_ref:ObjectId(user_id), {multi:true}, (err)-> if err? callback(err) db.users.remove {_id: ObjectId(user_id)}, callback getProject: (project_id, callback = (error, project)->) -> db.projects.findOne {_id: ObjectId(project_id.toString())}, callback createProject: (name, options, callback = (error, oroject_id) ->) -> if typeof options == "function" callback = options options = {} @request.post { url: "/project/new", json: Object.assign({projectName: name}, options) }, (error, response, body) -> return callback(error) if error? if !body?.project_id? error = new Error("SOMETHING WENT WRONG CREATING PROJECT", response.statusCode, response.headers["location"], body) callback error else callback(null, body.project_id) deleteProject: (project_id, callback=(error)) -> @request.delete { url: "/project/#{project_id}" }, (error, response, body) -> return callback(error) if error? callback(null) openProject: (project_id, callback=(error)) -> @request.get { url: "/project/#{project_id}" }, (error, response, body) -> return callback(error) if error? if response.statusCode != 200 err = new Error("Non-success response when opening project: #{response.statusCode}") return callback(err) callback(null) createDocInProject: (project_id, parent_folder_id, name, callback=(error, doc_id)->) -> @getCsrfToken (error) => return callback(error) if error? @request.post { url: "/project/#{project_id}/doc", json: { name: name, parent_folder_id: parent_folder_id } }, (error, response, body) => callback(null, body._id) addUserToProject: (project_id, user, privileges, callback = (error, user) ->) -> if privileges == 'readAndWrite' updateOp = {$addToSet: {collaberator_refs: user._id.toString()}} else if privileges == 'readOnly' updateOp = {$addToSet: {readOnly_refs: user._id.toString()}} db.projects.update {_id: db.ObjectId(project_id)}, updateOp, (err) -> callback(err) makePublic: (project_id, level, callback = (error) ->) -> @request.post { url: "/project/#{project_id}/settings/admin", json: publicAccessLevel: level }, (error, response, body) -> return callback(error) if error? callback(null) makePrivate: (project_id, callback = (error) ->) -> @request.post { url: "/project/#{project_id}/settings/admin", json: publicAccessLevel: 'private' }, (error, response, body) -> return callback(error) if error? callback(null) makeTokenBased: (project_id, callback = (error) ->) -> @request.post { url: "/project/#{project_id}/settings/admin", json: publicAccessLevel: 'tokenBased' }, (error, response, body) -> return callback(error) if error? callback(null) getCsrfToken: (callback = (error) ->) -> @request.get { url: "/dev/csrf" }, (err, response, body) => return callback(err) if err? @csrfToken = body @request = @request.defaults({ headers: "x-csrf-token": @csrfToken }) callback() changePassword: (callback = (error) ->) -> @getCsrfToken (error) => return callback(error) if error? @request.post { url: "/user/password/update" json: currentPassword: @password newPassword1: @password newPassword2: @password }, (error, response, body) => return callback(error) if error? db.users.findOne {email: @email}, (error, user) => return callback(error) if error? callback() getUserSettingsPage: (callback = (error, statusCode) ->) -> @getCsrfToken (error) => return callback(error) if error? @request.get { url: "/user/settings" }, (error, response, body) => return callback(error) if error? callback(null, response.statusCode) getProjectListPage: (callback=(error, statusCode)->) -> @getCsrfToken (error) => return callback(error) if error? @request.get { url: "/project" }, (error, response, body) => return callback(error) if error? callback(null, response.statusCode) isLoggedIn: (callback = (error, loggedIn) ->) -> @request.get "/user/personal_info", (error, response, body) -> return callback(error) if error? if response.statusCode == 200 return callback(null, true) else if response.statusCode == 302 return callback(null, false) else return callback(new Error("unexpected status code from /user/personal_info: #{response.statusCode}")) module.exports = User