inventory command handlers

commands/inventory-add.js

@@ -0,0 +1,160 @@
+const { MessageEmbed } = require('discord.js');
+
+module.exports = {
+  name: "inventory-add",
+  description: "add item to inventory",
+  usage: "[item]",
+  permissions: {
+    channel: ["VIEW_CHANNEL", "SEND_MESSAGES", "EMBED_LINKS"],
+    member: [],
+  },
+  SlashCommand: {
+    options: [
+      {
+        name: "firearm",
+        description: "add firearm to inventory",
+        value: "firearm",
+        type: 1,
+        options: [
+          {
+            name: "type",
+            description: "type of firearm",
+            value: "type",
+            type: 3,
+            required: true,
+            choices: [
+              { name: 'Pistol', value: 'Pistol' }, { name: 'Revolver', value: 'Revolver' }, { name: 'Shotgun', value: 'Shotgun' },
+              { name: 'Submachine Gun', value: 'Submachine Gun' }, { name: 'Light Machine Gun', value: 'Ligh Machine Gun' },
+              { name: 'Assault Rifle', value: 'Assault Rifle' }, { name: 'Sniper Rifle', value: 'Sniper Rifle' },
+            ]
+          },
+          {
+            name: "serialnumber",
+            description: "firearm serial number",
+            value: "serialnumber",
+            type: 3,
+            required: true,
+          },
+          {
+            name: "owner",
+            description: "Firearm Owner's Full Name",
+            value: "owner",
+            type: 3,
+            required: true,
+          },
+        ]
+      },
+      {
+        name: "generic",
+        description: "add a generic item to inventory",
+        value: "generic",
+        type: 1,
+        options: [
+          {
+            name: "name",
+            description: "item name",
+            value: "name",
+            type: 3,
+            required: true,
+          },
+          {
+            name: "description",
+            description: "item description",
+            value: "description",
+            type: 3,
+            required: true,
+          },
+        ]
+      }
+    ],  
+    /**
+     *
+     * @param {require("../structures/QuarksBot")} client
+     * @param {import("discord.js").MessageCreate} message
+     * @param {string[]} args
+     * @param {*} param3
+    */
+    run: async (client, interaction, args, { GuildDB }) => {
+      if (GuildDB.customChannelStatus==true&&!GuildDB.allowedChannels.includes(interaction.channel_id)) {
+        return interaction.send(`You are not allowed to use the bot in this channel.`);
+      }
+      
+      let inventory = await client.dbo.collection("inventories").findOne({"inventory.userID":interaction.member.user.id}).then(inventory => inventory.inventory);
+
+      if (!inventory) {
+        inventory = {
+          userID: interaction.member.user.id,
+          items: [{
+            name: 'Wallet',
+            type: 'wallet',
+            description: '`/inventory wallet` to view wallet',
+            cash: 0,
+            ID: {
+              firstname: '',
+              lastname: '',
+              dob: ''
+            }
+          }]
+        }
+    
+        // Register inventory for user  
+        let newInventory = new Inventory();
+        newInventory.createInventory(interaction.member.user.id);
+        newInventory.save()
+          .catch(err => {
+            if (err) {
+              client.log(err);
+              const channel = interaction.guild.channels.cache.find(channel => channel.id === GuildDB.tweetChannel);
+              return client.sendError(channel, err);
+            }
+          });
+      }
+
+      if (args[0].name == 'firearm') {
+        // Add firearm to inventory
+
+        let firearm = {
+          type: 'firearm',
+          weaponType: args[0].options[0].value,
+          serial: args[0].options[1].value,
+          ownerName: args[0].options[2].value
+        }
+
+        client.dbo.collection('inventories').updateOne({'inventory.userID': interaction.member.user.id}, {$push: {'inventory.items': firearm}}, function(err, res) {
+          if (err) {
+            client.log(err);
+            let embed = new MessageEmbed()
+              .setTitle(err)
+              .setColor("RED")
+
+            return interaction.send({ embeds: [embed] });
+          }
+
+          return interaction.send(`Added ${firearm.weaponType} to inventory`);
+        });
+
+      } else {
+        // Add generic item to inventory
+
+        let item = {
+          type: 'generic',
+          name: args[0].options[0].value,
+          description: args[0].options[1].value
+        }
+
+        client.dbo.collection('inventories').updateOne({'inventory.userID': interaction.member.user.id}, {$push: {'inventory.items': item}}, function(err, res) {
+          if (err) {
+            client.log(err);
+            let embed = new MessageEmbed()
+              .setTitle(err)
+              .setColor("RED")
+
+            return interaction.send({ embeds: [embed] });
+          }
+
+          return interaction.send(`Added ${item.name} to inventory`);
+        });
+      }
+    },
+  },
+}

