fix(GODT-2774): Add context parameter to Authorize() on Connector

ProtonMail · Jul 6, 2023 · a9327fb · a9327fb
1 parent 02a8a9f
commit a9327fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/connector/connector.go b/connector/connector.go
@@ -14,7 +14,7 @@ var ErrMessageSizeExceedsLimits = errors.New("message size exceeds limits")
 // Connector connects the gluon server to a remote mail store.
 type Connector interface {
 	// Authorize returns whether the given username/password combination are valid for this connector.
-	Authorize(username string, password []byte) bool
+	Authorize(ctx context.Context, username string, password []byte) bool
 
 	// CreateMailbox creates a mailbox with the given name.
 	CreateMailbox(ctx context.Context, name []string) (imap.Mailbox, error)

diff --git a/connector/dummy.go b/connector/dummy.go
@@ -104,7 +104,7 @@ func NewDummy(usernames []string, password []byte, period time.Duration, flags,
 	return conn
 }
 
-func (conn *Dummy) Authorize(username string, password []byte) bool {
+func (conn *Dummy) Authorize(_ context.Context, username string, password []byte) bool {
 	if bytes.Compare(password, conn.password) != 0 {
 		return false
 	}

diff --git a/internal/backend/backend.go b/internal/backend/backend.go
@@ -297,7 +297,7 @@ func (b *Backend) getUserID(ctx context.Context, username string, password []byt
 	b.loginWG.Wait()
 
 	for _, user := range b.users {
-		if user.connector.Authorize(username, password) {
+		if user.connector.Authorize(ctx, username, password) {
 			atomic.StoreInt32(&b.loginErrorCount, 0)
 			return user.userID, nil
 		}