Refactor path parsing to be more idiomatic

1 files changed, 10 insertions, 10 deletions

diff --git a/src/main.rs b/src/main.rs
index 436789e..94883ea 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -13,16 +13,16 @@ macro_rules! err {
 }
 
 fn main() {
-    let args: Vec<String> = env::args().collect();
-    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 {
-        String::from("app/")
-    };
+    let path = env::args()
+        .nth(1)
+        .map(|path| {
+            if !path.ends_with("/") {
+                path + "/"
+            } else {
+                path
+            }
+        })
+        .unwrap_or(String::from("app/"));
 
     let listener = match TcpListener::bind("0.0.0.0:8080") {
         Ok(listener) => listener,