From 14318cf2fb4f9a1d354b3a29d48f2d683c148587 Mon Sep 17 00:00:00 2001
From: Himadri Bhattacharjee <107522312+lavafroth@users.noreply.github.com>
Date: Sat, 21 Jun 2025 11:13:36 +0530
Subject: [PATCH] feat: better error handling for api v2

---
 src/v2/api.rs | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/src/v2/api.rs b/src/v2/api.rs
index c9a8f26..e4138fe 100644
--- a/src/v2/api.rs
+++ b/src/v2/api.rs
@@ -55,16 +55,15 @@ pub(crate) async fn get_snippet(
 
     println!("{prompt:?}");
 
-    let Ok(mut appstate) = data.inner.lock() else {
-        return Err(Error::Busy);
-    };
-
-    let Ok(target) = appstate.embed.embed(prompt) else {
-        return Err(Error::EmbedFailed);
-    };
-
+    let mut appstate = data.inner.lock().map_err(|_| Error::Busy)?;
+    let target = appstate
+        .embed
+        .embed(prompt)
+        .map_err(|_| Error::EmbedFailed)?;
     let mut parser = Parser::new();
-    parser.set_language(&langfn).unwrap();
+    parser
+        .set_language(&langfn)
+        .map_err(|_| Error::UnknownLang)?;
 
     let source_code = snippet_request.body.as_str();
     let source_bytes = source_code.as_bytes();