Fix #2

If a book has a description less than MAX_DESC_LEN, which is rare, the program will panic
thundergolfer · Oct 19, 2019 · c785552 · c785552
1 parent d17b9b6
commit c785552
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/models.rs b/src/models.rs
@@ -26,8 +26,14 @@ pub struct Author {
 
 impl Display for Book {
     fn fmt(&self, f: &mut Formatter) -> fmt::Result {
-        let shortened_desc = &self.description[..MAX_DESC_LEN];
-        write!(f, "{}: {}...", self.title, shortened_desc)
+        match self.description.len() {
+            0 => write!(f, "{}", self.title),
+            1..=MAX_DESC_LEN => write!(f, "{}: {}...", self.title, self.description),
+            _ => {
+                let shortened_desc = &self.description[..MAX_DESC_LEN];
+                write!(f, "{}: {}...", self.title, shortened_desc)
+            }
+        }
     }
 }