Implement the "dn" feature: Deleteing notes

2 files changed, 24 insertions, 0 deletions

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) {