harmony/src/structures/presence.ts

import { ActivityGame, ClientActivity, StatusType } from '../types/presence.ts'
import { StatusUpdatePayload } from '../types/gateway.ts'

enum ActivityTypes {
  PLAYING = 0,
  STREAMING = 1,
  LISTENING = 2,
  WATCHING = 3,
  CUSTOM_STATUS = 4,
  COMPETING = 5
}

export class ClientPresence {
  status: StatusType = 'online'
  activity?: ActivityGame | ActivityGame[]
  since?: number | null
  afk?: boolean

  constructor (data?: ClientActivity | StatusUpdatePayload | ActivityGame) {
    if (data !== undefined) {
      if ((data as ClientActivity).activity !== undefined) {
        Object.assign(this, data)
      } else if ((data as StatusUpdatePayload).activities !== undefined) {
      } else if ((data as ActivityGame).name !== undefined) {
        if (this.activity === undefined) {
          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: StatusUpdatePayload): ClientPresence {
    this.afk = payload.afk
    this.activity = payload.activities ?? undefined
    this.since = payload.since
    this.status = payload.status
    return this
  }

  static parse (payload: StatusUpdatePayload): ClientPresence {
    return new ClientPresence().parse(payload)
  }

  create (): StatusUpdatePayload {
    return {
      afk: this.afk === undefined ? false : this.afk,
      activities: this.createActivity(),
      since: this.since === undefined ? null : this.since,
      status: this.status === undefined ? 'online' : this.status
    }
  }

  createActivity (): ActivityGame[] | null {
    // eslint-disable-next-line @typescript-eslint/strict-boolean-expressions
    const activity =
      this.activity === undefined
        ? null
        : this.activity instanceof Array
        ? this.activity
        : [this.activity]
    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): ClientPresence {
    this.status = status
    return this
  }

  setActivity (activity: ActivityGame): ClientPresence {
    this.activity = activity
    return this
  }

  setActivities (activities: ActivityGame[]): ClientPresence {
    this.activity = activities
    return this
  }

  setAFK (afk: boolean): ClientPresence {
    this.afk = afk
    return this
  }

  removeAFK (): ClientPresence {
    this.afk = false
    return this
  }

  toggleAFK (): ClientPresence {
    this.afk = this.afk === undefined ? true : !this.afk
    return this
  }

  setSince (since?: number): ClientPresence {
    this.since = since
    return this
  }
}