feat(GODT-2617): Allow change send ability on GPA server addresses

Author: LBeernaertProton

server/backend/address.go

@@ -7,20 +7,21 @@ import (
 )
 
 type address struct {
-	addrID   string
-	email    string
-	order    int
-	status   proton.AddressStatus
-	addrType proton.AddressType
-	keys     []key
+	addrID    string
+	email     string
+	order     int
+	status    proton.AddressStatus
+	addrType  proton.AddressType
+	keys      []key
+	allowSend bool
 }
 
 func (add *address) toAddress() proton.Address {
 	return proton.Address{
 		ID:    add.addrID,
 		Email: add.email,
 
-		Send:    true,
+		Send:    proton.Bool(add.allowSend),
 		Receive: true,
 		Status:  add.status,
 		Type:    add.addrType,

server/backend/backend.go

@@ -209,14 +209,32 @@ func (b *Backend) RemoveUserKey(userID, keyID string) error {
 }
 
 func (b *Backend) CreateAddress(userID, email string, password []byte, withKey bool, status proton.AddressStatus, addrType proton.AddressType) (string, error) {
-	return b.createAddress(userID, email, password, withKey, status, addrType, false)
+	return b.createAddress(userID, email, password, withKey, status, addrType, false, true)
 }
 
 func (b *Backend) CreateAddressAsUpdate(userID, email string, password []byte, withKey bool, status proton.AddressStatus, addrType proton.AddressType) (string, error) {
-	return b.createAddress(userID, email, password, withKey, status, addrType, true)
+	return b.createAddress(userID, email, password, withKey, status, addrType, true, true)
 }
 
-func (b *Backend) createAddress(userID, email string, password []byte, withKey bool, status proton.AddressStatus, addrType proton.AddressType, issueUpdateInsteadOfCreate bool) (string, error) {
+func (b *Backend) CreateAddressWithSendDisabled(
+	userID, email string,
+	password []byte,
+	withKey bool,
+	status proton.AddressStatus,
+	addrType proton.AddressType,
+) (string, error) {
+	return b.createAddress(userID, email, password, withKey, status, addrType, false, false)
+}
+
+func (b *Backend) createAddress(
+	userID, email string,
+	password []byte,
+	withKey bool,
+	status proton.AddressStatus,
+	addrType proton.AddressType,
+	issueUpdateInsteadOfCreate bool,
+	allowSend bool,
+) (string, error) {
 	return withAcc(b, userID, func(acc *account) (string, error) {
 		var keys []key
 
@@ -257,12 +275,13 @@ func (b *Backend) createAddress(userID, email string, password []byte, withKey b
 		addressID := uuid.NewString()
 
 		acc.addresses[addressID] = &address{
-			addrID:   addressID,
-			email:    email,
-			order:    len(acc.addresses) + 1,
-			status:   status,
-			addrType: addrType,
-			keys:     keys,
+			addrID:    addressID,
+			email:     email,
+			order:     len(acc.addresses) + 1,
+			status:    status,
+			addrType:  addrType,
+			keys:      keys,
+			allowSend: allowSend,
 		}
 
 		var update update
@@ -295,6 +314,18 @@ func (b *Backend) ChangeAddressType(userID, addrId string, addrType proton.Addre
 	})
 }
 
+func (b *Backend) ChangeAddressAllowSend(userID, addrId string, allowSend bool) error {
+	return b.withAcc(userID, func(acc *account) error {
+		for _, addr := range acc.addresses {
+			if addr.addrID == addrId {
+				addr.allowSend = allowSend
+				return nil
+			}
+		}
+		return fmt.Errorf("no addrID matching %s for user %s", addrId, userID)
+	})
+}
+
 func (b *Backend) CreateAddressKey(userID, addrID string, password []byte) error {
 	return b.withAcc(userID, func(acc *account) error {
 		token, err := crypto.RandomToken(32)

server/server.go

@@ -155,6 +155,10 @@ func (s *Server) CreateAddressAsUpdate(userID, email string, password []byte) (s
 	return s.b.CreateAddressAsUpdate(userID, email, password, true, proton.AddressStatusEnabled, proton.AddressTypeOriginal)
 }
 
+func (s *Server) ChangeAddressAllowSend(userID, addrID string, allowSend bool) error {
+	return s.b.ChangeAddressAllowSend(userID, addrID, allowSend)
+}
+
 func (s *Server) ChangeAddressType(userID, addrId string, addrType proton.AddressType) error {
 	return s.b.ChangeAddressType(userID, addrId, addrType)
 }