From a0f7203996439210b2a40663eeeeb8563ab03d68 Mon Sep 17 00:00:00 2001
From: Grant Ammons <gammons@gmail.com>
Date: Mon, 25 Apr 2016 14:35:43 -0400
Subject: Get grouping working correctly

---
 todolist/app.go | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

(limited to 'todolist/app.go')

diff --git a/todolist/app.go b/todolist/app.go
index 3235f99..6122b6a 100644
--- a/todolist/app.go
+++ b/todolist/app.go
@@ -1,5 +1,7 @@
 package todolist
 
+import "regexp"
+
 type App struct {
 	TodoStore Store
 }
@@ -11,6 +13,26 @@ func NewApp() *App {
 }
 
 func (a *App) ListTodos(input string) {
-	formatter := NewFormatter(a.TodoStore.Todos())
+
+	grouped := a.getGroups(input)
+
+	formatter := NewFormatter(grouped)
 	formatter.Print()
 }
+
+func (a *App) getGroups(input string) *GroupedTodos {
+	grouper := &Grouper{}
+	contextRegex, _ := regexp.Compile(`by c.*$`)
+	projectRegex, _ := regexp.Compile(`by p.*$`)
+
+	var grouped *GroupedTodos
+
+	if contextRegex.MatchString(input) {
+		grouped = grouper.GroupByContext(a.TodoStore.Todos())
+	} else if projectRegex.MatchString(input) {
+		grouped = grouper.GroupByContext(a.TodoStore.Todos())
+	} else {
+		grouped = grouper.GroupByNothing(a.TodoStore.Todos())
+	}
+	return grouped
+}
-- 
cgit v1.3