catto.js 0.1.4 → 0.1.6

package/Bot.js

@@ -1,6 +1,7 @@
 var events = require("events");
 var Discord = require("discord.js");
 var User = require("./User");
+var MessageBuilder = require("./MessageBuilder");
 if (typeof EventEmitter !== "undefined") {} else {
   var { EventEmitter } = events;
 }
@@ -23,8 +24,9 @@ module.exports = class extends EventEmitter {
         }
       });
     }
-    this.slashCommands = new Map();
+    this.buttons = new Map();
     this.commands = new Map();
+    this.slashCommands = new Map();
     this.client.on("ready", () => {
       var cmds = [];
       for (var cmd of this.slashCommands.values()) {
@@ -130,6 +132,8 @@ module.exports = class extends EventEmitter {
           try {
             command.execute({
               Discord,
+              User,
+              MessageBuilder,
               interaction,
               "cmd": command.name,
               "bot": this
@@ -139,6 +143,26 @@ module.exports = class extends EventEmitter {
           }
         }
       }
+      if (interaction.isButton()) {
+        interaction.user = new User(interaction.user, this);
+        if (interaction.member) {
+          interaction.member.user = new User(interaction.member.user, this);
+        }
+        var button = this.buttons.get(interaction.customId);
+        if (button) {
+          try {
+            button.execute({
+              Discord,
+              User,
+              MessageBuilder,
+              interaction,
+              "bot": this
+            });
+          } catch(e) {
+            console.error(e);
+          }
+        }
+      }
     });
     this.client.on("messageCreate", message => {
       message.author = new User(message.author, this);
@@ -152,6 +176,8 @@ module.exports = class extends EventEmitter {
         try {
           command.execute({
             Discord,
+            User,
+            MessageBuilder,
             message,
             cmd,
             args,

package/MessageBuilder.js

@@ -0,0 +1,94 @@
+var Discord = require("discord.js");
+module.exports = class MessageBuilder {
+  constructor(bot) {
+    this.bot = bot;
+    this.data = {};
+  }
+  text(t) {
+    this.data.content += t;
+    return this;
+  }
+  link(text, link) {
+    this.data.content += `[${text}](${link})`;
+    return this;
+  }
+  button(basic, execute) {
+    var btn = new Discord.ButtonBuilder();
+    if (!basic.id && !basic.url) {
+      throw new Error("All buttons except URL must have ID.");
+    }
+    btn.setCustomId(basic.id);
+    btn.setDisabled(basic.disabled);
+    if (basic.emoji) {
+      btn.setEmoji(basic.emoji);
+    }
+    if (basic.text) {
+      btn.setLabel(basic.text);
+    }
+    if (!basic.emoji && !basic.text) {
+      throw new Error("Can't have an empty button.");
+    }
+    var colorlib = {
+      "blue": "Primary",
+      "gray": "Secondary",
+      "lime": "Success",
+      "red": "Danger"
+    };
+    basic.color = basic.color.toLowerCase();
+    if (!colorlib[basic.color] && !basic.url) {
+      throw new Error("Unknown color.");
+    }
+    if (basic.url) {
+      btn.setStyle(Discord.ButtonStyle.Link);
+      btn.setURL(basic.url);
+    } else {
+      btn.setStyle(Discord.ButtonStyle[colorlib[basic.color]]);
+    }
+    if (!this.data.components) {
+      this.data.components = [];
+    }
+    if (!this.data.components[0]) {
+      this.data.components[0] = new Discord.ActionRowBuilder();
+    }
+    if (this.data.components[0].components.length < 5) {
+      this.data.components[0].addComponents(btn);
+    } else {
+      if (!this.data.components[1]) {
+        this.data.components[1] = new Discord.ActionRowBuilder();
+      }
+      if (this.data.components[1].components.length < 5) {
+        this.data.components[1].addComponents(btn);
+      } else {
+        if (!this.data.components[2]) {
+          this.data.components[2] = new Discord.ActionRowBuilder();
+        }
+        if (this.data.components[2].components.length < 5) {
+          this.data.components[2].addComponents(btn);
+        } else {
+          if (!this.data.components[3]) {
+            this.data.components[3] = new Discord.ActionRowBuilder();
+          }
+          if (this.data.components[3].components.length < 5) {
+            this.data.components[3].addComponents(btn);
+          } else {
+            if (!this.data.components[4]) {
+              this.data.components[4] = new Discord.ActionRowBuilder();
+            }
+            if (this.data.components[4].components.length < 5) {
+              this.data.components[4].addComponents(btn);
+            } else {
+              throw new Error("Can't have more than 25 buttons in a single message.");
+            }
+          }
+        }
+      }
+    }
+    if (!bot.url) {
+      if (bot.buttons.get(basic.id)) {
+        throw new Error("ID must be unique.");
+      }
+      bot.buttons.set(basic.id, execute);
+    }
+    return this;
+  }
+};

package/index.js

@@ -1,5 +1,5 @@
 var mod = {};
-["random", "Server", "HTML", "request", "AuthClient", "utils", "GitHub", "Base64", "User", "Bitfield", "Flags", "Bot", "Application"].forEach(part => {
+["random", "Server", "HTML", "request", "AuthClient", "utils", "GitHub", "Base64", "User", "Bitfield", "Flags", "Bot", "Application", "MessageBuilder"].forEach(part => {
   mod[part] = require(`./${part}`);
 });
 module.exports = mod;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "catto.js",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "Universal module for everything.",
   "main": "index.js",
   "scripts": {