commands/inventory.js

@@ -0,0 +1,300 @@
+const { MessageEmbed } = require('discord.js');
+const Inventory = require('../structures/inventory');
+
+const ItemOption = {
+  name: "item-number",
+  description: "item to interact with",
+  value: 1,
+  type: 4,
+  required: false,
+}
+
+module.exports = {
+  name: "inventory",
+  description: "Manage your inventory",
+  usage: "[command]",
+  permissions: {
+    channel: ["VIEW_CHANNEL", "SEND_MESSAGES", "EMBED_LINKS"],
+    member: [],
+  },
+  SlashCommand: {
+    options: [
+      {
+        name: "list",
+        description: "View your inventory",
+        value: "list",
+        type: 1,
+      },
+      {
+        name: "wallet",
+        description: "View your wallet",
+        value: "wallet",
+        type: 1,
+      },
+      {
+        name: "item",
+        description: "view an item in your inventory",
+        value: "item",
+        type: 1,
+        options: [
+          {
+            name: "item-number",
+            description: "item to interact with",
+            value: 1,
+            type: 4,
+            required: true,
+          },
+        ],
+      },
+      {
+        name: "give",
+        description: "give user item from inventory",
+        value: "give",
+        type: 1,
+        options: [
+          {
+            name: "item-number",
+            description: "item to interact with",
+            value: 1,
+            type: 4,
+            required: true,
+          },
+          {
+            name: "user",
+            description: "user to give item",
+            value: "user",
+            type: 6,
+            required: true,
+          }
+        ],
+      },
+      {
+        name: "drop",
+        description: "drop item from inventory",
+        value: "drop",
+        type: 1,
+        options: [
+          {
+            name: "item-number",
+            description: "item to interact with",
+            value: 1,
+            type: 4,
+            required: true,
+          }
+        ],
+      },
+    ],
+    /**
+     *
+     * @param {require("../structures/QuarksBot")} client
+     * @param {import("discord.js").MessageCreate} message
+     * @param {string[]} args
+     * @param {*} param3
+    */
+    run: async (client, interaction, args, { GuildDB }) => {
+      if (GuildDB.customChannelStatus==true&&!GuildDB.allowedChannels.includes(interaction.channel_id)) {
+        return interaction.send(`You are not allowed to use the bot in this channel.`);
+      }
+
+      let inventory = await client.dbo.collection("inventories").findOne({"inventory.userID":interaction.member.user.id}).then(inventory => inventory);
+
+      if (!inventory) {
+        inventory = {
+          userID: interaction.member.user.id,
+          items: [{
+            name: 'Wallet',
+            type: 'wallet',
+            description: '`/inventory wallet` to view wallet',
+            cash: 0,
+            ID: {
+              firstname: '',
+              lastname: '',
+              dob: ''
+            }
+          }]
+        }
+    
+        // Register inventory for user  
+        let newInventory = new Inventory();
+        newInventory.createInventory(interaction.member.user.id);
+        newInventory.save()
+          .catch(err => {
+            if (err) {
+              client.log(err);
+              let embed = new MessageEmbed()
+                .setTitle("Please provide a valid item number")
+                .setColor("RED")
+
+              return interaction.send({ embeds: [embed] });
+            }
+          });
+      } else inventory = inventory.inventory
+
+      if (args[0].name == 'list') {
+        // Display Inventory
+      
+        let inventoryEmbed = new MessageEmbed()
+        .setColor(client.config.EmbedColor)
+        .setTitle('Your Inventory')
+        
+        let description = `To see detials on an Item use \`/inventory item [item number]\`\n\n\`\`\``
+        for (let i = 0; i < inventory.items.length; i++) {
+          description += `${i+1}. ${inventory.items[i].name} - ${inventory.items[i].description}\n`
+        }
+        description += "```"
+
+        inventoryEmbed.setDescription(description);
+
+        return interaction.send({ embeds: [inventoryEmbed] });
+      
+      } else if (args[0].name == 'wallet') {
+        // Display wallet
+        
+        let item = inventory.items[0];
+        let walletData = `**Cash:** $${item.cash}\n**ID:** ${item.ID.firstname ? client.exists(item.ID.firtname) : 'John'} ${item.ID.lastname ? client.exists(item.ID.lastname) : 'Doe'} | ${item.ID.dob ? client.exists(item.ID.dob) : 'N/A'}`
+
+        let walletEmbed = new MessageEmbed()
+          .setTitle(`Your Wallet`)
+          .setDescription(walletData);
+
+        return interaction.send({ embeds: [walletEmbed] });
+
+      } else if (args[0].name == 'item') {
+        // Display Item Data
+        
+        if (args[0].options[0].value < 1 || args[0].options[0].value > inventory.items.length) {
+          let embed = new MessageEmbed()
+            .setTitle("Please provide a valid item number")
+            .setColor("RED")
+
+          return interaction.send({ embeds: [embed] });
+        }
+
+        let item = inventory.items[args[0].options[0].value-1];
+        let itemEmbed = new MessageEmbed().setColor(client.config.EmbedColor);
+        if (item.type == 'wallet') {
+          // Display wallet
+        
+          let walletData = `**Cash:** $${item.cash}\n**ID:** ${item.ID.firstname ? client.exists(item.ID.firtname) : 'John'} ${item.ID.lastname ? client.exists(item.ID.lastname) : 'Doe'} | ${item.ID.dob ? client.exists(item.ID.dob) : 'N/A'}`
+
+          itemEmbed.setTitle(`Your Wallet`);
+          itemEmbed.setDescription(walletData);
+
+        } else if (item.type == 'firearm') {
+          // Display firearm
+          
+          itemEmbed.setTitle('Firearm');
+          itemEmbed.addFields(
+            { name: 'Serial #', value: item.serial, inline: false },
+            { name: 'Type', value: item.weaponType, inline: false },
+            { name: 'Owner', value: item.ownerName, inline: false }
+          );
+
+        } else {
+          // Display generic item here
+
+          itemEmbed.setTitle(`Item information`);
+          itemEmbed.addFields({ name: item.name, value: item.description, inline: false });
+        }
+
+        return interaction.send({ embeds: [itemEmbed] });
+
+      } else if (args[0].name == 'drop') {
+        // Remove Item from inventory
+        
+        if (args[0].options[0].value < 1 || args[0].options[0].value > inventory.items.length) {
+          let embed = new MessageEmbed()
+            .setTitle("Please provide a valid item number")
+            .setColor("RED")
+
+          return interaction.send({ embeds: [embed] });
+        }
+
+        if (args[0].options[0].value == 1) {
+          let embed = new MessageEmbed()
+            .setTitle("Invalid")
+            .setDescription("You're not allowed to drop your wallet")
+            .setColor(client.config.EmbedColor);
+
+          return interaction.send({ embeds: [embed] });
+        }
+
+        let item = inventory.items[args[0].options[0].value-1];
+
+        client.dbo.collection('inventories').updateOne({'inventory.userID': interaction.member.user.id}, {$pull: {'inventory.items': item}}, function(err, res) {
+          if (err) {
+            client.log(err);
+            let embed = new MessageEmbed()
+              .setTitle(err)
+              .setColor("RED")
+
+            return interaction.send({ embeds: [embed] });
+          }
+
+          return interaction.send(`Removed ${item.name} from inventory`);
+        });
+
+      } else if (args[0].name == 'give') {
+        // Give user Item from inventory
+        
+        if (args[0].options[0].value < 1 || args[0].options[0].value > inventory.items.length) {
+          let embed = new MessageEmbed()
+            .setTitle("Please provide a valid item number")
+            .setColor("RED")
+
+          return interaction.send({ embeds: [embed] });
+        }
+
+        let item = inventory.items[args[0].options[0].value-1];
+        let targetUserID = args[0].options[1].value.replace('<@!', '').replace('>', '');
+        let targetUserInventory = await client.dbo.collection("inventories").findOne({"inventory.userID":targetUserID}).then(user => user);
+
+        if (!targetUserInventory) {
+          // Register inventory for user with item givn  
+          let newInventory = new Inventory();
+          newInventory.createInventory(targetUserID);
+          newInventory.addItem(item);
+          newInventory.save()
+            .catch(err => {
+              if (err) {
+                client.log(err);
+                let embed = new MessageEmbed()
+                  .setTitle("Please provide a valid item number")
+                  .setColor("RED")
+
+                return interaction.send({ embeds: [embed] });
+              }
+            });
+        } else {
+          client.dbo.collection('inventories').updateOne({'inventory.userID': interaction.member.user.id}, {$pull: {'inventory.items': item}}, function(err, res) {
+            if (err) {
+              client.log(err);
+              let embed = new MessageEmbed()
+                .setTitle("Please provide a valid item number")
+                .setColor("RED")
+
+              return interaction.send({ embeds: [embed] });
+            }
+          });
+          client.dbo.collection('inventories').updateOne({'inventory.userID': targetUserID}, {$push: {'inventory.items': item}}, function(err, res) {
+            if (err) {
+              client.log(err);
+              let embed = new MessageEmbed()
+                .setTitle("Please provide a valid item number")
+                .setColor("RED")
+
+              return interaction.send({ embeds: [embed] });
+            }
+          });
+        }
+
+        let successEmbed = new MessageEmbed()
+          .setTitle('Success')
+          .setDescription(`Successfully gave <@${targetUserID}> ${item.name}`)
+          .setColor('GREEN');
+
+        return interaction.send({ embeds: [successEmbed] });
+      }
+    },
+  },
+}

