added the ability to get comments and priority

cmd/root.go

@@ -12,6 +12,7 @@ import (
     "github.com/spf13/cobra"
     "strings"
     "bufio"
+    "regexp"
 )
 
 
@@ -60,7 +61,6 @@ func init() {
     rootCmd.PersistentFlags().StringP("path", "p", "","assign the path to get comments from.")
 }
 func readFileTree(path string) {
-	var count int
     fsys := os.DirFS(path)
 
     // Walk through all files in the directory
@@ -83,64 +83,31 @@ func readFileTree(path string) {
             for scanner.Scan() {
                 line := scanner.Text()
 
+		//TODO: P:0 add support for TODO's not at start of line
                 // Check for C-style comments (// and /* */)
-				if strings.Contains(strings.TrimSpace(line), "//") {
+                if strings.HasPrefix(strings.TrimSpace(line), "//TODO:") {
-					count++
+                    // Regular expression to match "P:" followed by a digit
-					fmt.Printf("comment: %s | current comment count:%d\n",strings.TrimSpace(line),count)
+                    re := regexp.MustCompile(`P:(\d)`)
-				}
+
-				if strings.Contains(line, "/*") && strings.Contains(line, "*/") {
+                    // Trim spaces and remove the "//TODO:" prefix
-					count++
+                    trimmedLine := strings.TrimSpace(line)
+                    trimmedLine = strings.TrimPrefix(trimmedLine, "//TODO: ")
+
+                    // Try to find the priority
+                    matches := re.FindStringSubmatch(trimmedLine)
+    
+                    // If a priority is found, remove it and print the comment and priority
+                    if len(matches) > 1 {
+                        // Remove the "P:<digit>" from the beginning of the line
+                        trimmedLine = strings.Replace(trimmedLine, matches[0], "", 1) // Replace only the first occurrence
+                        fmt.Printf("comment: %s | priority: %s \n", strings.TrimSpace(trimmedLine), matches[1])
+                    } else {
+                        fmt.Printf("comment: %s | priority: 9 \n", trimmedLine)
                     }
                 }
 
-			// Check if there was any error reading the file
-			if err := scanner.Err(); err != nil {
-				return err
             }
-		}
-		if filepath.Ext(p) == ".lua" {
-			// Read the file and check for comments
-			file, err := os.Open(filepath.Join(path, p))
-			if err != nil {
-				return err
-			}
-			defer file.Close()
 
-			// Process each line in the file
-			scanner := bufio.NewScanner(file)
-			for scanner.Scan() {
-				line := scanner.Text()
-
-				// Check for Lua-style comments (--)
-				if strings.Contains(strings.TrimSpace(line), "--") {
-					count++
-					fmt.Printf("comment: %s | current comment count:%d\n",strings.TrimSpace(line),count)
-				}
-			}
-			// Check if there was any error reading the file
-			if err := scanner.Err(); err != nil {
-				return err
-			}
-		}
-		if filepath.Ext(p) == ".py" {
-			// Read the file and check for comments
-			file, err := os.Open(filepath.Join(path, p))
-			if err != nil {
-				return err
-			}
-			defer file.Close()
-			
-			// Process each line in the file
-			scanner := bufio.NewScanner(file)
-			for scanner.Scan() {
-				line := scanner.Text()
-
-				// Check for Python-style comments (#)
-				if strings.Contains(strings.TrimSpace(line), "#") {
-					count++ 
-					fmt.Printf("comment: %s | current comment count:%d\n",strings.TrimSpace(line),count)
-				}
-			}
             // Check if there was any error reading the file
             if err := scanner.Err(); err != nil {
                 return err
@@ -148,8 +115,4 @@ func readFileTree(path string) {
         }
         return nil
     })
-
-	// Print the number of comments found
-	fmt.Println("Comment count:", count)
 }
-