Load env in all cargo tasks

[rbdr/lyricli] / src / main.rs

diff --git a/src/main.rs b/src/main.rs
index a6119266f25581e98b17ff7530816455372b4d2e..cab96a53db35e17b7ae2fe3d587ce819615a7a25 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,17 +2,16 @@ mod configuration;
 mod lyrics_engine;
 mod sources;
 
-use std::io::{Result, Error, ErrorKind::Other};
 use clap::Parser;
+use std::io::{Error, ErrorKind::Other, Result};
 
 use configuration::Configuration;
-use sources::{enable, disable, get_track, reset, list};
 use lyrics_engine::print_lyrics;
+use sources::{disable, enable, get_track, list, reset};
 
 #[derive(Parser, Debug)]
 #[command(version, about, long_about = None)]
 struct Arguments {
-
     // Positional Arguments
     /// Specify the artist.
     artist: Option<String>,
@@ -41,11 +40,27 @@ struct Arguments {
 
 pub struct Track {
     pub name: String,
-    pub artist: String
+    pub artist: String,
 }
 
 #[tokio::main]
 async fn main() -> Result<()> {
+    let result = run().await;
+
+    if cfg!(debug_assertions) {
+        result
+    } else {
+        match result {
+            Ok(_) => Ok(()),
+            Err(e) => {
+                eprintln!("Error: {}", e);
+                std::process::exit(1);
+            }
+        }
+    }
+}
+
+async fn run() -> Result<()> {
     let mut configuration = Configuration::new();
     let arguments = Arguments::parse();
 
@@ -83,11 +98,11 @@ async fn main() -> Result<()> {
     if let Some(artist) = arguments.artist {
         current_track = Track {
             name: arguments.track_name.unwrap_or("".to_string()),
-            artist: artist
+            artist,
         };
     } else {
-        current_track = get_track()
-            .ok_or_else(|| Error::new(Other, "No Artist/Song could be found :("))?
+        current_track =
+            get_track().ok_or_else(|| Error::new(Other, "No Artist/Song could be found :("))?
     }
 
     print_lyrics(current_track, arguments.show_title).await