diff options
author | Koni Marti <koni.marti@gmail.com> | 2023-07-16 17:04:14 +0200 |
---|---|---|
committer | Robin Jarry <robin@jarry.cc> | 2023-08-03 22:29:18 +0200 |
commit | 0e09c05937913a938bc4987db2b6d193ed0501bd (patch) | |
tree | 34a54eb15b3bbc2ff507671c7b28f8943cac73df /worker/imap/extensions/xgmext/client.go | |
parent | 2fbce2e2c9aa782cc3d99a7232d78876b835e513 (diff) | |
download | aerc-0e09c05937913a938bc4987db2b6d193ed0501bd.tar.gz |
imap: support the Gmail extension (X-GM-EXT-1)
Support the IMAP Gmail extension (X-GM-EXT-1) to fetch all messages for
a given thread. This allows client-side threading to display a full
message thread. Obviously, it requires a Gmail account to work.
The extension is only used when requested in accounts.conf with:
"use-gmail-ext = true" (default: false)
Signed-off-by: Koni Marti <koni.marti@gmail.com>
Acked-by: Robin Jarry <robin@jarry.cc>
Tested-by: Tristan Partin <tristan@partin.io>
Diffstat (limited to 'worker/imap/extensions/xgmext/client.go')
-rw-r--r-- | worker/imap/extensions/xgmext/client.go | 86 |
1 files changed, 86 insertions, 0 deletions
diff --git a/worker/imap/extensions/xgmext/client.go b/worker/imap/extensions/xgmext/client.go new file mode 100644 index 00000000..3107e642 --- /dev/null +++ b/worker/imap/extensions/xgmext/client.go @@ -0,0 +1,86 @@ +package xgmext + +import ( + "errors" + "fmt" + + "git.sr.ht/~rjarry/aerc/log" + "github.com/emersion/go-imap" + "github.com/emersion/go-imap/client" + "github.com/emersion/go-imap/commands" + "github.com/emersion/go-imap/responses" +) + +type handler struct { + client *client.Client +} + +func NewHandler(c *client.Client) *handler { + return &handler{client: c} +} + +func (h handler) FetchEntireThreads(requested []uint32) ([]uint32, error) { + threadIds, err := h.fetchThreadIds(requested) + if err != nil { + return nil, + fmt.Errorf("faild to fetch thread IDs: %w", err) + } + uids, err := h.searchUids(threadIds) + if err != nil { + return nil, + fmt.Errorf("faild to search for thread IDs: %w", err) + } + return uids, nil +} + +func (h handler) fetchThreadIds(uids []uint32) ([]string, error) { + messages := make(chan *imap.Message) + done := make(chan error) + + thriditem := imap.FetchItem("X-GM-THRID") + items := []imap.FetchItem{ + thriditem, + } + + m := make(map[string]struct{}, len(uids)) + go func() { + defer log.PanicHandler() + for msg := range messages { + m[msg.Items[thriditem].(string)] = struct{}{} + } + done <- nil + }() + + var set imap.SeqSet + set.AddNum(uids...) + err := h.client.UidFetch(&set, items, messages) + <-done + + thrid := make([]string, 0, len(m)) + for id := range m { + thrid = append(thrid, id) + } + return thrid, err +} + +func (h handler) searchUids(thrid []string) ([]uint32, error) { + if len(thrid) == 0 { + return nil, errors.New("no thread IDs provided") + } + + if h.client.State() != imap.SelectedState { + return nil, errors.New("no mailbox selected") + } + + var cmd imap.Commander = NewThreadIDSearch(thrid) + cmd = &commands.Uid{Cmd: cmd} + + res := new(responses.Search) + + status, err := h.client.Execute(cmd, res) + if err != nil { + return nil, fmt.Errorf("imap execute failed: %w", err) + } + + return res.Ids, status.Err() +} |