42 lines
998 B
JavaScript
42 lines
998 B
JavaScript
const mongoose = require('mongoose');
|
|
|
|
let userSchema = mongoose.Schema({
|
|
user: {
|
|
userID: String,
|
|
guilds: {}
|
|
},
|
|
});
|
|
|
|
userSchema.methods.createUser = function (userID, guildID, startingBalance) {
|
|
this.user.userID = userID;
|
|
this.user.guilds = {};
|
|
this.user.guilds[guildID] = {
|
|
balance: startingBalance,
|
|
lastIncome: new Date('2000-01-01T00:00:00'),
|
|
};
|
|
};
|
|
|
|
/*
|
|
This function is to add a new guild specific user to an already existing
|
|
user document
|
|
or
|
|
can be used to reset a data back to default
|
|
*/
|
|
function addUser(guilds, guildID, userID, client, startingBalance) {
|
|
let updatedGuilds = guilds;
|
|
updatedGuilds[guildID] = {
|
|
balance: startingBalance,
|
|
lastIncome: new Date('2000-01-01T00:00:00')
|
|
}
|
|
|
|
client.dbo.collection("users").updateOne({"user.userID":userID}, {$set: {"user.guilds": updatedGuilds}}, (err, res) => {
|
|
if (err) return false
|
|
})
|
|
return true
|
|
}
|
|
|
|
module.exports = {
|
|
User: mongoose.model('User', userSchema),
|
|
addUser: addUser,
|
|
};
|