Added Presence

src/gateway/index.ts

@@ -10,6 +10,7 @@ import { gatewayHandlers } from './handlers/index.ts'
 import { GATEWAY_BOT } from '../types/endpoint.ts'
 import { GatewayBotPayload } from "../types/gatewayBot.ts"
 import { GatewayCache } from "../managers/GatewayCache.ts"
+import { ClientActivityPayload } from "../structures/presence.ts"
 
 /**
  * Handles Discord gateway connection.
@@ -79,12 +80,10 @@ class Gateway {
             return
           }
 
-          this.websocket.send(
-            JSON.stringify({
+          this.send({
             op: GatewayOpcodes.HEARTBEAT,
             d: this.sequenceID ?? null
           })
-          )
           this.lastPingTimestamp = Date.now()
         }, this.heartbeatInterval)
 
@@ -202,8 +201,7 @@ class Gateway {
         return this.sendResume()
       }
     }
-    this.websocket.send(
-      JSON.stringify({
+    this.send({
       op: GatewayOpcodes.IDENTIFY,
       d: {
         token: this.token,
@@ -218,15 +216,9 @@ class Gateway {
           (previous, current) => previous | current,
           0
         ),
-          presence: {
-            // TODO: User should can customize this
-            status: 'online',
-            since: null,
-            afk: false
-          }
+        presence: this.client.presence.create()
       }
     })
-    )
   }
 
   private async sendResume(): Promise<void> {
@@ -243,9 +235,7 @@ class Gateway {
         seq: this.sequenceID || null
       }
     }
-    this.websocket.send(
-      JSON.stringify(resumePayload)
-    )
+    this.send(resumePayload)
   }
 
   debug(msg: string) {
@@ -275,6 +265,24 @@ class Gateway {
   close(): void {
     this.websocket.close(1000)
   }
+
+  send(data: GatewayResponse) {
+    if (this.websocket.readyState != this.websocket.OPEN) return false
+    this.websocket.send(JSON.stringify({
+      op: data.op,
+      d: data.d,
+      s: typeof data.s == "number" ? data.s : null,
+      t: data.t || null,
+    }))
+    return true
+  }
+
+  sendPresence(data: ClientActivityPayload) {
+    this.send({
+      op: GatewayOpcodes.PRESENCE_UPDATE,
+      d: data
+    })
+  }
 }
 
 export type GatewayEventHandler = (gateway: Gateway, d: any) => void

src/models/client.ts

@@ -9,13 +9,15 @@ import { GuildManager } from "../managers/GuildsManager.ts"
 import { EmojisManager } from "../managers/EmojisManager.ts"
 import { ChannelsManager } from "../managers/ChannelsManager.ts"
 import { MessagesManager } from "../managers/MessagesManager.ts"
+import { ActivityGame, ClientActivity, ClientActivityPayload, ClientPresence } from "../structures/presence.ts"
 
 /** Some Client Options to modify behaviour */
 export interface ClientOptions {
   token?: string
   intents?: GatewayIntents[]
   cache?: ICacheAdapter,
-  forceNewSession?: boolean
+  forceNewSession?: boolean,
+  presence?: ClientPresence | ClientActivity | ActivityGame
 }
 
 /**
@@ -37,12 +39,15 @@ export class Client extends EventEmitter {
   messages: MessagesManager = new MessagesManager(this)
   emojis: EmojisManager = new EmojisManager(this)
 
+  presence: ClientPresence = new ClientPresence()
+
   constructor (options: ClientOptions = {}) {
     super()
     this.token = options.token
     this.intents = options.intents
     this.forceNewSession = options.forceNewSession
     if(options.cache) this.cache = options.cache
+    if(options.presence) this.presence = options.presence instanceof ClientPresence ? options.presence : new ClientPresence(options.presence)
   }
 
   setAdapter(adapter: ICacheAdapter) {
@@ -50,6 +55,13 @@ export class Client extends EventEmitter {
     return this
   }
 
+  setPresence(presence: ClientPresence | ClientActivity | ActivityGame) {
+    if(presence instanceof ClientPresence) {
+      this.presence = presence
+    } else this.presence = new ClientPresence(presence)
+    this.gateway?.sendPresence(this.presence.create())
+  }
+
   debug(tag: string, msg: string) {
     this.emit("debug", `[${tag}] ${msg}`)
   }

src/structures/presence.ts

@@ -0,0 +1,120 @@
+export type ActivityType = 'PLAYING' | 'STREAMING' | 'LISTENING' | 'WATCHING' | 'CUSTOM_STATUS' | 'COMPETING';
+export type StatusType = 'online' | 'invisible' | 'offline' | 'idle' | 'dnd';
+
+export enum ActivityTypes {
+  PLAYING = 0,
+  STREAMING = 1,
+  LISTENING = 2,
+  WATCHING = 3,
+  CUSTOM_STATUS = 4,
+  COMPETING = 5,
+}
+
+export interface ActivityGame {
+  name: string;
+  type: 0 | 1 | 2 | 3 | 4 | 5 | ActivityType;
+  url?: string;
+}
+
+export interface ClientActivity {
+  status?: StatusType
+  activity?: ActivityGame | ActivityGame[]
+  since?: number | null
+  afk?: boolean
+}
+
+export interface ClientActivityPayload {
+  status: StatusType
+  activities: ActivityGame[] | null
+  since: number | null
+  afk: boolean
+}
+
+export class ClientPresence {
+  status: StatusType = 'online'
+  activity?: ActivityGame | ActivityGame[] 
+  since?: number | null
+  afk?: boolean
+
+  constructor(data?: ClientActivity | ClientActivityPayload | ActivityGame) {
+    if (data) {
+      if((data as ClientActivity).activity !== undefined) {
+        Object.assign(this, data)
+      } else if((data as ClientActivityPayload).activities !== undefined) {
+        
+      } else if((data as ActivityGame).name !== undefined) {
+        if(!this.activity) {
+          this.activity = data as ActivityGame
+        } else if(this.activity instanceof Array) {
+          this.activity.push(data as ActivityGame)
+        } else this.activity = [ this.activity, data as ActivityGame ]
+      }
+    }
+  }
+
+  parse(payload: ClientActivityPayload) {
+    this.afk = payload.afk
+    this.activity = payload.activities ?? undefined
+    this.since = payload.since
+    this.status = payload.status
+  }
+
+  static parse(payload: ClientActivityPayload) {
+    return new ClientPresence().parse(payload)
+  }
+
+  create(): ClientActivityPayload {
+    return {
+      afk: this.afk || false,
+      activities: this.createActivity(),
+      since: this.since || null,
+      status: this.status || 'online'
+    }
+  }
+
+  createActivity() {
+    let activity = this.activity == undefined ? null : (this.activity instanceof Array ? this.activity : [this.activity]) || null
+    if(activity == null) return activity
+    else {
+      activity.map(e => {
+        if(typeof e.type == "string") e.type = ActivityTypes[e.type]
+        return e
+      })
+      return activity
+    }
+  }
+
+  setStatus(status: StatusType) {
+    this.status = status
+    return this
+  }
+
+  setActivity(activity: ActivityGame) {
+    this.activity = activity
+    return this
+  }
+
+  setActivities(activities: ActivityGame[]) {
+    this.activity = activities
+    return this
+  }
+
+  setAFK(afk: boolean) {
+    this.afk = afk
+  }
+
+  removeAFK() {
+    this.afk = false
+    return this
+  }
+
+  toggleAFK() {
+    this.afk = !(this.afk || true)
+    return this
+  }
+
+  setSince(since?: number) {
+    this.since = since
+    return this
+  }
+}

src/test/index.ts

@@ -8,8 +8,17 @@ import { Guild } from '../structures/guild.ts'
 import { User } from '../structures/user.ts'
 import { Message } from "../structures/message.ts"
 import { RedisCacheAdapter } from "../models/CacheAdapter.ts"
+import { ClientPresence } from "../structures/presence.ts"
 
-const bot = new Client()
+const bot = new Client({
+  presence: new ClientPresence({
+    activity: {
+      name: "Testing",
+      type: 'COMPETING'
+    }
+  }),
+  forceNewSession: true
+})
 
 bot.setAdapter(new RedisCacheAdapter(bot, {
   hostname: "127.0.0.1",
@@ -18,6 +27,10 @@ bot.setAdapter(new RedisCacheAdapter(bot, {
 
 bot.on('ready', () => {
   console.log(`[Login] Logged in as ${bot.user?.tag}!`)
+  bot.setPresence({
+    name: "Test After Ready",
+    type: 'COMPETING'
+  })
 })
 
 bot.on('debug', console.log)