feat: Add file status indicators to list command (v0.4.7)

- Add file status indicators showing source code file existence:
  🎯 Problem solved and source file exists locally
  🚧 Source file exists locally but problem not solved yet
  🌐 Problem solved but no local source file
  (blank) Problem not solved and no local source file
- Add --refresh-files flag to update file status cache by scanning local files
- Auto-update file status when creating files via edit command
- Database schema enhancement with `fstatus` field for efficient file status caching
- Improved performance by caching file status in database instead of scanning filesystem on each list command

Author: yousiki

CHANGELOG.md

@@ -1,3 +1,14 @@
+## v0.4.7
+
+- Add file status indicators to list command showing source code file existence
+  - 🎯 Problem solved and source file exists locally
+  - 🚧 Source file exists locally but problem not solved yet  
+  - 🌐 Problem solved but no local source file
+  - No indicator: Problem not solved and no local source file
+- Add `--refresh-files` flag to update file status cache by scanning local files
+- Auto-update file status when creating files via edit command
+- Database schema enhancement with `fstatus` field for efficient file status caching
+
 ## v0.4.1
 
 - Search problems by name

Cargo.toml

@@ -4,7 +4,7 @@ path = "src/bin/lc.rs"
 
 [package]
 name = "leetcode-cli"
-version = "0.4.6"
+version = "0.4.7"
 authors = ["clearloop <tianyi.gc@gmail.com>", "yousiki <you.siki@outlook.com>"]
 edition = "2021"
 description = "Leetcode command-line interface in rust. Forked by yousiki."

src/cache/mod.rs

@@ -384,6 +384,50 @@ impl Cache {
         Ok(res)
     }
 
+    /// Update file status for a single problem
+    pub fn update_problem_file_status(
+        &self,
+        problem_id: i32,
+        file_exists: bool,
+    ) -> Result<(), Error> {
+        let mut c = self.conn()?;
+        let target = problems.filter(id.eq(problem_id));
+        let file_status = if file_exists { "exists" } else { "missing" };
+        diesel::update(target)
+            .set(fstatus.eq(file_status))
+            .execute(&mut c)?;
+        Ok(())
+    }
+
+    /// Batch update file status for all problems
+    pub fn update_all_file_status(&self) -> Result<(), Error> {
+        use crate::helper::code_path;
+        use std::path::Path;
+
+        let mut c = self.conn()?;
+        let all_problems: Vec<Problem> = problems.load(&mut c)?;
+
+        for problem in all_problems {
+            // Skip non-algorithm problems
+            if problem.category != "algorithms" {
+                continue;
+            }
+
+            let file_exists = match code_path(&problem, None) {
+                Ok(path) => Path::new(&path).exists(),
+                Err(_) => false,
+            };
+
+            let file_status = if file_exists { "exists" } else { "missing" };
+            let target = problems.filter(id.eq(problem.id));
+            diesel::update(target)
+                .set(fstatus.eq(file_status))
+                .execute(&mut c)?;
+        }
+
+        Ok(())
+    }
+
     /// New cache
     pub fn new() -> Result<Self, Error> {
         let conf = Config::locate()?;

src/cache/models.rs

@@ -30,6 +30,7 @@ pub struct Problem {
     pub starred: bool,
     pub status: String,
     pub desc: String,
+    pub fstatus: String,
 }
 
 impl Problem {
@@ -51,6 +52,15 @@ impl Problem {
 
         res + "\n"
     }
+
+    pub fn file_status_indicator(&self) -> &str {
+        match (self.status.as_str(), self.fstatus.as_str()) {
+            ("ac", "exists") => "🎯", // Solved and file exists
+            (_, "exists") => "🚧",    // File exists but not solved
+            ("ac", _) => "🌐",        // Solved but no local file
+            _ => "  ",                // Not solved and no local file
+        }
+    }
 }
 
 static DONE: &str = " ✔";
@@ -134,10 +144,14 @@ impl std::fmt::Display for Problem {
         if pct.len() < 5 {
             pct.push_str(&"0".repeat(5 - pct.len()));
         }
+
+        let file_status_indicator = self.file_status_indicator();
+
         write!(
             f,
-            "  {} {} [{}] {} {} ({} %)",
+            "  {} {} {} [{}] {} {} ({} %)",
             lock,
+            file_status_indicator,
             done,
             id,
             name,

src/cache/parser.rs

@@ -35,6 +35,7 @@ pub fn problem(problems: &mut Vec<Problem>, v: Value) -> Option<()> {
             starred: p.get("is_favor")?.as_bool()?,
             status: p.get("status")?.as_str().unwrap_or("Null").to_string(),
             desc: String::new(),
+            fstatus: String::from("unknown"),
         });
     }
 

src/cache/schemas.rs

@@ -12,6 +12,7 @@ table! {
         starred -> Bool,
         status -> Text,
         desc -> Text,
+        fstatus -> Text,
     }
 }
 

src/cache/sql.rs

@@ -10,7 +10,8 @@ pub static CREATE_PROBLEMS_IF_NOT_EXISTS: &str = r#"
     slug TEXT NOT NULL,
     starred BOOLEAN NOT NULL DEFAULT 0,
     status TEXT NOT NULL,
-    desc TEXT NOT NULL DEFAULT ""
+    desc TEXT NOT NULL DEFAULT "",
+    fstatus TEXT NOT NULL DEFAULT "unknown"
   )
 "#;
 

src/cmds/edit.rs

@@ -221,6 +221,10 @@ impl Command for EditCommand {
             .envs(envs)
             .args(args)
             .status()?;
+
+        // Update file status since we just created/edited the file
+        cache.update_problem_file_status(problem.id, true)?;
+
         Ok(())
     }
 }

src/cmds/list.rs

@@ -66,6 +66,7 @@ static LIST_AFTER_HELP: &str = r#"EXAMPLES:
     leetcode list -q eD             List questions that with easy level and not done
     leetcode list -t linked-list    List questions that under tag "linked-list"
     leetcode list -r 50 100         List questions that has id in between 50 and 100
+    leetcode list -f                List questions with updated file status
 "#;
 
 /// implement Command trait for `list`
@@ -125,6 +126,13 @@ impl Command for ListCommand {
                     .num_args(1)
                     .help("Keyword in select query"),
             )
+            .arg(
+                Arg::new("refresh-files")
+                    .short('f')
+                    .long("refresh-files")
+                    .help("Update file status cache by scanning local files")
+                    .action(ArgAction::SetTrue),
+            )
     }
 
     /// `list` command handler
@@ -144,6 +152,14 @@ impl Command for ListCommand {
             return Self::handler(m).await;
         }
 
+        // refresh file status if requested
+        if m.contains_id("refresh-files") {
+            info!("Refreshing file status cache...");
+            cache.update_all_file_status()?;
+            // Reload problems to get updated file status
+            ps = cache.get_problems()?;
+        }
+
         // filtering...
         // pym scripts
         #[cfg(feature = "pym")]