homebridge-smartsystem 6.8.3 → 6.8.5

package/README.md

@@ -307,6 +307,8 @@ Homebridge UI version
 - 1: max/min values for hbValues
 - 2: HB up/down -> don't respond to motor unit's, only macro
 - 3: remember request before the login kicked in (and redirect to that one, once the login was succesful)
+- 4: now using only TCP sockets to the server
+- 5: making it more robust / inline with repo-service version
 
 ## Hardware 
 A Raspberry Pi that connects to a Duotecno IP Node 

package/config.json

@@ -461,11 +461,11 @@
   "p1": false,
   "proxy": {
     "cloudServer": "ws.duotecno.eu",
-    "cloudPort": 5098,
+    "cloudPort": 5097,
     "masterAddress": "192.168.0.97",
     "masterPort": 5001,
     "uniqueId": "",
     "debug": false,
-    "kind": "solo"
+    "kind": "gw"
   }
 }

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "homebridge-smartsystem",
   "displayname": "Duotecno Bridge",
-  "version": "6.8.3",
-  "description": "SmartServer (Proxy Websockets to TCP sockets, Smappee MQTT, Duotecno IP Nodes, Homekit interface)",
+  "version": "6.8.5",
+  "description": "SmartServer (Proxy TCP sockets to the cloud, Smappee MQTT, Duotecno IP Nodes, Homekit interface)",
   "main": "index.js",
   "author": "Johan Coppieters",
   "homepage": "http://www.duotecno.be",

package/server/proxy.js

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Context = exports.cleanStart = exports.makeNewCloudConnection = exports.setProxyConfig = exports.kEmptyProxy = void 0;
 const net = require("net");
