From 8c867a001a0d588b2aa367f89ae71f6fa548daa8 Mon Sep 17 00:00:00 2001
From: Chris Pearce <christopher.james.pearce@gmail.com>
Date: Fri, 12 Apr 2019 22:24:13 +0100
Subject: [PATCH] Remove unwrap on canonicalize result

---
 src/main.rs | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 5e0b658..01618b7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -67,14 +67,17 @@ fn main() {
             std::process::exit(1);
         });
 
-        let filepath = Path::new(filename).canonicalize().unwrap();
-        let exercise = exercises
-            .iter()
-            .find(|e| filepath.ends_with(&e.path))
-            .unwrap_or_else(|| {
-                println!("No exercise found for your file name!");
-                std::process::exit(1)
-            });
+        let matching_exercise = |e: &&Exercise| {
+            Path::new(filename)
+                .canonicalize()
+                .map(|p| p.ends_with(&e.path))
+                .unwrap_or(false)
+        };
+
+        let exercise = exercises.iter().find(matching_exercise).unwrap_or_else(|| {
+            println!("No exercise found for your file name!");
+            std::process::exit(1)
+        });
 
         run(&exercise).unwrap_or_else(|_| std::process::exit(1));
     }