import shared from '../../../common'; import _ from 'lodash'; import moment from 'moment'; import * as Tasks from '../task'; import Bluebird from 'bluebird'; import baseModel from '../../libs/baseModel'; import schema from './schema'; schema.plugin(baseModel, { // noSet is not used as updating uses a whitelist and creating only accepts specific params (password, email, username, ...) noSet: [], private: ['auth.local.hashed_password', 'auth.local.salt', '_cronSignature', '_ABtest'], toJSONTransform: function userToJSON (plainObj, originalDoc) { plainObj._tmp = originalDoc._tmp; // be sure to send down drop notifs delete plainObj.filters; return plainObj; }, }); schema.post('init', function postInitUser (doc) { shared.wrap(doc); }); function _populateDefaultTasks (user, taskTypes) { let tagsI = taskTypes.indexOf('tag'); if (tagsI !== -1) { user.tags = _.map(shared.content.userDefaults.tags, (tag) => { let newTag = _.cloneDeep(tag); // tasks automatically get _id=helpers.uuid() from TaskSchema id.default, but tags are Schema.Types.Mixed - so we need to manually invoke here newTag.id = shared.uuid(); // Render tag's name in user's language newTag.name = newTag.name(user.preferences.language); return newTag; }); } let tasksToCreate = []; if (tagsI !== -1) { taskTypes = _.clone(taskTypes); taskTypes.splice(tagsI, 1); } _.each(taskTypes, (taskType) => { let tasksOfType = _.map(shared.content.userDefaults[`${taskType}s`], (taskDefaults) => { let newTask = new Tasks[taskType](taskDefaults); newTask.userId = user._id; newTask.text = taskDefaults.text(user.preferences.language); if (newTask.notes) newTask.notes = taskDefaults.notes(user.preferences.language); if (taskDefaults.checklist) { newTask.checklist = _.map(taskDefaults.checklist, (checklistItem) => { checklistItem.text = checklistItem.text(user.preferences.language); return checklistItem; }); } return newTask.save(); }); tasksToCreate.push(...tasksOfType); }); return Bluebird.all(tasksToCreate) .then((tasksCreated) => { _.each(tasksCreated, (task) => { user.tasksOrder[`${task.type}s`].push(task._id); }); }); } function _setUpNewUser (user) { let taskTypes; let iterableFlags = user.flags.toObject(); // A/B Test 2016-09-26: Start with Armoire Enabled? if (Math.random() < 0.5) { user.flags.armoireEnabled = true; user._ABtest = '20160926-armoireEnabled'; } else { user._ABtest = '20160926-armoireDisabled'; } if (user.registeredThrough === 'habitica-web' || user.registeredThrough === 'habitica-android') { taskTypes = ['habit', 'daily', 'todo', 'reward', 'tag']; _.each(iterableFlags.tutorial.common, (val, section) => { user.flags.tutorial.common[section] = true; }); } else { taskTypes = ['todo', 'tag']; user.flags.showTour = false; _.each(iterableFlags.tour, (val, section) => { user.flags.tour[section] = -2; }); } return _populateDefaultTasks(user, taskTypes); } function _getFacebookName (fb) { if (!fb) { return; } let possibleName = fb.displayName || fb.name || fb.username; if (possibleName) { return possibleName; } if (fb.first_name && fb.last_name) { return `${fb.first_name} ${fb.last_name}`; } } function _setProfileName (user) { let google = user.auth.google; let localUsername = user.auth.local && user.auth.local.username; let googleUsername = google && google.displayName; let anonymous = 'Anonymous'; return localUsername || _getFacebookName(user.auth.facebook) || googleUsername || anonymous; } schema.pre('save', true, function preSaveUser (next, done) { next(); if (_.isNaN(this.preferences.dayStart) || this.preferences.dayStart < 0 || this.preferences.dayStart > 23) { this.preferences.dayStart = 0; } if (!this.profile.name) { this.profile.name = _setProfileName(this); } // Determines if Beast Master should be awarded let beastMasterProgress = shared.count.beastMasterProgress(this.items.pets); if (beastMasterProgress >= 90 || this.achievements.beastMasterCount > 0) { this.achievements.beastMaster = true; } // Determines if Mount Master should be awarded let mountMasterProgress = shared.count.mountMasterProgress(this.items.mounts); if (mountMasterProgress >= 90 || this.achievements.mountMasterCount > 0) { this.achievements.mountMaster = true; } // Determines if Triad Bingo should be awarded let dropPetCount = shared.count.dropPetsCurrentlyOwned(this.items.pets); let qualifiesForTriad = dropPetCount >= 90 && mountMasterProgress >= 90; if (qualifiesForTriad || this.achievements.triadBingoCount > 0) { this.achievements.triadBingo = true; } // Enable weekly recap emails for old users who sign in if (this.flags.lastWeeklyRecapDiscriminator) { // Enable weekly recap emails in 24 hours this.flags.lastWeeklyRecap = moment().subtract(6, 'days').toDate(); // Unset the field so this is run only once this.flags.lastWeeklyRecapDiscriminator = undefined; } // EXAMPLE CODE for allowing all existing and new players to be // automatically granted an item during a certain time period: // if (!this.items.pets['JackOLantern-Base'] && moment().isBefore('2014-11-01')) // this.items.pets['JackOLantern-Base'] = 5; // our own version incrementer if (_.isNaN(this._v) || !_.isNumber(this._v)) this._v = 0; this._v++; // Populate new users with default content if (this.isNew) { _setUpNewUser(this) .then(() => done()) .catch(done); } else { done(); } }); schema.pre('update', function preUpdateUser () { this.update({}, {$inc: {_v: 1}}); });