feat: Add ability to switch between different hub environments

cmd/config/init/consts.go

@@ -5,36 +5,41 @@ var FlagNames = struct {
 	DAEndpoint    string
 	Decimals      string
 	RollappBinary string
-	HubRPC        string
+	HubID         string
 }{
 	Home:          "home",
 	DAEndpoint:    "data-availability-endpoint",
 	Decimals:      "decimals",
 	RollappBinary: "rollapp-binary",
-	HubRPC:        "hub-rpc",
+	HubID:         "hub",
 }
 
+const TestnetHubID = "35-C"
+const StagingHubID = "internal-devnet"
+const LocalHubID = "local"
 
-var HubData = struct {
+var Hubs = map[string]HubData{
+	TestnetHubID: {
+		API_URL: "https://rest-hub-35c.dymension.xyz",
+		ID:      "35-C",
+		RPC_URL: "https://rpc-hub-35c.dymension.xyz:443",
+	},
+	StagingHubID: {
+		API_URL: "https://rest-hub-devnet.dymension.xyz",
+		ID:      "internal-devnet",
+		RPC_URL: "https://rpc-hub-devnet.dymension.xyz:443",
+	},
+	LocalHubID: {
+		API_URL: "http://localhost:1317",
+		ID:      "local",
+		RPC_URL: "http://localhost:36657",
+	},
+}
+
+type HubData = struct {
 	API_URL string
 	ID      string
 	RPC_URL string
-}{
-	API_URL: "https://rest-hub-35c.dymension.xyz",
-	ID:      "35-C",
-	RPC_URL: "https://rpc-hub-35c.dymension.xyz:443",
-}
-
-var Executables = struct {
-	Celestia  string
-	Rollapp   string
-	Relayer   string
-	Dymension string
-}{
-	Celestia:  "/usr/local/bin/roller_bins/celestia",
-	Rollapp:   "/usr/local/bin/rollapp_evm",
-	Relayer:   "/usr/local/bin/roller_bins/rly",
-	Dymension: "/usr/local/bin/roller_bins/dymd",
 }
 
 const defaultRollappRPC = "http://localhost:26657"

cmd/config/init/flags.go

@@ -1,18 +1,31 @@
 package initconfig
 
 import (
+	"fmt"
+
 	"github.com/dymensionxyz/roller/cmd/consts"
 	"github.com/spf13/cobra"
 
 	"github.com/dymensionxyz/roller/cmd/utils"
 )
 
 func addFlags(cmd *cobra.Command) {
-	cmd.Flags().StringP(FlagNames.HubRPC, "", HubData.RPC_URL, "Dymension Hub rpc endpoint")
+	cmd.Flags().StringP(FlagNames.HubID, "", TestnetHubID, fmt.Sprintf("The ID of the Dymension hub. %s", getAvailableHubsMessage()))
 	cmd.Flags().StringP(FlagNames.DAEndpoint, "", "", "The data availability light node endpoint. Runs an Arabica Celestia light node if not provided")
 	cmd.Flags().StringP(FlagNames.RollappBinary, "", "", "The rollapp binary. Should be passed only if you built a custom rollapp")
 	cmd.Flags().Uint64P(FlagNames.Decimals, "", 18, "The number of decimal places a rollapp token supports")
 	cmd.Flags().StringP(FlagNames.Home, "", utils.GetRollerRootDir(), "The directory of the roller config files")
+
+	cmd.PreRunE = func(cmd *cobra.Command, args []string) error {
+		hubID, err := cmd.Flags().GetString(FlagNames.HubID)
+		if err != nil {
+			return err
+		}
+		if _, ok := Hubs[hubID]; !ok {
+			return fmt.Errorf("invalid hub ID: %s. %s", hubID, getAvailableHubsMessage())
+		}
+		return nil
+	}
 }
 
 func getDecimals(cmd *cobra.Command) uint64 {
@@ -38,13 +51,18 @@ func GetInitConfig(cmd *cobra.Command, args []string) InitConfig {
 	createLightNode := !cmd.Flags().Changed(FlagNames.DAEndpoint)
 	rollappBinaryPath := getRollappBinaryPath(cmd)
 	decimals := getDecimals(cmd)
+	hubID := cmd.Flag(FlagNames.HubID).Value.String()
 	return InitConfig{
 		Home:              home,
 		RollappID:         rollappId,
 		RollappBinary:     rollappBinaryPath,
 		CreateDALightNode: createLightNode,
 		Denom:             denom,
-		HubID:             HubData.ID,
 		Decimals:          decimals,
+		HubData:           Hubs[hubID],
 	}
 }
+
+func getAvailableHubsMessage() string {
+	return fmt.Sprintf("Acceptable values are '%s', '%s' or '%s'", TestnetHubID, StagingHubID, LocalHubID)
+}

cmd/config/init/init.go

@@ -3,9 +3,9 @@ package initconfig
 import (
 	"os"
 
-	"github.com/spf13/cobra"
-	"github.com/dymensionxyz/roller/cmd/utils"
 	"github.com/dymensionxyz/roller/cmd/consts"
+	"github.com/dymensionxyz/roller/cmd/utils"
+	"github.com/spf13/cobra"
 )
 
 type InitConfig struct {
@@ -15,7 +15,7 @@ type InitConfig struct {
 	CreateDALightNode bool
 	Denom             string
 	Decimals          uint64
-	HubID             string
+	HubData           HubData
 }
 
 func InitCmd() *cobra.Command {
@@ -24,7 +24,7 @@ func InitCmd() *cobra.Command {
 		Short: "Initialize a RollApp configuration on your local machine.",
 		Run: func(cmd *cobra.Command, args []string) {
 			initConfig := GetInitConfig(cmd, args)
-			utils.PrettifyErrorIfExists(VerifyUniqueRollappID(initConfig.RollappID))
+			utils.PrettifyErrorIfExists(VerifyUniqueRollappID(initConfig.RollappID, initConfig))
 			isRootExist, err := dirNotEmpty(initConfig.Home)
 			utils.PrettifyErrorIfExists(err)
 			if isRootExist {
@@ -49,8 +49,8 @@ func InitCmd() *cobra.Command {
 				Denom:         initConfig.Denom,
 				AddressPrefix: consts.AddressPrefixes.Rollapp,
 			}, ChainConfig{
-				ID:            HubData.ID,
-				RPC:           cmd.Flag(FlagNames.HubRPC).Value.String(),
+				ID:            initConfig.HubData.ID,
+				RPC:           initConfig.HubData.RPC_URL,
 				Denom:         "udym",
 				AddressPrefix: consts.AddressPrefixes.Hub,
 			}, initConfig))

cmd/config/init/keys.go

@@ -66,7 +66,7 @@ func getDefaultKeysConfig(initConfig InitConfig) []utils.KeyConfig {
 			Prefix:   consts.AddressPrefixes.Rollapp,
 		},
 		{
-			Dir:      path.Join(consts.ConfigDirName.Relayer, KeysDirName, HubData.ID),
+			Dir:      path.Join(consts.ConfigDirName.Relayer, KeysDirName, initConfig.HubData.ID),
 			ID:       consts.KeyNames.HubRelayer,
 			CoinType: consts.CoinTypes.Cosmos,
 			Prefix:   consts.AddressPrefixes.Hub,

cmd/config/init/unique_rollapp_id.go

@@ -5,8 +5,8 @@ import (
 	"net/http"
 )
 
-func IsRollappIDUnique(rollappID string) (bool, error) {
-	url := HubData.API_URL + "/dymensionxyz/dymension/rollapp/rollapp/" + rollappID
+func IsRollappIDUnique(rollappID string, initConfig InitConfig) (bool, error) {
+	url := initConfig.HubData.API_URL + "/dymensionxyz/dymension/rollapp/rollapp/" + rollappID
 
 	req, err := http.NewRequest("GET", url, nil)
 	if err != nil {
@@ -31,8 +31,8 @@ func IsRollappIDUnique(rollappID string) (bool, error) {
 	}
 }
 
-func VerifyUniqueRollappID(rollappID string) error {
-	isUniqueRollapp, err := IsRollappIDUnique(rollappID)
+func VerifyUniqueRollappID(rollappID string, initConfig InitConfig) error {
+	isUniqueRollapp, err := IsRollappIDUnique(rollappID, initConfig)
 	if err != nil {
 		return err
 	}

cmd/register/register.go

@@ -24,7 +24,7 @@ func RegisterCmd() *cobra.Command {
 			home := cmd.Flag(initconfig.FlagNames.Home).Value.String()
 			rollappConfig, err := initconfig.LoadConfigFromTOML(home)
 			utils.PrettifyErrorIfExists(err)
-			utils.PrettifyErrorIfExists(initconfig.VerifyUniqueRollappID(rollappConfig.RollappID))
+			utils.PrettifyErrorIfExists(initconfig.VerifyUniqueRollappID(rollappConfig.RollappID, rollappConfig))
 			utils.PrettifyErrorIfExists(registerRollapp(rollappConfig))
 		},
 	}
@@ -81,6 +81,6 @@ func getRegisterRollappCmd(rollappConfig initconfig.InitConfig) *exec.Cmd {
 		"--keyring-backend", "test",
 		"--keyring-dir", filepath.Join(rollappConfig.Home, consts.ConfigDirName.Rollapp),
 		rollappConfig.RollappID, "stamp1", "genesis-path/1", "3", "3", `{"Addresses":[]}`, "--output", "json",
-		"--node", initconfig.HubData.RPC_URL, "--yes", "--broadcast-mode", "block",
+		"--node", rollappConfig.HubData.RPC_URL, "--yes", "--broadcast-mode", "block",
 	)
 }

test/config/init/init_test.go

@@ -57,7 +57,7 @@ func TestInitCmd(t *testing.T) {
 			assert.NoError(testutils.VerifyRollerConfig(initConfig))
 			assert.NoError(os.Remove(filepath.Join(tempDir, initconfig.RollerConfigFileName)))
 			assert.NoError(testutils.VerifyRollappKeys(tempDir))
-			assert.NoError(testutils.VerifyRelayerKeys(tempDir, rollappID, initconfig.HubData.ID))
+			assert.NoError(testutils.VerifyRelayerKeys(tempDir, rollappID, initConfig.HubData.ID))
 			if !testutils.Contains(tc.optionalFlags, "--"+initconfig.FlagNames.DAEndpoint) {
 				assert.NoError(testutils.VerifyLightNodeKeys(tempDir))
 			}