node-red-contrib-home-assistant-websocket
Version:
Node-RED integration with Home Assistant through websocket and REST API
150 lines (149 loc) • 7.08 kB
JavaScript
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.default = triggerState;
const joi_1 = __importDefault(require("joi"));
const const_1 = require("../../common/const");
const helpers_1 = require("../../common/controllers/helpers");
const ConfigError_1 = __importDefault(require("../../common/errors/ConfigError"));
const inputErrorHandler_1 = require("../../common/errors/inputErrorHandler");
const ClientEvents_1 = __importDefault(require("../../common/events/ClientEvents"));
const Events_1 = __importDefault(require("../../common/events/Events"));
const ComparatorService_1 = __importDefault(require("../../common/services/ComparatorService"));
const InputService_1 = __importDefault(require("../../common/services/InputService"));
const State_1 = __importDefault(require("../../common/State"));
const EventStatus_1 = __importDefault(require("../../common/status/EventStatus"));
const Status_1 = require("../../common/status/Status");
const TransformState_1 = __importDefault(require("../../common/TransformState"));
const globals_1 = require("../../globals");
const migrate_1 = require("../../helpers/migrate");
const node_1 = require("../../helpers/node");
const homeAssistant_1 = require("../../homeAssistant");
const Websocket_1 = require("../../homeAssistant/Websocket");
const const_2 = require("./const");
const helpers_2 = require("./helpers");
const TriggerStateController_1 = __importDefault(require("./TriggerStateController"));
const TriggerStateStatus_1 = __importDefault(require("./TriggerStateStatus"));
const testingSchema = joi_1.default.object({
payload: joi_1.default.object({
entity_id: joi_1.default.string().required(),
old_state: joi_1.default.object({
state: joi_1.default.string().required(),
}).unknown(true),
new_state: joi_1.default.object({
state: joi_1.default.string().required(),
}).unknown(true),
}),
}).unknown(true);
const enableSchema = joi_1.default.object({
payload: joi_1.default.string().valid(const_2.ENABLE, const_2.DISABLE),
}).unknown(true);
function triggerState(config) {
var _a, _b, _c, _d, _e, _f;
globals_1.RED.nodes.createNode(this, config);
this.config = (0, migrate_1.migrate)(config);
if (!((_b = (_a = this.config) === null || _a === void 0 ? void 0 : _a.entities[const_1.IdSelectorType.Entity]) === null || _b === void 0 ? void 0 : _b.length) &&
!((_d = (_c = this.config) === null || _c === void 0 ? void 0 : _c.entities[const_1.IdSelectorType.Substring]) === null || _d === void 0 ? void 0 : _d.length) &&
!((_f = (_e = this.config) === null || _e === void 0 ? void 0 : _e.entities[const_1.IdSelectorType.Regex]) === null || _f === void 0 ? void 0 : _f.length)) {
const error = new ConfigError_1.default('trigger-state.error.entity_required');
this.status({
fill: Status_1.StatusColor.Red,
shape: Status_1.StatusShape.Ring,
text: error.statusMessage,
});
throw error;
}
const serverConfigNode = (0, node_1.getServerConfigNode)(this.config.server);
const homeAssistant = (0, homeAssistant_1.getHomeAssistant)(serverConfigNode);
const exposeAsConfigNode = (0, node_1.getExposeAsConfigNode)(this.config.exposeAsEntityConfig);
const clientEvents = new ClientEvents_1.default({
node: this,
emitter: homeAssistant.eventBus,
});
let status;
let state;
if (exposeAsConfigNode) {
status = new EventStatus_1.default({
clientEvents,
config: serverConfigNode.config,
exposeAsEntityConfigNode: exposeAsConfigNode,
node: this,
});
}
else {
// create a state to track the enabled state of the node
// only needed if the node is not exposed to HA
const events = new Events_1.default({
node: this,
emitter: this,
});
state = new State_1.default(this);
status = new TriggerStateStatus_1.default({
clientEvents,
config: serverConfigNode.config,
events,
exposeAsEntityConfigNode: exposeAsConfigNode,
node: this,
state,
});
}
clientEvents.setStatus(status);
exposeAsConfigNode === null || exposeAsConfigNode === void 0 ? void 0 : exposeAsConfigNode.integration.setStatus(status);
const controllerDeps = (0, helpers_1.createControllerDependencies)(this, homeAssistant);
const transformState = new TransformState_1.default(serverConfigNode.config.ha_boolean);
const comparatorService = new ComparatorService_1.default({
nodeRedContextService: controllerDeps.nodeRedContextService,
homeAssistant,
jsonataService: controllerDeps.jsonataService,
transformState,
});
const inputService = new InputService_1.default({
nodeConfig: this.config,
});
const controller = new TriggerStateController_1.default({
comparatorService,
inputService,
node: this,
state,
status,
transformState,
...controllerDeps,
});
controller.setExposeAsConfigNode(exposeAsConfigNode);
// register custom inputs if inputs are enabled
if (this.config.enableInput) {
controller.addOptionalInput('enable', enableSchema, controller.onInputEnabled.bind(controller));
controller.addOptionalInput('testing', testingSchema, controller.onInputTesting.bind(controller));
}
if (this.config.entities[const_1.IdSelectorType.Substring].length === 0 &&
this.config.entities[const_1.IdSelectorType.Regex].length === 0) {
for (const entity of this.config.entities[const_1.IdSelectorType.Entity]) {
const eventTopic = `ha_events:state_changed:${entity}`;
clientEvents.addListener(eventTopic, controller.onEntityStateChanged.bind(controller));
}
}
else {
clientEvents.addListener('ha_events:state_changed', controller.onEntityStateChanged.bind(controller));
}
if (controller.isEnabled && this.config.outputInitially) {
const generateStateChanges = async () => {
const events = (0, helpers_2.createStateChangeEvents)(homeAssistant);
for (const event of events) {
await controller.onEntityStateChanged(event).catch((e) => {
const { error, statusMessage } = (0, inputErrorHandler_1.getErrorData)(e);
status.setError(statusMessage);
this.error(error);
});
}
};
// Here for when the node is deploy without the server config being deployed
if (homeAssistant.isHomeAssistantRunning) {
generateStateChanges();
}
else {
clientEvents.addListener(Websocket_1.ClientEvent.InitialConnectionReady, generateStateChanges);
}
}
}