initial refactor

2 files changed, 8 insertions, 1 deletions

diff --git a/internal/entry/entry.go b/internal/entry/entry.go
index 1c5dbad..0fe203e 100644
--- a/internal/entry/entry.go
+++ b/internal/entry/entry.go
@@ -23,7 +23,7 @@ const (
 )
 
 var (
-	validTags            = []string{"ask", "prio", "now"}
+	validTags            = []string{"ask", "prio", "now"} // TODO: Document all tags
 	ErrSizeLimitExceeded = errors.New("message size limit exceeded")
 )
 
@@ -162,6 +162,10 @@ func (e Entry) Edit() error {
 	return nil
 }
 
+func (e Entry) Remove() error {
+	return os.Remove(e.Path)
+}
+
 func extractURLs(input string) []string {
 	urlPattern := `(http://|https://|ftp://)[^\s]+`
 	re := regexp.MustCompile(urlPattern)
diff --git a/internal/entry/entry_test.go b/internal/entry/entry_test.go
index c59dc16..ac97f04 100644
--- a/internal/entry/entry_test.go
+++ b/internal/entry/entry_test.go
@@ -94,6 +94,9 @@ func TestHasTag(t *testing.T) {
 		if err != nil {
 			t.Error(err)
 		}
+		if len(expectedTags) != len(ent.tags) {
+			t.Errorf("expected '%d' tags but got '%d'", len(expectedTags), len(ent.tags))
+		}
 		for _, tag := range expectedTags {
 			if !ent.HasTag(tag) {
 				t.Errorf("expected tag '%s' but got '%s'", tag, ent.tags)