feat: draft implementation

Author: sbackend123

openapi/Swarm.yaml

@@ -1,7 +1,7 @@
 openapi: 3.0.3
 
 info:
-  version: 8.1.1
+  version: 8.2.0
   title: Bee API
   description: "API endpoints for interacting with the Swarm network, supporting file operations, messaging, and node management"
 
@@ -2426,6 +2426,37 @@ paths:
         default:
           description: Default response
 
+  "/redistribution":
+    put:
+      summary: Enable or disable participation in new redistribution rounds
+      description: >
+        Controls whether the node will commit in a new redistribution round.
+        Disabling does not abort an in-flight commit and does not skip reveal or claim
+        for a round that already committed. Re-enabling during a commit phase that was
+        already skipped does not retry that round; participation resumes on the next
+        sample/commit cycle. After a node restart participation is enabled again.
+      tags:
+        - RedistributionState
+      requestBody:
+        required: true
+        content:
+          application/json:
+            schema:
+              $ref: "SwarmCommon.yaml#/components/schemas/RedistributionEnableRequest"
+      responses:
+        "200":
+          description: Participation flag updated
+          content:
+            application/json:
+              schema:
+                $ref: "SwarmCommon.yaml#/components/schemas/RedistributionEnableResponse"
+        "400":
+          $ref: "SwarmCommon.yaml#/components/responses/400"
+        "500":
+          $ref: "SwarmCommon.yaml#/components/responses/500"
+        default:
+          description: Default response
+
   "/redistributionstate":
     get:
       summary: Get the node's redistribution game status

openapi/SwarmCommon.yaml

@@ -825,6 +825,9 @@ components:
           type: boolean
         isHealthy:
           type: boolean
+        enabled:
+          type: boolean
+          description: Whether the node will commit in new redistribution rounds. A disabled node still finishes a round that already has an on-chain commit. Re-enabling during a commit phase that was already skipped does not retry that round; participation resumes on the next sample/commit cycle.
         phase:
           type: string
         round:
@@ -846,6 +849,21 @@ components:
         fees:
           $ref: "#/components/schemas/BigInt"
 
+    RedistributionEnableRequest:
+      type: object
+      required:
+        - enabled
+      properties:
+        enabled:
+          type: boolean
+          description: Whether the node should enter new redistribution rounds.
+
+    RedistributionEnableResponse:
+      type: object
+      properties:
+        enabled:
+          type: boolean
+
     PendingTransactionsResponse:
       type: object
       properties:

pkg/api/api_test.go

