Dont panic if invalid desc

pawurb · Oct 25, 2024 · ade8020 · ade8020
1 parent 9fdae7e
commit ade8020
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/bin/main.rs b/bin/main.rs
@@ -220,12 +220,14 @@ async fn execute() -> Result<(), PgExtrasError> {
 }
 
 fn extract_desc(desc: &str) -> String {
-    let start = desc.find("/*").unwrap();
-    let end = desc.find("*/").unwrap();
-    let extracted = &desc[start + 2..end];
-    let mut trimmed = extracted.trim().to_string();
-    if trimmed.ends_with('.') {
-        trimmed.pop();
+    if let (Some(start), Some(end)) = (desc.find("/*"), desc.find("*/")) {
+        let extracted = &desc[start + 2..end];
+        let mut trimmed = extracted.trim().to_string();
+        if trimmed.ends_with('.') {
+            trimmed.pop();
+        }
+        trimmed
+    } else {
+        desc.to_string()
     }
-    trimmed
 }