fix up migration to only save clean fields

This commit is contained in:
Tyler Renelle 2013-02-04 20:12:48 -05:00
parent ca907a75df
commit 629ba2df79
2 changed files with 94 additions and 51 deletions

View file

@ -1,56 +1,99 @@
db.users.find({}).forEach(function(user){
// %mongo server:27017/dbname underscore.js my_commands.js
// %mongo server:27017/dbname underscore.js --shell
//db.users.find({'auth.facebook.email': 'tylerrenelle@gmail.com'}).forEach(function(user){
db.users.find().forEach(function(user){
if (!user._id) {
print("User has null _id");
return; // need to figure out how to delete these buggers if they don't have an id to delete from
}
if (user._id.indexOf("$") === 0) {
print("User id starts with $ (" + user._id + ")")
return;
}
// even though we're clobbering user later, sometimes these are undefined and crash the script
// this saves us some ternaries
user.stats = user.stats || {};
user.items = user.items || {};
user.preferences = user.preferences || {};
user.notifications = user.notifications || {};
user.flags = user.flags || {};
user.habitIds = user.habitIds || [];
user.dailyIds = user.dailyIds || [];
user.todoIds = user.todoIds || [];
user.rewardIds = user.rewardIds|| [];
_.each(user.tasks, function(task, key){
if (!task.type) {
delete user.tasks[key];
// idList will take care of itself on page-load
return
}
if (key == '$spec') {
print("$spec was found: " + user._id);
return
}
if (key.indexOf("$_") === 0) {
var newKey = key.replace("$_", ''),
index = user[task.type + "Ids"].indexOf(key)
user[task.type + "Ids"][index] = newKey;
task.id = newKey
user.tasks[newKey] = task
// TODO make sure this is ok, that we're not deleting the original
// Otherwise use lodash.cloneDeep
delete user.tasks[key]
}
});
// New user schema has public and private paths, so we can setup proper access control with racer
// Note 'public' and 'private' are reserved words
var newUser = {
pub:{},
priv: {}
auth: user.auth, // we need this top-level due to derby-auth
pub:{
party: null,
invitations: [],
items: {
armor: user.items.armor || 0,
weapon: user.items.weapon || 0
},
stats: {
gp: user.stats.money || 0,
hp: user.stats.hp || 50,
exp: user.stats.exp || 0,
lvl: user.stats.lvl || 1
},
preferences: {
armorSet: user.preferences.armorSet || 'v1',
gender: user.preferences.gender || 'm'
}
},
priv: {
balance: user.balance || 2,
lastCron: user.lastCron || +new Date,
tasks: user.tasks || {},
history: user.history || [],
apiToken: user.preferences.api_token || null, // set on update, we need derby.uuid()
idLists: {
habit:user.habitList || [],
daily:user.dailyList || [],
todo:user.todoList || [],
reward:user.rewardList || []
},
flags: {
partyEnabled: false,
itemsEnabled: user.items.itemsEnabled || false,
kickstarter: user.notifications.kickstarter || 'show',
ads: user.flags.ads || null // null because it's set on registration
}
}
};
newUser.priv.lastCron = user.lastCron;
user.priv.balance = user.balance;
user.priv.tasks = user.tasks;
// ------------ Stats
// rename money to gp
user.stats.gp = user.stats.money;
delete user.stats.money;
newUser.pub.stats = user.stats;
// ------------ Party
newUser.pub.party = null;
// ------------ ID lists
newUser.priv.idLists = {habit:user.habitList, daily:user.dailyList, todo:user.todoList, reward:user.rewardList};
// ------------ Flags
newUser.priv.flags = user.flags || {};
user.priv.partyEnabled = false;
user.priv.flags.itemsEnabled = user.items.itemsEnabled;
delete user.items.itemsEnabled;
// Items
user.pub.items = user.items;
// kickstarter
if (!user.notifications || !user.notifications.kickstarter) user.notifications = {kickstarter:'show'}
newUser.priv.flags.kickstarter = user.notifications.kickstarter;
// Ads
newUser.priv.flags.ads = user.flags.ads;
// ------------ API Token
newUser.priv.apiToken = user.preferences.api_token;
delete user.preferences.api_token;
// ------------ Preferences
newUser.pub.preferences = user.preferences;
db.users.update({_id:user.id}, newUser, {multi:true});
try {
db.users.update({_id:user._id}, newUser);
} catch(e) {
print(e);
}
})

View file

@ -8,8 +8,8 @@ userSchema =
# _id
pub:
stats: { gp: 0, exp: 0, lvl: 1, hp: 50 }
# party: null
# TODO invitations
party: null
invitations: []
items: { armor: 0, weapon: 0 }
preferences: { gender: 'm', armorSet: 'v1' }
priv: