From e692205b8b135c1b7b7b9233fd3650a024aed3ec Mon Sep 17 00:00:00 2001 From: Shravan Mandava Date: Sun, 22 Feb 2026 13:37:27 +0000 Subject: Refactor source path generation --- src/main.rs | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'src') diff --git a/src/main.rs b/src/main.rs index e780ed1..436789e 100644 --- a/src/main.rs +++ b/src/main.rs @@ -14,12 +14,15 @@ macro_rules! err { fn main() { let args: Vec = env::args().collect(); - let mut path = String::new(); - if args.len() == 2 && args[1].ends_with("/") { - path.push_str(&args[1]); + let path = if args.len() == 2 { + if !(args[1].ends_with("/")) { + String::from(args[1].as_str()) + "/" + } else { + String::from(args[1].as_str()) + } } else { - path.push_str("app/"); - } + String::from("app/") + }; let listener = match TcpListener::bind("0.0.0.0:8080") { Ok(listener) => listener, -- cgit