show solved count for searched link

Author: halshar

Cargo.lock

@@ -1,12 +1,12 @@
 [package]
 name = "abhyas"
-version = "1.2.0"
+version = "1.4.0"
 edition = "2021"
 authors = ["Harshal Jadhav <bestt1315@gmail.com>"]
 homepage = "https://github.com/halshar/abhyas"
 description = "Abhyas is a Rust command-line application for managing and interacting with a database of links."
 readme = "README.md"
-repository = "https://github.com/halshar/abhyas" 
+repository = "https://github.com/halshar/abhyas"
 license = "GPL-3.0-or-later"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

Cargo.toml

@@ -273,6 +273,14 @@ fn search_link_options(db: &Db) -> Result<(), CustomErrors> {
         }
     };
 
+    match db.get_searched_link_count(&link) {
+        Ok(val) => match val {
+            Some(solved_count) => pretty_print(&[(link.clone(), solved_count)]),
+            None => pretty_print(&[(link.clone(), 0)]),
+        },
+        Err(e) => return Err(e),
+    };
+
     single_link_options(db, &link)?;
 
     Ok(())

src/cli.rs

@@ -136,6 +136,27 @@ impl Db {
         }
     }
 
+    pub fn get_searched_link_count(&self, link: &str) -> Result<Option<i32>, CustomErrors> {
+        let mut stmt = match self.conn.prepare(
+            "SELECT solved_count FROM links
+            WHERE link = ?1;",
+        ) {
+            Ok(val) => val,
+            Err(_) => return Err(CustomErrors::StatementFailed),
+        };
+
+        match stmt.query_row([&link], |row| {
+            let solved_count: i32 = row.get(0)?;
+            Ok(solved_count)
+        }) {
+            Ok(solved_count) => Ok(Some(solved_count)),
+            Err(rusqlite::Error::QueryReturnedNoRows) => Ok(None),
+            Err(_) => Err(CustomErrors::Others(
+                "Error: Something went wrong while fetching link count".to_owned(),
+            )),
+        }
+    }
+
     pub fn mark_as_complete(&self, link: &str) -> Result<(), CustomErrors> {
         match self.conn.execute(
             "UPDATE links