From 48ea90800c3aa6055247fe2c56bc8e9d63024bd3 Mon Sep 17 00:00:00 2001
From: Ruben Beltran del Rio <ruben@unlimited.pizza>
Date: Sun, 16 Apr 2023 16:56:10 +0200
Subject: Handle layout error better

---
 src/main.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index fdfe863..c6de435 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,6 +3,7 @@ mod file_finder;
 mod file_handler;
 
 use std::io::Result;
+use std::process::exit;
 use std::env::current_dir;
 use std::fs::{create_dir_all, remove_dir_all};
 
@@ -27,7 +28,13 @@ fn main() -> Result<()> {
 
     // Step 3. Load the layout
     let mut file_handler = FileHandler::default();
-    file_handler.get_layout_or_panic(&files);
+    match file_handler.get_layout_or_panic(&files) {
+        Ok(_) => {},
+        Err(error) => {
+            eprintln!("{}", error);
+            exit(1);
+        }
+    }
 
     // Step 4. Process all files
     file_handler.handle_all(&source, &destination, &files);
-- 
cgit