chore(screenshot): print no title if url doesn't have one // closes #30

pwnwriter · Oct 17, 2023 · b062793 · b062793
1 parent 661e700
commit b062793
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/cli/screenshot.rs b/src/cli/screenshot.rs
@@ -155,10 +155,14 @@ async fn take_screenshot(
     .await?;
 
     if !silent {
-        let title = page.get_title().await.unwrap_or_default().unwrap();
-        show_info(url, title, res.status());
+        match page.get_title().await {
+            Ok(Some(title)) => show_info(url.clone(), title, res.status()),
+            _ => {
+                let title = "No title".to_string();
+                show_info(url.clone(), title, res.status());
+            }
+        }
     }
-
     page.close().await?;
 
     Ok(())