config/config.js

@@ -1,5 +1,5 @@
 module.exports = {
-	Dev: "PRODUCTION",
+	Dev: "PROD",
 	Version: "0.1.0",
   Admins: ["362791661274660874"], // Admins of the bot
 	DefaultPrefix: "?",

package.json

@@ -14,7 +14,10 @@
   "author": "Braeden Sowinski",
   "license": "ISC",
   "dependencies": {
+    "colors": "^1.4.0",
     "discord.js": "^13.8.1",
-    "mongodb": "^4.7.0"
+    "mongodb": "^4.7.0",
+    "mongoose": "^6.4.2",
+    "winston": "^3.8.1"
   }
 }

structures/QuarksBot.js

@@ -1,5 +1,6 @@
 const { Collection, Client, MessageEmbed } = require('discord.js');
 const MongoClient = require('mongodb').MongoClient;
+const mongoose = require('mongoose');
 const Logger = require("../util/Logger");
 const path = require("path");
 const fs = require('fs');
@@ -61,8 +62,11 @@ class QuarksBot extends Client {
   }
 
   async connectMongo(mongoURI, dbo) {
+    // Connect to MongoDB database.
     this.db = await MongoClient.connect(mongoURI);
     this.dbo = this.db.db(dbo);
+    
+    mongoose.connect(`${mongoURI}/${dbo}`);
     this.log('Successfully connected to mongoDB');
   }
 

