From 1377ceefd2999d268d1ae58065cd9136d2bb02ab Mon Sep 17 00:00:00 2001
From: MegaBrutal <megabrutal+github@megabrutal.com>
Date: Sat, 18 Feb 2023 01:40:00 +0100
Subject: [PATCH] Query multiple addresses from command line

---
 src/main.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 0255860..4bd31f3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -4,13 +4,13 @@ use a2s::A2SClient;
 
 fn main() {
     let client = A2SClient::new().unwrap();
-    let addresses = args().nth(1)
-        .unwrap_or_else(|| String::from("localhost:27015"))
-        .to_socket_addrs().unwrap()
-        .flat_map(|sa| match sa {
-            SocketAddr::V4(sa4) => Some(sa4),
-            _ => None
-        });
+    let addresses = args().skip(1)
+        .flat_map(|arg| arg.to_socket_addrs())
+        .flat_map(|iter_addr| iter_addr.flat_map(|sa| match sa {
+                SocketAddr::V4(sa4) => Some(sa4),
+                _ => None
+            })
+        );
 
     for address in addresses {
 
-- 
2.34.1