search by tag, show all

src/main.rs

@@ -3,18 +3,19 @@ use clap::Parser;
 use home::home_dir;
 use rusqlite::{Connection, Result};
 
-/// Simple program to greet a person
 #[derive(Parser, Debug)]
 #[command(version, about, long_about = None)]
 struct Args {
-    // Quickly throw in a note
+    /// Create a new note: --create "Hello World"
     #[arg(short, long, default_value_t = String::new())]
     create: String,
-    // Number of times to greet
-    //#[arg(short, long, default_value_t = 1)]
-    //count: u8,
+    /// Show the latest entries
     #[arg(short, long, default_value_t = false)]
     show: bool,
+    /// Show all entries
+    #[arg(long, default_value_t = false)]
+    showall: bool,
+    /// Append or show entries with tag
     #[arg(short, long, default_value_t = String::new())]
     tag: String,
 }
@@ -28,28 +29,38 @@ struct Note {
 fn main() {
     let args = Args::parse();
 
-    //for _ in 0..args.count {
-    //    println!("Hello {}!", args.name)
-    //}
-
     match init_db() {
         Ok(_) => println!(),
         Err(e) => println!("Failed to initialize DB: {}", e),
     };
 
     if args.show {
-        match show_notes() {
+        match show_notes(false, &args.tag) {
             Ok(_) => println!(),
             Err(e) => println!("Failed to show DB: {}", e),
         }
     }
 
     if args.create != String::new() {
-        match create_note(args.create, args.tag) {
+        match create_note(&args.create, &args.tag) {
             Ok(_) => println!(),
             Err(e) => println!("Failed to throw into DB: {}", e),
         }
     }
+
+    if args.showall {
+        match show_notes(true, &args.tag) {
+            Ok(_) => println!(),
+            Err(e) => println!("Failed to show DB: {}", e),
+        }
+    }
+
+    if args.tag != String::new() && args.create == String::new() {
+        match show_notes(true, &args.tag) {
+            Ok(_) => println!(),
+            Err(e) => println!("Failed to show DB: {}", e),
+        }
+    }
 }
 
 fn init_db() -> Result<()> {
@@ -73,15 +84,20 @@ fn init_db() -> Result<()> {
     Ok(())
 }
 
-fn create_note(content: String, tag: String) -> Result<()> {
+fn create_note(content: &String, tag: &String) -> Result<()> {
     let home = home_dir().unwrap().join(".snotes.db");
     let connection = Connection::open(home)?;
     let date = Local::now();
-    let date_string = date.format("%m-%d %H:%M").to_string();
+    let date_string = date.format("%m-%d-%y %H:%M").to_string();
+    let tag_string = if tag.to_string() == String::new() {
+        String::from("quick")
+    } else {
+        tag.to_string()
+    };
 
     let query_insert = "INSERT INTO notes (content, date, tag) VALUES (?1, ?2, ?3)";
 
-    match connection.execute(query_insert, [&content, &date_string, &tag]) {
+    match connection.execute(query_insert, [&content, &date_string, &tag_string]) {
         Ok(v) => println!("CREATE OK {}", v),
         Err(e) => println!("CREATE ERR {}", e),
     };
@@ -89,13 +105,21 @@ fn create_note(content: String, tag: String) -> Result<()> {
     Ok(())
 }
 
-fn show_notes() -> Result<()> {
+fn show_notes(all: bool, tag: &String) -> Result<()> {
     let home = home_dir().unwrap().join(".snotes.db");
     let connection = Connection::open(home)?;
 
-    let query_latest_10 = "SELECT * FROM notes LIMIT 10";
+    let mut query = String::from("SELECT * FROM notes LIMIT 10");
 
-    let mut prepare = connection.prepare(query_latest_10)?;
+    if all {
+        query = String::from("SELECT * FROM notes");
+    }
+
+    if tag != &String::new() {
+        query = format!("SELECT * FROM notes WHERE tag IS '{}'", tag);
+    }
+
+    let mut prepare = connection.prepare(&query)?;
 
     let notes = prepare.query_map([], |row| {
         Ok(Note {