Merge pull request #105 from pwnwriter/feat/js

chore(exec): check if the ss was successful or not and then only log

src/cli/exec.rs

@@ -93,7 +93,7 @@ pub async fn run(
         }
     }
 
-    if stdin {
+    let is_screenshot_taken = if stdin {
         env::set_current_dir(dump_dir)?;
         let urls = super::hxn_helper::read_urls_from_stdin(ports)?;
         take_screenshot_in_bulk(
@@ -108,7 +108,7 @@ pub async fn run(
             accept_invalid_certs,
             javascript,
         )
-        .await?;
+        .await
     } else {
         match (url, file_path) {
             (None, Some(file_path)) => {
@@ -126,7 +126,7 @@ pub async fn run(
                     accept_invalid_certs,
                     javascript,
                 )
-                .await?;
+                .await
             }
             (Some(url), None) => {
                 let urls = if let Some(ports) = ports {
@@ -147,16 +147,20 @@ pub async fn run(
                     accept_invalid_certs,
                     javascript,
                 )
-                .await?;
+                .await
             }
             _ => unreachable!(),
         }
-    }
-
-    info(
-        format!("Screenshots Taken and saved in directory {}", outdir.bold()),
-        colored::Color::Cyan,
-    );
+    };
 
-    Ok(())
+    match is_screenshot_taken {
+        Ok(_) => {
+            info(
+                format!("Screenshots Taken and saved in directory {}", outdir.bold()),
+                colored::Color::Cyan,
+            );
+            Ok(())
+        }
+        Err(e) => Err(e),
+    }
 }