Here's my Todo List program
use clap::{Parser, Subcommand};
use colored::Colorize;
use rusqlite::{params, Connection};
use rusqlite::types::Null;
use time::{PrimitiveDateTime, OffsetDateTime, Duration, Date};
use time::macros::format_description;
fn slice_to_string(slice: &[i32]) -> String {
let mut str = String::new();
for (i, item) in slice.iter().enumerate() {
str.push_str(item.to_string().as_str());
if i <= slice.len() - 2 {
str.push_str(", ");
}
}
str
}
#[derive(Parser)]
#[command(version, about, long_about = None)]
pub struct Cli {
#[command(subcommand)]
command: Option<Commands>,
}
#[derive(Subcommand)]
enum Commands {
Add { item: String },
Complete { id: i32 },
Remove { id: i32 },
}
fn add_item(item: String, conn: &Connection) -> Result<(), Box<dyn std::error::Error>> {
println!("Do you want to add a due date? (y/n)");
let str = read_line()?;
if str.chars().nth(0) == Some('y'){
println!("What date do you want to add a due date (YYYY-MM-DD)");
let format = format_description!("[year]-[month]-[day]");
let date_time = Date::parse(&read_line(), format)?.midnight();
conn.execute("INSERT INTO items (ITEM, DUE_DATE, IS_COMPLETED) VALUES (?, ?, ?)", params![item, date_time, false])?;
} else {
conn.execute("INSERT INTO items (ITEM, DUE_DATE, IS_COMPLETED) VALUES (?, ?, ?)", params![item, Null, false])?;
}
Ok(())
}
fn read_line() -> Result<String, Box<dyn std::error::Error>> {
let mut buf = String::new();
std::io::stdin().read_line(&mut buf)?;
Ok(buf)
}
fn list_items(conn: &Connection) -> Result<(), Box<dyn std::error::Error>> {
let mut stmt = conn.prepare("SELECT * FROM items")?;
let rows = stmt.query_map((), |rows| Ok((rows.get::<_, i32>(0)?, rows.get::<_, String>(1)?, rows.get::<_, PrimitiveDateTime>(2)?, rows.get::<_, bool>(3)?)))?;
println!("{0: <10} | {1: <50} | {2: <20} | {3: <10}", "ID", "Description", "Due Date", "Status");
let row_vec = rows.map(|rows| rows.unwrap()).collect::<Vec<_>>();
let now = PrimitiveDateTime::new(OffsetDateTime::now_local()?.date(), OffsetDateTime::now_local()?.time());
for row in row_vec.iter() {
let row = row;
let id = row.0;
let description = &row.1;
let due_date = row.2;
let is_completed = row.3;
if !is_completed {
if now < due_date && due_date < now + Duration::days(3) {
println!("{0: <10} | {1: <50} | {2: <20} | {3: <10}", id.to_string().yellow(), description.yellow(), due_date.to_string().yellow(), is_completed.to_string().yellow());
} else if now > due_date {
println!("{0: <10} | {1: <50} | {2: <20} | {3: <10}", id.to_string().red(), description.red(), due_date.to_string().red(), is_completed.to_string().red());
} else {
println!("{0: <10} | {1: <50} | {2: <20} | {3: <10}", id, description, due_date, is_completed);
}
}
};
let overdue_items = &row_vec.iter().filter(|row| row.2 < now).map(|row| row.0).collect::<Vec<_>>()[..];
let close_items = &row_vec.iter().filter(|row| now + Duration::days(3) > row.2 && row.2 > now).map(|row| row.0).collect::<Vec<_>>()[..];
if overdue_items.len() > 0 { println!("{} {:?} {}", "Items".red(), slice_to_string(overdue_items).red(), "are overdue! Finish them quickly!".red()); }
if close_items.len() > 0 { println!("{} {:?} {}", "Items".yellow(), slice_to_string(close_items).yellow(), "are close to deadline! Finish them quickly!".yellow()); }
Ok(())
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let cli = Cli::parse();
let conn = Connection::open("./todo.db")?;
conn.execute("CREATE TABLE IF NOT EXISTS items (\
ID INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE,\
ITEM TEXT NOT NULL UNIQUE,\
DUE_DATE TEXT,\
IS_COMPLETED INTEGER NOT NULL\
)", ())?;
match cli.command {
Some(Commands::Add { item, }) => {
add_item(item, &conn)?;
}
Some(Commands::Complete { id }) => {
conn.execute("UPDATE items SET IS_COMPLETED = 1 WHERE ID = ?", params!(id))?;
},
Some(Commands::Remove { id }) => {
conn.execute("DELETE FROM items WHERE ID = ?", params!(id))?;
}
None => {
list_items(&conn)?;
}
}
Ok(())
}