feat: integarate polkadot kurtosis package in DIVE-CLI to run polkado…

…t testnet-mainnet setup
HugoByte · Dec 14, 2023 · 1a14fc9 · 1a14fc9
1 parent 07ab525
commit 1a14fc9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/cli/cmd/chains/polkadot/cmd.go b/cli/cmd/chains/polkadot/cmd.go
@@ -53,6 +53,7 @@ func polkadot(cmd *cobra.Command, args []string) {
 	if err != nil {
 		cliContext.Fatal(err)
 	}
+
 	serviceFileName := fmt.Sprintf(common.ServiceFilePath, common.EnclaveName)
 
 	fmt.Print(response.Dive)
@@ -63,6 +64,6 @@ func polkadot(cmd *cobra.Command, args []string) {
 			cliContext.Fatal(err)
 		}
 	}
-
-	cliContext.StartSpinnerIfNotVerbose("Polkadot Node Started. Please find the service details in current working directory(services.json)", common.DiveLogs)
+	stopMessage := fmt.Sprintf("Polkadot Node Started. Please find the service details in current working directory (%s)", serviceFileName)
+	cliContext.StopSpinnerIfNotVerbose(stopMessage, common.DiveLogs)
 }
diff --git a/cli/cmd/chains/polkadot/run.go b/cli/cmd/chains/polkadot/run.go
@@ -68,7 +68,6 @@ func RunPolkadot(cli *common.Cli) (*common.DiveMultipleServiceResponse, error) {
 	polkadotResponseData := &common.DiveMultipleServiceResponse{}
 	result, err := polkadotResponseData.Decode([]byte(responseData))
 
-
 	if err != nil {
 
 		errRemove := cli.Context().RemoveServicesByServiceNames(services, common.DiveEnclave)
@@ -107,8 +106,11 @@ func configureService(serviceConfig *utils.PolkadotServiceConfig) {
 }
 
 func configureMetrics(serviceConfig *utils.PolkadotServiceConfig) {
-	for _, node := range append(serviceConfig.RelayChain.Nodes, serviceConfig.Para[0].Nodes...) {
-		node.Prometheus = true
+	for i := range serviceConfig.RelayChain.Nodes {
+		serviceConfig.RelayChain.Nodes[i].Prometheus = true
+	}
+	for i := range serviceConfig.Para[0].Nodes {
+		serviceConfig.Para[0].Nodes[i].Prometheus = true
 	}
 }
 

diff --git a/cli/common/types.go b/cli/common/types.go
@@ -18,7 +18,7 @@ type DiveServiceResponse struct {
 	PrometheusEndpoint string `json:"endpoint_prometheus,omitempty"`
 	Prometheus         bool   `json:"prometheus,omitempty"`
 	IpAddress          string `json:"ip_address,omitempty"`
-	Node               string `json:"node,omitempty"`
+	Node               string `json:"node-type,omitempty"`
 }
 
 type DiveMultipleServiceResponse struct {