feat: add ability to print todos in terminal

src/main.rs

@@ -2,8 +2,6 @@ use std::fs;
 use std::env;
 use colored::Colorize;
 
-// \x1b is used to print in red
-
 fn main() {
     let args: Vec<String> = env::args().collect();
     if args.len() != 2 {
@@ -23,11 +21,49 @@ fn main() {
         return
     }
 
-    let objects = fs::read_dir(args[1].to_owned()).unwrap();
+    let todos = traverse_dir(&args[1]).unwrap();
-    for result in objects {
+    for todo in todos {
-        let object = result.unwrap().path();
+        println!("TODO: {}", todo.green());
-        let metadata = fs::metadata(object.to_str().unwrap()).unwrap();
-        let type_ = if metadata.is_dir() { "Dirc" } else { "File" };
-        println!("{}: {}", type_, object.display())
     }
+}
+
+fn traverse_dir(path: &str) -> Result<Vec<String>, std::io::Error> {
+    let mut todos = Vec::new();
+
+    let objects = fs::read_dir(path.to_owned()).unwrap();
+    for result in objects {
+        let obj_path = result.unwrap().path();
+        let obj_str = obj_path.to_str().unwrap();
+        let obj_metadata = fs::metadata(obj_str).unwrap();
+
+        if obj_metadata.is_dir() {
+            println!("DIRC: {}", obj_str.yellow());
+            todos.append(&mut traverse_dir(obj_str).unwrap());
+        }
+        else if obj_metadata.is_file() {
+            if obj_str.ends_with(".py") {
+                println!("FILE: {}", obj_str.yellow());
+                todos.append(&mut get_todos(obj_str).unwrap());
+            }
+        }
+    }
+    
+    Ok(todos)
+}
+
+fn get_todos(path: &str) -> Result<Vec<String>, std::io::Error> {
+    let mut todos = Vec::new();
+
+    let contents = fs::read_to_string(path).unwrap();
+    for line in contents.lines() {
+        if line.contains("TODO") {
+            let (_, comment) = line.split_once("TODO").unwrap();
+            let cleaned_comment = comment.replace(":", "")
+                                                .trim()
+                                                .to_owned();
+            todos.push(cleaned_comment.to_owned())
+        }
+    }
+
+    Ok(todos)
 }