From c3c5ff4ebe0243886007c2b26e211f355eaa9b8e Mon Sep 17 00:00:00 2001
From: clerie <git@clerie.de>
Date: Sun, 14 Jul 2024 20:21:07 +0200
Subject: [PATCH] Add option to add port to listen on

---
 README.md   |  2 +-
 src/main.rs | 26 +++++++++++++++++++++++++-
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 455e156..677b549 100644
--- a/README.md
+++ b/README.md
@@ -7,5 +7,5 @@ There is also a Map: <https://www.berlinerbaeder.de/baeder/sommerbaeder-auslastu
 ## Run
 
 ```
-berlinerbaeder-exporter
+berlinerbaeder-exporter --listen "[::]:1234"
 ```
diff --git a/src/main.rs b/src/main.rs
index 3964542..a0b7647 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -29,11 +29,35 @@ struct TrafficDataItem {
 
 #[tokio::main]
 async fn main() {
+
+    let mut args = std::env::args();
+    let _name = args.next().unwrap();
+
+    let mut listen = String::from("[::]:3000");
+
+    loop {
+        let arg = if let Some(arg) = args.next() {
+            arg
+        } else {
+            break;
+        };
+
+        match arg.as_str() {
+            "--listen" => {
+                listen = args.next().unwrap();
+            }
+            unknown => {
+                println!("unknown option: {}", unknown);
+                std::process::exit(1)
+            }
+        }
+    }
+
     let app = Router::new()
         .route("/", get(route_index))
         .route("/metrics", get(route_metrics));
 
-    let listener = tokio::net::TcpListener::bind("[::]:3000").await.unwrap();
+    let listener = tokio::net::TcpListener::bind(listen).await.unwrap();
     println!("Server listening on: http://{}", listener.local_addr().unwrap());
 
     axum::serve(listener, app).await.unwrap();