use mnemonic in registrar cli

Author: Eslam-Nawara

registrar-cli/cmd/account_create.go

@@ -12,7 +12,7 @@ var accountCreateCmd = &cobra.Command{
 	Use:   "create",
 	Short: "create new account in node registrar",
 	RunE: func(cobraCmd *cobra.Command, args []string) error {
-		seed, err := cobraCmd.Flags().GetString("seed")
+		mnemonic, err := cobraCmd.Flags().GetString("menmonic")
 		if err != nil {
 			return err
 		}
@@ -32,10 +32,11 @@ var accountCreateCmd = &cobra.Command{
 			return err
 		}
 
-		account, err := cmd.CreaeteAccount(seed, network, relays, rmbEncKey)
+		account, mnemonic, err := cmd.CreaeteAccount(network, relays, rmbEncKey, mnemonic)
 		if err != nil {
 			return err
 		}
+		log.Info().Str("mnemonic", mnemonic).Msg("new account is created with mnemonic")
 
 		log.Info().Uint64("twinID", account.TwinID).Msg("account is created successfully")
 
@@ -45,7 +46,7 @@ var accountCreateCmd = &cobra.Command{
 
 func init() {
 	accountCmd.AddCommand(accountCreateCmd)
-	accountCreateCmd.Flags().StringP("seed", "s", "", "account seed key")
+	accountCreateCmd.Flags().StringP("menmonic", "m", "", "account menmonic")
 	accountCreateCmd.Flags().StringP("network", "n", "", "network (dev, qa, test, main)")
 	accountCreateCmd.Flags().StringArrayP("relays", "r", nil, "relays urls")
 	accountCreateCmd.Flags().StringP("rmb-enc-key", "k", "", "rmb encryption key")

registrar-cli/cmd/account_update.go

@@ -12,7 +12,7 @@ var accountUpdateCmd = &cobra.Command{
 	Use:   "update",
 	Short: "update account in node registrar",
 	RunE: func(cobraCmd *cobra.Command, args []string) error {
-		seed, err := cobraCmd.Flags().GetString("seed")
+		seed, err := cobraCmd.Flags().GetString("menmonic")
 		if err != nil {
 			return err
 		}
@@ -45,7 +45,7 @@ var accountUpdateCmd = &cobra.Command{
 
 func init() {
 	accountCmd.AddCommand(accountUpdateCmd)
-	accountUpdateCmd.Flags().StringP("seed", "s", "", "account seed key")
+	accountUpdateCmd.Flags().StringP("menmonic", "m", "", "account menmonic")
 	accountUpdateCmd.Flags().StringP("network", "n", "", "network (dev, qa, test, main)")
 	accountUpdateCmd.Flags().StringArrayP("relays", "r", nil, "relays urls")
 	accountUpdateCmd.Flags().StringP("rmb-enc-key", "k", "", "rmb encryption key")

registrar-cli/cmd/farm_create.go

@@ -12,7 +12,7 @@ var farmCreateCmd = &cobra.Command{
 	Use:   "create",
 	Short: "create new farm in node registrar",
 	RunE: func(cobraCmd *cobra.Command, args []string) error {
-		seed, err := cobraCmd.Flags().GetString("seed")
+		mnemonic, err := cobraCmd.Flags().GetString("menmonic")
 		if err != nil {
 			return err
 		}
@@ -27,12 +27,17 @@ var farmCreateCmd = &cobra.Command{
 			return err
 		}
 
+		stellarAddrss, err := cobraCmd.Flags().GetString("stellar-address")
+		if err != nil {
+			return err
+		}
+
 		dedicated, err := cobraCmd.Flags().GetBool("dedicated")
 		if err != nil {
 			return err
 		}
 
-		farmID, err := cmd.CreaeteFarm(seed, network, farmName, dedicated)
+		farmID, err := cmd.CreaeteFarm(mnemonic, network, farmName, stellarAddrss, dedicated)
 		if err != nil {
 			return err
 		}
@@ -45,8 +50,9 @@ var farmCreateCmd = &cobra.Command{
 
 func init() {
 	farmCmd.AddCommand(farmCreateCmd)
-	farmCreateCmd.Flags().StringP("seed", "s", "", "account seed key")
+	farmCreateCmd.Flags().StringP("menmonic", "m", "", "account menmonic")
 	farmCreateCmd.Flags().StringP("network", "n", "", "network (dev, qa, test, main)")
 	farmCreateCmd.Flags().StringP("farm-name", "f", "", "farm name")
+	farmCreateCmd.Flags().StringP("stellar-address", "s", "", "stellar address")
 	farmCreateCmd.Flags().BoolP("dedicated", "d", false, "is farm dedicated")
 }

registrar-cli/cmd/farm_update.go

@@ -12,7 +12,7 @@ var farmUpdateCmd = &cobra.Command{
 	Use:   "update",
 	Short: "update farm in node registrar",
 	RunE: func(cobraCmd *cobra.Command, args []string) error {
-		seed, err := cobraCmd.Flags().GetString("seed")
+		mnemonic, err := cobraCmd.Flags().GetString("menmonic")
 		if err != nil {
 			return err
 		}
@@ -32,12 +32,17 @@ var farmUpdateCmd = &cobra.Command{
 			return err
 		}
 
+		stellarAddrss, err := cobraCmd.Flags().GetString("stellar-address")
+		if err != nil {
+			return err
+		}
+
 		dedicated, err := cobraCmd.Flags().GetBool("dedicated")
 		if err != nil {
 			return err
 		}
 
-		err = cmd.UpdateFarm(seed, network, farmID, farmName, dedicated)
+		err = cmd.UpdateFarm(farmID, mnemonic, network, farmName, stellarAddrss, dedicated)
 		if err != nil {
 			return err
 		}
@@ -50,9 +55,10 @@ var farmUpdateCmd = &cobra.Command{
 
 func init() {
 	farmCmd.AddCommand(farmUpdateCmd)
-	farmUpdateCmd.Flags().StringP("seed", "s", "", "account seed key")
+	farmUpdateCmd.Flags().StringP("menmonic", "m", "", "account menmonic")
 	farmUpdateCmd.Flags().StringP("network", "n", "", "network (dev, qa, test, main)")
 	farmUpdateCmd.Flags().Uint64P("farm-id", "i", 0, "farm id")
 	farmUpdateCmd.Flags().String("farm-name", "", "new farm name")
+	farmUpdateCmd.Flags().StringP("stellar-address", "s", "", "stellar address")
 	farmUpdateCmd.Flags().BoolP("dedicated", "d", false, "farm is dedicated")
 }

registrar-cli/cmd/root.go

@@ -11,13 +11,6 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var urls = map[string]string{
-	"dev":  "https://registrar.dev4.grid.tf/v1",
-	"qa":   "https://registrar.qa4.grid.tf/v1",
-	"test": "https://registrar.test4.grid.tf/v1",
-	"main": "https://registrar.prod4.grid.tf/v1",
-}
-
 // rootCmd represents the base command when called without any subcommands
 var rootCmd = &cobra.Command{
 	Use:   "registrar-cli",

registrar-cli/cmd/zos_version_update.go

@@ -12,7 +12,7 @@ var UpdateZosVersionCmd = &cobra.Command{
 	Use:   "update",
 	Short: "update zos version in node registrar",
 	RunE: func(cobraCmd *cobra.Command, args []string) error {
-		seed, err := cobraCmd.Flags().GetString("seed")
+		seed, err := cobraCmd.Flags().GetString("menmonic")
 		if err != nil {
 			return err
 		}
@@ -45,7 +45,7 @@ var UpdateZosVersionCmd = &cobra.Command{
 
 func init() {
 	zosVersionCmd.AddCommand(UpdateZosVersionCmd)
-	UpdateZosVersionCmd.Flags().StringP("seed", "s", "", "account seed key")
+	UpdateZosVersionCmd.Flags().StringP("menmonic", "m", "", "account menmonic")
 	UpdateZosVersionCmd.Flags().StringP("network", "n", "", "network (dev, qa, test, main)")
 	UpdateZosVersionCmd.Flags().StringP("version", "v", "v0.0.0", "new zos version")
 	UpdateZosVersionCmd.Flags().BoolP("safe-to-upgrade", "u", false, "safe to upgrade")

registrar-cli/internal/cmd/account.go

@@ -0,0 +1,82 @@
+package cmd
+
+import (
+	"encoding/hex"
+	"fmt"
+
+	"github.com/threefoldtech/tfgrid4-sdk-go/node-registrar/client"
+)
+
+var urls = map[string]string{
+	"dev":  "https://registrar.dev4.grid.tf/v1",
+	"qa":   "https://registrar.qa4.grid.tf/v1",
+	"test": "https://registrar.test4.grid.tf/v1",
+	"main": "https://registrar.prod4.grid.tf/v1",
+}
+
+func CreaeteAccount(network string, relays []string, rmbEncKey string, mnemonicOrSeed ...string) (account client.Account, mnemonic string, err error) {
+	u, ok := urls[network]
+	if !ok {
+		return account, "", fmt.Errorf("invalid network %s", network)
+	}
+
+	var cli client.RegistrarClient
+	cli, err = client.NewRegistrarClient(u, mnemonicOrSeed...)
+	if err != nil {
+		return
+	}
+
+	return cli.CreateAccount(relays, rmbEncKey)
+}
+
+func GetAccount(network string, twinID uint64, pk string) (account client.Account, err error) {
+	u, ok := urls[network]
+	if !ok {
+		return account, fmt.Errorf("invalid network %s", network)
+	}
+
+	publicKey, err := hex.DecodeString(pk)
+	if err != nil {
+		return
+	}
+
+	cli, err := client.NewRegistrarClient(u)
+	if err != nil {
+		return
+	}
+
+	if twinID != 0 {
+		return cli.GetAccount(twinID)
+	} else if len(publicKey) != 0 {
+		return cli.GetAccountByPK(publicKey)
+	}
+
+	return account, fmt.Errorf("you need to provide either twin id or public key to load an account")
+}
+
+func UpdateAccount(mnemonic string, network string, relays []string, rmbEncKey string) (err error) {
+	u, ok := urls[network]
+	if !ok {
+		return fmt.Errorf("invalid network %s", network)
+	}
+
+	if len(mnemonic) == 0 {
+		return fmt.Errorf("can not initialize registrar client with no mnemonic")
+	}
+
+	cli, err := client.NewRegistrarClient(u, mnemonic)
+	if err != nil {
+		return err
+	}
+
+	var opts []client.UpdateAccountOpts
+	if len(relays) > 0 {
+		opts = append(opts, client.UpdateAccountWithRelays(relays))
+	}
+
+	if len(rmbEncKey) > 0 {
+		opts = append(opts, client.UpdateAccountWithRMBEncKey(rmbEncKey))
+	}
+
+	return cli.UpdateAccount(opts...)
+}

registrar-cli/internal/cmd/create.go

@@ -0,0 +1,69 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/threefoldtech/tfgrid4-sdk-go/node-registrar/client"
+)
+
+func CreaeteFarm(mnemonic, network, farmName, stellarAddress string, dedicated bool) (farmID uint64, err error) {
+	u, ok := urls[network]
+	if !ok {
+		return farmID, fmt.Errorf("invalid network %s", network)
+	}
+
+	if len(mnemonic) == 0 {
+		return farmID, fmt.Errorf("can not initialize registrar client with no seed")
+	}
+
+	cli, err := client.NewRegistrarClient(u, mnemonic)
+	if err != nil {
+		return
+	}
+
+	return cli.CreateFarm(farmName, stellarAddress, dedicated)
+}
+
+func GetFarm(network string, farmID uint64) (farm client.Farm, err error) {
+	u, ok := urls[network]
+	if !ok {
+		return farm, fmt.Errorf("invalid network %s", network)
+	}
+
+	cli, err := client.NewRegistrarClient(u)
+	if err != nil {
+		return
+	}
+
+	return cli.GetFarm(farmID)
+}
+
+func UpdateFarm(farmID uint64, mnemonic, network, farmName, stellarAddress string, dedicated bool) (err error) {
+	u, ok := urls[network]
+	if !ok {
+		return fmt.Errorf("invalid network %s", network)
+	}
+
+	if len(mnemonic) == 0 {
+		return fmt.Errorf("can not initialize registrar client with no mnemonic")
+	}
+
+	cli, err := client.NewRegistrarClient(u, mnemonic)
+	if err != nil {
+		return err
+	}
+
+	var opts []client.UpdateFarmOpts
+	if len(farmName) > 0 {
+		opts = append(opts, client.UpdateFarmWithName(farmName))
+	}
+
+	if dedicated {
+		opts = append(opts, client.UpdateFarmWithDedicated())
+	}
+	if len(stellarAddress) != 0 {
+		// opts = append(opts, client.Updfarm)
+	}
+
+	return cli.UpdateFarm(farmID, opts...)
+}