From 42458d378093a738fb564a91263a3a45aaaa5677 Mon Sep 17 00:00:00 2001
From: darkfi <darkfi@darkfi>
Date: Sat, 20 Jul 2024 11:31:03 +0200
Subject: [PATCH] script/ping: allow specifying endpoint using argv

---
 script/ping/src/main.rs | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/script/ping/src/main.rs b/script/ping/src/main.rs
index f7f78530a..bcaae0ec7 100644
--- a/script/ping/src/main.rs
+++ b/script/ping/src/main.rs
@@ -23,9 +23,13 @@ use url::Url;
 
 const ENDPOINT: &str = "tcp+tls://lilith1.dark.fi:5262";
 
-async fn ping() {
-    println!("Pinging {ENDPOINT}");
-    let endpoint = Url::parse(ENDPOINT).unwrap();
+async fn ping(endpoint: &str) {
+    let Ok(endpoint) = Url::parse(endpoint) else {
+        println!("Invalid endpoint {endpoint}");
+        return
+    };
+    println!("Pinging {endpoint}");
+
     let dialer = net::transport::Dialer::new(endpoint, None).await.unwrap();
     let timeout = std::time::Duration::from_secs(5);
 
@@ -51,11 +55,14 @@ async fn ping() {
 }
 
 fn main() {
+    let args: Vec<String> = std::env::args().collect();
+    let endpoint = if args.len() == 1 { ENDPOINT } else { &args[1] };
+
     let (signal, shutdown) = smol::channel::unbounded::<()>();
 
     let ex = Arc::new(smol::Executor::new());
     let _task = ex.spawn(async {
-        ping().await;
+        ping(endpoint).await;
         let _ = signal.send(()).await;
     });