some general refactoring

Author: patrickdevivo

cmd/commands/todos.go

@@ -51,12 +51,9 @@ var todosCmd = &cobra.Command{
 		commit, err := r.CommitObject(ref.Hash())
 		handleError(err)
 
-		comments, err := comments.SearchCommit(commit)
+		comments, err := comments.SearchDir(dir)
 		handleError(err)
 
-		// comments, err := comments.SearchDir(dir)
-		// handleError(err)
-
 		t := todos.NewToDos(comments)
 
 		ctx := context.Background()
@@ -67,7 +64,6 @@ var todosCmd = &cobra.Command{
 			total := len(t)
 			s.Suffix = fmt.Sprintf(" (%d/%d) %s: %s", total-remaining, total, commit.Hash, commit.Author.When)
 		})
-
 		sort.Sort(&t)
 
 		handleError(err)

go.mod

@@ -13,10 +13,12 @@ require (
 	github.com/spf13/cobra v0.0.5
 	github.com/spf13/pflag v1.0.5 // indirect
 	github.com/src-d/enry/v2 v2.1.0
+	github.com/stretchr/testify v1.4.0 // indirect
 	golang.org/x/crypto v0.0.0-20191029031824-8986dd9e96cf // indirect
 	golang.org/x/net v0.0.0-20191028085509-fe3aa8a45271 // indirect
 	golang.org/x/sys v0.0.0-20191029155521-f43be2a4598c // indirect
 	gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 // indirect
 	gopkg.in/src-d/go-billy.v4 v4.3.2
 	gopkg.in/src-d/go-git.v4 v4.13.1
+	gopkg.in/yaml.v2 v2.2.4 // indirect
 )

go.sum

@@ -114,6 +114,8 @@ github.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoH
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.3.0 h1:TivCn/peBQ7UY8ooIcPgZFpTNSz0Q2U6UrFlUfqbe0Q=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
+github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
 github.com/toqueteos/trie v1.0.0 h1:8i6pXxNUXNRAqP246iibb7w/pSFquNTQ+uNfriG7vlk=
 github.com/toqueteos/trie v1.0.0/go.mod h1:Ywk48QhEqhU1+DwhMkJ2x7eeGxDHiGkAdc9+0DYcbsM=
 github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0=
@@ -173,3 +175,5 @@ gopkg.in/warnings.v0 v0.1.2 h1:wFXVbFY8DY5/xOe1ECiWdKCzZlxgshcYVNkBHstARME=
 gopkg.in/warnings.v0 v0.1.2/go.mod h1:jksf8JmL6Qr/oQM2OXTHunEvvTAsrWBLb6OOjuVWRNI=
 gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
 gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.4 h1:/eiJrUcujPVeJ3xlSWaiNi3uSVmDGBK1pDHUHAnao1I=
+gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

pkg/comments/comments.go

@@ -70,7 +70,7 @@ type Comment struct {
 }
 
 // SearchFile searches a file for comments. It infers the language
-func SearchFile(filePath string, reader io.ReadCloser) (Comments, error) {
+func SearchFile(filePath string, reader io.Reader) (Comments, error) {
 	src, err := ioutil.ReadAll(reader)
 	if err != nil {
 		return nil, err
@@ -120,7 +120,7 @@ func SearchDir(dirPath string) (Comments, error) {
 		if matched {
 			return nil
 		}
-		if !info.IsDir() {
+		if info.Mode().IsRegular() {
 			p, err := filepath.Abs(path)
 			if err != nil {
 				return err
@@ -129,6 +129,7 @@ func SearchDir(dirPath string) (Comments, error) {
 			if err != nil {
 				return err
 			}
+			defer f.Close()
 			t, err := SearchFile(p, f)
 			if err != nil {
 				return err

pkg/todos/todos.go

@@ -2,6 +2,7 @@ package todos
 
 import (
 	"context"
+	"fmt"
 	"regexp"
 	"strings"
 	"sync"
@@ -34,6 +35,10 @@ type Author struct {
 	When  time.Time
 }
 
+func (a *Author) String() string {
+	return fmt.Sprintf("%s <%s>", a.Name, a.Email)
+}
+
 // ToDos represents a list of ToDo items
 type ToDos []*ToDo
 
@@ -72,30 +77,30 @@ func NewToDos(comments comments.Comments) ToDos {
 }
 
 // Len returns the number of todos
-func (t *ToDos) Len() int {
-	return len(*t)
+func (t ToDos) Len() int {
+	return len(t)
 }
 
 // Less compares two todos by their creation time
-func (t *ToDos) Less(i, j int) bool {
-	first := (*t)[i]
-	second := (*t)[j]
+func (t ToDos) Less(i, j int) bool {
+	first := t[i]
+	second := t[j]
 	if first.Commit == nil || second.Commit == nil {
 		return false
 	}
 	return first.Commit.Author.When.Before(second.Commit.Author.When)
 }
 
 // Swap swaps two todoss
-func (t *ToDos) Swap(i, j int) {
-	temp := (*t)[i]
-	(*t)[i] = (*t)[j]
-	(*t)[j] = temp
+func (t ToDos) Swap(i, j int) {
+	temp := t[i]
+	t[i] = t[j]
+	t[j] = temp
 }
 
 // CountWithCommits returns the number of todos with an associated commit (in which that todo was added)
-func (t *ToDos) CountWithCommits() (count int) {
-	for _, todo := range *t {
+func (t ToDos) CountWithCommits() (count int) {
+	for _, todo := range t {
 		if todo.Commit != nil {
 			count++
 		}