plejd.js

const noble = require('@abandonware/noble');
const crypto = require('crypto');
const xor = require('buffer-xor');
const _ = require('lodash');
const EventEmitter = require('events');
const sleep = require('sleep');

let debug = '';

const getLogger = () => {
  const consoleLogger = msg => console.log('plejd', msg);
  if (debug === 'console') {
    return consoleLogger;
  }

  // > /dev/null
  return _.noop;
};

const logger = getLogger();

// UUIDs
const PLEJD_SERVICE = "31ba000160854726be45040c957391b5"
const DATA_UUID = "31ba000460854726be45040c957391b5"
const LAST_DATA_UUID = "31ba000560854726be45040c957391b5"
const AUTH_UUID = "31ba000960854726be45040c957391b5"
const PING_UUID = "31ba000a60854726be45040c957391b5"

class Controller extends EventEmitter {
  constructor(cryptoKey, keepAlive = false) {
    super();

    this.cryptoKey = Buffer.from(cryptoKey.replace(/-/g, ''), 'hex');
    this.peripheral = null;
    this.peripheral_address = null;

    this.isScanning = false;
    this.isConnecting = false;
    this.isConnected = false;
    this.keepAlive = keepAlive;
    this.writeQueue = [];
    this.peripherals = [];

    // Holds a reference to the connected peripheral from the peripheral list.
    // In case the peripheral we're connecting to, disconnects us, we can then reinitiate the connection
    // by increasing the connectedIndex and by that, connect to the next in line.
    this.connectedIndex = 0;
  }

  async init() {
    const self = this;

    noble.on('stateChange', async (state) => {
      logger('ble state changed: ' + state);

      if (state === 'poweredOn') {
        await this.scan();
      }
    });

    noble.on('discover', (peripheral) => {
      logger('found ' + peripheral.advertisement.localName + ' with addr ' + peripheral.address);
      if (peripheral.advertisement.localName === 'P mesh') {
        self.peripherals.push(peripheral);
      }
    });

    noble.on('disconnect', async () => {
      if (self.peripherals.length) {
        logger('peripherals already scanned.');
        this.connectedIndex = 0;
        await self.connect();
      }
    });
  }

  async reinit() {
    console.log('reinitializing the Plejd add-on.');
    this.once('scanComplete', async (peripherals) => {
      console.log('found Plejd devices, reconnecting');
      await this.connect();
    });

    await this.scan();
  }

  async scan() {
    const self = this;
    this.isScanning = true;
    noble.startScanning([PLEJD_SERVICE]);

    setTimeout(() => {
      noble.stopScanning();
      this.isScanning = false;

      self.peripherals.sort((a, b) => a.rssi > b.rssi);
      this.emit('scanComplete', self.peripherals);

    }, 5000);
  }

  async connect() {
    const self = this;

    if (this.isScanning) {
      logger('already scanning, waiting.');
      return Promise.resolve(false);
    }

    // if (!this.peripherals.length) {
    //   await this.scan();
    // }

    this.isConnecting = true;

    return await this._internalConnect(this.connectedIndex);
  }

