MIgrate adapter to Typescript
This commit is contained in:
GermanBluefox
+248
@@ -0,0 +1,248 @@
|
||||
import { EventEmitter } from 'node:events';
|
||||
import WebSocket from 'ws';
|
||||
|
||||
interface HassOptions {
|
||||
host: string;
|
||||
port: number;
|
||||
password?: string;
|
||||
secure?: boolean;
|
||||
}
|
||||
|
||||
interface HassRequest {
|
||||
type: string;
|
||||
ts: number;
|
||||
cb?: (err: boolean | string | null, result?: any) => void;
|
||||
}
|
||||
|
||||
/*
|
||||
const ERRORS: Record<number, string> = {
|
||||
1: 'ERR_CANNOT_CONNECT',
|
||||
2: 'ERR_INVALID_AUTH',
|
||||
3: 'ERR_CONNECTION_LOST',
|
||||
};
|
||||
*/
|
||||
export default class HASS extends EventEmitter {
|
||||
private socket: WebSocket | null = null;
|
||||
private readonly options: HassOptions;
|
||||
private readonly log: ioBroker.Logger;
|
||||
private currentId: number = 1;
|
||||
private readonly requests: Record<number, HassRequest> = {};
|
||||
private _connected: boolean = false;
|
||||
private connectTimeout: ReturnType<typeof setTimeout> | null = null;
|
||||
private closed: boolean = false;
|
||||
|
||||
constructor(options: HassOptions, log: ioBroker.Logger) {
|
||||
super();
|
||||
this.options = {
|
||||
host: options.host || '127.0.0.1',
|
||||
port: parseInt(String(options.port), 10) || 8123,
|
||||
password: options.password,
|
||||
secure: options.secure,
|
||||
};
|
||||
this.log = log;
|
||||
}
|
||||
|
||||
private subscribeEvents(socket: WebSocket, callback?: (err: boolean | string | null) => void): void {
|
||||
if (socket && typeof socket.send === 'function') {
|
||||
const id = this.currentId++;
|
||||
this.requests[id] = { type: 'subscribe_events', ts: Date.now(), cb: callback };
|
||||
socket.send(
|
||||
JSON.stringify({
|
||||
id,
|
||||
type: 'subscribe_events',
|
||||
}),
|
||||
);
|
||||
} else {
|
||||
callback?.('not connected');
|
||||
}
|
||||
}
|
||||
|
||||
private sendCommand(
|
||||
socket: WebSocket | null,
|
||||
type: string,
|
||||
callback?: (err: boolean | string | null, result?: any) => void,
|
||||
extra?: Record<string, any>,
|
||||
): void {
|
||||
if (socket && typeof socket.send === 'function') {
|
||||
const id = this.currentId++;
|
||||
this.requests[id] = { type, cb: callback, ts: Date.now() };
|
||||
socket.send(
|
||||
JSON.stringify({
|
||||
id,
|
||||
type,
|
||||
...extra,
|
||||
}),
|
||||
);
|
||||
} else {
|
||||
callback?.('not connected');
|
||||
}
|
||||
}
|
||||
|
||||
private sendAuth(socket: WebSocket, pass: string): void {
|
||||
if (socket && typeof socket.send === 'function') {
|
||||
socket.send(
|
||||
JSON.stringify({
|
||||
type: 'auth',
|
||||
access_token: pass,
|
||||
}),
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
private initSocket(socket: WebSocket): void {
|
||||
socket.on('message', (msg: WebSocket.Data): void => {
|
||||
const msgStr = (msg as Buffer).toString();
|
||||
this.log.silly(msgStr);
|
||||
|
||||
const response = JSON.parse(msgStr);
|
||||
if (response.type === 'event') {
|
||||
if (response.event?.data && response.event.event_type === 'system_log_event') {
|
||||
if (response.event.data.level === 'WARNING') {
|
||||
this.log.warn(`EVENT: ${response.event.data.message}`);
|
||||
} else if (response.event.data.level === 'ERROR') {
|
||||
this.log.error(`EVENT: ${response.event.data.message}`);
|
||||
} else {
|
||||
this.log.debug(`EVENT: ${response.event.data.message}`);
|
||||
}
|
||||
} else if (response.event?.event_type === 'state_changed') {
|
||||
this.emit('state_changed', response.event.data.new_state);
|
||||
}
|
||||
} else if (response.type === 'auth_required') {
|
||||
if (!this.options.password) {
|
||||
this.emit('error', 'Password required. Connection closed');
|
||||
socket.terminate();
|
||||
} else {
|
||||
setTimeout(() => this.sendAuth(socket, this.options.password!), 50);
|
||||
}
|
||||
} else if (response.type === 'auth_ok') {
|
||||
setImmediate(() =>
|
||||
this.subscribeEvents(socket, err => {
|
||||
if (!err) {
|
||||
this._connected = true;
|
||||
this.emit('connected');
|
||||
}
|
||||
}),
|
||||
);
|
||||
} else if (response.id === undefined) {
|
||||
this.log.error(`Invalid answer: ${msgStr}`);
|
||||
} else {
|
||||
if (response.type === 'result' && this.requests[response.id]) {
|
||||
this.log.debug(
|
||||
`got answer for ${this.requests[response.id].type} success = ${response.success}, result = ${JSON.stringify(response.result)}`,
|
||||
);
|
||||
if (typeof this.requests[response.id].cb === 'function') {
|
||||
this.requests[response.id].cb!(!response.success, response.result);
|
||||
delete this.requests[response.id];
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
socket.on('error', (err: Error) => {
|
||||
this.socket = null;
|
||||
if (err?.message?.indexOf('RSV2 and RSV3 must be clear') !== -1) {
|
||||
// ignore deflate error
|
||||
} else {
|
||||
this.log.error(err.toString());
|
||||
}
|
||||
});
|
||||
|
||||
socket.on('open', () => {
|
||||
// connection opened
|
||||
});
|
||||
|
||||
socket.on('close', () => {
|
||||
this.socket = null;
|
||||
if (this._connected) {
|
||||
this._connected = false;
|
||||
this.emit('disconnected');
|
||||
}
|
||||
if (!this.connectTimeout && !this.closed) {
|
||||
this.connectTimeout = setTimeout(() => {
|
||||
this.connectTimeout = null;
|
||||
this.connect();
|
||||
}, 3000);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
isConnected(): boolean {
|
||||
return this._connected;
|
||||
}
|
||||
|
||||
getConfig(callback: (err: boolean | string | null, result?: any) => void): void {
|
||||
if (!this._connected) {
|
||||
callback('not connected');
|
||||
} else {
|
||||
this.sendCommand(this.socket, 'get_config', callback);
|
||||
}
|
||||
}
|
||||
|
||||
getStates(callback: (err: boolean | string | null, result?: any) => void): void {
|
||||
if (!this._connected) {
|
||||
callback('not connected');
|
||||
} else {
|
||||
this.sendCommand(this.socket, 'get_states', callback);
|
||||
}
|
||||
}
|
||||
|
||||
getServices(callback: (err: boolean | string | null, result?: any) => void): void {
|
||||
if (!this._connected) {
|
||||
callback('not connected');
|
||||
} else {
|
||||
this.sendCommand(this.socket, 'get_services', callback);
|
||||
}
|
||||
}
|
||||
|
||||
getPanels(callback: (err: boolean | string | null, result?: any) => void): void {
|
||||
if (!this._connected) {
|
||||
callback('not connected');
|
||||
} else {
|
||||
this.sendCommand(this.socket, 'get_panels', callback);
|
||||
}
|
||||
}
|
||||
|
||||
callService(
|
||||
service: string,
|
||||
domain: string,
|
||||
serviceData: Record<string, any>,
|
||||
target: Record<string, any>,
|
||||
callback: (err: boolean | string | null, result?: any) => void,
|
||||
): void {
|
||||
if (!this._connected) {
|
||||
callback('not connected');
|
||||
} else {
|
||||
this.sendCommand(this.socket, 'call_service', callback, {
|
||||
domain: domain || '',
|
||||
service,
|
||||
service_data: serviceData,
|
||||
target,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
connect(): void {
|
||||
if (this.connectTimeout) {
|
||||
clearTimeout(this.connectTimeout);
|
||||
this.connectTimeout = null;
|
||||
}
|
||||
|
||||
this.socket = new WebSocket(
|
||||
`ws${this.options.secure ? 's' : ''}://${this.options.host}:${this.options.port}/api/websocket`,
|
||||
{ perMessageDeflate: false },
|
||||
);
|
||||
|
||||
this.initSocket(this.socket);
|
||||
}
|
||||
|
||||
close(): void {
|
||||
if (this.connectTimeout) {
|
||||
clearTimeout(this.connectTimeout);
|
||||
this.connectTimeout = null;
|
||||
}
|
||||
this.closed = true;
|
||||
if (this.socket) {
|
||||
this.socket.close();
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user