diff --git a/bin/server.js b/bin/server.js index 2c3231a4..95244b3c 100644 --- a/bin/server.js +++ b/bin/server.js @@ -19,8 +19,8 @@ async function startServer () { // Connect to the Mongo Database. mongoose.Promise = global.Promise - await mongoose.connect(config.database, { useNewUrlParser: true }) mongoose.set('useCreateIndex', true) // Stop deprecation warning. + await mongoose.connect(config.database, { useNewUrlParser: true }) // MIDDLEWARE START diff --git a/util/README.md b/util/README.md new file mode 100644 index 00000000..31994a6d --- /dev/null +++ b/util/README.md @@ -0,0 +1 @@ +This directory contains utility functions for managing the database. diff --git a/util/users/createUsers.js b/util/users/createUsers.js new file mode 100644 index 00000000..52c36d8a --- /dev/null +++ b/util/users/createUsers.js @@ -0,0 +1,36 @@ +const mongoose = require('mongoose') + +const config = require('../../config') + +const USERNAME = 'test' +const PASSWORD = 'pass' + +// Connect to the Mongo Database. +mongoose.Promise = global.Promise + +async function addUser () { + await mongoose.connect(config.database) + + const User = require('../../src/models/users') + + const userData = { + username: USERNAME, + password: PASSWORD + } + + const user = new User(userData) + + // Enforce default value of 'user' + user.type = 'user' + + await user.save() + + await mongoose.connection.close() + + console.log(`User ${USERNAME} created.`) +} +addUser() + +module.exports = { + addUser +} diff --git a/util/users/getUsers.js b/util/users/getUsers.js new file mode 100644 index 00000000..cd00d31c --- /dev/null +++ b/util/users/getUsers.js @@ -0,0 +1,18 @@ +const mongoose = require('mongoose') + +const config = require('../../config') + +const User = require('../../src/models/users') + +async function getUsers () { + // Connect to the Mongo Database. + mongoose.Promise = global.Promise + mongoose.set('useCreateIndex', true) // Stop deprecation warning. + await mongoose.connect(config.database, { useNewUrlParser: true }) + + const users = await User.find({}, '-password') + console.log(`users: ${JSON.stringify(users, null, 2)}`) + + mongoose.connection.close() +} +getUsers() diff --git a/util/wipe-db.js b/util/wipe-db.js new file mode 100644 index 00000000..4340e5d1 --- /dev/null +++ b/util/wipe-db.js @@ -0,0 +1,34 @@ +const mongoose = require('mongoose') + +const config = require('../config') + +// Connect to the Mongo Database. +mongoose.Promise = global.Promise +mongoose.connect(config.database, () => { + // mongoose.connection.db.dropDatabase() +}) + +console.log(`config: ${JSON.stringify(config, null, 2)}`) + +/* +// Wipe the DB. +function cleanDb () { + for (const collection in mongoose.connection.collections) { + if (mongoose.connection.collections.hasOwnProperty(collection)) { + mongoose.connection.collections[collection].remove() + } + } + console.log(`Database wiped.`) +} +cleanDb() +*/ + +mongoose.connection.close() + +console.log(` +Here's how to wipe the db: +1. mongo +2. use p2pvps-server-dev +3. db.dropDatabase() +4. exit +`)