mas: handle 3pids from mas

userapi/api/api.go

@@ -111,6 +111,7 @@ type ClientUserAPI interface {
 	QueryLocalpartForThreePID(ctx context.Context, req *QueryLocalpartForThreePIDRequest, res *QueryLocalpartForThreePIDResponse) error
 	PerformForgetThreePID(ctx context.Context, req *PerformForgetThreePIDRequest, res *struct{}) error
 	PerformSaveThreePIDAssociation(ctx context.Context, req *PerformSaveThreePIDAssociationRequest, res *struct{}) error
+	PerformBulkSaveThreePIDAssociation(ctx context.Context, req *PerformBulkSaveThreePIDAssociationRequest, res *struct{}) error
 }
 
 type KeyBackupAPI interface {
@@ -653,6 +654,12 @@ type PerformSaveThreePIDAssociationRequest struct {
 	Medium     string
 }
 
+type PerformBulkSaveThreePIDAssociationRequest struct {
+	ThreePIDs  []authtypes.ThreePID
+	Localpart  string
+	ServerName spec.ServerName
+}
+
 type QueryAccountByLocalpartRequest struct {
 	Localpart  string
 	ServerName spec.ServerName

userapi/internal/user_api.go

@@ -987,4 +987,8 @@ func (a *UserInternalAPI) PerformSaveThreePIDAssociation(ctx context.Context, re
 	return a.DB.SaveThreePIDAssociation(ctx, req.ThreePID, req.Localpart, req.ServerName, req.Medium)
 }
 
+func (a *UserInternalAPI) PerformBulkSaveThreePIDAssociation(ctx context.Context, req *api.PerformBulkSaveThreePIDAssociationRequest, res *struct{}) error {
+	return a.DB.BulkSaveThreePIDAssociation(ctx, req.ThreePIDs, req.Localpart, req.ServerName)
+}
+
 const pushRulesAccountDataType = "m.push_rules"

userapi/storage/interface.go

@@ -120,6 +120,7 @@ type Pusher interface {
 
 type ThreePID interface {
 	SaveThreePIDAssociation(ctx context.Context, threepid, localpart string, serverName spec.ServerName, medium string) (err error)
+	BulkSaveThreePIDAssociation(ctx context.Context, threePIDs []authtypes.ThreePID, localpart string, serverName spec.ServerName) (err error)
 	RemoveThreePIDAssociation(ctx context.Context, threepid string, medium string) (err error)
 	GetLocalpartForThreePID(ctx context.Context, threepid string, medium string) (localpart string, serverName spec.ServerName, err error)
 	GetThreePIDsForLocalpart(ctx context.Context, localpart string, serverName spec.ServerName) (threepids []authtypes.ThreePID, err error)

userapi/storage/shared/storage.go

@@ -353,6 +353,41 @@ func (d *Database) SaveThreePIDAssociation(
 	})
 }
 
+// BulkSaveThreePIDAssociation recreates 3PIDs for a user.
+// If the third-party identifier is already part of an association, returns Err3PIDInUse.
+// Returns an error if there was a problem talking to the database.
+func (d *Database) BulkSaveThreePIDAssociation(ctx context.Context, threePIDs []authtypes.ThreePID, localpart string, serverName spec.ServerName) (err error) {
+	return d.Writer.Do(d.DB, nil, func(txn *sql.Tx) error {
+		oldThreePIDs, err := d.ThreePIDs.SelectThreePIDsForLocalpart(ctx, localpart, serverName)
+		if err != nil {
+			return err
+		}
+		for _, t := range oldThreePIDs {
+			if err := d.ThreePIDs.DeleteThreePID(ctx, txn, t.Address, t.Medium); err != nil {
+				return err
+			}
+		}
+		for _, t := range threePIDs {
+			// if 3PID is associated with another user, return Err3PIDInUse
+			user, _, err := d.ThreePIDs.SelectLocalpartForThreePID(
+				ctx, txn, t.Address, t.Medium,
+			)
+			if err != nil {
+				return err
+			}
+
+			if len(user) > 0 && user != localpart {
+				return Err3PIDInUse
+			}
+
+			if err = d.ThreePIDs.InsertThreePID(ctx, txn, t.Address, t.Medium, localpart, serverName); err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
 // RemoveThreePIDAssociation removes the association involving a given third-party
 // identifier.
 // If no association exists involving this third-party identifier, returns nothing.