homebridge-homeassistant-mbmccormick/accessories/switch.js

Homebridge plugin for Home Assistant: https://home-assistant.io

'use strict';

let Service;
let Characteristic;
let communicationError;

function HomeAssistantSwitch(log, data, client, type, firmware) {
  // device info
  this.domain = type;
  this.data = data;
  this.entity_id = data.entity_id;
  this.uuid_base = data.entity_id;
  this.firmware = firmware;
  if (data.attributes && data.attributes.friendly_name) {
    this.name = data.attributes.friendly_name;
  } else {
    this.name = data.entity_id.split('.').pop().replace(/_/g, ' ');
  }
  if (data.attributes && data.attributes.homebridge_manufacturer) {
    this.manufacturer = String(data.attributes.homebridge_manufacturer);
  } else {
    this.manufacturer = 'Home Assistant';
  }
  if (data.attributes && data.attributes.homebridge_serial) {
    this.serial = String(data.attributes.homebridge_serial);
  } else {
    this.serial = data.entity_id;
  }
  this.client = client;
  this.log = log;
}

HomeAssistantSwitch.prototype = {
  onEvent(oldState, newState) {
    if (newState.state) {
      this.service.getCharacteristic(Characteristic.On)
        .setValue(newState.state === 'on', null, 'internal');
    }
  },
  getPowerState(callback) {
    this.client.fetchState(this.entity_id, (data) => {
      if (data) {
        const powerState = data.state === 'on';
        callback(null, powerState);
      } else {
        callback(communicationError);
      }
    });
  },
  setPowerState(powerOn, callback, context) {
    if (context === 'internal') {
      callback();
      return;
    }

    const that = this;
    const serviceData = {};
    serviceData.entity_id = this.entity_id;
    var callDomain = this.domain === 'group' ? 'homeassistant' : this.domain;

    if (powerOn) {
      this.log(`Setting power state on the '${this.name}' to on`);

      this.client.callService(callDomain, 'turn_on', serviceData, (data) => {
        if (this.domain === 'scene' || (this.domain === 'script' && !(this.data.attributes.can_cancel))) {
          setTimeout(() => {
            this.service.getCharacteristic(Characteristic.On)
              .setValue(false, null, 'internal');
          }, 500);
        }
        if (data) {
          that.log(`Successfully set power state on the '${that.name}' to on`);
          callback();
        } else {
          callback(communicationError);
        }
      });
    } else {
      this.log(`Setting power state on the '${this.name}' to off`);

      this.client.callService(callDomain, 'turn_off', serviceData, (data) => {
        if (data) {
          that.log(`Successfully set power state on the '${that.name}' to off`);
          callback();
        } else {
          callback(communicationError);
        }
      });
    }
  },
  getServices() {
    let model;

    switch (this.domain) {
      case 'scene':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Scene';
        }
        break;
      case 'input_boolean':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Input Boolean';
        }
        break;
      case 'group':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Group';
        }
        break;
      case 'switch':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Switch';
        }
        break;
      case 'remote':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Remote';
        }
        break;
      case 'automation':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Automation';
        }
        break;
      case 'vacuum':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Vacuum';
        }
        break;
      case 'script':
        if (this.data.attributes && this.data.attributes.homebridge_model) {
          model = String(this.data.attributes.homebridge_model);
        } else {
          model = 'Script';
        }
        break;
      default:
        model = 'Switch';
    }

    this.service = new Service.Switch();
    if (this.data && this.data.attributes && this.data.attributes.homebridge_switch_type === 'outlet') {
      this.service = new Service.Outlet();
      if (this.data.attributes && this.data.attributes.homebridge_model) {
        model = String(this.data.attributes.homebridge_model);
      } else {
        model = 'Outlet';
      }
      this.service
        .getCharacteristic(Characteristic.OutletInUse)
        .on('get', this.getPowerState.bind(this));
    }
    const informationService = new Service.AccessoryInformation();

    informationService
      .setCharacteristic(Characteristic.Manufacturer, this.manufacturer)
      .setCharacteristic(Characteristic.Model, model)
      .setCharacteristic(Characteristic.SerialNumber, this.serial)
      .setCharacteristic(Characteristic.FirmwareRevision, this.firmware);

    if (this.domain === 'remote' || this.domain === 'switch' || this.domain === 'input_boolean' || this.domain === 'group' || this.domain === 'automation' || this.domain === 'vacuum' || (this.domain === 'script' && this.data.attributes.can_cancel)) {
      this.service
        .getCharacteristic(Characteristic.On)
        .on('get', this.getPowerState.bind(this))
        .on('set', this.setPowerState.bind(this));
    } else {
      this.service
        .getCharacteristic(Characteristic.On)
        .on('set', this.setPowerState.bind(this));
    }

    return [informationService, this.service];
  },

};

function HomeAssistantSwitchPlatform(oService, oCharacteristic, oCommunicationError) {
  Service = oService;
  Characteristic = oCharacteristic;
  communicationError = oCommunicationError;

  return HomeAssistantSwitch;
}

module.exports = HomeAssistantSwitchPlatform;
module.exports.HomeAssistantSwitch = HomeAssistantSwitch;