fix: address reviewer feedbacks

multiaccounts/database.go

@@ -344,8 +344,12 @@ func (db *Database) UpdateAccountTimestamp(keyUID string, loginTimestamp int64)
 	return err
 }
 
-func (db *Database) UpdateAccountCustomizationColor(keyUID string, color string, clock uint64) (sql.Result, error) {
-	return db.db.Exec("UPDATE accounts SET customizationColor = ?, customizationColorClock = ? WHERE keyUid = ? AND customizationColorClock < ?", color, clock, keyUID, clock)
+func (db *Database) UpdateAccountCustomizationColor(keyUID string, color string, clock uint64) (int64, error) {
+	result, err := db.db.Exec("UPDATE accounts SET customizationColor = ?, customizationColorClock = ? WHERE keyUid = ? AND customizationColorClock < ?", color, clock, keyUID, clock)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
 }
 
 func (db *Database) DeleteAccount(keyUID string) error {

protocol/messenger_handler.go

@@ -2415,15 +2415,11 @@ func (m *Messenger) HandleSyncSetting(messageState *ReceivedMessageState, messag
 }
 
 func (m *Messenger) HandleSyncAccountCustomizationColor(state *ReceivedMessageState, message *protobuf.SyncAccountCustomizationColor, statusMessage *v1protocol.StatusMessage) error {
-	result, err := m.multiAccounts.UpdateAccountCustomizationColor(message.GetKeyUid(), message.GetCustomizationColor(), message.GetUpdatedAt())
+	affected, err := m.multiAccounts.UpdateAccountCustomizationColor(message.GetKeyUid(), message.GetCustomizationColor(), message.GetUpdatedAt())
 	if err != nil {
 		return err
 	}
 
-	affected, err := result.RowsAffected()
-	if err != nil {
-		return err
-	}
 	if affected > 0 {
 		state.Response.CustomizationColor = message.GetCustomizationColor()
 	}

protocol/messenger_messages.go

@@ -5,13 +5,13 @@ import (
 	"crypto/ecdsa"
 	"errors"
 
-	"go.uber.org/zap"
-
 	"github.com/golang/protobuf/proto"
 
 	"github.com/status-im/status-go/protocol/common"
 	"github.com/status-im/status-go/protocol/protobuf"
 	"github.com/status-im/status-go/protocol/requests"
+
+	"go.uber.org/zap"
 )
 
 var ErrInvalidEditOrDeleteAuthor = errors.New("sender is not the author of the message")

protocol/messenger_settings.go

@@ -2,6 +2,7 @@ package protocol
 
 import (
 	"context"
+
 	"github.com/status-im/status-go/nodecfg"
 	"github.com/status-im/status-go/protocol/requests"
 	"github.com/status-im/status-go/timesource"
@@ -24,6 +25,9 @@ func (m *Messenger) SetCustomNodes(request *requests.SetCustomNodes) error {
 }
 
 func (m *Messenger) SetCustomizationColor(ctx context.Context, request *requests.SetCustomizationColor) error {
+	if err := request.Validate(); err != nil {
+		return err
+	}
 
 	acc, err := m.multiAccounts.GetAccount(request.KeyUID)
 	if err != nil {
@@ -32,20 +36,24 @@ func (m *Messenger) SetCustomizationColor(ctx context.Context, request *requests
 
 	acc.CustomizationColor = request.CustomizationColor
 
+	//Use a combination of wall clock + lamport timestamp, just like Chat#NextClockAndTimestamp
 	tNow := timesource.GetCurrentTimeInMillis()
 	if acc.CustomizationColorClock >= tNow {
 		acc.CustomizationColorClock++
 	} else {
 		acc.CustomizationColorClock = tNow
 	}
 
-	_, err = m.multiAccounts.UpdateAccountCustomizationColor(request.KeyUID, string(acc.CustomizationColor), acc.CustomizationColorClock)
+	affected, err := m.multiAccounts.UpdateAccountCustomizationColor(request.KeyUID, string(acc.CustomizationColor), acc.CustomizationColorClock)
 	if err != nil {
 		return err
 	}
-	err = m.syncAccountCustomizationColor(ctx, acc)
-	if err != nil {
-		return err
+
+	if affected > 0 {
+		err = m.syncAccountCustomizationColor(ctx, acc)
+		if err != nil {
+			return err
+		}
 	}
 	return nil
 }

protocol/messenger_settings_test.go

@@ -5,12 +5,14 @@ import (
 	"testing"
 
 	gethbridge "github.com/status-im/status-go/eth-node/bridge/geth"
+
 	"github.com/status-im/status-go/eth-node/crypto"
 	"github.com/status-im/status-go/multiaccounts/common"
 	"github.com/status-im/status-go/protocol/encryption/multidevice"
 	"github.com/status-im/status-go/protocol/requests"
 	"github.com/status-im/status-go/protocol/tt"
 	"github.com/status-im/status-go/waku"
+
 	"github.com/stretchr/testify/suite"
 )
 
@@ -94,7 +96,9 @@ func (s *MessengerSettingsSuite) TestSetCustomizationColor() {
 	}, "message syncAccountCustomizationColor not received")
 	s.Require().NoError(err)
 	acc, err = s.alice2.multiAccounts.GetAccount(s.alice.account.KeyUID)
+	s.Require().NoError(err)
 	acc2, err = s.alice2.multiAccounts.GetAccount(s.alice2.account.KeyUID)
 	s.Require().NoError(err)
+	s.Require().Equal(common.CustomizationColorBlue, acc.CustomizationColor)
 	s.Require().Equal(acc.CustomizationColor, acc2.CustomizationColor)
 }

protocol/requests/set_customization_color.go

@@ -1,8 +1,27 @@
 package requests
 
-import "github.com/status-im/status-go/multiaccounts/common"
+import (
+	"errors"
+
+	"github.com/status-im/status-go/multiaccounts/common"
+)
+
+var ErrSetCustomizationColorInvalidColor = errors.New("customizationColor: invalid color")
+var ErrSetCustomizationColorInvalidKeyUID = errors.New("keyUid: invalid id")
 
 type SetCustomizationColor struct {
 	CustomizationColor common.CustomizationColor `json:"customizationColor"`
 	KeyUID             string                    `json:"keyUid"`
 }
+
+func (a *SetCustomizationColor) Validate() error {
+	if len(a.CustomizationColor) == 0 {
+		return ErrSetCustomizationColorInvalidColor
+	}
+
+	if len(a.KeyUID) == 0 {
+		return ErrSetCustomizationColorInvalidKeyUID
+	}
+
+	return nil
+}