-const WebSocket = require("ws");
 const child_process_1 = require("child_process");
 const process_1 = require("process");
 exports.kEmptyProxy = {
@@ -61,36 +60,38 @@ function error(str) {
 ///////////////////////////////
 // Start up the proxy server //
 ///////////////////////////////
-// if (config.kind != "gw") {
-//   makeNewCloudConnection(config.masterAddress, config.masterPort, config.cloudServer, config.cloudPort, config.uniqueId);
-// }
-function makeNewCloudConnection(master, masterPort, server, serverPort, uniqueId) {
-    const cloudSocket = new WebSocket('ws://' + server + ':' + serverPort + "/" + uniqueId);
-    cloudSocket.on('open', () => {
+if (config.uniqueId) {
+    makeNewCloudConnection(config.masterAddress, config.masterPort, config.cloudServer, config.cloudPort, config.uniqueId);
+}
+function makeNewCloudConnection(master, masterPort, server, serverPort, uniqueId, count = 0) {
+    const cloudSocket = new net.Socket();
+    cloudSocket.connect(serverPort, server, () => {
         log(`[PROXY] → [CLOUD] Connected to Cloud at ${server}:${serverPort}`);
+        // Send unique ID as the first message to identify this proxy/device
+        cloudSocket.write(`[${uniqueId}]`);
+        debug(`[PROXY] → [CLOUD] Sent unique ID: ${uniqueId}`);
         const context = new Context(cloudSocket, master, masterPort, server, serverPort, uniqueId);
         gCloudConnections.push(context);
         log(`[PROXY] → [CLOUD] New free connection #${gCloudConnections.length} to the Cloud at ${server}:${serverPort}`);
     });
-    log(`[PROXY] → [CLOUD] Attempting to start a new free connection for ${config.uniqueId} to the Cloud at ${server}:${serverPort}`);
+    cloudSocket.on('error', (err) => {
+        error(`[PROXY] → [CLOUD] Failed to connect ${count + 1} times to the Cloud server: ${err.message}`);
+        if (count < 3) {
+            setTimeout(() => {
+                makeNewCloudConnection(master, masterPort, server, serverPort, uniqueId, count + 1);
+            }, 1000);
+        }
+    });
+    log(`[PROXY] → [CLOUD] Attempt ${count + 1} to start a new free connection for ${config.uniqueId} to the Cloud at ${server}:${serverPort}`);
 }
 exports.makeNewCloudConnection = makeNewCloudConnection;
 ////////////////
 // Restarting //
 ////////////////
 function cleanStart(restart = false) {
-    if (gCloudConnections.length) {
-        gCloudConnections.forEach(context => {
-            if (context.deviceSocket) {
-                context.deviceSocket.end();
-                context.deviceSocket = null;
-            }
-            context.cloudSocket.close();
-            context.cloudSocket = null;
-        });
-        gCloudConnections.splice(0, gCloudConnections.length); // clear the array
-        log(`[PROXY] → [CLOUD] Cleaned up all cloud connections.`);
-    }
+    gCloudConnections.forEach(ctx => ctx.cleanupSockets());
+    log(`[PROXY] → [CLOUD] Cleaned up all ${gCloudConnections.length} cloud connections.`);
+    gCloudConnections.splice(0, gCloudConnections.length); // clear the array
     if (restart) {
         if (config.uniqueId) {
             makeNewCloudConnection(config.masterAddress, config.masterPort, config.cloudServer, config.cloudPort, config.uniqueId);
@@ -117,11 +118,11 @@ class Context {
     }
     setupCloudSocket() {
         // set up handlers for the cloud socket
-        this.cloudSocket.on('message', message => {
-            this.handleDataFromCloud(message);
+        this.cloudSocket.on('data', (data) => {
+            this.handleDataFromCloud(data);
         });
-        this.cloudSocket.on('close', (code, reason) => {
-            warning('[CLOUD] Connection closed: ' + code + ' ' + reason);
+        this.cloudSocket.on('close', () => {
+            warning('[CLOUD] Connection closed');
             if (this.deviceSocket) {
                 this.deviceSocket.end();
                 this.deviceSocket = null;
@@ -136,27 +137,32 @@ class Context {
         });
     }
     handleDataFromCloud(data) {
-        debug(`[CLOUD] → [PROXY]: Data from Cloud: ${data.toString().slice(0, -1)}`);
+        debug(`[CLOUD] → [PROXY]: Data from Cloud: ${data.toString()}`);
         if (!this.deviceSocket) {
             // we either have a real client that wants to connect to the device,
             // or we receive a heartbeat from the cloud server
+            // or we receive a connection response [OK], [OK-=2], [ERROR-...]
             if (this.isHeartbeatRequest(data)) {
                 debug(`[CLOUD] → [PROXY]: Received heartbeat from Cloud, returning response.`);
                 // respond to the heartbeat of the cloud server
-                this.cloudSocket.send("[72,3]"); // command([Command.HeartbeatStatus, CommandMethod.Heartbeat.server])
+                this.cloudSocket.write("[72,3]"); // command([Command.HeartbeatStatus, CommandMethod.Heartbeat.server])
+            }
+            else if (this.isConnectionResponse(data)) {
+                debug(`[CLOUD] → [PROXY]: Received connection response from Cloud: ${data.toString()}`);
+                // Handle connection response from the cloud server
             }
             else {
-                // There is incomming data, it's a fresh connection, so: a new client wants to connect to the device
+                // There is real incomming data, it's a fresh connection, so: a new client wants to connect to the device
                 // 1) create a new (free) connection to the cloud server for the next client
                 // 2) connect to the device and send the initial data
-                log(`[PROXY] → [CLOUD] New client connection, creating new free connection for this proxy/device: ${this.uniqueId}.`);
+                log(`[PROXY] → [CLOUD] New client connection, creating new free connection for this proxy/device: ${config.uniqueId}.`);
                 makeNewCloudConnection(this.master, this.masterPort, this.server, this.serverPort, this.uniqueId);
                 this.makeDeviceConnection(this.master, this.masterPort, data);
             }
         }
         else if (this.deviceSocket.readyState === 'open') {
             this.deviceSocket.write(data);
-            debug(`[PROXY] → [DEVICE] forwarding data to device: ${data.toString().slice(0, -1)}`);
+            debug(`[PROXY] → [DEVICE] forwarding data to device: ${data.toString().trim()}`);
         }
         else {
             warning(`[PROXY] → [DEVICE] Device socket is not open yet, waiting for connection... what to do with the data??`);
@@ -168,9 +174,14 @@ class Context {
             this.deviceSocket = new net.Socket();
             // Connect to local device and send the data that came in from the cloud
             this.deviceSocket.connect(port, address, () => {
+                // Check if socket is still valid (not cleaned up by removeConnection)
+                if (!this.deviceSocket) {
+                    warning(`[PROXY] → [DEVICE] Device socket was cleaned up before connection completed`);
+                    return;
+                }
                 log(`[PROXY] → [DEVICE] Connected to local device at ${address}:${port}`);
                 this.setUpDeviceSocket();
-                log(`[PROXY] → [DEVICE] Sending initial message: ${data.toString().slice(0, -1)}`);
+                log(`[PROXY] → [DEVICE] Sending initial message: ${data.toString().trim()}`);
                 this.deviceSocket.write(data);
             });
         }
@@ -183,10 +194,14 @@ class Context {
         }
     }
     setUpDeviceSocket() {
+        if (!this.deviceSocket) {
+            error(`[PROXY] → [DEVICE] Cannot set up device socket - socket is null`);
+            return;
+        }
         this.deviceSocket.on('data', (data) => {
             const message = data.toString('utf-8');
-            debug(`[DEVICE] → [PROXY] forwarding data to Cloud: ${message.slice(0, -1)}`);
-            this.cloudSocket.send(message);
+            debug(`[DEVICE] → [PROXY] forwarding data to Cloud: ${message.trim()}`);
+            this.cloudSocket.write(data);
         });
         this.deviceSocket.on('close', () => {
             log(`[DEVICE] → [PROXY] Device socket closed`);
@@ -197,27 +212,43 @@ class Context {
             this.removeConnection();
         });
     }
+    cleanupSockets() {
+        // Clean up the device socket
+        if (this.deviceSocket) {
+            if (!this.deviceSocket.destroyed) {
+                this.deviceSocket.removeAllListeners();
+                this.deviceSocket.end();
+                this.deviceSocket.destroy();
+            }
+            this.deviceSocket = null;
+        }
+        // Clean up the cloud socket
+        if (this.cloudSocket) {
+            if (!this.cloudSocket.destroyed) {
+                this.cloudSocket.removeAllListeners();
+                this.cloudSocket.end();
+                this.cloudSocket.destroy();
+            }
+            this.cloudSocket = null;
+        }
+    }
     removeConnection() {
-        this.deviceSocket = null;
-        // remove this context and close the cloud socket
         const index = gCloudConnections.indexOf(this);
-        if (index !== -1) {
-            this.cloudSocket.close();
-            this.cloudSocket = null;
+        if (index !== -1)
             gCloudConnections.splice(index, 1);
-            log(`[PROXY] → [CLOUD] Closed socket to Cloud and removed context for device ${this.master}`);
-        }
+        this.cleanupSockets();
+        log(`[PROXY] → [CLOUD] Closed socket to Cloud and removed context for device ${this.master}`);
         if (gCloudConnections.length === 0) {
             log(`[PROXY] → [CLOUD] No more cloud connections, restarting proxy.`);
             // just to be sure: restart...
             if (config.kind === "pm2") {
                 // PM2 should restart us.
                 log(`[PROXY] → [CLOUD] PM2 should restart us, exiting now.`);
-                process.exit();
+                setTimeout(() => process.exit(0), 100);
             }
             else if (config.kind === "solo") {
                 log(`[PROXY] → [CLOUD] Running solo, restarting proxy.`);
-                this.restart();
+                restart();
             }
             else {
                 log(`[PROXY] → [CLOUD] Running under gateway/homebridge, not restarting... not sure what to do here.`);
@@ -227,31 +258,71 @@ class Context {
     ///////////////
     // Utilities //
     ///////////////
-    restart() {
-        console.log('🔄 Restarting...');
+    // check if it's a heartbeat request from the server
+    isHeartbeatRequest(data) {
+        const msg = data.toString('utf-8').trim();
+        return msg === '[215,3]';
+        // hearbeatKind = (poll = 1, poll_old = 2, serve = 3)
+        // return ((data[0] === 91) &&           // [
+        //         (data[1] === 50) &&           // 2
+        //         (data[2] === 49) &&           // 1
+        //         (data[3] === 53) &&           // 5
+        //         (data[4] === 44) &&           // ,
+        //         (data[5] === 51) &&           // 3
+        //         (data[6] === 93));            // ]
+    }
+    // check if it's a server response after connecting and sending our uniqueID
+    isConnectionResponse(data) {
+        const message = data.toString();
+        // Check for various server response patterns
+        return (message.startsWith('[OK')) || (message.startsWith('[ERROR'));
+    }
+}
+exports.Context = Context;
+function restart() {
+    console.log('🔄 Restarting...');
+    setTimeout(() => {
         (0, child_process_1.spawn)(process_1.execPath, process_1.argv.slice(1), {
             cwd: (0, process_1.cwd)(),
             detached: true,
             stdio: 'inherit',
         });
-        process.exit();
+        process.exit(0);
+    }, 100);
+}
+// Graceful shutdown handler
+function handleShutdown(signal) {
+    log(`[PROXY] → [SYSTEM] Received ${signal}, shutting down gracefully...`);
+    // Close all cloud connections
+    gCloudConnections.forEach(ctx => ctx.cleanupSockets());
+    gCloudConnections.splice(0, gCloudConnections.length);
+    log(`[PROXY] → [SYSTEM] All connections closed. Exiting.`);
+    if (config.kind === "solo") {
+        log(`[PROXY] → [SYSTEM] Restarting proxy in solo mode...`);
+        restart();
     }
-    // check if it's a heartbeat request from the server
-    isHeartbeatRequest(data) {
-        // heartbeatKind = (poll = 1, poll_old = 2, serve = 3)
-        if (typeof data === 'string') {
-            return data === "[215,3]";
-        }
-        else {
-            return ((data[0] === 91) && // [
-                (data[1] === 50) && // 2
-                (data[2] === 49) && // 1
-                (data[3] === 53) && // 5
-                (data[4] === 44) && // ,
-                (data[5] === 51) && // 3
-                (data[6] === 93)); // ]
-        }
+    else {
+        log(`[PROXY] → [SYSTEM] Exiting without restart.`);
+        setTimeout(() => process.exit(0), 100);
     }
 }
-exports.Context = Context;
+// Listen for termination signals
+process.on('SIGINT', () => handleShutdown('SIGINT'));
+process.on('SIGTERM', () => handleShutdown('SIGTERM'));
+process.on('SIGINT', (err) => {
+    error(`[SYSTEM] received SIGINT`);
+    handleShutdown('SIGINT');
+});
+process.on('SIGTERM', (err) => {
+    error(`[SYSTEM] received SIGTERM`);
+    handleShutdown('SIGTERM');
+});
+process.on('uncaughtException', (err) => {
+    error(`[SYSTEM] Uncaught Exception: ${err.message}`);
+    handleShutdown('uncaughtException');
+});
+process.on('unhandledRejection', (reason) => {
+    error(`[SYSTEM] Unhandled Rejection: ${reason}`);
+    handleShutdown('unhandledRejection');
+});
 //# sourceMappingURL=proxy.js.map

package/server/proxy.ts

@@ -1,10 +1,9 @@
 import * as net from 'net';
-import * as WebSocket from 'ws';
 import { spawn } from 'child_process';
 import { argv, cwd, execPath } from 'process';
 
 //
-// [ device (TCP) ] ←→ [ Proxy = WebSocket client (ws) ] ←→ [ Cloud = 2 x WebSocket server (ws) ] ←→ [ browser client ]
+// [ device (TCP) ] ←→ [ Proxy = TCP client ] ←→ [ Cloud = TCP server ] ←→ [ browser client ]
 //
 
 export interface ProxyConfig {
@@ -16,6 +15,7 @@ export interface ProxyConfig {
   debug?: boolean;       // debug mode
   kind?: "pm2" | "gw" | "solo"; 
 }
+
 export const kEmptyProxy: ProxyConfig = {
   cloudServer: "ws.duotecno.eu",
   cloudPort: 5098,
@@ -44,12 +44,14 @@ export function setProxyConfig(c?: ProxyConfig): ProxyConfig {
 
 let config: ProxyConfig;
 try {
-  setProxyConfig(require('../config-proxy.json'))
+  setProxyConfig(require('../config-proxy.json'));
 } catch (e) {
   setProxyConfig();
   console.log("No config-proxy.json found, using default values from homebridge or other.");
 }
 
+
+
 const gCloudConnections: Array<Context> = [];
 const kDebug = config.debug || false; // enable debug mode from config
 
@@ -75,42 +77,47 @@ function error(str: string) {
 ///////////////////////////////
 // Start up the proxy server //
 ///////////////////////////////
+if (config.uniqueId) {
+  makeNewCloudConnection(config.masterAddress, config.masterPort, config.cloudServer, config.cloudPort, config.uniqueId);
+}
 
-// if (config.kind != "gw") {
-//   makeNewCloudConnection(config.masterAddress, config.masterPort, config.cloudServer, config.cloudPort, config.uniqueId);
-// }
-
-
-export function makeNewCloudConnection(master: string, masterPort: number, server: string, serverPort: number, uniqueId: string) {
-  const cloudSocket = new WebSocket('ws://'+server+':'+serverPort+"/"+uniqueId);
 
-  cloudSocket.on('open', ()  => {
+export function makeNewCloudConnection(master: string, masterPort: number, server: string, serverPort: number, uniqueId: string, count = 0) {
+  const cloudSocket = new net.Socket();
+  
+  cloudSocket.connect(serverPort, server, () => {
     log(`[PROXY] → [CLOUD] Connected to Cloud at ${server}:${serverPort}`);
+    
+    // Send unique ID as the first message to identify this proxy/device
+    cloudSocket.write(`[${uniqueId}]`);
+    debug(`[PROXY] → [CLOUD] Sent unique ID: ${uniqueId}`);
 
     const context = new Context(cloudSocket, master, masterPort, server, serverPort, uniqueId);
     gCloudConnections.push(context);
     log(`[PROXY] → [CLOUD] New free connection #${gCloudConnections.length} to the Cloud at ${server}:${serverPort}`);
   });
 
-  log(`[PROXY] → [CLOUD] Attempting to start a new free connection for ${config.uniqueId} to the Cloud at ${server}:${serverPort}`);
+  cloudSocket.on('error', (err) => {
+    error(`[PROXY] → [CLOUD] Failed to connect ${count+1} times to the Cloud server: ${err.message}`);
+    if (count < 3) {
+      setTimeout(() => {
+        makeNewCloudConnection(master, masterPort, server, serverPort, uniqueId, count + 1);
+      }, 1000);
+    }
+  });
+
+  log(`[PROXY] → [CLOUD] Attempt ${count+1} to start a new free connection for ${config.uniqueId} to the Cloud at ${server}:${serverPort}`);
 }
 
+
 ////////////////
 // Restarting //
 ////////////////
 export function cleanStart(restart: boolean = false) {
-  if (gCloudConnections.length) {
-    gCloudConnections.forEach(context => {
-      if (context.deviceSocket) {
-        context.deviceSocket.end();
-        context.deviceSocket = null;
-      }
-      context.cloudSocket.close();
-      context.cloudSocket = null;
-    });
-    gCloudConnections.splice(0, gCloudConnections.length); // clear the array
-    log(`[PROXY] → [CLOUD] Cleaned up all cloud connections.`);
-  }
+  gCloudConnections.forEach(ctx => ctx.cleanupSockets());
+  log(`[PROXY] → [CLOUD] Cleaned up all ${gCloudConnections.length} cloud connections.`);
+
+  gCloudConnections.splice(0, gCloudConnections.length); // clear the array
 
   if (restart) {
     if (config.uniqueId) {
@@ -125,18 +132,18 @@ export function cleanStart(restart: boolean = false) {
 
 
 export class Context {
-  deviceSocket: net.Socket;
+  deviceSocket: net.Socket | null;
   master: string;
   masterPort: number;
 
-  cloudSocket: WebSocket;
+  cloudSocket: net.Socket | null;
   server: string;
   serverPort: number;
 
   // unique identifier for this device-gateway combination
   uniqueId: string; 
 
-  constructor(cloudSocket: WebSocket, master: string, masterPort: number, server: string, serverPort: number, uniqueId: string) {
+  constructor(cloudSocket: net.Socket, master: string, masterPort: number, server: string, serverPort: number, uniqueId: string) {
     this.deviceSocket = null;
     this.master = master;
     this.masterPort = masterPort;
@@ -152,12 +159,12 @@ export class Context {
 
   setupCloudSocket() {
     // set up handlers for the cloud socket
-    this.cloudSocket.on('message', message => {
-      this.handleDataFromCloud(message as Buffer);
+    this.cloudSocket.on('data', (data: Buffer) => {
+      this.handleDataFromCloud(data);
     });
 
-    this.cloudSocket.on('close', (code, reason) => {
-      warning('[CLOUD] Connection closed: ' + code + ' ' + reason);
+    this.cloudSocket.on('close', () => {
+      warning('[CLOUD] Connection closed');
       if (this.deviceSocket) {
         this.deviceSocket.end();
         this.deviceSocket = null;
@@ -174,29 +181,34 @@ export class Context {
   }
 
   handleDataFromCloud(data: Buffer) {
-    debug(`[CLOUD] → [PROXY]: Data from Cloud: ${data.toString().slice(0, -1)}`);
+    debug(`[CLOUD] → [PROXY]: Data from Cloud: ${data.toString()}`);
 
     if (!this.deviceSocket) {
       // we either have a real client that wants to connect to the device,
       // or we receive a heartbeat from the cloud server
+      // or we receive a connection response [OK], [OK-=2], [ERROR-...]
 
       if (this.isHeartbeatRequest(data)) {
         debug(`[CLOUD] → [PROXY]: Received heartbeat from Cloud, returning response.`);
          // respond to the heartbeat of the cloud server
-        this.cloudSocket.send("[72,3]"); // command([Command.HeartbeatStatus, CommandMethod.Heartbeat.server])
+        this.cloudSocket.write("[72,3]"); // command([Command.HeartbeatStatus, CommandMethod.Heartbeat.server])
 
+      } else if (this.isConnectionResponse(data)) {
+        debug(`[CLOUD] → [PROXY]: Received connection response from Cloud: ${data.toString()}`);
+        // Handle connection response from the cloud server
+        
       } else {
-        // There is incomming data, it's a fresh connection, so: a new client wants to connect to the device
+        // There is real incomming data, it's a fresh connection, so: a new client wants to connect to the device
         // 1) create a new (free) connection to the cloud server for the next client
         // 2) connect to the device and send the initial data
-        log(`[PROXY] → [CLOUD] New client connection, creating new free connection for this proxy/device: ${this.uniqueId}.`);
+        log(`[PROXY] → [CLOUD] New client connection, creating new free connection for this proxy/device: ${config.uniqueId}.`);
         makeNewCloudConnection(this.master, this.masterPort, this.server, this.serverPort, this.uniqueId);
         this.makeDeviceConnection(this.master, this.masterPort, data);
       }
 
     } else if (this.deviceSocket.readyState === 'open') {
       this.deviceSocket.write(data);
-      debug(`[PROXY] → [DEVICE] forwarding data to device: ${data.toString().slice(0, -1)}`);
+      debug(`[PROXY] → [DEVICE] forwarding data to device: ${data.toString().trim()}`);
 
     } else {
       warning(`[PROXY] → [DEVICE] Device socket is not open yet, waiting for connection... what to do with the data??`);
@@ -212,11 +224,17 @@ export class Context {
     
       // Connect to local device and send the data that came in from the cloud
       this.deviceSocket.connect(port, address, () => {
+        // Check if socket is still valid (not cleaned up by removeConnection)
+        if (!this.deviceSocket) {
+          warning(`[PROXY] → [DEVICE] Device socket was cleaned up before connection completed`);
+          return;
+        }
+        
         log(`[PROXY] → [DEVICE] Connected to local device at ${address}:${port}`);
 
         this.setUpDeviceSocket();
-  
-        log(`[PROXY] → [DEVICE] Sending initial message: ${data.toString().slice(0, -1)}`);
+
+        log(`[PROXY] → [DEVICE] Sending initial message: ${data.toString().trim()}`);
         this.deviceSocket.write(data);
       });
 
@@ -230,10 +248,15 @@ export class Context {
   }
 
   setUpDeviceSocket() {
+    if (!this.deviceSocket) {
+      error(`[PROXY] → [DEVICE] Cannot set up device socket - socket is null`);
+      return;
+    }
+
     this.deviceSocket.on('data', (data: Buffer) => {
       const message = data.toString('utf-8');
-      debug(`[DEVICE] → [PROXY] forwarding data to Cloud: ${message.slice(0, -1)}`);
-      this.cloudSocket.send(message);
+      debug(`[DEVICE] → [PROXY] forwarding data to Cloud: ${message.trim()}`);
+      this.cloudSocket.write(data);
     });
 
     this.deviceSocket.on('close', () => {
@@ -247,30 +270,46 @@ export class Context {
     });
   }
 
-  removeConnection() {
-    this.deviceSocket = null;
+  cleanupSockets() {
+    // Clean up the device socket
+    if (this.deviceSocket) {
+      if (!this.deviceSocket.destroyed) {
+        this.deviceSocket.removeAllListeners();
+        this.deviceSocket.end();
+        this.deviceSocket.destroy();
+      }
+      this.deviceSocket = null;
+    }
 
-    // remove this context and close the cloud socket
-    const index = gCloudConnections.indexOf(this);
-    if (index !== -1) {
-      this.cloudSocket.close();
+    // Clean up the cloud socket
+    if (this.cloudSocket) {
+      if (!this.cloudSocket.destroyed) {
+        this.cloudSocket.removeAllListeners();
+        this.cloudSocket.end();
+        this.cloudSocket.destroy();
+      }
       this.cloudSocket = null;
-      gCloudConnections.splice(index, 1);
-      log(`[PROXY] → [CLOUD] Closed socket to Cloud and removed context for device ${this.master}`);
     }
+  }
+
+  removeConnection() {
+    const index = gCloudConnections.indexOf(this);
+    if (index !== -1) gCloudConnections.splice(index, 1);
+
+    this.cleanupSockets();
+    log(`[PROXY] → [CLOUD] Closed socket to Cloud and removed context for device ${this.master}`);
 
     if (gCloudConnections.length === 0) {
         log(`[PROXY] → [CLOUD] No more cloud connections, restarting proxy.`);
       // just to be sure: restart...
-
       if (config.kind === "pm2") {
         // PM2 should restart us.
         log(`[PROXY] → [CLOUD] PM2 should restart us, exiting now.`);
-        process.exit(); 
+        setTimeout(() => process.exit(0), 100); 
 
       } else if (config.kind === "solo"){
         log(`[PROXY] → [CLOUD] Running solo, restarting proxy.`);
-        this.restart();
+        restart();
 
       } else {
         log(`[PROXY] → [CLOUD] Running under gateway/homebridge, not restarting... not sure what to do here.`);
@@ -283,33 +322,84 @@ export class Context {
   // Utilities //
   ///////////////
 
-  restart() {
-    console.log('🔄 Restarting...');
+  // check if it's a heartbeat request from the server
+  isHeartbeatRequest(data: Buffer): boolean {
+    const msg = data.toString('utf-8').trim();
+    return msg === '[215,3]';
+    // hearbeatKind = (poll = 1, poll_old = 2, serve = 3)
+    // return ((data[0] === 91) &&           // [
+    //         (data[1] === 50) &&           // 2
+    //         (data[2] === 49) &&           // 1
+    //         (data[3] === 53) &&           // 5
+    //         (data[4] === 44) &&           // ,
+    //         (data[5] === 51) &&           // 3
+    //         (data[6] === 93));            // ]
+  }
+
+  // check if it's a server response after connecting and sending our uniqueID
+  isConnectionResponse(data: Buffer): boolean {
+    const message = data.toString();
+    
+    // Check for various server response patterns
+    return (message.startsWith('[OK')) || (message.startsWith('[ERROR'));
+  }
+}
+
+function restart() {
+  console.log('🔄 Restarting...');
 
+  setTimeout(() => {
     spawn(execPath, argv.slice(1), {
       cwd: cwd(),
       detached: true,
       stdio: 'inherit',
     });
 
-    process.exit();
-  }
+    process.exit(0);
+  }, 100);
+}
 
-  // check if it's a heartbeat request from the server
-  isHeartbeatRequest(data: Buffer): boolean {
-    // heartbeatKind = (poll = 1, poll_old = 2, serve = 3)
 
-    if (typeof data === 'string') {
-      return data === "[215,3]";
+// Graceful shutdown handler
+function handleShutdown(signal: string) {
+  log(`[PROXY] → [SYSTEM] Received ${signal}, shutting down gracefully...`);
 
-    } else {
-      return ((data[0] === 91) &&           // [
-              (data[1] === 50) &&           // 2
-              (data[2] === 49) &&           // 1
-              (data[3] === 53) &&           // 5
-              (data[4] === 44) &&           // ,
-              (data[5] === 51) &&           // 3
-              (data[6] === 93));            // ]
-    }
+  // Close all cloud connections
+  gCloudConnections.forEach(ctx => ctx.cleanupSockets());
+  gCloudConnections.splice(0, gCloudConnections.length);
+
+  log(`[PROXY] → [SYSTEM] All connections closed. Exiting.`);
+
+  if (config.kind === "solo") {
+    log(`[PROXY] → [SYSTEM] Restarting proxy in solo mode...`);
+    restart();
+  } else {
+    log(`[PROXY] → [SYSTEM] Exiting without restart.`);
+    setTimeout(() => process.exit(0), 100);
   }
 }
+
+// Listen for termination signals
+process.on('SIGINT', () => handleShutdown('SIGINT'));
+process.on('SIGTERM', () => handleShutdown('SIGTERM'));
+
+
+process.on('SIGINT', (err) => {
+  error(`[SYSTEM] received SIGINT`);
+  handleShutdown('SIGINT');
+});
+
+process.on('SIGTERM', (err) => {
+  error(`[SYSTEM] received SIGTERM`);
+  handleShutdown('SIGTERM');
+});
+
+process.on('uncaughtException', (err) => {
+  error(`[SYSTEM] Uncaught Exception: ${err.message}`);
+  handleShutdown('uncaughtException');
+});
+
+process.on('unhandledRejection', (reason) => {
+  error(`[SYSTEM] Unhandled Rejection: ${reason}`);
+  handleShutdown('unhandledRejection');
+});