Allow adding additional start args (#381)

strangelove-ventures · Oct 27, 2023 · ff3610b · ff3610b
1 parent 41a210e
commit ff3610b
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 0 deletions.
diff --git a/api/v1/cosmosfullnode_types.go b/api/v1/cosmosfullnode_types.go
@@ -542,6 +542,10 @@ type ChainSpec struct {
 	// If not provided, the operator will not upgrade the chain, and will use the image specified in the pod spec.
 	// +optional
 	Versions []ChainVersion `json:"versions"`
+
+	// Additional arguments to pass to the chain start command.
+	// +optional
+	AdditionalStartArgs []string `json:"additionalStartArgs"`
 }
 
 type ChainVersion struct {

diff --git a/api/v1/zz_generated.deepcopy.go b/api/v1/zz_generated.deepcopy.go
diff --git a/config/crd/bases/cosmos.strange.love_cosmosfullnodes.yaml b/config/crd/bases/cosmos.strange.love_cosmosfullnodes.yaml
@@ -45,6 +45,11 @@ spec:
               chain:
                 description: Blockchain-specific configuration.
                 properties:
+                  additionalStartArgs:
+                    description: Additional arguments to pass to the chain start command.
+                    items:
+                      type: string
+                    type: array
                   addrbookScript:
                     description: 'Specify shell (sh) script commands to properly download
                       and save the address book file. Prefer AddrbookURL if the file

diff --git a/internal/fullnode/pod_builder.go b/internal/fullnode/pod_builder.go
@@ -473,6 +473,9 @@ func startCommandArgs(crd *cosmosv1.CosmosFullNode) []string {
 	if format := cfg.LogFormat; format != nil {
 		args = append(args, "--log_format", *format)
 	}
+	if len(crd.Spec.ChainSpec.AdditionalStartArgs) > 0 {
+		args = append(args, crd.Spec.ChainSpec.AdditionalStartArgs...)
+	}
 	return args
 }
 

diff --git a/internal/fullnode/pod_builder_test.go b/internal/fullnode/pod_builder_test.go
@@ -202,6 +202,20 @@ func TestPodBuilder(t *testing.T) {
 		test.RequireValidMetadata(t, pod)
 	})
 
+	t.Run("additional args", func(t *testing.T) {
+		crd := defaultCRD()
+
+		crd.Spec.ChainSpec.AdditionalStartArgs = []string{"--foo", "bar"}
+
+		builder := NewPodBuilder(&crd)
+		pod, err := builder.WithOrdinal(0).Build()
+		require.NoError(t, err)
+
+		test.RequireValidMetadata(t, pod)
+
+		require.Equal(t, []string{"start", "--home", "/home/operator/cosmos", "--foo", "bar"}, pod.Spec.Containers[0].Args)
+	})
+
 	t.Run("containers", func(t *testing.T) {
 		crd := defaultCRD()
 		const wantWrkDir = "/home/operator"