node_helper.js

const NodeHelper = require("node_helper");
const HomeAssistant = require("homeassistant");
const HomeAssistantWS = require("homeassistant-ws");
const Logger = require("./helpers/Logger");

module.exports = NodeHelper.create({
  start,
  stop,
  socketNotificationReceived,
  connect,
  connectWs,
  getState,
  toggleState,
  setCoverPosition,
  setMediaPlayerVolume,
  mediaPlayerPlayPause,
  mediaPlayerNext,
  mediaPlayerPrevious,
  onStateChangedEvent,
});

function start() {
  this.logger = new Logger(this.name);
  this.connections = {};
}

function stop() {
  for (const connection in this.connections) {
    this.connections[connection].websocket.unsubscribeFromEvent(
      "state_changed"
    );
  }
}

function socketNotificationReceived(notification, payload) {
  this.logger.debug(`Recieved notification ${notification}`, payload);
  if (
    notification !== "CONNECT" &&
    (!payload.identifier || !this.connections[payload.identifier])
  ) {
    this.logger.error(`No connection for ${payload.identifier} found`);
    return;
  }

  switch (notification) {
    case "CONNECT":
      this.connect(payload);
      break;
    case "GET_STATE":
      this.getState(payload);
      break;
    case "TOGGLE_STATE":
      this.toggleState(payload);
      break;
    case "SET_COVER_POSITION":
      this.setCoverPosition(payload);
      break;
    case "SET_MEDIAPLAYER_VOLUME":
      this.setMediaPlayerVolume(payload);
      break;
    case "MEDIA_PLAYER_PLAYPAUSE":
      this.mediaPlayerPlayPause(payload);
      break;
    case "MEDIA_PLAYER_PREVIOUS":
      this.mediaPlayerPrevious(payload);
      break;
    case "MEDIA_PLAYER_NEXT":
      this.mediaPlayerNext(payload);
      break;
  }
}

async function connect(payload) {
  const connectionConfig = {
    host: payload.host,
    port: payload.port,
    token: payload.token,
    ignoreCert: payload.ignoreCert,
  };
  const hass = new HomeAssistant(connectionConfig);
  this.logger.info(`HomeAssistant connected for ${payload.identifier}`);
  this.connections[payload.identifier] = {
    hass,
    entities: [],
  };

  this.connectWs(connectionConfig, payload)
}

async function connectWs(connectionConfig, payload) {
  const self = this;
  HomeAssistantWS.default({
    ...connectionConfig,
    host: new URL(connectionConfig.host).host,
  })
    .then((hassWs) => {
      this.connections[payload.identifier].websocket = hassWs;
      hassWs.on("state_changed", onStateChangedEvent.bind(self));
      hassWs.on('ws_close', () => {
        this.logger.debug(`Lost connection for ${payload.identifier}... Trying to reconnect in 2secs`)
        setTimeout(() => {
          this.connectWs(connectionConfig, payload)
        }, 2000)
      })
    })
    .catch((err) => {
      this.logger.error(
        `WS connection for ${payload.identifier} failed...`
      );
      this.logger.debug(`Trying to reconnect for ${payload.identifier} in 2secs`)
      setTimeout(() => {
        this.connectWs(connectionConfig, payload)
      }, 2000)
    });
}

async function getState(payload) {
  this.logger.debug(`Getting state for ${payload.entity}`);
  const hass = this.connections[payload.identifier].hass;
  const [domain, entity] = payload.entity.split(".");
  const response = await hass.states.get(domain, entity);
  this.sendSocketNotification("GOT_STATE", {
    identifier: payload.identifier,
    data: response,
  });

  if (!this.connections[payload.identifier].entities.includes(payload.entity)) {
    this.connections[payload.identifier].entities.push(payload.entity);
  }
}

async function toggleState(payload) {
  this.logger.debug(`Toggling state for ${payload.entity}`);
  const hass = this.connections[payload.identifier].hass;
  const [domain, entity] = payload.entity.split(".");
  await hass.services.call("toggle", domain, entity);
  this.getState(payload);
}

async function setCoverPosition(payload) {
  this.logger.debug(
    `Setting position for cover ${payload.entity} to ${payload.position}`
  );
  const hass = this.connections[payload.identifier].hass;
  await hass.services.call("set_cover_position", "cover", {
    entity_id: payload.entity,
    position: payload.position,
  });
  this.getState(payload);
}

async function setMediaPlayerVolume(payload) {
  this.logger.debug(
    `Setting volume for media_player ${payload.entity} to ${payload.volume_level}`
  );
  const hass = this.connections[payload.identifier].hass;
  await hass.services.call("volume_set", "media_player", {
    entity_id: payload.entity,
    volume_level: payload.volume_level,
  });
}

async function mediaPlayerPlayPause(payload) {
  this.logger.debug(`Play/Pause for media_player ${payload.entity}`);
  const hass = this.connections[payload.identifier].hass;
  await hass.services.call("media_play_pause", "media_player", {
    entity_id: payload.entity
  });
}

async function mediaPlayerNext(payload) {
  this.logger.debug(`Next for media_player ${payload.entity}`);
  const hass = this.connections[payload.identifier].hass;
  await hass.services.call("media_next_track", "media_player", {
    entity_id: payload.entity
  });
}

async function mediaPlayerPrevious(payload) {
  this.logger.debug(`Next for media_player ${payload.entity}`);
  const hass = this.connections[payload.identifier].hass;
  await hass.services.call("media_previous_track", "media_player", {
    entity_id: payload.entity
  });
}


function onStateChangedEvent(event) {
  //this.logger.debug(`Got state change for ${event.data.entity_id}`);
  for (const connection in this.connections) {
    if (this.connections[connection].entities.includes(event.data.entity_id)) {
      this.logger.debug(
        `Found listening connection (${connection}) for entity ${event.data.entity_id}`
      );
      this.sendSocketNotification("CHANGED_STATE", {
        identifier: connection,
        data: event.data.new_state,
      });
    }
  }
}