Implement interface for my custom IMAP flags

1 files changed, 34 insertions, 1 deletions

diff --git a/lib/format/format.go b/lib/format/format.go
index 30e8be7..96a19b9 100644
--- a/lib/format/format.go
+++ b/lib/format/format.go
@@ -297,6 +297,15 @@ func ParseMessageFormat(format string, timeFmt string, ctx Ctx) (string,
 			seen := false
 			recent := false
 			answered := false
+			// NOTE(shivesh): my custom flags
+			var importantFlag = ""
+			var listFlag = ""
+			var foodFlag = ""
+			var personalFlag = ""
+			var workFlag = ""
+			var physicsFlag = ""
+			var todoFlag = ""
+			var laterFlag = ""
 			for _, flag := range ctx.MsgInfo.Flags {
 				if flag == models.SeenFlag {
 					seen = true
@@ -305,6 +314,30 @@ func ParseMessageFormat(format string, timeFmt string, ctx Ctx) (string,
 				} else if flag == models.AnsweredFlag {
 					answered = true
 				}
+				if flag == models.ImportantFlag {
+					importantFlag = "I"
+				}
+				if flag == models.ListFlag {
+					listFlag = "l"
+				}
+				if flag == models.FoodFlag {
+					foodFlag = "F"
+				}
+				if flag == models.PersonalFlag {
+					personalFlag = "P"
+				}
+				if flag == models.WorkFlag {
+					workFlag = "W"
+				}
+				if flag == models.PhysicsFlag {
+					physicsFlag = "X"
+				}
+				if flag == models.TodoFlag {
+					todoFlag = "T"
+				}
+				if flag == models.LaterFlag {
+					laterFlag = "L"
+				}
 				if flag == models.DeletedFlag {
 					delFlag = "D"
 					// TODO: check if attachments
@@ -329,7 +362,7 @@ func ParseMessageFormat(format string, timeFmt string, ctx Ctx) (string,
 				markedFlag = "*"
 			}
 			retval = append(retval, '4', 's')
-			args = append(args, readReplyFlag+delFlag+flaggedFlag+markedFlag)
+			args = append(args, readReplyFlag+delFlag+flaggedFlag+markedFlag+importantFlag+listFlag+foodFlag+personalFlag+workFlag+physicsFlag+todoFlag+laterFlag)
 
 		// Move the below cases to proper alphabetical positions once
 		// implemented