simple-discord-bot.js 0.1.0

package/api.js

@@ -0,0 +1,61 @@
+const https = require("https")
+
+class API {
+  constructor(token) {
+    this.token = token
+  }
+
+  request(method, path, body = null) {
+    return new Promise((resolve, reject) => {
+      const data = body ? JSON.stringify(body) : null
+
+      const headers = {
+        Authorization: `Bot ${this.token}`,
+        "Content-Type": "application/json"
+      }
+
+      if (data) {
+        headers["Content-Length"] = Buffer.byteLength(data)
+      }
+
+      const req = https.request({
+        hostname: "discord.com",
+        path: `/api/v10${path}`,
+        method,
+        headers
+      }, res => {
+        let result = ""
+
+        res.on("data", chunk => {
+          result += chunk
+        })
+
+        res.on("end", () => {
+          try {
+            resolve(JSON.parse(result))
+          } catch {
+            resolve(result)
+          }
+        })
+      })
+
+      req.on("error", reject)
+
+      if (data) {
+        req.write(data)
+      }
+
+      req.end()
+    })
+  }
+
+  registerCommands(applicationId, commands) {
+    return this.request(
+      "PUT",
+      `/applications/${applicationId}/commands`,
+      commands
+    )
+  }
+}
+
+module.exports = API

package/client.js

@@ -0,0 +1,166 @@
+const EventEmitter = require("events")
+const Gateway = require("./gateway")
+const API = require("./api")
+const Message = require("./structures/Message")
+
+const PermissionFlags = {
+  CreateInstantInvite: 1n << 0n,
+  KickMembers: 1n << 1n,
+  BanMembers: 1n << 2n,
+  Administrator: 1n << 3n,
+  ManageChannels: 1n << 4n,
+  ManageGuild: 1n << 5n,
+  AddReactions: 1n << 6n,
+  ViewAuditLog: 1n << 7n,
+  PrioritySpeaker: 1n << 8n,
+  Stream: 1n << 9n,
+  ViewChannel: 1n << 10n,
+  SendMessages: 1n << 11n,
+  SendTTSMessages: 1n << 12n,
+  ManageMessages: 1n << 13n,
+  EmbedLinks: 1n << 14n,
+  AttachFiles: 1n << 15n,
+  ReadMessageHistory: 1n << 16n,
+  MentionEveryone: 1n << 17n,
+  UseExternalEmojis: 1n << 18n,
+  ViewGuildInsights: 1n << 19n,
+  Connect: 1n << 20n,
+  Speak: 1n << 21n,
+  MuteMembers: 1n << 22n,
+  DeafenMembers: 1n << 23n,
+  MoveMembers: 1n << 24n,
+  UseVAD: 1n << 25n,
+  ChangeNickname: 1n << 26n,
+  ManageNicknames: 1n << 27n,
+  ManageRoles: 1n << 28n,
+  ManageWebhooks: 1n << 29n,
+  ManageGuildExpressions: 1n << 30n,
+  UseApplicationCommands: 1n << 31n
+}
+
+class Permissions {
+  constructor(value) {
+    this.bitfield = BigInt(value || 0)
+  }
+
+  has(permission) {
+    const flag = PermissionFlags[permission]
+
+    if (flag === undefined) {
+      return false
+    }
+
+    if ((this.bitfield & PermissionFlags.Administrator) !== 0n) {
+      return true
+    }
+
+    return (this.bitfield & flag) !== 0n
+  }
+}
+
+class Client extends EventEmitter {
+  constructor() {
+    super()
+
+    this.token = null
+    this.user = null
+    this.api = null
+    this.gateway = new Gateway(this)
+    this.commands = new Map()
+    this.stopping = false
+
+    this.on("rawMessage", data => {
+      const message = new Message(this, data)
+      this.emit("message", message)
+    })
+
+    this.on("rawInteraction", data => {
+      if (data.type !== 2) return
+
+      const command = this.commands.get(data.data.name)
+
+      if (!command) return
+
+      const member = data.member
+
+      const interaction = {
+        id: data.id,
+        token: data.token,
+
+        commandName: data.data.name,
+
+        user: member?.user || data.user,
+
+        member: member ? {
+          ...member,
+
+          permissions: new Permissions(member.permissions)
+        } : null,
+
+        channelId: data.channel_id,
+
+        guildId: data.guild_id || null,
+
+        options: {},
+
+        reply: content => {
+          return this.api.request(
+            "POST",
+            `/interactions/${data.id}/${data.token}/callback`,
+            {
+              type: 4,
+              data: {
+                content: content
+              }
+            }
+          )
+        }
+      }
+
+      for (const option of data.data.options || []) {
+        interaction.options[option.name] = option.value
+      }
+
+      command.handler(interaction)
+    })
+
+    process.on("SIGINT", () => {
+      if (this.stopping) return
+
+      this.stopping = true
+
+      console.log("Stopping bot...")
+
+      this.destroy()
+
+      setTimeout(() => {
+        process.exit(0)
+      }, 1000)
+    })
+  }
+
+  login(token) {
+    this.token = token
+    this.api = new API(token)
+
+    return this.gateway.connect(token)
+  }
+
+  destroy() {
+    this.gateway.destroy()
+    this.emit("disconnect")
+  }
+
+  command(name, handler, options = {}) {
+    this.commands.set(name, {
+      command: {
+        name: name,
+        description: options.description || "No description",
+        options: options.options || []
+      },
+      handler
+    })
+  }
+}
+
+module.exports = Client

