-
Notifications
You must be signed in to change notification settings - Fork 4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'develop' into negue/fix/roundTaskDifficulty
- Loading branch information
Showing
536 changed files
with
16,823 additions
and
15,116 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,117 @@ | ||
import each from 'lodash/each'; | ||
import keys from 'lodash/keys'; | ||
import content from '../../website/common/script/content/index'; | ||
const migrationName = 'full-stable.js'; | ||
const authorName = 'Sabe'; // in case script author needs to know when their ... | ||
const authorUuid = '7f14ed62-5408-4e1b-be83-ada62d504931'; // ... own data is done | ||
|
||
/* | ||
* Award users every extant pet and mount | ||
*/ | ||
const connectionString = 'mongodb://localhost:27017/habitrpg?auto_reconnect=true'; // FOR TEST DATABASE | ||
|
||
let monk = require('monk'); | ||
let dbUsers = monk(connectionString).get('users', { castIds: false }); | ||
|
||
function processUsers (lastId) { | ||
// specify a query to limit the affected users (empty for all users): | ||
let query = { | ||
'profile.name': 'SabreCat', | ||
}; | ||
|
||
if (lastId) { | ||
query._id = { | ||
$gt: lastId, | ||
}; | ||
} | ||
|
||
dbUsers.find(query, { | ||
sort: {_id: 1}, | ||
limit: 250, | ||
fields: [ | ||
], // specify fields we are interested in to limit retrieved data (empty if we're not reading data): | ||
}) | ||
.then(updateUsers) | ||
.catch((err) => { | ||
console.log(err); | ||
return exiting(1, `ERROR! ${ err}`); | ||
}); | ||
} | ||
|
||
let progressCount = 1000; | ||
let count = 0; | ||
|
||
function updateUsers (users) { | ||
if (!users || users.length === 0) { | ||
console.warn('All appropriate users found and modified.'); | ||
displayData(); | ||
return; | ||
} | ||
|
||
let userPromises = users.map(updateUser); | ||
let lastUser = users[users.length - 1]; | ||
|
||
return Promise.all(userPromises) | ||
.then(() => { | ||
processUsers(lastUser._id); | ||
}); | ||
} | ||
|
||
function updateUser (user) { | ||
count++; | ||
let set = { | ||
migration: migrationName, | ||
}; | ||
|
||
each(keys(content.pets), (pet) => { | ||
set[`items.pets.${pet}`] = 5; | ||
}); | ||
each(keys(content.premiumPets), (pet) => { | ||
set[`items.pets.${pet}`] = 5; | ||
}); | ||
each(keys(content.questPets), (pet) => { | ||
set[`items.pets.${pet}`] = 5; | ||
}); | ||
each(keys(content.specialPets), (pet) => { | ||
set[`items.pets.${pet}`] = 5; | ||
}); | ||
each(keys(content.mounts), (mount) => { | ||
set[`items.mounts.${mount}`] = true; | ||
}); | ||
each(keys(content.premiumMounts), (mount) => { | ||
set[`items.mounts.${mount}`] = true; | ||
}); | ||
each(keys(content.questMounts), (mount) => { | ||
set[`items.mounts.${mount}`] = true; | ||
}); | ||
each(keys(content.specialMounts), (mount) => { | ||
set[`items.mounts.${mount}`] = true; | ||
}); | ||
|
||
dbUsers.update({_id: user._id}, {$set: set}); | ||
|
||
if (count % progressCount === 0) console.warn(`${count } ${ user._id}`); | ||
if (user._id === authorUuid) console.warn(`${authorName } processed`); | ||
} | ||
|
||
function displayData () { | ||
console.warn(`\n${ count } users processed\n`); | ||
return exiting(0); | ||
} | ||
|
||
function exiting (code, msg) { | ||
code = code || 0; // 0 = success | ||
if (code && !msg) { | ||
msg = 'ERROR!'; | ||
} | ||
if (msg) { | ||
if (code) { | ||
console.error(msg); | ||
} else { | ||
console.log(msg); | ||
} | ||
} | ||
process.exit(code); | ||
} | ||
|
||
module.exports = processUsers; |
Oops, something went wrong.