Database Models

Author: MrAugu

index.js

@@ -5,6 +5,13 @@ const { promisify } = require("util");
 const readdir = promisify(require("fs").readdir);
 const klaw = require("klaw");
 const path = require("path");
+const mongoose = require("mongoose");
+const dbUrl = require("./config.js").mongo;
+
+mongoose.connect(dbUrl, {
+  useNewUrlParser: true,
+  useUnifiedTopology: true
+});
 
 class Bot extends Client {
   constructor (options) {
@@ -70,7 +77,7 @@ class Bot extends Client {
 
       for (var i = 0; i < size; i++) {
         if (fromTrue === true) {
-          from -= fromRem; 
+          from -= fromRem;
         }
 
         if (toTrue === true) {
@@ -83,11 +90,11 @@ class Bot extends Client {
     this.clean = async (client, text) => {
       if (text && text.constructor.name == "Promise") text = await text;
       if (typeof evaled !== "string") text = require("util").inspect(text, {depth: 0});
-    
+
       text = text.replace(/`/g, "`" + String.fromCharCode(8203))
         .replace(/@/g, "@" + String.fromCharCode(8203))
         .replace(client.token, "nani");
-    
+
       return text;
     };
   }
@@ -153,7 +160,7 @@ const init = async () => {
     const response = client.loadCommand(cmdFile.dir, `${cmdFile.name}${cmdFile.ext}`);
     if (response) client.logger.error(response);
   });
-    
+
   const evtFiles = await readdir("./events/");
   client.logger.log(`Loading a total of ${evtFiles.length} events.`, "log");
   evtFiles.forEach(file => {
@@ -164,7 +171,7 @@ const init = async () => {
     client.on(eventName, (...args) => event.run(...args));
     delete require.cache[require.resolve(`./events/${file}`)];
   });
-    
+
   client.levelCache = {};
   for (let i = 0; i < client.config.permLevels.length; i++) {
     const thisLevel = client.config.permLevels[i];
@@ -185,4 +192,4 @@ process.on("uncaughtException", (err) => {
   const errorMsg = err.stack.replace(new RegExp(`${__dirname}/`, "g"), "./");
   console.error("Uncaught Exception: ", errorMsg);
   process.exit(1);
-});
+});

models/message.js

@@ -0,0 +1,13 @@
+const mongoose = require("mongoose");
+
+const Schema = new mongoose.Schema({
+  thread: { type: Number },
+  message: { type: String },
+  recipient: { type: String },
+  channel: { type: String },
+  content: { type: String },
+  author: { type: String },
+  timestamp: { type: Number }
+});
+
+module.exports = mongoose.model("messages", Schema);

models/posts.js

@@ -0,0 +1,8 @@
+const mongoose = require("mongoose");
+
+const Schema = new mongoose.Schema({
+  keyword: { type: String },
+  content: { type: String }
+});
+
+module.exports = mongoose.model("snippets", Schema);

models/profile.js

@@ -0,0 +1,8 @@
+const mongoose = require("mongoose");
+
+const Schema = new mongoose.Schema({
+  "keyword": { type: String },
+  "content": { type: String }
+});
+
+module.exports = mongoose.model("tags", Schema);

models/snippet.js

@@ -0,0 +1,14 @@
+const mongoose = require("mongoose");
+
+const Schema = new mongoose.Schema({
+  id: { type: Number },
+  recipient: { type: String },
+  channel: { type: String },
+  guild: { type: String },
+  subscribedRoles: { type: Array, default: null },
+  subscribedUsers: { type: Array, default: null },
+  nsfw: { type: Boolean, default: false },
+  closed: { type: Boolean, default: false }
+});
+
+module.exports = mongoose.model("threads", Schema);

models/tags.js

@@ -0,0 +1,8 @@
+const mongoose = require("mongoose");
+
+const Schema = new mongoose.Schema({
+  id: { type: String },
+  blocked: { type: Boolean, default: false }
+});
+
+module.exports = mongoose.model("users", Schema);