diff --git a/cmd/stakepool.go b/cmd/stakepool.go index f38b97b9..8383077b 100644 --- a/cmd/stakepool.go +++ b/cmd/stakepool.go @@ -78,21 +78,34 @@ var spInfo = &cobra.Command{ Args: cobra.MinimumNArgs(0), Run: func(cmd *cobra.Command, args []string) { var ( - flags = cmd.Flags() - blobberID string - err error + flags = cmd.Flags() + err error + providerID string + providerType sdk.ProviderType ) doJSON, _ := cmd.Flags().GetBool("json") if flags.Changed("blobber_id") { - if blobberID, err = flags.GetString("blobber_id"); err != nil { - log.Fatalf("can't get 'blobber_id' flag: %v", err) + if providerID, err = flags.GetString("blobber_id"); err != nil { + log.Fatalf("invalid 'blobber_id' flag: %v", err) + } else { + providerType = sdk.ProviderBlobber + } + } else if flags.Changed("validator_id") { + if providerID, err = flags.GetString("validator_id"); err != nil { + log.Fatalf("invalid 'validator_id' flag: %v", err) + } else { + providerType = sdk.ProviderValidator } } + if providerType == 0 || providerID == "" { + log.Fatal("missing flag: one of 'blobber_id' or 'validator_id' is required") + } + var info *sdk.StakePoolInfo - if info, err = sdk.GetStakePoolInfo(blobberID); err != nil { + if info, err = sdk.GetStakePoolInfo(providerType, providerID); err != nil { log.Fatalf("Failed to get stake pool info: %v", err) } if doJSON { @@ -267,7 +280,9 @@ func init() { rootCmd.AddCommand(spUnlock) spInfo.PersistentFlags().String("blobber_id", "", - "for given blobber, default is current client") + "for given blobber") + spInfo.PersistentFlags().String("validator_id", "", + "for given validator") spInfo.PersistentFlags().Bool("json", false, "pass this option to print response as json data") spUserInfo.PersistentFlags().Bool("json", false, "pass this option to print response as json data") diff --git a/go.mod b/go.mod index 5b89b3d6..5d843bb7 100644 --- a/go.mod +++ b/go.mod @@ -4,7 +4,7 @@ go 1.18 require ( github.com/0chain/errors v1.0.3 - github.com/0chain/gosdk v1.8.9 + github.com/0chain/gosdk v1.8.10-0.20221009195923-7af8225e3229 github.com/icza/bitio v1.1.0 github.com/olekukonko/tablewriter v0.0.5 github.com/spf13/cobra v1.5.0 diff --git a/go.sum b/go.sum index 823d9862..bd674621 100644 --- a/go.sum +++ b/go.sum @@ -38,8 +38,8 @@ cloud.google.com/go/storage v1.14.0/go.mod h1:GrKmX003DSIwi9o29oFT7YDnHYwZoctc3f dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU= github.com/0chain/errors v1.0.3 h1:QQZPFxTfnMcRdt32DXbzRQIfGWmBsKoEdszKQDb0rRM= github.com/0chain/errors v1.0.3/go.mod h1:xymD6nVgrbgttWwkpSCfLLEJbFO6iHGQwk/yeSuYkIc= -github.com/0chain/gosdk v1.8.9 h1:h4kQOdd8gS0dVJ6evnj2AnlbZ4FftpYLUuK0DDMXxLA= -github.com/0chain/gosdk v1.8.9/go.mod h1:74ICmmW3mprJWhCPemYIJrw/jNO8pza2HHk5aw+EzN0= +github.com/0chain/gosdk v1.8.10-0.20221009195923-7af8225e3229 h1:ELCAamOPnCNBQVbVH9dkQLURg0qOztnPCSkpc8wOS4U= +github.com/0chain/gosdk v1.8.10-0.20221009195923-7af8225e3229/go.mod h1:74ICmmW3mprJWhCPemYIJrw/jNO8pza2HHk5aw+EzN0= github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo= github.com/VictoriaMetrics/fastcache v1.6.0 h1:C/3Oi3EiBCqufydp1neRZkqcwmEiuRT9c3fqvvgKm5o=