From 3c5325283ffe173f325a7e6b8e2375e7481e79b7 Mon Sep 17 00:00:00 2001 From: Vincent Batts Date: Mon, 11 Aug 2014 08:28:07 -0400 Subject: [PATCH] maybe TODO is the end of the line --- cmd/find-todos/main.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmd/find-todos/main.go b/cmd/find-todos/main.go index 9fe72be..14abeb4 100644 --- a/cmd/find-todos/main.go +++ b/cmd/find-todos/main.go @@ -58,7 +58,7 @@ func main() { 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, ":") { + if strings.HasSuffix(trimmed, ":") || strings.HasSuffix(trimmed, "TODO") { i := strings.IndexRune(line, rune(trimmed[0])) for { buf, err := rdr.Peek(i + 1)