diff options
| -rw-r--r-- | src/main.rs | 13 |
1 files changed, 8 insertions, 5 deletions
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<String> = 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, |