Implement the "dn" feature: Deleteing notes

3 files changed, 27 insertions, 1 deletions

diff --git a/todo.go b/todo.go
index e48eda2..9619bdb 100644
--- a/todo.go
+++ b/todo.go
@@ -127,6 +127,8 @@ func usage() {
 	fmt.Println("\tAdds notes \"check http://this.web.site\" to the todo with id 12\n")
 	yellow.Println("\ttodo ln 12")
 	fmt.Println("\tLists notes of the todo with id 12\n")
+	yellow.Println("\ttodo dn 12 3")
+	fmt.Println("\tDeletes the 3rd note of the todo with id 12\n")
 
 	blueBold.Println("\nGarbage Collection")
 	yellow.Println("\ttodo gc")
@@ -159,7 +161,7 @@ func routeInput(command string, input string) {
 		app.EditTodo(input)
 	case "ex", "expand":
 		app.ExpandTodo(input)
-	case "an", "ln":
+	case "an", "ln", "dn":
 		app.ManipulateNotes(input)
 	case "gc":
 		app.GarbageCollect()
diff --git a/todolist/parser.go b/todolist/parser.go
index 1c3b7f8..3883454 100644
--- a/todolist/parser.go
+++ b/todolist/parser.go
@@ -90,12 +90,29 @@ func (p *Parser) ParseNotes(todo *Todo, input string) string {
 	case "an":
 		todo.Notes = append(todo.Notes, matches[2])
 		return "add"
+
 	case "ln":
 		groups := map[string][]*Todo{}
 		groups[""] = append(groups[""], todo)
 		formatter := NewFormatter(&GroupedTodos{Groups: groups})
 		formatter.PrintNotes()
 		return "list"
+
+	case "dn":
+		rmid, err := strconv.Atoi(matches[2])
+		if err != nil {
+			fmt.Println("wrong note id")
+			return ""
+		}
+
+		for id, _ := range todo.Notes {
+			if id == rmid {
+				todo.Notes = append(todo.Notes[:rmid], todo.Notes[rmid+1:]...)
+				return "delete"
+			}
+		}
+		fmt.Println("Could not found note id")
+		return ""
 	}
 
 	fmt.Println("Could not match command or id")
diff --git a/todolist/parser_test.go b/todolist/parser_test.go
index 9412bdf..e5c7007 100644
--- a/todolist/parser_test.go
+++ b/todolist/parser_test.go
@@ -84,6 +84,13 @@ func TestParseNotes(t *testing.T) {
 	if todo.Notes[0] != "www.google.com" {
 		t.Error("Expected note 1 to be 'www.google.com' but got", todo.Notes[0])
 	}
+
+	if parser.ParseNotes(todo, "dn 1 0") != "delete" {
+		t.Error("Expected Notes to be deleted")
+	}
+	if len(todo.Notes) != 0 {
+		t.Error("Expected no note")
+	}
 }
 
 func TestDueToday(t *testing.T) {