ring-mqtt/lib/tokenapp.js

import { RingRestClient } from 'ring-client-api/rest-client'
import utils from './utils.js'
import { dirname } from 'path'
import { fileURLToPath } from 'url'
import express from 'express'
import bodyParser from 'body-parser'
import chalk from 'chalk'
import debugModule from 'debug'
const debug = debugModule('ring-mqtt')

export default new class TokenApp {
    constructor() {
        this.app = express()
        this.listener = false
        this.ringConnected = false
        this.systemId = ''

        if (process.env.RUNMODE === 'addon') {
            this.start()
        }

        utils.event.on('ring_api_state', async (state) => {
            if (state === 'connected') {
                this.ringConnected = true

                // Only the addon leaves the web UI running all the time
                if (process.env.RUNMODE !== 'addon') {
                    this.stop()
                }
            } else {
                this.ringConnected = false
            }
        })
    }

    setSystemId(systemId) {
        if (systemId) {
            this.systemId = systemId
        }
    }

    // Super simple web service to acquire authentication tokens from Ring
    async start() {
        if (this.listener) {
            return
        }

        const webdir = dirname(fileURLToPath(new URL('.', import.meta.url)))+'/web'
        let restClient

        this.listener = this.app.listen(55123, () => {
            debug('Succesfully started the token generator web UI')
        })

        this.app.use(bodyParser.urlencoded({ extended: false }))

        this.app.get('/', (req, res) => {
            res.cookie('systemId', `${process.env.RUNMODE === 'addon' ? 'ring-mqtt-addon' : 'ring-mqtt'}-${this.systemId.slice(-5)}`, { maxAge: 3600000, encode: String })
            if (this.ringConnected) {
                res.sendFile('connected.html', {root: webdir})
            } else {
                res.sendFile('account.html', {root: webdir})
            }
        })

        this.app.get(/.*force-token-generation$/, (req, res) => {
            res.cookie('systemId', `${process.env.RUNMODE === 'addon' ? 'ring-mqtt-addon' : 'ring-mqtt'}-${this.systemId.slice(-5)}`, { maxAge: 3600000, encode: String })
            res.sendFile('account.html', {root: webdir})
        })

        this.app.post(/.*submit-account$/, async (req, res) => {
            res.cookie('systemId', `${process.env.RUNMODE === 'addon' ? 'ring-mqtt-addon' : 'ring-mqtt'}-${this.systemId.slice(-5)}`, { maxAge: 3600000, encode: String })
            const email = req.body.email
            const password = req.body.password
            restClient = await new RingRestClient({
                email,
                password,
                controlCenterDisplayName: `${process.env.RUNMODE === 'addon' ? 'ring-mqtt-addon' : 'ring-mqtt'}-${this.systemId.slice(-5)}`,
                systemId: this.systemId
            })
            // Check if the user/password was accepted
            try {
                await restClient.getCurrentAuth()
            } catch(error) {
                if (restClient.using2fa) {
                    debug('Username/Password was accepted, waiting for 2FA code to be entered.')
                    res.sendFile('code.html', {root: webdir})
                } else {
                    const errmsg = error.message ? error.message : 'Null response, you may be temporarily throttled/blocked. Please shut down ring-mqtt and try again in a few hours.'
                    debug(chalk.red(errmsg))
                    res.cookie('error', errmsg, { maxAge: 1000, encode: String })
                    res.sendFile('account.html', {root: webdir})
                }
            }
        })

        this.app.post(/.*submit-code$/, async (req, res) => {
            res.cookie('systemId', `${process.env.RUNMODE === 'addon' ? 'ring-mqtt-addon' : 'ring-mqtt'}-${this.systemId.slice(-5)}`, { maxAge: 3600000, encode: String })
            let generatedToken
            const code = req.body.code
            try {
                generatedToken = await restClient.getAuth(code)
            } catch {
                generatedToken = false
                const errormsg = 'The 2FA code was not accepted, please verify the code and try again.'
                debug(errormsg)
                res.cookie('error', errormsg, { maxAge: 1000, encode: String })
                res.sendFile('code.html', {root: webdir})
            }
            if (generatedToken) {
                res.sendFile('restart.html', {root: webdir})
                utils.event.emit('generated_token', generatedToken.refresh_token)
            }
        })
    }

    async stop() {
        if (this.listener) {
            await this.listener.close()
            this.listener = false
        }
    }
}