structures/inventory.js

@@ -0,0 +1,35 @@
+const mongoose = require('mongoose');
+
+let inventorySchema = mongoose.Schema({
+  inventory: {
+      userID: String,
+      items: [],
+      createdAt: Date,
+      updatedAt: Date
+  }
+});
+
+inventorySchema.methods.createInventory = function (userID) {
+    this.inventory.userID = userID;
+    this.inventory.items = [];
+    let walletData = {
+      name: 'Wallet',
+      type: 'wallet',
+      description: '`/inventory wallet` to view wallet',
+      cash: 0.00,
+      ID: {
+        firstname: '',
+        lastname: '',
+        dob: ''
+      }
+    }
+    this.inventory.items.push(walletData);
+    this.inventory.createdAt = new Date();
+    this.inventory.updatedAt = this.inventory.createdAt;
+};
+
+inventorySchema.methods.addItem = function (item) {
+  this.inventory.items.push(item);
+};
+
+module.exports = mongoose.model('Inventory', inventorySchema);

util/RegisterSlashCommands.js

@@ -28,6 +28,7 @@ module.exports = (client, guild) => {
       try {
         await GuildAPI.commands.post({ data: dataStuff });
       } catch (e) {
+        if (client.config.Dev == 'DEV') console.log(e);
         client.log('Error: API missing permissions, re-invite the bot');
 
         // Forces bot to leave server