tuya-mqtt/tuya-mqtt.js

#!/usr/bin/env node
const fs = require('fs')
const mqtt = require('mqtt')
const json5 = require('json5')
const debug = require('debug')('tuya-mqtt:mqtt')
const debugError = require('debug')('tuya-mqtt:error')
const SimpleSwitch = require('./devices/simple-switch')
const SimpleDimmer = require('./devices/simple-dimmer')
const RGBTWLight = require('./devices/rgbtw-light')
const GenericDevice = require('./devices/generic-device')

var CONFIG = undefined
var tuyaDevices = new Array()

function getDevice(configDevice, mqttClient) {
    const deviceInfo = {
        configDevice: configDevice,
        mqttClient: mqttClient,
        topic: CONFIG.topic
    }
    switch (configDevice.type) {
        case 'SimpleSwitch':
            return new SimpleSwitch(deviceInfo)
            break;
        case 'SimpleDimmer':
            return new SimpleDimmer(deviceInfo)
            break;
        case 'RGBTWLight':
            return new RGBTWLight(deviceInfo)
            break;
        case 'GenericDevice':
            return new GenericDevice(deviceInfo)
            break;
    }
    return null
}

function initDevices(configDevices, mqttClient) {
    for (let configDevice of configDevices) {
        if (!configDevice.type) {
            debug('Device type not specified, skipping creation of this device')
        } else {
            const newDevice = getDevice(configDevice, mqttClient)
            if (newDevice) {
                tuyaDevices.push(newDevice)
            }
        }
    }
}

// Main code function
const main = async() => {
    let configDevices
    let mqttClient

    try {
        CONFIG = require('./config')
    } catch (e) {
        console.error('Configuration file not found!')
        debugError(e)
        process.exit(1)
    }

    if (typeof CONFIG.qos == 'undefined') {
        CONFIG.qos = 2
    }
    if (typeof CONFIG.retain == 'undefined') {
        CONFIG.retain = false
    }

    try {
        configDevices = fs.readFileSync('./devices.conf', 'utf8')
        configDevices = json5.parse(configDevices)
    } catch (e) {
        console.error('Devices file not found!')
        debugError(e)
        process.exit(1)
    }

    if (!configDevices.length) {
        console.error('No devices found in devices file!')
        process.exit(1)
    }

    mqttClient = mqtt.connect({
        host: CONFIG.host,
        port: CONFIG.port,
        username: CONFIG.mqtt_user,
        password: CONFIG.mqtt_pass,
    })

    mqttClient.on('connect', function (err) {
        debug('Connection established to MQTT server')
        let topic = CONFIG.topic + '#'
        mqttClient.subscribe(topic, {
            retain: CONFIG.retain,
            qos: CONFIG.qos
        })
        initDevices(configDevices, mqttClient)
    })

    mqttClient.on('reconnect', function (error) {
        if (mqttClient.connected) {
            debug('Connection to MQTT server lost. Attempting to reconnect...')
        } else {
            debug('Unable to connect to MQTT server')
        }
    })

    mqttClient.on('error', function (error) {
        debug('Unable to connect to MQTT server', error)
    })

    mqttClient.on('message', function (topic, message) {
        try {
            message = message.toString()
            const splitTopic = topic.split('/')
            const topicLength = splitTopic.length
            const commandTopic = splitTopic[topicLength - 1]
            const deviceTopicLevel = splitTopic[1]

            // If it looks like a valid command topic try to process it
            if (commandTopic.includes('command')) {
                debug('Received MQTT message -> ', JSON.stringify({
                    topic: topic,
                    message: message
                }))

                // Use device topic level to find matching device
                const device = tuyaDevices.find(d => d.options.name === deviceTopicLevel || d.options.id === deviceTopicLevel)
                switch (topicLength) {
                    case 3:
                        device.processCommand(message, commandTopic)
                        break;
                    case 4:
                        device.processDpsCommand(message)
                        break;
                    case 5:
                        const dpsKey = splitTopic[topicLength-2]
                        device.processDpsKeyCommand(message, dpsKey)
                        break;
                }
            }
        } catch (e) {
            debugError(e)
        }
    })
}

// Call the main code
main()