feat: change staking overlay

pkg/node/node.go

@@ -273,6 +273,7 @@ func NewBee(
 		}
 	}
 
+	var changedOverlay bool
 	if targetNeighborhood != "" {
 		neighborhood, err := swarm.ParseBitStrAddress(targetNeighborhood)
 		if err != nil {
@@ -311,6 +312,7 @@ func NewBee(
 			if err != nil {
 				return nil, fmt.Errorf("statestore: save new overlay: %w", err)
 			}
+			changedOverlay = true
 		}
 	}
 
@@ -966,6 +968,21 @@ func NewBee(
 
 	stakingContract := staking.New(overlayEthAddress, stakingContractAddress, abiutil.MustParseABI(chainCfg.StakingABI), bzzTokenAddress, transactionService, common.BytesToHash(nonce))
 
+	if chainEnabled && changedOverlay {
+		stake, err := stakingContract.GetStake(ctx)
+		if err != nil {
+			return nil, errors.New("getting stake balance")
+		}
+		if stake.Cmp(big.NewInt(0)) > 0 {
+			logger.Debug("changing overlay address in staking contract")
+			tx, err := stakingContract.ChangeStakeOverlay(ctx, common.BytesToHash(nonce))
+			if err != nil {
+				return nil, fmt.Errorf("cannot change staking overlay address: %v", err.Error())
+			}
+			logger.Info("overlay address changed in staking contract", "transaction", tx)
+		}
+	}
+
 	var (
 		pullerService *puller.Puller
 		agent         *storageincentives.Agent

pkg/storageincentives/staking/contract.go

@@ -37,6 +37,7 @@ var (
 
 type Contract interface {
 	DepositStake(ctx context.Context, stakedAmount *big.Int) (common.Hash, error)
+	ChangeStakeOverlay(ctx context.Context, nonce common.Hash) (common.Hash, error)
 	GetStake(ctx context.Context) (*big.Int, error)
 	WithdrawAllStake(ctx context.Context) (common.Hash, error)
 	RedistributionStatuser
@@ -223,6 +224,16 @@ func (c *contract) DepositStake(ctx context.Context, stakedAmount *big.Int) (com
 	return receipt.TxHash, nil
 }
 
+// ChangeStakeOverlay only changes the overlay address used in the redistribution game.
+func (c *contract) ChangeStakeOverlay(ctx context.Context, nonce common.Hash) (common.Hash, error) {
+	receipt, err := c.sendDepositStakeTransaction(ctx, new(big.Int), c.overlayNonce)
+	if err != nil {
+		return common.Hash{}, err
+	}
+
+	return receipt.TxHash, nil
+}
+
 func (c *contract) GetStake(ctx context.Context) (*big.Int, error) {
 	stakedAmount, err := c.getStake(ctx)
 	if err != nil {

pkg/storageincentives/staking/mock/contract.go

@@ -23,6 +23,10 @@ func (s *stakingContractMock) DepositStake(ctx context.Context, stakedAmount *bi
 	return s.depositStake(ctx, stakedAmount)
 }
 
+func (s *stakingContractMock) ChangeStakeOverlay(_ context.Context, h common.Hash) (common.Hash, error) {
+	return h, nil
+}
+
 func (s *stakingContractMock) GetStake(ctx context.Context) (*big.Int, error) {
 	return s.getStake(ctx)
 }