Implement interface for my custom IMAP flags

1 files changed, 17 insertions, 0 deletions

diff --git a/worker/imap/search.go b/worker/imap/search.go
index f866b1c..d27648b 100644
--- a/worker/imap/search.go
+++ b/worker/imap/search.go
@@ -66,6 +66,23 @@ func getParsedFlag(name string) (string, error) {
 		return imap.FlaggedFlag, nil
 	case "answered":
 		return imap.AnsweredFlag, nil
+	// NOTE(shivesh): my custom flags
+	case "important":
+		return name, nil
+	case "list":
+		return name, nil
+	case "food":
+		return name, nil
+	case "personal":
+		return name, nil
+	case "work":
+		return name, nil
+	case "physics":
+		return name, nil
+	case "todo":
+		return name, nil
+	case "later":
+		return name, nil
 	}
 	return imap.FlaggedFlag, errors.New("Flag not suppored")
 }