From ef420db3ab72b1a49a7706fdbabce1a179b41bf9 Mon Sep 17 00:00:00 2001 From: Rosyid Haryadi Date: Sat, 15 Mar 2025 17:01:27 +0700 Subject: trying macro --- src/utilities.rs | 30 ++++++++++++++++++++++++++---- 1 file changed, 26 insertions(+), 4 deletions(-) (limited to 'src/utilities.rs') diff --git a/src/utilities.rs b/src/utilities.rs index 54f1292..aa7a44e 100644 --- a/src/utilities.rs +++ b/src/utilities.rs @@ -2,7 +2,29 @@ use std::collections::HashMap; pub type ConfigHashMap = HashMap>>; -pub fn die(message: &str) { - println!("{}", message); - std::process::exit(1); -} \ No newline at end of file +#[macro_export] +macro_rules! die { + ($message:expr) => { + println!("{}", $message); + std::process::exit(1); + } +} + +#[macro_export] +macro_rules! create_path_or_die { + (dir: $path:expr, $message:expr) => { + if let Err(_) = std::fs::create_dir($path) { + die!($message); + } + }; + + (file: $path:expr, $content:expr, $message:expr) => { + if let Ok(mut file) = std::fs::File::create($path) { + if let Err(_) = file.write_all($content.as_ref()) { + die!($message); + } + } else { + die!($message); + } + }; +} -- cgit v1.2.3-70-g09d2