feat(user): added DM fetching

src/client/ClientUser.ts

@@ -2,6 +2,7 @@ import Room from "../room/Room";
 import User from "../user/User";
 import Client from "../client/Client";
 import { Handling, Presence } from "..";
+import { DirectMessage } from "..";
 
 export default class ClientUser extends User {
   constructor(data: any, client: Client) {
@@ -92,17 +93,7 @@ export default interface ClientUser {
   /**
    * Emitted when the ClientUser receives a message from another User
    */
-  on(
-    event: "message",
-    callback: (message: {
-      content: string;
-      content_safe: string;
-      author: User | undefined;
-      systemMessage: boolean;
-      id: string;
-      ts: string;
-    }) => void
-  ): this;
+  on(event: "message", callback: (message: DirectMessage) => void): this;
 
   /**
    * Emitted when the ClientUser is invited to a room

src/index.ts

@@ -7,6 +7,27 @@ import { TetraChannel } from "./channel/Channel";
 
 export { Client, ClientUser, Room, User, UserManager, TetraChannel };
 
+export interface DirectMessage {
+  /** Contents of the message. */
+  content: string;
+  /** Contents of the message after being passed through TETR.IO's profanity filter. */
+  content_safe: string;
+
+  /** Whether this is a system message. */
+  system: boolean;
+
+  /** The ISO 8601-formatted timestamp of the message. */
+  ts: string;
+
+  /** The id of the message. */
+  id: string;
+
+  /**
+   * The author of the message.
+   */
+  author?: User;
+}
+
 export interface CacheData {
   cache: {
     status: string;

src/user/User.ts

@@ -1,4 +1,6 @@
 import EventEmitter from "events";
+import fetch from "node-fetch";
+import { DirectMessage } from "..";
 import Client from "../client/Client";
 
 export default class User extends EventEmitter {
@@ -298,4 +300,48 @@ export default class User extends EventEmitter {
       data: this._id,
     });
   }
+
+  /**
+   * Fetches the DMs between the bot and the user
+   */
+  public async getDMs(): Promise<DirectMessage[]> {
+    const id = this._id;
+    const token = this.client.token;
+    const data: {
+      success: boolean;
+      error?: any;
+      dms: {
+        _id: string;
+        ts: string;
+        stream: string;
+        data: {
+          content: string;
+          content_safe: string;
+          system: boolean;
+          user: string;
+          userdata: any;
+        };
+      }[];
+    } = await (
+      await fetch(`https://tetr.io/api/dms/${encodeURIComponent(id)}`, {
+        headers: { Authorization: `Bearer ${token}` },
+      })
+    ).json();
+    if (!data.success) throw new Error(data.error);
+    else {
+      const dms: DirectMessage[] = [];
+      data.dms.forEach(async (message) => {
+        dms.push({
+          content: message.data.content,
+          content_safe: message.data.content_safe,
+          system: message.data.system,
+          id: message._id,
+          ts: message.ts,
+          author:
+            message.data.user == this._id ? this : await this.client.users.fetch(message.data.user),
+        });
+      });
+      return dms;
+    }
+  }
 }