feat: add custom listen addresses

Author: maschad

src/cli/commands.rs

@@ -9,6 +9,8 @@ pub fn get_args() -> Command {
         .subcommand_required(true)
         .arg_required_else_help(true)
         .arg(arg!(-v --debug "Print debug information"))
+        .arg(arg!(-a --address <IP_ADDRESS> "IP address to listen on"))
+        .arg(arg!(-p --port <PORT> "Port number to listen on"))
         .subcommand(
             Command::new("share")
                 .about("Send a file or directory to another peer")

src/main.rs

@@ -424,12 +424,35 @@ async fn start_network(
     spawn(event_loop.run());
     spawn(handle_network_events(event_stream, app.clone()));
 
-    client
-        .start_listening("/ip4/0.0.0.0/udp/0/quic-v1".parse().unwrap())
-        .await
-        .expect("Listening not to fail.");
+    // Get IP and port from command line args, defaulting to 0.0.0.0:0
+    let matches = cli::commands::get_args().get_matches();
+    let default_ip = "0.0.0.0".to_string();
+    let default_port = "0".to_string();
+    let ip = matches.get_one::<String>("address").unwrap_or(&default_ip);
+    let port = matches.get_one::<String>("port").unwrap_or(&default_port);
+
+    let addr = format!("/ip4/{ip}/udp/{port}/quic-v1")
+        .parse()
+        .unwrap_or_else(|e| {
+            tracing::error!("Failed to parse listening address: {}", e);
+            eprintln!("Error: Invalid listening address format. Please check your IP and port.");
+            std::process::exit(1);
+        });
+
+    client.start_listening(addr).await.unwrap_or_else(|e| {
+        tracing::error!("Failed to start listening: {}", e);
+        eprintln!(
+            "Error: Could not start listening on the specified address. The port might be in use."
+        );
+        std::process::exit(1);
+    });
+
+    let listening_addrs: Vec<Multiaddr> = client.get_listening_addrs().await.unwrap_or_else(|e| {
+        tracing::error!("Failed to get listening addresses: {}", e);
+        eprintln!("Error: Could not get listening addresses. Please try again.");
+        std::process::exit(1);
+    });
 
-    let listening_addrs: Vec<Multiaddr> = client.get_listening_addrs().await.unwrap();
     {
         let mut app = app.lock();
         app.listening_addrs = listening_addrs;