Get WS_URL from ENV

bokub · Dec 23, 2023 · 9c96185 · 9c96185
1 parent 4f4f494
commit 9c96185
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/ha.ts b/src/ha.ts
@@ -4,7 +4,8 @@ import { auth } from 'home-assistant-js-websocket/dist/messages.js';
 import { debug, warn } from './log.js';
 import dayjs from 'dayjs';
 
-const WS_URL = 'ws://supervisor/core/websocket';
+const WS_URL = process.env.WS_URL || 'ws://supervisor/core/websocket';
+const TOKEN = process.env.SUPERVISOR_TOKEN;
 
 export type SuccessMessage = {
   id: string;
@@ -62,7 +63,7 @@ export class HomeAssistantClient {
               }
             });
 
-            connection.sendUTF(JSON.stringify(auth(process.env.SUPERVISOR_TOKEN)));
+            connection.sendUTF(JSON.stringify(auth(TOKEN)));
           }
         });
       });