poorchat.js

const WebSocket = require('ws')
const ReconnectingWebSocket= require('reconnecting-websocket')
const EventEmitter = require('events')
const parse = require('irc-message').parse

class Poorchat extends EventEmitter {
    constructor(options) {
        super()
        this.ws = new ReconnectingWebSocket(options.websocket, ['base64'], {
                WebSocket: WebSocket
            })
        this.login = options.login
        this.password = options.password
        this.cap = options.cap
        this.channel = options.channel
        this.debug = options.debug 
        this.options = options
        this.PONGInterval = null
    }

    messageEncode(data) {
        return Buffer.from(`${data}\r\n`).toString('base64')
    }

    messageDecode(data) {
        return Buffer.from(data, 'base64').toString('utf-8')
    }

    sendMessage(message) {
        const encodedMessage = this.messageEncode(message)
        this.ws.send(encodedMessage)
    }

    readMessage(data) {
        const decodedMessage = this.messageDecode(data)
        return parse(decodedMessage)
    }

    say(data) {
        const encodedMessage = this.messageEncode(`PRIVMSG ${this.channel} :${data}`)
        this.ws.send(encodedMessage)
    }

    pm(channel, data) {
        const encodedMessage = this.messageEncode(`PRIVMSG ${channel} :${data}`)
        this.ws.send(encodedMessage)
    }

    connect() {
        return new Promise((resolve) => {
            console.log('Connecting to IRC...')
            this.ws.addEventListener('open', () => {
                this.sendMessage(`NICK ${this.login}`)
                this.sendMessage(`USER ${this.login} ${this.options.irc} Poorchat ${this.login}`)
                for (const cap of this.cap) {
                    this.sendMessage(cap)
                }
                this.sendMessage('CAP END')
            })
            this.ws.addEventListener('message', ({ data }) => {
                const message = this.readMessage(data)

                if (message.command === '422') {
                this.sendMessage(`PRIVMSG Poorchat :LOGIN ${this.login} ${this.password}`)
                this.sendMessage(`JOIN ${this.channel}`)
                }
    
                if (message.command === 'JOIN' && message.prefix.split('!')[0] === this.login) {
                    clearInterval(this.PONGInterval)
                    setInterval(() => {
                        this.sendMessage(`PONG irc.poorchat.net`)
                    }, 30000)
                    console.log('IRC Conntected!')
		    resolve()
                }

                this.messageHandler(message)
                if (this.debug) {
                    console.log(message)
                }
            })
        })
    }

    messageHandler(message) {
        switch (message.command) {
            case 'PING':
                this.sendMessage(`PONG ${message.params[0]}`)
                break
            case 'PRIVMSG':
                this.emit('message', message)
                break
            case 'JOIN':
                this.emit('join', message)
                break
            case 'PART':
                this.emit('part', message)
                break
            case 'EMBED':
                this.emit('message', message)
                break
            case 'NOTICE':
                this.emit('message', message)
                break
            case 'MODE':
                this.emit('mode', message)
            default: 
                if (this.debug) {
                    console.log(message.raw)
                }
        }
    }
}

module.exports = Poorchat