ring-mqtt/lib/mqtt.js

const mqttApi = require('mqtt')
const debug = require('debug')('ring-mqtt')
const colors = require('colors/safe')
const utils = require('./utils.js')

class Mqtt {
    constructor() {
        this.connected = false
    }

    async init(ring, config) {
        this.ring = ring
        this.config = config
        // Initiate connection to MQTT broker
        try {
            debug('Starting connection to MQTT broker...')
            this.client = await this.connect()
            if (this.client.connected) {
                this.connected = true
                debug('MQTT connection established, sending config/state information in 5 seconds.')
            }
            // Monitor configured/default Home Assistant status topic
            this.client.subscribe(this.config.hass_topic)
            // Monitor legacy Home Assistant status topics
            this.client.subscribe('hass/status')
            this.client.subscribe('hassio/status')
            this.start()
        } catch (error) {
            debug(error)
            debug(colors.red('Couldn\'t authenticate to MQTT broker. Please check the broker and configuration settings.'))
            process.exit(1)
        }
    }

    // Initiate the connection to MQTT broker
    connect() {
        const mqtt_user = this.config.mqtt_user ? this.config.mqtt_user : null
        const mqtt_pass = this.config.mqtt_pass ? this.config.mqtt_pass : null
        const mqtt = mqttApi.connect({
            host: this.config.host,
            port: this.config.port,
            username: mqtt_user,
            password: mqtt_pass
        });
        return mqtt
    }

    // MQTT initialization successful, setup actions for MQTT events
    start() {
        const mqtt = this
        // On MQTT connect/reconnect send config/state information after delay
        this.client.on('connect', async function () {
            if (!mqtt.connected) {
                mqtt.connected = true
                debug('MQTT connection established, processing locations...')
            }
            mqtt.ring.processLocations(mqtt)
        })

        this.client.on('reconnect', function () {
            if (mqtt.connected) {
                debug('Connection to MQTT broker lost. Attempting to reconnect...')
            } else {
                debug('Attempting to reconnect to MQTT broker...')
            }
            mqtt.connected = false
        })

        this.client.on('error', function (error) {
            debug('Unable to connect to MQTT broker.', error.message)
            mqtt.connected = false
        })

        // Process MQTT messages from subscribed command topics
        this.client.on('message', async function (topic, message) {
            mqtt.processMessage(topic, message)
        })
    }

    // Process received MQTT command
    async processMessage(topic, message) {
        message = message.toString()
        if (topic === this.config.hass_topic || topic === 'hass/status' || topic === 'hassio/status') {
            debug('Home Assistant state topic '+topic+' received message: '+message)
            if (message === 'online') {
                this.ring.republishDevices()
            }
        } else {
            this.ring.processDeviceCommand(topic, message)
        }
    }
}

module.exports = new Mqtt()