adding find-todo

cmd/find-todos/main.go

@@ -0,0 +1,91 @@
+package main
+
+import (
+	"bufio"
+	"flag"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+var (
+	NoteDir   string
+	flNoteDir *string
+	flNotePat *string
+	err       error
+)
+
+func main() {
+	if len(os.Getenv("NOTEDIR")) > 0 {
+		NoteDir, err = filepath.Abs(os.Getenv("NOTEDIR"))
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+	} else {
+		NoteDir = filepath.Join(os.Getenv("HOME"), "Notes")
+	}
+	flNoteDir = flag.String("d", NoteDir, "directory of notes")
+	flNotePat = flag.String("p", "Tasks*.txt", "file pattern")
+	flag.Parse()
+
+	matches, err := filepath.Glob(filepath.Join(*flNoteDir, *flNotePat))
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+
+	for _, match := range matches {
+		fh, err := os.Open(match)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		rdr := bufio.NewReader(fh)
+		count := int64(0)
+		for {
+			line, err := rdr.ReadString('\n')
+			if err == io.EOF {
+				break
+			}
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
+			count += 1
+			if strings.Contains(line, "TODO") {
+				fmt.Printf("%s:%d\t%s\n", filepath.Base(match), count, strings.TrimRight(line, " \n"))
+				trimmed := strings.TrimSpace(line)
+				if strings.HasSuffix(trimmed, ":") {
+					i := strings.IndexRune(line, rune(trimmed[0]))
+					for {
+						buf, err := rdr.Peek(i + 1)
+						if err == io.EOF {
+							break
+						}
+						if err != nil {
+							fmt.Println(err)
+							os.Exit(1)
+						}
+						if strings.Count(string(buf), " ") > 0 {
+							nextLine, err := rdr.ReadString('\n')
+							if err == io.EOF {
+								break
+							}
+							if err != nil {
+								fmt.Println(err)
+								os.Exit(1)
+							}
+							fmt.Printf("%s:%d\t%s\n", filepath.Base(match), count, strings.TrimRight(nextLine, " \n"))
+						} else {
+							break
+						}
+					}
+				}
+			}
+		}
+		fh.Close()
+	}
+}