peertube-plugin-livechat/bots/lib/bot/component.ts
2021-12-07 18:57:08 +01:00

83 lines
2.1 KiB
TypeScript

/* eslint-disable no-void */
import { logger } from '../logger'
import { XMPP, XMPPXmlFunction, XMPPStanza, XMPPAddress } from './types'
const { component, xml } = require('@xmpp/component')
interface ComponentConnectionConfig {
service: string
domain: string
password: string
}
abstract class ComponentBot {
protected xmpp?: XMPP
protected address?: XMPPAddress
constructor (
public readonly botName: string,
protected readonly connectionConfig: ComponentConnectionConfig
) {}
protected xml: XMPPXmlFunction = (...args) => xml(...args)
public async connect (): Promise<void> {
this.xmpp = component({
service: this.connectionConfig.service,
domain: this.connectionConfig.domain,
password: this.connectionConfig.password
}) as XMPP
this.xmpp.on('error', (err: any) => {
logger.error(err)
})
this.xmpp.on('offline', () => {
logger.info(`${this.botName} is now offline.`)
})
this.xmpp.on('stanza', (stanza: XMPPStanza) => {
logger.debug('stanza received' + (stanza?.toString ? ': ' + stanza.toString() : ''))
if (stanza.is('message')) {
void this.onMessage(stanza)
}
if (stanza.is('presence')) {
void this.onPresence(stanza)
}
if (stanza.is('iq')) {
void this.onIq(stanza)
}
})
this.xmpp.on('online', (address: XMPPAddress) => {
logger.debug('Online with address' + address.toString())
this.address = address
void this.onOnline()
})
this.xmpp.start()
}
public async stop (): Promise<any> {
const p = new Promise((resolve) => {
this.xmpp?.on('offline', () => {
logger.info(`Stoppping process: ${this.botName} is now offline.`)
resolve(true)
})
})
await this.xmpp?.stop()
await p
}
protected async onMessage (_stanza: XMPPStanza): Promise<void> {}
protected async onIq (_stanza: XMPPStanza): Promise<void> {}
protected async onPresence (_stanza: XMPPStanza): Promise<void> {}
protected async onOnline (): Promise<void> {}
}
export {
ComponentConnectionConfig,
ComponentBot
}