rewrite in typescript

Author: RedDaedalus

src/dispatch/.gitignore

@@ -0,0 +1,4 @@
+dist
+node_modules
+package-lock.json
+*/.DS_Store

src/dispatch/package.json

@@ -0,0 +1,36 @@
+{
+  "name": "dispatch",
+  "version": "1.0.0",
+  "description": "A tool for creating Discord modifications at the server side.",
+  "main": "dist/index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1",
+    "start": "node ./dist/index.js"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/RedDaedalus/dispatch.git"
+  },
+  "keywords": [
+    "discord",
+    "dispatch",
+    "proxy"
+  ],
+  "author": "Daedalus",
+  "license": "MIT",
+  "bugs": {
+    "url": "https://github.com/RedDaedalus/dispatch/issues"
+  },
+  "homepage": "https://github.com/RedDaedalus/dispatch#readme",
+  "dependencies": {
+    "@types/flake-idgen": "^0.1.30",
+    "@types/pako": "^1.0.1",
+    "@types/ws": "^7.4.0",
+    "discord-api-types": "^0.12.1",
+    "erlpack": "^0.1.3",
+    "flake-idgen": "^1.4.0",
+    "pako": "^1.0.11",
+    "ws": "^7.4.4",
+    "zlib-sync": "^0.1.7"
+  }
+}

src/dispatch/src/gateway.ts

@@ -0,0 +1,70 @@
+import EventListener from "events"
+import WebSocket from "ws"
+import erlpack from "erlpack"
+import pako from "pako"
+import querystring from "querystring"
+
+export class GatewayProxy extends EventListener {
+
+    constructor() {
+        super()
+
+        const server = new WebSocket.Server({ port: 11000 })
+        server.on("connection", (ws, req) => {
+            const { encoding } = querystring.parse(req.url.substring(2))
+            console.log("Starting new session")
+            console.log(encoding)
+            const session = new GatewaySession(ws, encoding as string)
+            this.emit("session", session)
+            console.log("New session")
+        })
+    }
+}
+
+export class GatewaySession {
+
+    private inflate = new pako.Inflate()
+    private deflate = new pako.Deflate()
+    private encoding: string
+    private wsRemote: WebSocket
+    private wsLocal: WebSocket
+    
+    constructor(ws: WebSocket, encoding: string) {
+        this.encoding = encoding
+        this.wsRemote = new WebSocket(`wss://gateway.discord.gg/?v=8&encoding=${encoding}&compress=zlib-stream`)
+        this.wsLocal = ws
+
+        this.wsRemote.on("message", (chunk: Buffer) => {
+            this.inflate.push(chunk, 2)
+            const result = Buffer.from(this.inflate.result)
+
+            this.deflate.push(result, 2)
+            const outgoing = Buffer.from(this.deflate.result)
+
+            this.wsLocal.send(outgoing, e => e ? console.error(e) : undefined)
+        })
+        this.wsRemote.on("close", (code, reason) => console.log(`Closed: ${code} ${reason}`))
+        this.wsLocal.on("message", (chunk: Buffer) => {
+            this.wsRemote.send(chunk, e => e ? console.error(e) : undefined)
+        })
+    }
+
+    send(event: string, data: any) {
+        if (!data.op) data = { op: 0, d: data, s: null, t: event }
+
+        let message
+        if (this.encoding === "etf") {
+            message = erlpack.pack(data)
+        } else if (this.encoding === "json") {
+            message = Buffer.from(JSON.stringify(data))
+        }
+        this.deflate.push(message, 2)
+        this.wsLocal.send(Buffer.from(this.deflate.result))
+    }
+}
+
+export enum PacketDirection {
+
+    Incoming = 0,
+    Outgoing = 1
+}

src/dispatch/src/index.ts

@@ -0,0 +1,78 @@
+import * as ws from "./gateway"
+import FlakeId from "flake-idgen"
+
+const proxy = new ws.GatewayProxy()
+const snowflakeGen = new FlakeId()
+
+proxy.on("session", session => {
+    setTimeout(() => { // Settimeout bc I don't have an events system to wait for ready lol
+        console.log("Sending artificial message")
+        session.send("MESSAGE_CREATE", {
+            id: snowflakeGen.next().readBigInt64BE().toString(),
+            channel_id: "821523237409914910",
+            guild_id: "758105437244489799",
+            author: {
+                id: "820091309565280306",
+                username: "Dispatch",
+                discriminator: "0001",
+                avatar: "52abf38a25c1a8ca15333a6eeaa727e1",
+                bot: true,
+                system: true,
+                flags: (1 << 12) | (1 << 16)
+            },
+            member: {
+                roles: [],
+                joined_at: new Date().toString(),
+                deaf: false,
+                mute: false
+            },
+            timestamp: new Date().toString(),
+            edited_timestamp: null,
+            tts: false,
+            mention_everyone: false,
+            mentions: [],
+            mention_roles: [],
+            mention_channels: [],
+            attachments: [],
+            embeds: [{
+                title: "» Dispatch",
+                description: "Dispatch is a modding engine for Discord that doesn't require modification of the client.",
+                type: "rich",
+                fields: [{
+                    name: "Current Version",
+                    value: "0.0.1",
+                    inline: true
+                }, {
+                    name: "Website",
+                    value: "https://dispatch.co",
+                    inline: true
+                }, {
+                    name: "Source",
+                    value: "[GitHub Repo](https://github.com/discord-dispatch/dispatch)",
+                    inline: true
+                }],
+                footer: {
+                    text: "Developed by Daedalus#0001 under the MIT license"
+                },
+                color: 0x7289DA,
+                timestamp: new Date().toString()
+            }],
+            pinned: false,
+            type: 0,
+            flags: 1 << 6,
+            interaction: {
+                id: snowflakeGen.next().readBigInt64BE().toString(),
+                type: 2,
+                name: "info",
+                user: {
+                    id: "268071134057070592",
+                    username: "Daedalus",
+                    discriminator: "0001",
+                    avatar: "a_4ae268f6dbb3adfb9c1c45461263ce8d",
+                    bot: false,
+                    flags: (1 << 8) | (1 << 9) | (1 << 17),
+                }
+            }
+        })
+    }, 10_000)
+})

src/dispatch/tsconfig.json

@@ -0,0 +1,11 @@
+{
+    "compilerOptions": {
+        "module": "commonjs",
+        "esModuleInterop": true,
+        "target": "es2020",
+        "moduleResolution": "node",
+        "sourceMap": true,
+        "outDir": "dist"
+    },
+    "lib": ["es2020"]
+}

src/index.js

@@ -41,7 +41,7 @@ async function handleRequest(request) {
       return quoteMessage(body)
     } else {
       return new Response(JSON.stringify({
-        type: 3,
+        type: 4,
         data: {
           flags: 64,
           content: `
@@ -68,7 +68,7 @@ Click [here](https://discord.com/api/oauth2/authorize?client_id=8128922189257769
  */
 async function error(message) {
   return new Response(JSON.stringify({ 
-    type: 3,
+    type: 4,
     data: {
       flags: 64,
       content: `**Error:** ${message}`