From 87843b62a71261a2cb0405ca42d1214b7154818e Mon Sep 17 00:00:00 2001
From: djairoh <ljxjah@mykolab.com>
Date: Wed, 8 Feb 2023 17:34:43 +0100
Subject: [PATCH] implemented writing to file

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

diff --git a/src/output.rs b/src/output.rs
index d478f90..e12b6f3 100644
--- a/src/output.rs
+++ b/src/output.rs
@@ -1,6 +1,7 @@
 use std::fs::File;
 use std::io::Write;
 use std::path::PathBuf;
+use std::process::exit;
 use log::error;
 use crate::model_rgb_ascii::Ascii;
 
@@ -14,7 +15,7 @@ pub fn print_terminal(art: Vec<Vec<Ascii>>, in_colour: bool) {
     }
 }
 
-pub fn print_file(art: Vec<Vec<Ascii>>, out: PathBuf) -> std::io::Result<()> {
+fn _print_file(art: Vec<Vec<Ascii>>, out: PathBuf) -> std::io::Result<()> {
     let mut file = File::create_new(out)?;
     for line in art {
         for ascii in line {
@@ -23,4 +24,11 @@ pub fn print_file(art: Vec<Vec<Ascii>>, out: PathBuf) -> std::io::Result<()> {
         writeln!(file)?;
     }
     Ok(())
+}
+
+pub fn print_file(art: Vec<Vec<Ascii>>, out: PathBuf) {
+    if let Err(e) = _print_file(art, out) {
+        error!("Failed to write to file: {}", e.to_string());
+        exit(1);
+    }
 }
\ No newline at end of file