added some color

wagoodman · Jun 11, 2018 · 23289cc · 23289cc
1 parent 3602d7d
commit 23289cc
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/filetree/node.go b/filetree/node.go
@@ -3,6 +3,8 @@ package filetree
 import (
 	"sort"
 	"strings"
+
+	"github.com/fatih/color"
 )
 
 type FileNode struct {
@@ -63,7 +65,20 @@ func (node *FileNode) Remove() error {
 }
 
 func (node *FileNode) String() string {
-	return node.Name
+	var style *color.Color
+	switch node.Data.DiffType {
+	case Added:
+		style = color.New(color.FgGreen)
+	case Removed:
+		style = color.New(color.FgRed)
+	case Changed:
+		style = color.New(color.FgYellow)
+	case Unchanged:
+		style = color.New(color.Reset)
+	default:
+		style = color.New(color.BgMagenta)
+	}
+	return style.Sprint(node.Name)
 }
 
 func (node *FileNode) Visit(visiter Visiter) error {