@@ -22,6 +22,9 @@ import (
 	"time"
 
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/gorilla/websocket"
+	"resenje.org/web"
+
 	"github.com/ethersphere/bee/v2/pkg/accesscontrol"
 	mockac "github.com/ethersphere/bee/v2/pkg/accesscontrol/mock"
 	accountingmock "github.com/ethersphere/bee/v2/pkg/accounting/mock"
@@ -70,8 +73,6 @@ import (
 	"github.com/ethersphere/bee/v2/pkg/transaction/backendmock"
 	transactionmock "github.com/ethersphere/bee/v2/pkg/transaction/mock"
 	"github.com/ethersphere/bee/v2/pkg/util/testutil"
-	"github.com/gorilla/websocket"
-	"resenje.org/web"
 )
 
 var (
@@ -126,17 +127,18 @@ type testServerOptions struct {
 	BatchStore postage.Storer
 	SyncStatus func() (bool, error)
 
-	BackendOpts         []backendmock.Option
-	Erc20Opts           []erc20mock.Option
-	BeeMode             api.BeeNodeMode
-	RedistributionAgent *storageincentives.Agent
-	NodeStatus          *status.Service
-	PinIntegrity        api.PinIntegrity
-	WhitelistedAddr     string
-	FullAPIDisabled     bool
-	ChequebookDisabled  bool
-	SwapDisabled        bool
-	Erc20ServiceNil     bool
+	BackendOpts                 []backendmock.Option
+	Erc20Opts                   []erc20mock.Option
+	BeeMode                     api.BeeNodeMode
+	RedistributionAgent         *storageincentives.Agent
+	RedistributionAgentDisabled bool
+	NodeStatus                  *status.Service
+	PinIntegrity                api.PinIntegrity
+	WhitelistedAddr             string
+	FullAPIDisabled             bool
+	ChequebookDisabled          bool
+	SwapDisabled                bool
+	Erc20ServiceNil             bool
 }
 
 func newTestServer(t *testing.T, o testServerOptions) (*http.Client, *websocket.Conn, string, *chanStorer) {
@@ -223,11 +225,13 @@ func newTestServer(t *testing.T, o testServerOptions) (*http.Client, *websocket.
 
 	s.SetP2P(o.P2P)
 
-	if o.RedistributionAgent == nil {
-		o.RedistributionAgent, _ = createRedistributionAgentService(t, o.Overlay, o.StateStorer, erc20, transaction, backend, o.BatchStore)
-		s.SetRedistributionAgent(o.RedistributionAgent)
+	if !o.RedistributionAgentDisabled {
+		if o.RedistributionAgent == nil {
+			o.RedistributionAgent, _ = createRedistributionAgentService(t, o.Overlay, o.StateStorer, erc20, transaction, backend, o.BatchStore)
+			s.SetRedistributionAgent(o.RedistributionAgent)
+		}
+		testutil.CleanupCloser(t, o.RedistributionAgent)
 	}
-	testutil.CleanupCloser(t, o.RedistributionAgent)
 
 	s.SetSwarmAddress(&o.Overlay)
 	s.SetProbe(o.Probe)

pkg/api/export_test.go

@@ -98,6 +98,8 @@ type (
 	StakeTransactionReponse           = stakeTransactionReponse
 	StatusSnapshotResponse            = statusSnapshotResponse
 	StatusResponse                    = statusResponse
+	RedistributionStatusResponse      = redistributionStatusResponse
+	RedistributionToggleResponse      = redistributionToggleResponse
 )
 
 var (

pkg/api/redistribution.go

@@ -5,6 +5,7 @@
 package api
 
 import (
+	"encoding/json"
 	"net/http"
 
 	"github.com/ethersphere/bee/v2/pkg/bigint"
@@ -28,6 +29,15 @@ type redistributionStatusResponse struct {
 	Reward                    *bigint.BigInt `json:"reward"`
 	Fees                      *bigint.BigInt `json:"fees"`
 	IsHealthy                 bool           `json:"isHealthy"`
+	Enabled                   bool           `json:"enabled"`
+}
+
+type redistributionToggleRequest struct {
+	Enabled *bool `json:"enabled"`
+}
+
+type redistributionToggleResponse struct {
+	Enabled bool `json:"enabled"`
 }
 
 func (s *Service) redistributionStatusHandler(w http.ResponseWriter, r *http.Request) {
@@ -70,5 +80,30 @@ func (s *Service) redistributionStatusHandler(w http.ResponseWriter, r *http.Req
 		Reward:                    bigint.Wrap(status.Reward),
 		Fees:                      bigint.Wrap(status.Fees),
 		IsHealthy:                 status.IsHealthy,
+		Enabled:                   s.redistributionAgent.IsEnabled(),
 	})
 }
+
+func (s *Service) redistributionToggleHandler(w http.ResponseWriter, r *http.Request) {
+	logger := tracing.NewLoggerWithTraceID(r.Context(), s.logger.WithName("put_redistribution").Build())
+
+	if s.beeMode != FullMode {
+		jsonhttp.BadRequest(w, errOperationSupportedOnlyInFullMode)
+		return
+	}
+
+	var body redistributionToggleRequest
+	if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
+		logger.Debug("decode body failed", "error", err)
+		logger.Error(nil, "decode body failed")
+		jsonhttp.BadRequest(w, "invalid request body")
+		return
+	}
+	if body.Enabled == nil {
+		jsonhttp.BadRequest(w, "enabled is required")
+		return
+	}
+
+	s.redistributionAgent.SetEnabled(*body.Enabled)
+	jsonhttp.OK(w, redistributionToggleResponse{Enabled: *body.Enabled})
+}