From eb5e1121c526841221af81b44f89c17e21316c4f Mon Sep 17 00:00:00 2001 From: Evgenii Stratonikov Date: Thu, 1 Sep 2022 10:49:22 +0300 Subject: [PATCH] [#1748] neofs-adm: Move `dump-hashes` command to a separate file I have just lost 3 minutes trying to find it. Signed-off-by: Evgenii Stratonikov --- .../internal/modules/morph/config.go | 125 ---------------- .../internal/modules/morph/dump_hashes.go | 140 ++++++++++++++++++ 2 files changed, 140 insertions(+), 125 deletions(-) create mode 100644 cmd/neofs-adm/internal/modules/morph/dump_hashes.go diff --git a/cmd/neofs-adm/internal/modules/morph/config.go b/cmd/neofs-adm/internal/modules/morph/config.go index 9f969a52..5c125906 100644 --- a/cmd/neofs-adm/internal/modules/morph/config.go +++ b/cmd/neofs-adm/internal/modules/morph/config.go @@ -12,140 +12,15 @@ import ( "github.com/nspcc-dev/neo-go/pkg/io" "github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag" - "github.com/nspcc-dev/neo-go/pkg/util" "github.com/nspcc-dev/neo-go/pkg/vm/emit" "github.com/nspcc-dev/neo-go/pkg/vm/stackitem" "github.com/nspcc-dev/neo-go/pkg/vm/vmstate" - "github.com/nspcc-dev/neofs-contract/nns" "github.com/spf13/cobra" "github.com/spf13/viper" ) -const lastGlagoliticLetter = 41 - const forceConfigSet = "force" -type contractDumpInfo struct { - hash util.Uint160 - name string - version string -} - -func dumpContractHashes(cmd *cobra.Command, _ []string) error { - c, err := getN3Client(viper.GetViper()) - if err != nil { - return fmt.Errorf("can't create N3 client: %w", err) - } - - cs, err := c.GetContractStateByID(1) - if err != nil { - return err - } - - infos := []contractDumpInfo{{name: nnsContract, hash: cs.Hash}} - - irSize := 0 - for ; irSize < lastGlagoliticLetter; irSize++ { - ok, err := nnsIsAvailable(c, cs.Hash, getAlphabetNNSDomain(irSize)) - if err != nil { - return err - } else if ok { - break - } - } - - buf := bytes.NewBuffer(nil) - bw := io.NewBufBinWriter() - - if irSize != 0 { - bw.Reset() - for i := 0; i < irSize; i++ { - emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly, - getAlphabetNNSDomain(i), - int64(nns.TXT)) - } - - alphaRes, err := c.InvokeScript(bw.Bytes(), nil) - if err != nil { - return fmt.Errorf("can't fetch info from NNS: %w", err) - } - - for i := 0; i < irSize; i++ { - info := contractDumpInfo{name: fmt.Sprintf("alphabet %d", i)} - if h, err := parseNNSResolveResult(alphaRes.Stack[i]); err == nil { - info.hash = h - } - infos = append(infos, info) - } - } - - for _, ctrName := range contractList { - bw.Reset() - emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly, - ctrName+".neofs", int64(nns.TXT)) - - res, err := c.InvokeScript(bw.Bytes(), nil) - if err != nil { - return fmt.Errorf("can't fetch info from NNS: %w", err) - } - - info := contractDumpInfo{name: ctrName} - if len(res.Stack) != 0 { - if h, err := parseNNSResolveResult(res.Stack[0]); err == nil { - info.hash = h - } - } - infos = append(infos, info) - } - - bw.Reset() - for i := range infos { - if infos[i].hash.Equals(util.Uint160{}) { - emit.Int(bw.BinWriter, 0) - } else { - emit.AppCall(bw.BinWriter, infos[i].hash, "version", callflag.NoneFlag) - } - } - - res, err := c.InvokeScript(bw.Bytes(), nil) - if err != nil { - return fmt.Errorf("can't fetch info from NNS: %w", err) - } - - if res.State == vmstate.Halt.String() { - for i := range res.Stack { - infos[i].version = parseContractVersion(res.Stack[i]) - } - } - - tw := tabwriter.NewWriter(buf, 0, 2, 2, ' ', 0) - for _, info := range infos { - if info.version == "" { - info.version = "unknown" - } - _, _ = tw.Write([]byte(fmt.Sprintf("%s\t(%s):\t%s\n", - info.name, info.version, info.hash.StringLE()))) - } - _ = tw.Flush() - - cmd.Print(buf.String()) - - return nil -} - -func parseContractVersion(item stackitem.Item) string { - bi, err := item.TryInteger() - if err != nil || bi.Sign() == 0 || !bi.IsInt64() { - return "unknown" - } - - v := bi.Int64() - major := v / 1_000_000 - minor := (v % 1_000_000) / 1000 - patch := v % 1_000 - return fmt.Sprintf("v%d.%d.%d", major, minor, patch) -} - func dumpNetworkConfig(cmd *cobra.Command, _ []string) error { c, err := getN3Client(viper.GetViper()) if err != nil { diff --git a/cmd/neofs-adm/internal/modules/morph/dump_hashes.go b/cmd/neofs-adm/internal/modules/morph/dump_hashes.go new file mode 100644 index 00000000..9fa92a59 --- /dev/null +++ b/cmd/neofs-adm/internal/modules/morph/dump_hashes.go @@ -0,0 +1,140 @@ +package morph + +import ( + "bytes" + "fmt" + "text/tabwriter" + + "github.com/nspcc-dev/neo-go/pkg/io" + "github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag" + "github.com/nspcc-dev/neo-go/pkg/util" + "github.com/nspcc-dev/neo-go/pkg/vm/emit" + "github.com/nspcc-dev/neo-go/pkg/vm/stackitem" + "github.com/nspcc-dev/neo-go/pkg/vm/vmstate" + "github.com/nspcc-dev/neofs-contract/nns" + "github.com/spf13/cobra" + "github.com/spf13/viper" +) + +const lastGlagoliticLetter = 41 + +type contractDumpInfo struct { + hash util.Uint160 + name string + version string +} + +func dumpContractHashes(cmd *cobra.Command, _ []string) error { + c, err := getN3Client(viper.GetViper()) + if err != nil { + return fmt.Errorf("can't create N3 client: %w", err) + } + + cs, err := c.GetContractStateByID(1) + if err != nil { + return err + } + + infos := []contractDumpInfo{{name: nnsContract, hash: cs.Hash}} + + irSize := 0 + for ; irSize < lastGlagoliticLetter; irSize++ { + ok, err := nnsIsAvailable(c, cs.Hash, getAlphabetNNSDomain(irSize)) + if err != nil { + return err + } else if ok { + break + } + } + + buf := bytes.NewBuffer(nil) + bw := io.NewBufBinWriter() + + if irSize != 0 { + bw.Reset() + for i := 0; i < irSize; i++ { + emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly, + getAlphabetNNSDomain(i), + int64(nns.TXT)) + } + + alphaRes, err := c.InvokeScript(bw.Bytes(), nil) + if err != nil { + return fmt.Errorf("can't fetch info from NNS: %w", err) + } + + for i := 0; i < irSize; i++ { + info := contractDumpInfo{name: fmt.Sprintf("alphabet %d", i)} + if h, err := parseNNSResolveResult(alphaRes.Stack[i]); err == nil { + info.hash = h + } + infos = append(infos, info) + } + } + + for _, ctrName := range contractList { + bw.Reset() + emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly, + ctrName+".neofs", int64(nns.TXT)) + + res, err := c.InvokeScript(bw.Bytes(), nil) + if err != nil { + return fmt.Errorf("can't fetch info from NNS: %w", err) + } + + info := contractDumpInfo{name: ctrName} + if len(res.Stack) != 0 { + if h, err := parseNNSResolveResult(res.Stack[0]); err == nil { + info.hash = h + } + } + infos = append(infos, info) + } + + bw.Reset() + for i := range infos { + if infos[i].hash.Equals(util.Uint160{}) { + emit.Int(bw.BinWriter, 0) + } else { + emit.AppCall(bw.BinWriter, infos[i].hash, "version", callflag.NoneFlag) + } + } + + res, err := c.InvokeScript(bw.Bytes(), nil) + if err != nil { + return fmt.Errorf("can't fetch info from NNS: %w", err) + } + + if res.State == vmstate.Halt.String() { + for i := range res.Stack { + infos[i].version = parseContractVersion(res.Stack[i]) + } + } + + tw := tabwriter.NewWriter(buf, 0, 2, 2, ' ', 0) + for _, info := range infos { + if info.version == "" { + info.version = "unknown" + } + _, _ = tw.Write([]byte(fmt.Sprintf("%s\t(%s):\t%s\n", + info.name, info.version, info.hash.StringLE()))) + } + _ = tw.Flush() + + cmd.Print(buf.String()) + + return nil +} + +func parseContractVersion(item stackitem.Item) string { + bi, err := item.TryInteger() + if err != nil || bi.Sign() == 0 || !bi.IsInt64() { + return "unknown" + } + + v := bi.Int64() + major := v / 1_000_000 + minor := (v % 1_000_000) / 1000 + patch := v % 1_000 + return fmt.Sprintf("v%d.%d.%d", major, minor, patch) +}