trakteer-websocket-wrapper/src/Client/client.ts

A wrapper for saweria websocket endpoint.

import WebSocket from "ws";
import EventEmitter from "events"
import {
  ClientParameter,
  WebsocketEvent,
  Donation,
} from "../../index"
/**
 * Initiate a Trakteer Websocket Wrapper Client
 * @param socketURL
 */
class TrakteerSocket extends EventEmitter implements TrakteerSocket{
  public TrakteerSocket:WebSocket
  public PingIntervalTime:number = 25000
  public PingInterval:ReturnType<typeof setInterval>
  public mainChannelID:string
  public testChannelID:string
  constructor(PARAMETER:ClientParameter){
    super()
    if(!PARAMETER.mainChannelID || !PARAMETER.testChannelID){
      throw "Insufficient parameters provided. mainChannelID and testChannelID required."
    }
    this.mainChannelID = PARAMETER.mainChannelID
    this.testChannelID = PARAMETER.testChannelID
    //Handle websocket connections
    this.TrakteerSocket = new WebSocket("wss://socket.trakteer.id/app/2ae25d102cc6cd41100a")
    if(!this.TrakteerSocket){
      throw "Socket URL Invalid"
    }
    this.TrakteerSocket.on('error', (data) => {
      this.emitEvent("websocketerror", data)
    })
    this.TrakteerSocket.on('close', (data) => {
      this.emitEvent("websocketclosed", data)
    })
    this.TrakteerSocket.on('message', (data:Buffer) => {
      const eventObject:WebsocketEvent = JSON.parse(data.toString())
      console.log(eventObject)
      //Handle connection to Trakteer WebSocket
      if(eventObject.event == "pusher:connection_established"){
        this.emitEvent("connected", eventObject)
        this.TrakteerSocket.send(JSON.stringify({
          event: "pusher:subscribe",
          data:{
            auth: "",
            channel: PARAMETER.mainChannelID
          }
        }))
        this.TrakteerSocket.send(JSON.stringify({
          event: "pusher:subscribe",
          data: {
            auth: "",
            channel: PARAMETER.testChannelID
          }
        }))
        this.PingInterval = setInterval(()=>{
          this.TrakteerSocket.send(JSON.stringify({
            event: "pusher:ping",
            data: {}
          }))
        }, this.PingIntervalTime)
      }
      if(eventObject.event == "pusher_internal:subscription_succeeded"){
        this.emitEvent("subscribed", eventObject)
      }
      if(eventObject.event == "pusher:pong"){
        this.emitEvent("pong", eventObject)
      }
      if(eventObject.event == "pusher:subscription_error"){
        this.emitEvent("subscription_error", eventObject)
      }
      if(eventObject.event == "pusher:error"){
        this.emitEvent("error", eventObject)
      }
      if(eventObject.event == "Illuminate\\Notifications\\Events\\BroadcastNotificationCreated"){
        const donationData:Donation = JSON.parse(eventObject.data as string)
        this.emitEvent("donation", donationData)
      } 
    })
    this.TrakteerSocket.on('close', (data) => {
      this.emit('close', {
        event: "close",
        data: data
      })
      this.TrakteerSocket.close()
    })
  }
  close = ():void => {
    this.TrakteerSocket.send(JSON.stringify({
      event: "pusher:close",
      data: {
        channel: this.mainChannelID
      }
    }))
    this.TrakteerSocket.send(JSON.stringify({
      event: "pusher:close",
      data: {
        channel: this.testChannelID
      }
    }))
    clearInterval(this.PingInterval)
    this.TrakteerSocket.close()
    this.emitEvent("close", {
      event: "close",
      data: ""
    })
  }
  interval = (newInterval:number):ReturnType<typeof setInterval> => {
    clearInterval(this.PingInterval)
    this.PingInterval = setInterval(()=>{
      this.TrakteerSocket.send(JSON.stringify({
        event: "pusher:ping",
        data: {}
      }))
    }, newInterval)
    return this.PingInterval
  }
  emitEvent = (eventName:string, eventObject:WebsocketEvent|Donation|Error|number) => {
    this.emit(eventName, {
      event: eventName,
      data: eventObject
    })
  }
}
module.exports = TrakteerSocket