homebridge-broadlink-rm/helpers/learnRFData.js

Broadlink RM plugin (including the mini and pro) for homebridge: https://github.com/nfarina/homebridge

const { getDevice } = require('./getDevice');

let closeClient = null;
let isClosingClient = false;
let timeout = null;
let getDataTimeout = null;
let getDataTimeout2 = null;
let getDataTimeout3 = null;

let currentDevice

const stop = (log, device) => {
  // Reset existing learn requests
  if (!closeClient || isClosingClient) return;

  isClosingClient = true;

  if (currentDevice) currentDevice.cancelLearn();

  setTimeout(() => {
  if (!closeClient) return;

    closeClient();
    closeClient = null;
    isClosingClient = false;

    if (log) log(`\x1b[35m[INFO]\x1b[0m Scan RF (stopped)`);
  }, 500)
}

const start = (host, callback, turnOffCallback, log, disableTimeout) => {
  stop()

  // Get the Broadlink device
  const device = getDevice({ host, log, learnOnly: true })
  if (!device) { 
    return log(`\x1b[35m[INFO]\x1b[0m Learn Code (Couldn't learn code, device not found)`);
  }

  if (!device.enterLearning) return log(`\x1b[31m[ERROR]\x1b[0m Learn Code (IR/RF learning not supported for device at ${host})`);
  if (!device.enterRFSweep) return log(`\x1b[31m[ERROR]\x1b[0m Scan RF (RF learning not supported for device (${device.type}) at ${host})`);

  currentDevice = device

  let onRawData;
  let onRawData3;

  closeClient = (err) => {
    if (timeout) clearTimeout(timeout);
    timeout = null;

    if (getDataTimeout) clearTimeout(getDataTimeout);
    getDataTimeout = null;

    if (getDataTimeout2) clearTimeout(getDataTimeout2);
    getDataTimeout2 = null;

    if (getDataTimeout3) clearTimeout(getDataTimeout3);
    getDataTimeout2 = null;


    device.removeListener('rawRFData', onRawData);
    device.removeListener('rawData', onRawData3);
  };

  onRawData = (message) => {
    if (!closeClient) return;

    if (getDataTimeout) clearTimeout(getDataTimeout);
    getDataTimeout = null;

    log(`\x1b[35m[INFO]\x1b[0m Scan RF (found frequency - 1 of 2)`);
    log(`\x1b[35m[ACTION]\x1b[0m Keep holding that button!`)

    getDataTimeout2 = setTimeout(() => {
      getData2(device);
    }, 1000);

    // Temporary work-around. 
    setTimeout(() => {
      log(`\x1b[35m[INFO]\x1b[0m Scan RF (found frequency - 2 of 2)`)
      log(`\x1b[35m[ACTION]\x1b[0m Press the RF button multiple times with a pause between them.`);

      getDataTimeout3 = setTimeout(() => {
        getData3(device);
      }, 1000);
    }, 8000)


    return device.enterLearning();
  };

  onRawData3 = (message) => {
    console.log('!!! onRawData3')
    if (!closeClient) return;

    const hex = message.toString('hex');
    log(`\x1b[35m[INFO]\x1b[0m Scan RF (complete)`);
    log(`\x1b[35m[RESULT]\x1b[0m Hex Code: ${hex}`);

    device.cancelLearn();

    closeClient();

    turnOffCallback();
  };

  device.on('rawRFData', onRawData);
  device.on('rawData', onRawData3);

  device.enterRFSweep();
  log(`\x1b[35m[INFO]\x1b[0m Scan RF (scanning)`);
  log(`\x1b[35m[ACTION]\x1b[0m Hold down the button that sends the RF frequency.`);

  if (callback) callback();

  getDataTimeout = setTimeout(() => {
    getData(device);
  }, 1000);

  if (disableTimeout) return;

  // Timeout the client after 20 seconds
  timeout = setTimeout(() => {
    device.cancelLearn()

    setTimeout(() => {
      log('\x1b[35m[INFO]\x1b[0m Scan RF (stopped - 20s timeout)');
      closeClient();

      turnOffCallback();
    }, 1000);
  }, 60 * 1000); // 20s
}

const getData = (device) => {
  if (getDataTimeout) clearTimeout(getDataTimeout);
  if (!closeClient) return;

  device.checkRFData();

  getDataTimeout = setTimeout(() => {
    getData(device);
  }, 1000);
}

const getData2 = (device) => {
  if (getDataTimeout2) clearTimeout(getDataTimeout2);
  if (!closeClient) return;

  device.checkRFData2();

  getDataTimeout2 = setTimeout(() => {
    getData2(device);
  }, 1000);
}

const getData3 = (device) => {
  if (getDataTimeout3) clearTimeout(getDataTimeout3);
  if (!closeClient) return;

  device.checkData()

  getDataTimeout3 = setTimeout(() => {
    getData3(device);
  }, 1000);
}

module.exports = { start, stop }