Have a util function to pluralize todo.

3 files changed, 21 insertions, 1 deletions

diff --git a/todolist/app.go b/todolist/app.go
index d4513ac..f63c41b 100644
--- a/todolist/app.go
+++ b/todolist/app.go
@@ -44,7 +44,7 @@ func (a *App) DeleteTodo(input string) {
 	}
 	a.TodoList.Delete(ids...)
 	a.Save()
-	fmt.Println("Todo deleted.")
+	fmt.Printf("%s deleted.\n", pluralize(len(ids), "Todo", "Todos"))
 }
 
 func (a *App) CompleteTodo(input string) {
diff --git a/todolist/util.go b/todolist/util.go
index 34f9dd1..ff3dbba 100644
--- a/todolist/util.go
+++ b/todolist/util.go
@@ -52,3 +52,10 @@ func getNearestMonday(t time.Time) time.Time {
 		}
 	}
 }
+
+func pluralize(count int, singular, plural string) string {
+	if count > 1 {
+		return plural
+	}
+	return singular
+}
diff --git a/todolist/util_test.go b/todolist/util_test.go
new file mode 100644
index 0000000..40ecb54
--- /dev/null
+++ b/todolist/util_test.go
@@ -0,0 +1,13 @@
+package todolist
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPluralize(t *testing.T) {
+	assert := assert.New(t)
+	assert.Equal("todo", pluralize(1, "todo", "todos"))
+	assert.Equal("todos", pluralize(2, "todo", "todos"))
+}