From 79b35966851c52552fcaa11594dbc07e1a18bae2 Mon Sep 17 00:00:00 2001
From: Chris Troutner <chris.troutner@gmail.com>
Date: Sun, 3 Feb 2019 17:33:47 -0800
Subject: [PATCH] Added util dir

---
 bin/server.js             |  2 +-
 util/README.md            |  1 +
 util/users/createUsers.js | 36 ++++++++++++++++++++++++++++++++++++
 util/users/getUsers.js    | 18 ++++++++++++++++++
 util/wipe-db.js           | 34 ++++++++++++++++++++++++++++++++++
 5 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 util/README.md
 create mode 100644 util/users/createUsers.js
 create mode 100644 util/users/getUsers.js
 create mode 100644 util/wipe-db.js

diff --git a/bin/server.js b/bin/server.js
index 2c3231a..95244b3 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 0000000..31994a6
--- /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 0000000..52c36d8
--- /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 0000000..cd00d31
--- /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 0000000..4340e5d
--- /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
+`)