From 36468e4bafb4cec24fbc769f10b26e9a8c4626e9 Mon Sep 17 00:00:00 2001
From: William Desportes <williamdes@wdes.fr>
Date: Fri, 27 Sep 2024 21:02:49 +0200
Subject: [PATCH] Support ENV WORKER_URL for CLI

---
 snow-scanner/src/worker/worker.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/snow-scanner/src/worker/worker.rs b/snow-scanner/src/worker/worker.rs
index 0dae5b1..6dc30f3 100644
--- a/snow-scanner/src/worker/worker.rs
+++ b/snow-scanner/src/worker/worker.rs
@@ -171,9 +171,13 @@ fn main() -> () {
         })
         .start();
     info!("Running the worker");
-    let url = "ws://127.0.0.1:8800";
+    let url = match env::var("WORKER_URL") {
+        Ok(worker_url) => worker_url,
+        Err(_) => "ws://127.0.0.1:8800".to_string(),
+    };
+
     let mut worker = Worker::initial();
-    match ws2::connect(url) {
+    match ws2::connect(&url) {
         Ok(mut ws_client) => {
             let connected = ws_client.is_open();
             if connected {