mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-13 05:45:02 +00:00
124 lines
3.6 KiB
Go
124 lines
3.6 KiB
Go
package smartcontract
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/nspcc-dev/neo-go/cli/cmdargs"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/binding"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/urfave/cli/v2"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var generatorFlags = []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "config",
|
|
Aliases: []string{"c"},
|
|
Usage: `Configuration bindings file to use (*.yml). Configuration file is
|
|
generated by 'contract compile' command with --bindings flag`,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "manifest",
|
|
Aliases: []string{"m"},
|
|
Required: true,
|
|
Usage: "Read contract manifest (*.manifest.json) file",
|
|
Action: cmdargs.EnsureNotEmpty("manifest"),
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "out",
|
|
Aliases: []string{"o"},
|
|
Required: true,
|
|
Usage: "Output of the compiled wrapper",
|
|
Action: cmdargs.EnsureNotEmpty("out"),
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "hash",
|
|
Usage: "Smart-contract hash. If not passed, the wrapper will be designed for dynamic hash usage",
|
|
},
|
|
}
|
|
|
|
var generateWrapperCmd = &cli.Command{
|
|
Name: "generate-wrapper",
|
|
Usage: "Generate wrapper to use in other contracts",
|
|
UsageText: "neo-go contract generate-wrapper --manifest <file.json> --out <file.go> [--hash <hash>] [--config <config>]",
|
|
Description: `Generates a Go wrapper to use it in other smart contracts. If the
|
|
--hash flag is provided, CALLT instruction is used for the target contract
|
|
invocation as an optimization of the wrapper contract code. If omitted, the
|
|
generated wrapper will be designed for dynamic hash usage, allowing
|
|
the hash to be specified at runtime.
|
|
`,
|
|
Action: contractGenerateWrapper,
|
|
Flags: generatorFlags,
|
|
}
|
|
|
|
var generateRPCWrapperCmd = &cli.Command{
|
|
Name: "generate-rpcwrapper",
|
|
Usage: "Generate RPC wrapper to use for data reads",
|
|
UsageText: "neo-go contract generate-rpcwrapper --manifest <file.json> --out <file.go> [--hash <hash>] [--config <config>]",
|
|
Action: contractGenerateRPCWrapper,
|
|
Flags: generatorFlags,
|
|
}
|
|
|
|
func contractGenerateWrapper(ctx *cli.Context) error {
|
|
return contractGenerateSomething(ctx, binding.Generate)
|
|
}
|
|
|
|
func contractGenerateRPCWrapper(ctx *cli.Context) error {
|
|
return contractGenerateSomething(ctx, rpcbinding.Generate)
|
|
}
|
|
|
|
// contractGenerateSomething reads generator parameters and calls the given callback.
|
|
func contractGenerateSomething(ctx *cli.Context, cb func(binding.Config) error) error {
|
|
if err := cmdargs.EnsureNone(ctx); err != nil {
|
|
return err
|
|
}
|
|
var (
|
|
h util.Uint160
|
|
err error
|
|
)
|
|
if hStr := ctx.String("hash"); len(hStr) != 0 {
|
|
h, err = util.Uint160DecodeStringLE(strings.TrimPrefix(hStr, "0x"))
|
|
if err != nil {
|
|
return cli.Exit(fmt.Errorf("invalid contract hash: %w", err), 1)
|
|
}
|
|
}
|
|
m, _, err := readManifest(ctx.String("manifest"), h)
|
|
if err != nil {
|
|
return cli.Exit(fmt.Errorf("can't read contract manifest: %w", err), 1)
|
|
}
|
|
|
|
cfg := binding.NewConfig()
|
|
if cfgPath := ctx.String("config"); cfgPath != "" {
|
|
bs, err := os.ReadFile(cfgPath)
|
|
if err != nil {
|
|
return cli.Exit(fmt.Errorf("can't read config file: %w", err), 1)
|
|
}
|
|
decoder := yaml.NewDecoder(bytes.NewReader(bs))
|
|
decoder.KnownFields(true)
|
|
|
|
err = decoder.Decode(&cfg)
|
|
if err != nil {
|
|
return cli.Exit(fmt.Errorf("can't parse config file: %w", err), 1)
|
|
}
|
|
}
|
|
cfg.Manifest = m
|
|
cfg.Hash = h
|
|
|
|
f, err := os.Create(ctx.String("out"))
|
|
if err != nil {
|
|
return cli.Exit(fmt.Errorf("can't create output file: %w", err), 1)
|
|
}
|
|
defer f.Close()
|
|
|
|
cfg.Output = f
|
|
|
|
err = cb(cfg)
|
|
if err != nil {
|
|
return cli.Exit(fmt.Errorf("error during generation: %w", err), 1)
|
|
}
|
|
return nil
|
|
}
|