add rust project

1 files changed, 137 insertions, 0 deletions

diff --git a/rust/todo/src/main.rs b/rust/todo/src/main.rs
new file mode 100644
index 0000000..e009712
--- /dev/null
+++ b/rust/todo/src/main.rs
@@ -0,0 +1,137 @@
+use clap::{Arg, Command};
+use std::fs::{self, OpenOptions};
+use std::io::{self, Write};
+use std::path::Path;
+
+#[derive(Debug)]
+struct Task {
+    id: usize,
+    description: String,
+    completed: bool,
+}
+
+impl Task {
+    fn to_string(&self) -> String {
+        let status = if self.completed { "[X]" } else { "[ ]" };
+        format!("{} {}", status, self.description)
+    }
+}
+
+fn main() {
+    let matches = Command::new("Task Echo")
+        .version("1.0")
+        .about("Echoes tasks to a file")
+        .arg(
+            Arg::new("task")
+                .short('t')
+                .long("task")
+                .value_name("TASK")
+                .help("Sets the task to be saved")
+                .num_args(1..),
+        )
+        .arg(
+            Arg::new("done")
+                .short('d')
+                .long("done")
+                .value_name("TASK_ID")
+                .help("Marks the specified task(s) as complete")
+                .num_args(1..),
+        )
+        .get_matches();
+
+    let file_path = "tasks.txt";
+
+    // Handle task completion
+    if let Some(ids) = matches.get_many::<String>("done") {
+        let mut tasks = load_tasks(file_path);
+        let mut completed_tasks = Vec::new();
+        for id in ids {
+            if let Ok(index) = id.parse::<usize>() {
+                if index > 0 && index <= tasks.len() {
+                    tasks[index - 1].completed = true; // Mark task as complete
+                } else {
+                    completed_tasks.push(id.clone()); // Collect invalid task IDs
+                }
+            }
+        }
+        save_tasks(file_path, &tasks);
+
+        // Alert for nonexistent tasks
+        if !completed_tasks.is_empty() {
+            for id in completed_tasks {
+                println!("Task {} doesn't exist.", id);
+            }
+        } else {
+            println!("Tasks marked as complete.");
+        }
+
+        // Display the updated task list
+        display_tasks(file_path);
+    }
+
+    // Handle new tasks
+    if let Some(new_tasks) = matches.get_many::<String>("task") {
+        let mut tasks = load_tasks(file_path);
+        let next_id = tasks.len() + 1; // Determine the next task ID
+        for task_desc in new_tasks {
+            tasks.push(Task {
+                id: next_id,
+                description: task_desc.to_string(),
+                completed: false,
+            });
+        }
+        save_tasks(file_path, &tasks);
+        println!("Tasks saved to {}", file_path);
+
+        // Display the updated task list
+        display_tasks(file_path);
+    }
+
+    // Display saved tasks if no new tasks or done tasks were provided
+    if !matches.contains_id("task") && !matches.contains_id("done") {
+        display_tasks(file_path);
+    }
+}
+
+fn load_tasks(file_path: &str) -> Vec<Task> {
+    let mut tasks = Vec::new();
+    if Path::new(file_path).exists() {
+        let contents = fs::read_to_string(file_path).expect("Unable to read file");
+        for (id, line) in contents.lines().enumerate() {
+            let completed = line.starts_with("[X]");
+            let description = line[4..].trim().to_string();
+            tasks.push(Task {
+                id: id + 1,
+                description,
+                completed,
+            });
+        }
+    }
+    tasks
+}
+
+fn save_tasks(file_path: &str, tasks: &[Task]) {
+    let mut file = OpenOptions::new()
+        .create(true)
+        .write(true)
+        .truncate(true) // Clear the file before writing
+        .open(file_path)
+        .expect("Unable to open file");
+
+    for task in tasks {
+        writeln!(file, "{}", task.to_string()).expect("Unable to write to file");
+    }
+}
+
+fn display_tasks(file_path: &str) {
+    let tasks = load_tasks(file_path);
+    if tasks.is_empty() {
+        println!("No tasks saved yet.");
+    } else {
+        println!("Saved tasks:");
+        for task in tasks {
+            // Pad the task ID to a width of 3 for alignment
+            println!("{:>3}: {}", task.id, task.to_string()); // Display task ID
+        }
+    }
+}