diff options
Diffstat (limited to 'worker/maildir/message.go')
| -rw-r--r-- | worker/maildir/message.go | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/worker/maildir/message.go b/worker/maildir/message.go index 9d622b4..5c6c930 100644 --- a/worker/maildir/message.go +++ b/worker/maildir/message.go @@ -53,6 +53,26 @@ func (m Message) SetFlags(flags []maildir.Flag) error { return m.dir.SetFlags(m.key, flags) } +// MarkReplied either adds or removes the maildir.FlagReplied flag from the +// message. +func (m Message) MarkReplied(answered bool) error { + flags, err := m.Flags() + if err != nil { + return fmt.Errorf("could not read previous flags: %v", err) + } + if answered { + flags = append(flags, maildir.FlagReplied) + return m.SetFlags(flags) + } + var newFlags []maildir.Flag + for _, flag := range flags { + if flag != maildir.FlagReplied { + newFlags = append(newFlags, flag) + } + } + return m.SetFlags(newFlags) +} + // MarkRead either adds or removes the maildir.FlagSeen flag from the message. func (m Message) MarkRead(seen bool) error { flags, err := m.Flags() |