refactor: 🪛 use function to exit

Author: pwnwriter

src/colors.rs

@@ -9,6 +9,6 @@ pub const YELLOW: &str = "\x1B[32m"; //( bright, medium yellow color)
 pub const YELLOW_BRIGHT: &str = "\x1B[93m"; //( very bright, light yellow color)
 pub const RESET: &str = "\x1B[0m"; //( resets the text color to the default)
 
-// unused colors rn :- 
-                                    //pub const MAGENTAA: &str = "\x1b[35m"; //( bright, medium pinkish-purple color)
-                                    //pub const WHITE: &str = "\x1B[37m"; //( bright, pale gray color)
+// unused colors rn :-
+//pub const MAGENTAA: &str = "\x1b[35m"; //( bright, medium pinkish-purple color)
+//pub const WHITE: &str = "\x1B[37m"; //( bright, pale gray color)

src/main.rs

@@ -72,6 +72,11 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 
     Ok(())
 }
+
+fn exit_on_error() {
+    std::process::exit(0);
+}
+
 async fn run(
     url: String,
     outdir: Option<String>,
@@ -81,12 +86,11 @@ async fn run(
     height: Option<u32>,
     silent: bool,
 ) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
-        // Check if the browser binary path is valid
+    // Check if the browser binary path is valid
     if !Path::new(&binary_path).exists() {
         println!("{RED}[   ]Browser binary not found at path {}, You should try manually pasing the binary path !!{RESET}", binary_path);
         println!("{BLUE}[ ﯦ  ]{RESET}{CYAN} $ hxn -b $(which brave) <url> or use --help flag");
-        std::process::exit(0);
-
+        exit_on_error();
     }
     let outdir = match outdir {
         Some(dir) => dir,
@@ -161,6 +165,7 @@ async fn run(
         handle
             .await?
             .expect("Something went wrong while waiting for taking screenshot and saving to file");
+        exit_on_error();
     }
 
     println!("{RED}♥ {GREEN} {YELLOW_BRIGHT}Yoo 1337!! Screenshots saved in dir {outdir}{RED} ♥ {GREEN}{RESET} ");
@@ -211,4 +216,3 @@ async fn take_screenshots(
 
     Ok(())
 }
-