added option to check status

Author: halshar

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "abhyas"
-version = "1.1.0"
+version = "1.2.0"
 edition = "2021"
 authors = ["Harshal Jadhav <bestt1315@gmail.com>"]
 homepage = "https://github.com/halshar/abhyas"

Cargo.toml

@@ -4,6 +4,7 @@ Abhyas is a Rust command-line application for managing and interacting with a da
 
 ### Features
 
+- **Check Status**: Get the total, completed, and skipped links count.
 - **Get Link**: Get random link from the database.
 - **Add Link**: Add new links to the database.
 - **Search Link**: Search link from the database.

README.md

@@ -1,10 +1,11 @@
-use crate::print::pretty_print;
+use crate::print::{pretty_print, pretty_status};
 use crate::utility::show_red;
 use crate::CustomErrors;
 use crate::{database::Db, utility::show_green};
 use inquire::{required, validator::Validation, Select, Text};
 
 enum MainMenuOptions {
+    Status,
     GetLink,
     AddLink,
     SearchLink,
@@ -30,7 +31,14 @@ enum OtherOptions {
 }
 
 pub fn show_options(db: &Db) -> Result<(), CustomErrors> {
-    let options = vec!["Get Link", "Add Link", "Search Link", "Other", "Exit"];
+    let options = vec![
+        "Check Status",
+        "Get Link",
+        "Add Link",
+        "Search Link",
+        "Other",
+        "Exit",
+    ];
 
     let user_option = match Select::new("select your option", options).prompt() {
         Ok(val) => val,
@@ -50,6 +58,7 @@ pub fn show_options(db: &Db) -> Result<(), CustomErrors> {
     };
 
     let selected_item = match user_option {
+        "Check Status" => MainMenuOptions::Status,
         "Get Link" => MainMenuOptions::GetLink,
         "Add Link" => MainMenuOptions::AddLink,
         "Search Link" => MainMenuOptions::SearchLink,
@@ -59,6 +68,7 @@ pub fn show_options(db: &Db) -> Result<(), CustomErrors> {
     };
 
     match selected_item {
+        MainMenuOptions::Status => get_status(&db)?,
         MainMenuOptions::GetLink => get_link_options(&db)?,
         MainMenuOptions::AddLink => add_link_options(&db)?,
         MainMenuOptions::SearchLink => search_link_options(&db)?,
@@ -69,6 +79,21 @@ pub fn show_options(db: &Db) -> Result<(), CustomErrors> {
     Ok(())
 }
 
+fn get_status(db: &Db) -> Result<(), CustomErrors> {
+    match db.get_status() {
+        Ok(val) => {
+            match val {
+                Some((total_links, completed_links, skipped_links)) => {
+                    pretty_status(total_links, completed_links, skipped_links)
+                }
+                None => pretty_status(0, 0, 0),
+            };
+        }
+        Err(e) => return Err(e),
+    };
+    Ok(())
+}
+
 fn get_link_options(db: &Db) -> Result<(), CustomErrors> {
     let link = match db.get_single_link() {
         Ok(val) => match val {

src/cli.rs

@@ -12,6 +12,35 @@ impl Db {
         Db { conn }
     }
 
+    /// get total, completed, and skipped links count
+    pub fn get_status(&self) -> Result<Option<(i32, i32, i32)>, CustomErrors> {
+        let mut stmt = match self.conn.prepare(
+            "SELECT
+                COALESCE(COUNT(*), 0) AS total_links,
+                COALESCE(SUM(CASE WHEN is_solved = 1 THEN 1 ELSE 0 END), 0) AS completed_links,
+                COALESCE(SUM(CASE WHEN is_skipped = 1 THEN 1 ELSE 0 END), 0) AS skipped_links
+            FROM links",
+        ) {
+            Ok(val) => val,
+            Err(_) => return Err(CustomErrors::StatementFailed),
+        };
+
+        match stmt.query_row([], |row| {
+            let total_links: i32 = row.get(0)?;
+            let completed_links: i32 = row.get(1)?;
+            let skipped_links: i32 = row.get(2)?;
+            Ok((total_links, completed_links, skipped_links))
+        }) {
+            Ok((total_links, completed_links, skipped_links)) => {
+                Ok(Some((total_links, completed_links, skipped_links)))
+            }
+            Err(rusqlite::Error::QueryReturnedNoRows) => Ok(None),
+            Err(_) => Err(CustomErrors::Others(
+                "Error: Something went wrong while checking the status".to_owned(),
+            )),
+        }
+    }
+
     /// add new links into the db
     pub fn add_link(&self, link: String) -> Result<(), CustomErrors> {
         match self.conn.execute(

src/database.rs

@@ -1,5 +1,7 @@
-use tabled::settings::Style;
-use tabled::{Table, Tabled};
+use tabled::{
+    settings::{object::Segment, Alignment, Modify, Style},
+    Table, Tabled,
+};
 
 #[derive(Tabled)]
 struct Links {
@@ -8,6 +10,33 @@ struct Links {
     solved_count: i32,
 }
 
+#[derive(Tabled)]
+struct Status {
+    total_links: i32,
+    completed_links: i32,
+    skipped_links: i32,
+}
+
+pub fn pretty_status(total_links: i32, completed_links: i32, skipped_links: i32) {
+    let data = vec![Status {
+        total_links,
+        completed_links,
+        skipped_links,
+    }];
+
+    let mut table = Table::new(data);
+    table.with(
+        Modify::new(Segment::all())
+            .with(Alignment::center())
+            .with(Alignment::top()),
+    );
+
+    let table = table.with(Style::modern());
+    let table_string = table.to_string();
+
+    println!("{}", table_string);
+}
+
 pub fn pretty_print(data: &[(String, i32)]) {
     let new_data: Vec<Links> = data
         .iter()