1
0
Fork 0
mirror of https://github.com/vbatts/freezing-octo-hipster.git synced 2024-11-21 22:45:39 +00:00

adding find-todo

This commit is contained in:
Vincent Batts 2014-06-12 23:14:29 -04:00
commit 4c3cae91dd

91
cmd/find-todos/main.go Normal file
View file

@ -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()
}
}