1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
package backend
import (
"context"
"time"
"github.com/ProtonMail/gluon/connector"
"github.com/ProtonMail/gluon/imap"
"github.com/ProtonMail/gluon/internal/state"
)
type stateConnectorImpl struct {
connector connector.Connector
metadata map[string]any
user *user
}
func newStateConnectorImpl(u *user) state.Connector {
return &stateConnectorImpl{
connector: u.connector,
metadata: make(map[string]any),
user: u,
}
}
func (sc *stateConnectorImpl) SetConnMetadataValue(key string, value any) {
sc.metadata[key] = value
}
func (sc *stateConnectorImpl) ClearConnMetadataValue(key string) {
delete(sc.metadata, key)
}
func (sc *stateConnectorImpl) ClearAllConnMetadata() {
sc.metadata = make(map[string]any)
}
func (sc *stateConnectorImpl) CreateMailbox(ctx context.Context, name []string) (imap.Mailbox, error) {
ctx = sc.newContextWithMetadata(ctx)
mbox, err := sc.connector.CreateMailbox(ctx, name)
if err != nil {
return imap.Mailbox{}, err
}
return mbox, nil
}
func (sc *stateConnectorImpl) UpdateMailbox(ctx context.Context, mboxID imap.MailboxID, newName []string) error {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.UpdateMailboxName(ctx, mboxID, newName)
}
func (sc *stateConnectorImpl) DeleteMailbox(ctx context.Context, mboxID imap.MailboxID) error {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.DeleteMailbox(ctx, mboxID)
}
func (sc *stateConnectorImpl) CreateMessage(
ctx context.Context,
mboxID imap.MailboxID,
literal []byte,
flags imap.FlagSet,
date time.Time,
) (imap.InternalMessageID, imap.Message, []byte, error) {
ctx = sc.newContextWithMetadata(ctx)
msg, newLiteral, err := sc.connector.CreateMessage(ctx, mboxID, literal, flags, date)
if err != nil {
return imap.InternalMessageID{}, imap.Message{}, nil, err
}
return imap.NewInternalMessageID(), msg, newLiteral, nil
}
func (sc *stateConnectorImpl) GetMessageLiteral(ctx context.Context, id imap.MessageID) ([]byte, error) {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.GetMessageLiteral(ctx, id)
}
func (sc *stateConnectorImpl) AddMessagesToMailbox(
ctx context.Context,
messageIDs []imap.MessageID,
mboxID imap.MailboxID,
) error {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.AddMessagesToMailbox(ctx, messageIDs, mboxID)
}
func (sc *stateConnectorImpl) RemoveMessagesFromMailbox(
ctx context.Context,
messageIDs []imap.MessageID,
mboxID imap.MailboxID,
) error {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.RemoveMessagesFromMailbox(ctx, messageIDs, mboxID)
}
func (sc *stateConnectorImpl) MoveMessagesFromMailbox(
ctx context.Context,
messageIDs []imap.MessageID,
mboxFromID imap.MailboxID,
mboxToID imap.MailboxID,
) (bool, error) {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.MoveMessages(ctx, messageIDs, mboxFromID, mboxToID)
}
func (sc *stateConnectorImpl) SetMessagesSeen(ctx context.Context, messageIDs []imap.MessageID, seen bool) error {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.MarkMessagesSeen(ctx, messageIDs, seen)
}
func (sc *stateConnectorImpl) SetMessagesFlagged(ctx context.Context, messageIDs []imap.MessageID, flagged bool) error {
ctx = sc.newContextWithMetadata(ctx)
return sc.connector.MarkMessagesFlagged(ctx, messageIDs, flagged)
}
func (sc *stateConnectorImpl) GetMailboxVisibility(ctx context.Context, id imap.MailboxID) imap.MailboxVisibility {
return sc.connector.GetMailboxVisibility(ctx, id)
}
func (sc *stateConnectorImpl) getMetadataValue(key string) any {
v, ok := sc.metadata[key]
if !ok {
return nil
}
return v
}
func (sc *stateConnectorImpl) newContextWithMetadata(ctx context.Context) context.Context {
if v := sc.getMetadataValue(imap.IMAPIDConnMetadataKey); v != nil {
switch x := v.(type) {
case imap.IMAPID:
ctx = imap.NewContextWithIMAPID(ctx, x)
}
}
return ctx
}
|
