Add open browser argument

Cargo.toml

@@ -22,6 +22,7 @@ tokio = { version = "1.35.0", features = ["full"] }
 axum = { version = "0.7.2", features = ["ws"]}
 futures = "0.3.29"
 mime_guess = "2.0.3"
+open = "5.0.1"
 
 [dev-dependencies]
 reqwest = "0.11.23"

src/lib.rs

@@ -17,9 +17,10 @@
 mod server;
 mod watcher;
 
-use std::{error::Error, path::PathBuf};
+use std::{error::Error, net::IpAddr, path::PathBuf};
 
 use axum::Router;
+use local_ip_address::local_ip;
 use notify::ReadDirectoryChangesWatcher;
 use notify_debouncer_full::{DebouncedEvent, Debouncer, FileIdMap};
 use server::{create_listener, create_server};
@@ -43,7 +44,7 @@ pub struct Listener {
 
 impl Listener {
     /// Start live-server
-    /// 
+    ///
     /// ```
     /// use live_server::listen;
     ///
@@ -52,9 +53,9 @@ impl Listener {
     /// }
     /// ```
     pub async fn start(self) -> Result<(), Box<dyn Error>> {
-        ROOT.set(self.root_path.clone()).unwrap();
+        ROOT.set(self.root_path.clone())?;
         let (tx, _) = broadcast::channel(16);
-        TX.set(tx).unwrap();
+        TX.set(tx)?;
 
         let watcher_future = tokio::spawn(watcher::watch(self.root_path, self.debouncer, self.rx));
         let server_future = tokio::spawn(server::serve(self.tcp_listener, self.router));
@@ -63,11 +64,25 @@ impl Listener {
 
         Ok(())
     }
-}
 
+    pub fn link(&self) -> Result<String, Box<dyn Error>> {
+        let addr = self.tcp_listener.local_addr()?;
+        let port = addr.port();
+        let host = addr.ip();
+        let host = match host.is_unspecified() {
+            true => local_ip()?,
+            false => host,
+        };
+
+        Ok(match host {
+            IpAddr::V4(host) => format!("http://{host}:{port}"),
+            IpAddr::V6(host) => format!("http://[{host}]:{port}"),
+        })
+    }
+}
 
 /// Create live-server listener
-/// 
+///
 /// ```
 /// use live_server::listen;
 ///

src/main.rs

@@ -15,16 +15,30 @@ struct Args {
     /// Set the listener port
     #[clap(short, long, default_value = "0")]
     port: u16,
+    /// Open the page in browser automatically
+    #[clap(short, long)]
+    open: bool,
 }
 
 #[tokio::main]
 async fn main() {
     let env = Env::new().default_filter_or("info");
     env_logger::init_from_env(env);
 
-    let Args { host, port, root } = Args::parse();
+    let Args {
+        host,
+        port,
+        root,
+        open,
+    } = Args::parse();
 
     let addr = format!("{}:{}", host, port);
+    let listener = listen(addr, root).await.unwrap();
 
-    listen(addr, root).await.unwrap().start().await.unwrap();
+    if open {
+        let link = listener.link().unwrap();
+        open::that(link).unwrap();
+    }
+
+    listener.start().await.unwrap();
 }