package/gateway.js

@@ -0,0 +1,91 @@
+const WebSocket = require("ws")
+
+class Gateway {
+  constructor(client) {
+    this.client = client
+    this.ws = null
+    this.heartbeat = null
+  }
+
+  connect(token) {
+    this.ws = new WebSocket(
+      "wss://gateway.discord.gg/?v=10&encoding=json"
+    )
+
+    this.ws.on("open", () => {
+      this.ws.send(JSON.stringify({
+        op: 2,
+        d: {
+          token: token,
+          intents: 33281,
+          properties: {
+            os: "android",
+            browser: "SimpleDiscord.js",
+            device: "SimpleDiscord.js"
+          }
+        }
+      }))
+    })
+
+    this.ws.on("message", data => {
+      const packet = JSON.parse(data)
+
+      if (packet.op === 10) {
+        this.startHeartbeat(packet.d.heartbeat_interval)
+      }
+
+      if (packet.t === "READY") {
+        this.client.user = packet.d.user
+        this.client.emit("ready", packet.d)
+      }
+
+      if (packet.t === "MESSAGE_CREATE") {
+        this.client.emit("rawMessage", packet.d)
+      }
+
+      if (packet.t === "INTERACTION_CREATE") {
+        this.client.emit("rawInteraction", packet.d)
+      }
+    })
+
+    this.ws.on("error", error => {
+      this.client.emit("error", error)
+    })
+
+    this.ws.on("close", () => {
+      this.stopHeartbeat()
+      this.ws = null
+      this.client.emit("disconnect")
+    })
+  }
+
+  startHeartbeat(interval) {
+    this.stopHeartbeat()
+
+    this.heartbeat = setInterval(() => {
+      if (!this.ws) return
+
+      this.ws.send(JSON.stringify({
+        op: 1,
+        d: null
+      }))
+    }, interval)
+  }
+
+  stopHeartbeat() {
+    if (this.heartbeat) {
+      clearInterval(this.heartbeat)
+      this.heartbeat = null
+    }
+  }
+
+  destroy() {
+    this.stopHeartbeat()
+
+    if (this.ws) {
+      this.ws.close(1000, "Bot stopped")
+    }
+  }
+}
+
+module.exports = Gateway

package/index.js

@@ -0,0 +1,5 @@
+const Client = require("./client")
+
+module.exports = {
+  Client
+}

package/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "simple-discord-bot.js",
+  "version": "0.1.0",
+  "description": "A simple Discord bot library",
+  "main": "index.js",
+  "license": "MIT",
+  "keywords": [
+    "discord",
+    "discord-bot",
+    "discord-library",
+    "discord-api"
+  ],
+  "dependencies": {
+    "ws": "^8.18.3"
+  }
+}

package/structures/Message.js

@@ -0,0 +1,23 @@
+class Message {
+  constructor(client, data) {
+    this.client = client
+
+    this.id = data.id
+    this.content = data.content
+    this.channelId = data.channel_id
+    this.guildId = data.guild_id
+    this.author = data.author
+  }
+
+  reply(content) {
+    return this.client.api.request(
+      "POST",
+      `/channels/${this.channelId}/messages`,
+      {
+        content: content
+      }
+    )
+  }
+}
+
+module.exports = Message