New Feature: Expanding existing todos

2 files changed, 39 insertions, 0 deletions

diff --git a/todolist/app.go b/todolist/app.go
index 58b4801..e67605f 100644
--- a/todolist/app.go
+++ b/todolist/app.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"regexp"
 	"strconv"
+	"strings"
 	"time"
 )
 
@@ -102,6 +103,33 @@ func (a *App) EditTodoDue(input string) {
 	fmt.Println("Todo due date updated.")
 }
 
+func (a *App) ExpandTodo(input string) {
+	a.Load()
+	id, _ := a.getId(input)
+	parser := &Parser{}
+	if id == -1 {
+		return
+	}
+
+	commonProject := parser.ExpandProject(input)
+	todos := strings.LastIndex(input, ":")
+	if commonProject == "" || len(input) <= todos+1 || todos == -1 {
+		fmt.Println("I'm expecting a format like \"todolist ex <project>: <todo1>, <todo2>, ...\"")
+		return
+	}
+
+	newTodos := strings.Split(input[todos+1:], ",")
+
+	for _, todo := range newTodos {
+		args := []string{"add +", commonProject, " ", todo}
+		a.AddTodo(strings.Join(args, ""))
+	}
+
+	a.TodoList.Delete(id)
+	a.Save()
+	fmt.Println("Todo expanded.")
+}
+
 func (a *App) ArchiveCompleted() {
 	a.Load()
 	for _, todo := range a.TodoList.Todos() {
diff --git a/todolist/parser.go b/todolist/parser.go
index 62d864d..3ad3aae 100644
--- a/todolist/parser.go
+++ b/todolist/parser.go
@@ -39,6 +39,17 @@ func (p *Parser) Subject(input string) string {
 	}
 }
 
+func (p *Parser) ExpandProject(input string) string {
+	r, _ := regexp.Compile(`(ex|expand) +\d+ +\+\w+[^:]`)
+	newProject := r.FindString(input)
+	if len(newProject) == 0 {
+		return ""
+	}
+
+	project := strings.Split(newProject, " ")
+	return project[len(project)-1]
+}
+
 func (p *Parser) Projects(input string) []string {
 	r, _ := regexp.Compile(`\+\w+`)
 	return p.matchWords(input, r)