diff --git a/migrations/users/full-gear.js b/migrations/users/full-gear.js new file mode 100644 index 0000000000..e47b6f8539 --- /dev/null +++ b/migrations/users/full-gear.js @@ -0,0 +1,64 @@ +/* eslint-disable no-console */ +import each from 'lodash/each'; +import keys from 'lodash/keys'; +import content from '../../website/common/script/content/index'; + +import { model as User } from '../../website/server/models/user'; + +const MIGRATION_NAME = 'full-gear'; + +const progressCount = 1000; +let count = 0; + +/* + * Award users every extant pet and mount + */ + +async function updateUser (user) { + count += 1; + + const set = {}; + + set.migration = MIGRATION_NAME; + + each(keys(content.gear.flat), gearItem => { + set[`items.gear.owned.${gearItem}`] = true; + }); + + if (count % progressCount === 0) console.warn(`${count} ${user._id}`); + + return User.update({ _id: user._id }, { $set: set }).exec(); +} + +export default async function processUsers () { + const query = { + migration: { $ne: MIGRATION_NAME }, + 'auth.local.lowerCaseUsername': 'olson1', + }; + + const fields = { + _id: 1, + }; + + while (true) { // eslint-disable-line no-constant-condition + const users = await User // eslint-disable-line no-await-in-loop + .find(query) + .limit(250) + .sort({ _id: 1 }) + .select(fields) + .lean() + .exec(); + + if (users.length === 0) { + console.warn('All appropriate users found and modified.'); + console.warn(`\n${count} users processed\n`); + break; + } else { + query._id = { + $gt: users[users.length - 1], + }; + } + + await Promise.all(users.map(updateUser)); // eslint-disable-line no-await-in-loop + } +}