feat(GODT-2801): Add Key Decryption helpers to identityservice.State

This commit is contained in:
Leander Beernaert
2023-07-25 09:59:26 +02:00
parent 1b0f930471
commit da528f2d9b

View File

@ -21,6 +21,7 @@ import (
"context"
"crypto/subtle"
"fmt"
"github.com/ProtonMail/gopenpgp/v2/crypto"
"strings"
"github.com/ProtonMail/go-proton-api"
@ -243,3 +244,16 @@ func (s *State) CheckAuth(email string, password []byte, bridgePassProvider Brid
return "", fmt.Errorf("invalid email")
}
func (s *State) WithAddrKR(addrID string, keyPass []byte, fn func(userKR, addrKR *crypto.KeyRing) error) error {
addr, ok := s.Addresses[addrID]
if !ok {
return fmt.Errorf("address not found")
}
return usertypes.WithAddrKR(s.User, addr, keyPass, fn)
}
func (s *State) WithAddrKRs(keyPass []byte, fn func(*crypto.KeyRing, map[string]*crypto.KeyRing) error) error {
return usertypes.WithAddrKRs(s.User, s.Addresses, keyPass, fn)
}