  async _internalConnect(idx) {
    const self = this;

    if (idx >= this.peripherals.length) {
      logger('reached end of list.');
      return Promise.resolve(false);
    }

    logger('connecting to Plejd device');
    try {
      this.peripherals[idx].connect(async (err) => {
        if (err) {
          console.log('error: failed to connect to Plejd device: ' + err);
          return await self._internalConnect(idx + 1);
        }

        self.peripheral = self.peripherals[idx];
        console.log('connected to Plejd device with addr ' + self.peripheral.address + ' with rssi ' + self.peripheral.rssi);

        self.peripheral_address = self._reverseBuffer(Buffer.from(String(self.peripheral.address).replace(/\-/g, '').replace(/\:/g, ''), 'hex'));

        let successfullyReadCharacteristics = false;
        logger('discovering services and characteristics');

        setTimeout(async () => {
          if (!successfullyReadCharacteristics) {
            logger('error: timed out when reading characteristics. moving on to next device.');
            return await self._internalConnect(idx + 1);
          }
        }, 10000);

        await self.peripheral.discoverSomeServicesAndCharacteristics([PLEJD_SERVICE], [], async (err, services, characteristics) => {
        //await self.peripheral.discoverAllServicesAndCharacteristics(async (err, services, characteristics) => {
          if (err) {
            console.log('error: failed to discover services: ' + err);
            return;
          }

          characteristics.forEach((ch) => {
            if (DATA_UUID == ch.uuid) {
              logger('found DATA characteristic.');
              self.dataCharacteristic = ch;
            }
            else if (LAST_DATA_UUID == ch.uuid) {
              logger('found LAST_DATA characteristic.');
              self.lastDataCharacteristic = ch;
            }
            else if (AUTH_UUID == ch.uuid) {
              logger('found AUTH characteristic.');
              self.authCharacteristic = ch;
            }
            else if (PING_UUID == ch.uuid) {
              logger('found PING characteristic.');
              self.pingCharacteristic = ch;
            }
          });

          if (this.dataCharacteristic
            && this.lastDataCharacteristic
            && this.authCharacteristic
            && this.pingCharacteristic) {

            successfullyReadCharacteristics = true;

            this.once('authenticated', () => {
              logger('Plejd is connected and authenticated.');
              this.connectedIndex = idx;

              if (self.keepAlive) {
                self.startPing();
              }

              self.subscribe();

              self.emit('connected');
            });

            try {
              await this.authenticate();
            }
            catch (error) {
              this.isConnecting = false;
              console.log('error: failed to authenticate: ' + error);
              return Promise.resolve(false);
            }

            this.isConnected = true;
            this.isConnecting = false;

            // make sure to write any queued up messages to the Plejd devices
            if (this.writeQueue && this.writeQueue.length > 0) {
              this.flush();
            }
          }

          return Promise.resolve(true);
        });
      });
    }
    catch (error) {
      this.isConnecting = false;

      console.log('error: failed to connect to Plejd device: ' + error);
    }

    return Promise.resolve(true);
  }

  subscribe() {
    const self = this;

    self.lastDataCharacteristic.subscribe((err) => {
      if (err) {
        console.log('error: couldnt subscribe to notification characteristic.');
        return;
      }

      // subscribe to last data event
      self.lastDataCharacteristic.on('data', (data, isNotification) => {
        const decoded = self._encryptDecrypt(self.cryptoKey, self.peripheral_address, data);

        let state = 0;
        let dim = 0;
        let device = parseInt(decoded[0], 10);

        if (decoded.toString('hex', 3, 5) === '00c8' || decoded.toString('hex', 3, 5) === '0098') {
          state = parseInt(decoded.toString('hex', 5, 6), 10);
          dim = parseInt(decoded.toString('hex', 6, 8), 16) >> 8;

          logger('d: ' + device + ' got state+dim update: ' + state + ' - ' + dim);
          this.emit('dimChanged', device, state, dim);
        }
        else if (decoded.toString('hex', 3, 5) === '0097') {
          state = parseInt(decoded.toString('hex', 5, 6), 10);
          logger('d: ' + device + ' got state update: ' + state);
          this.emit('stateChanged', device, state);
        }
      });
    });
  }

  async disconnect() {
    logger('disconnecting from Plejd');

    if (this.isConnected) {
      clearInterval(this.pingRef);

      if (this.peripheral) {
        try {
          // disconnect
          await this.peripheral.disconnect();

          // we need to reset the ble adapter too
          noble._bindings._hci.reset();

          // wait 200 ms for reset command to take effect :)
          sleep.msleep(200);

          // now we're ready to connect again
        }
        catch (error) {
          console.log('error: unable to disconnect from Plejd: ' + error);
          return Promise.resolve(false);
        }

        this.isConnected = false;
        logger('disconnected from Plejd');

        return Promise.resolve(true);
      }
    }
    else {
      clearInterval(this.pingRef);
      this.isConnected = false;
      logger('disconnected from Plejd');

      return Promise.resolve(true);
    }
  }

  async turnOn(id, brightness) {
    // if (this.peripheral.state !== 'connected') {
    //   console.log('warning: not connected, will connect. might take a few seconds.');
    //   await this.reinit();
    // }

    logger('turning on ' + id + ' at brightness ' + brightness);

    var payload;

    if (!brightness) {
      payload = Buffer.from((id).toString(16).padStart(2, '0') + '0110009701', 'hex');
    } else {
      brightness = brightness << 8 | brightness;
      payload = Buffer.from((id).toString(16).padStart(2, '0') + '0110009801' + (brightness).toString(16).padStart(4, '0'), 'hex');
    }

    this.write(payload);
  }

  async turnOff(id) {
    // if (this.peripheral.state !== 'connected') {
    //   console.log('warning: not connected, will connect. might take a few seconds.');
    //   await this.reinit();
    // }

    logger('turning off ' + id);

    var payload = Buffer.from((id).toString(16).padStart(2, '0') + '0110009700', 'hex');
    this.write(payload);
  }

  startPing() {
    const self = this;

    clearInterval(this.pingRef);
    logger('starting ping');
    this.pingRef = setInterval(async () => {
      logger('ping');
      if (self.peripheral.state == 'connected') {
        await self.plejdPing(async (pingOk) => {

          if (!pingOk) {
            console.log('error: ping failed');
            await self.disconnect();
            // await self.reinit();
          }
          else {
            logger('pong');
          }
        });
      }
      else {
        await self.disconnect();
      //   await self.reinit();
      }
    }, 3000);
  }

  async plejdPing(callback) {
    var ping = crypto.randomBytes(1);

    try {
      // make sure we're connected, otherwise, return false and reconnect
      // if (this.peripheral.state != 'connected') {
      //   callback(false);
      //   return;
      // }

      this.pingCharacteristic.write(ping, false, (err) => {
        if (err) {
          console.log('error: unable to send ping: ' + err);
          callback(false);
        }

        this.pingCharacteristic.read((err, data) => {
          if (err) {
            console.log('error: unable to read ping: ' + err);
            callback(false);
          }

          if (((ping[0] + 1) & 0xff) !== data[0]) {
            callback(false);
          }
          else {
            callback(true);
          }
        });
      });
    }
    catch (error) {
      console.log('error: writing to plejd: ' + error);
      callback(false);
    }
  }

  async authenticate() {
    const self = this;

    logger('authenticating connection');
    this.authCharacteristic.write(Buffer.from([0]), false, (err) => {
      if (err) {
        console.log('error: failed to authenticate: ' + err);
        return;
      }

      this.authCharacteristic.read(async (err2, data) => {
        if (err2) {
          console.log('error: challenge request failed: ' + err2);
          return;
        }

        var resp = self._challengeResponse(self.cryptoKey, data);

        this.authCharacteristic.write(resp, false, (err3) => {
          if (err3) {
            console.log('error: challenge failed: ' + err2);
            return;
          }

          this.emit('authenticated');
        });
      });
    });
  }

  async write(data) {
    const self = this;

    try {
      if (this.peripheral.state !== 'connected') {
        logger('adding message to queue.');
        this.writeQueue.push(data);
        return Promise.resolve(true);
      }

      if (!this.keepAlive) {
        logger('not connected to Plejd. reconnecting.');
        await this.connect();
      }

      logger('writing ' + data + ' to ' + this.peripheral.address);
      this.dataCharacteristic.write(this._encryptDecrypt(this.cryptoKey, this.peripheral_address, data), false);
      this.flush();

      if (!this.keepAlive) {
        clearTimeout(this.disconnectIntervalRef);
        this.disconnectIntervalRef = setTimeout(async () => {
          await self.disconnect();
        }, 5000);
      }
    }
    catch (error) {
      console.log('error: writing to plejd: ' + error);
      await self.disconnect();
      // await self.connect();
    }
  }

  async flush() {
    let writeData;
    while ((writeData = this.writeQueue.shift()) !== undefined) {
      this.dataCharacteristic.write(this._encryptDecrypt(this.cryptoKey, this.peripheral_address, writeData), false);
    }
  }

  _challengeResponse(key, challenge) {
    const intermediate = crypto.createHash('sha256').update(xor(key, challenge)).digest();
    const part1 = intermediate.subarray(0, 16);
    const part2 = intermediate.subarray(16);

    const resp = xor(part1, part2);

    return resp;
  }

  _encryptDecrypt(key, addr, data) {
    var buf = Buffer.concat([addr, addr, addr.subarray(0, 4)]);

    var cipher = crypto.createCipheriv("aes-128-ecb", key, '');
    cipher.setAutoPadding(false);

    var ct = cipher.update(buf).toString('hex');
    ct += cipher.final().toString('hex');
    ct = Buffer.from(ct, 'hex');

    var output = "";
    for (var i = 0, length = data.length; i < length; i++) {
      output += String.fromCharCode(data[i] ^ ct[i % 16]);
    }

    return Buffer.from(output, 'ascii');
  }

  _reverseBuffer(src) {
    var buffer = Buffer.allocUnsafe(src.length)

    for (var i = 0, j = src.length - 1; i <= j; ++i, --j) {
      buffer[i] = src[j]
      buffer[j] = src[i]
    }

    return buffer
  }
}

module.exports = { Controller };