2022-10-26 20:27:24 +00:00
|
|
|
package rpcbinding
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"sort"
|
2022-11-25 10:04:24 +00:00
|
|
|
"strings"
|
2022-10-26 20:27:24 +00:00
|
|
|
"text/template"
|
2023-05-24 08:52:14 +00:00
|
|
|
"unicode"
|
2022-10-26 20:27:24 +00:00
|
|
|
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/binding"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest/standard"
|
2023-05-11 10:25:45 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
2022-10-26 20:27:24 +00:00
|
|
|
)
|
|
|
|
|
2023-05-24 08:44:31 +00:00
|
|
|
// The set of constants containing parts of RPC binding template. Each block of code
|
|
|
|
// including template definition and var/type/method definitions contain new line at the
|
|
|
|
// start and ends with a new line. On adding new block of code to the template, please,
|
|
|
|
// ensure that this block has new line at the start and in the end of the block.
|
|
|
|
const (
|
2023-05-24 08:52:14 +00:00
|
|
|
eventDefinition = `{{ define "EVENT" }}
|
2023-05-23 11:17:39 +00:00
|
|
|
// {{.Name}} represents "{{.ManifestName}}" event emitted by the contract.
|
|
|
|
type {{.Name}} struct {
|
2023-05-24 08:52:14 +00:00
|
|
|
{{- range $index, $arg := .Parameters}}
|
2023-08-10 17:23:54 +00:00
|
|
|
{{ upperFirst .Name}} {{.Type}}
|
2023-05-24 08:52:14 +00:00
|
|
|
{{- end}}
|
|
|
|
}
|
|
|
|
{{ end }}`
|
|
|
|
|
2023-05-24 08:44:31 +00:00
|
|
|
safemethodDefinition = `{{ define "SAFEMETHOD" }}
|
2022-10-26 20:27:24 +00:00
|
|
|
// {{.Name}} {{.Comment}}
|
|
|
|
func (c *ContractReader) {{.Name}}({{range $index, $arg := .Arguments -}}
|
|
|
|
{{- if ne $index 0}}, {{end}}
|
|
|
|
{{- .Name}} {{.Type}}
|
|
|
|
{{- end}}) {{if .ReturnType }}({{ .ReturnType }}, error) {
|
2023-10-18 15:09:58 +00:00
|
|
|
return {{if and (not .ItemTo) (eq .Unwrapper "Item")}}func(item stackitem.Item, err error) ({{ .ReturnType }}, error) {
|
2022-12-02 07:20:55 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2022-12-02 12:52:52 +00:00
|
|
|
return {{addIndent (etTypeConverter .ExtendedReturn "item") "\t"}}
|
2022-12-02 07:20:55 +00:00
|
|
|
} ( {{- end -}} {{if .ItemTo -}} itemTo{{ .ItemTo }}( {{- end -}}
|
2023-05-11 10:25:45 +00:00
|
|
|
unwrap.{{.Unwrapper}}(c.invoker.Call(c.hash, "{{ .NameABI }}"
|
2022-12-02 07:20:55 +00:00
|
|
|
{{- range $arg := .Arguments -}}, {{.Name}}{{end -}} )) {{- if or .ItemTo (eq .Unwrapper "Item") -}} ) {{- end}}
|
2022-10-26 20:27:24 +00:00
|
|
|
{{- else -}} (*result.Invoke, error) {
|
2023-05-11 10:25:45 +00:00
|
|
|
c.invoker.Call(c.hash, "{{ .NameABI }}"
|
2022-10-26 20:27:24 +00:00
|
|
|
{{- range $arg := .Arguments -}}, {{.Name}}{{end}})
|
|
|
|
{{- end}}
|
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{ if eq .Unwrapper "SessionIterator" }}
|
2022-11-09 08:58:01 +00:00
|
|
|
// {{.Name}}Expanded is similar to {{.Name}} (uses the same contract
|
|
|
|
// method), but can be useful if the server used doesn't support sessions and
|
|
|
|
// doesn't expand iterators. It creates a script that will get the specified
|
|
|
|
// number of result items from the iterator right in the VM and return them to
|
|
|
|
// you. It's only limited by VM stack and GAS available for RPC invocations.
|
|
|
|
func (c *ContractReader) {{.Name}}Expanded({{range $index, $arg := .Arguments}}{{.Name}} {{.Type}}, {{end}}_numOfIteratorItems int) ([]stackitem.Item, error) {
|
2023-05-11 10:25:45 +00:00
|
|
|
return unwrap.Array(c.invoker.CallAndExpandIterator(c.hash, "{{.NameABI}}", _numOfIteratorItems{{range $arg := .Arguments}}, {{.Name}}{{end}}))
|
2022-11-09 08:58:01 +00:00
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{ end }}{{ end }}`
|
|
|
|
methodDefinition = `{{ define "METHOD" }}{{ if eq .ReturnType "bool"}}
|
2023-06-05 14:47:17 +00:00
|
|
|
func (c *Contract) scriptFor{{.Name}}({{range $index, $arg := .Arguments -}}
|
2022-11-07 19:13:47 +00:00
|
|
|
{{- if ne $index 0}}, {{end}}
|
|
|
|
{{- .Name}} {{.Type}}
|
|
|
|
{{- end}}) ([]byte, error) {
|
2023-05-11 10:25:45 +00:00
|
|
|
return smartcontract.CreateCallWithAssertScript(c.hash, "{{ .NameABI }}"{{- range $index, $arg := .Arguments -}}, {{.Name}}{{end}})
|
2022-11-07 19:13:47 +00:00
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{ end }}
|
|
|
|
// {{.Name}} {{.Comment}}
|
2022-10-27 13:20:35 +00:00
|
|
|
// This transaction is signed and immediately sent to the network.
|
|
|
|
// The values returned are its hash, ValidUntilBlock value and error if any.
|
|
|
|
func (c *Contract) {{.Name}}({{range $index, $arg := .Arguments -}}
|
|
|
|
{{- if ne $index 0}}, {{end}}
|
|
|
|
{{- .Name}} {{.Type}}
|
|
|
|
{{- end}}) (util.Uint256, uint32, error) {
|
2023-05-11 10:25:45 +00:00
|
|
|
{{if ne .ReturnType "bool"}}return c.actor.SendCall(c.hash, "{{ .NameABI }}"
|
2023-06-05 14:47:17 +00:00
|
|
|
{{- range $index, $arg := .Arguments -}}, {{.Name}}{{end}}){{else}}script, err := c.scriptFor{{.Name}}({{- range $index, $arg := .Arguments -}}{{- if ne $index 0}}, {{end}}{{.Name}}{{end}})
|
2022-11-07 19:13:47 +00:00
|
|
|
if err != nil {
|
|
|
|
return util.Uint256{}, 0, err
|
|
|
|
}
|
|
|
|
return c.actor.SendRun(script){{end}}
|
2022-10-27 13:20:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// {{.Name}}Transaction {{.Comment}}
|
|
|
|
// This transaction is signed, but not sent to the network, instead it's
|
|
|
|
// returned to the caller.
|
|
|
|
func (c *Contract) {{.Name}}Transaction({{range $index, $arg := .Arguments -}}
|
|
|
|
{{- if ne $index 0}}, {{end}}
|
|
|
|
{{- .Name}} {{.Type}}
|
|
|
|
{{- end}}) (*transaction.Transaction, error) {
|
2023-05-11 10:25:45 +00:00
|
|
|
{{if ne .ReturnType "bool"}}return c.actor.MakeCall(c.hash, "{{ .NameABI }}"
|
2023-06-05 14:47:17 +00:00
|
|
|
{{- range $index, $arg := .Arguments -}}, {{.Name}}{{end}}){{else}}script, err := c.scriptFor{{.Name}}({{- range $index, $arg := .Arguments -}}{{- if ne $index 0}}, {{end}}{{.Name}}{{end}})
|
2022-11-07 19:13:47 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return c.actor.MakeRun(script){{end}}
|
2022-10-27 13:20:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// {{.Name}}Unsigned {{.Comment}}
|
|
|
|
// This transaction is not signed, it's simply returned to the caller.
|
|
|
|
// Any fields of it that do not affect fees can be changed (ValidUntilBlock,
|
|
|
|
// Nonce), fee values (NetworkFee, SystemFee) can be increased as well.
|
|
|
|
func (c *Contract) {{.Name}}Unsigned({{range $index, $arg := .Arguments -}}
|
|
|
|
{{- if ne $index 0}}, {{end}}
|
|
|
|
{{- .Name}} {{.Type}}
|
|
|
|
{{- end}}) (*transaction.Transaction, error) {
|
2023-05-11 10:25:45 +00:00
|
|
|
{{if ne .ReturnType "bool"}}return c.actor.MakeUnsignedCall(c.hash, "{{ .NameABI }}", nil
|
2023-06-05 14:47:17 +00:00
|
|
|
{{- range $index, $arg := .Arguments -}}, {{.Name}}{{end}}){{else}}script, err := c.scriptFor{{.Name}}({{- range $index, $arg := .Arguments -}}{{- if ne $index 0}}, {{end}}{{.Name}}{{end}})
|
2022-11-07 19:13:47 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return c.actor.MakeUnsignedRun(script, nil){{end}}
|
2022-10-27 13:20:35 +00:00
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{end}}`
|
|
|
|
|
|
|
|
bindingDefinition = `// Package {{.PackageName}} contains RPC wrappers for {{.ContractName}} contract.
|
2023-10-18 17:32:51 +00:00
|
|
|
//
|
|
|
|
// Code generated by neo-go contract generate-rpcwrapper --manifest <file.json> --out <file.go> [--hash <hash>] [--config <config>]; DO NOT EDIT.
|
2022-10-26 20:27:24 +00:00
|
|
|
package {{.PackageName}}
|
|
|
|
|
|
|
|
import (
|
|
|
|
{{range $m := .Imports}} "{{ $m }}"
|
|
|
|
{{end}})
|
2023-05-11 10:25:45 +00:00
|
|
|
{{if len .Hash}}
|
2022-10-26 20:27:24 +00:00
|
|
|
// Hash contains contract hash.
|
|
|
|
var Hash = {{ .Hash }}
|
2023-05-11 10:25:45 +00:00
|
|
|
{{end -}}
|
2023-11-21 18:46:21 +00:00
|
|
|
{{- range $index, $typ := .NamedTypes }}
|
|
|
|
// {{toTypeName $typ.Name}} is a contract-specific {{$typ.Name}} type used by its methods.
|
|
|
|
type {{toTypeName $typ.Name}} struct {
|
2022-11-25 10:04:24 +00:00
|
|
|
{{- range $m := $typ.Fields}}
|
2023-08-10 17:23:54 +00:00
|
|
|
{{ upperFirst .Field}} {{etTypeToStr .ExtendedType}}
|
2022-11-25 10:04:24 +00:00
|
|
|
{{- end}}
|
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{end}}
|
2023-05-24 08:52:14 +00:00
|
|
|
{{- range $e := .CustomEvents }}{{template "EVENT" $e }}{{ end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- if .HasReader}}
|
|
|
|
// Invoker is used by ContractReader to call various safe methods.
|
2022-10-26 20:27:24 +00:00
|
|
|
type Invoker interface {
|
2022-11-08 12:43:32 +00:00
|
|
|
{{if or .IsNep11D .IsNep11ND}} nep11.Invoker
|
2022-11-09 08:58:01 +00:00
|
|
|
{{else -}}
|
|
|
|
{{ if .IsNep17}} nep17.Invoker
|
2023-04-03 10:34:24 +00:00
|
|
|
{{else if len .SafeMethods}} Call(contract util.Uint160, operation string, params ...any) (*result.Invoke, error)
|
2022-11-08 12:43:32 +00:00
|
|
|
{{end -}}
|
2023-04-03 10:34:24 +00:00
|
|
|
{{if .HasIterator}} CallAndExpandIterator(contract util.Uint160, method string, maxItems int, params ...any) (*result.Invoke, error)
|
2022-11-09 08:58:01 +00:00
|
|
|
TerminateSession(sessionID uuid.UUID) error
|
|
|
|
TraverseIterator(sessionID uuid.UUID, iterator *result.Iterator, num int) ([]stackitem.Item, error)
|
|
|
|
{{end -}}
|
|
|
|
{{end -}}
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
2022-11-08 13:36:38 +00:00
|
|
|
{{end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- if .HasWriter}}
|
|
|
|
// Actor is used by Contract to call state-changing methods.
|
2022-10-27 13:20:35 +00:00
|
|
|
type Actor interface {
|
2022-11-08 13:36:38 +00:00
|
|
|
{{- if .HasReader}}
|
2022-10-27 13:20:35 +00:00
|
|
|
Invoker
|
2022-11-08 13:36:38 +00:00
|
|
|
{{end}}
|
|
|
|
{{- if or .IsNep11D .IsNep11ND}}
|
|
|
|
nep11.Actor
|
|
|
|
{{else if .IsNep17}}
|
|
|
|
nep17.Actor
|
|
|
|
{{end}}
|
|
|
|
{{- if len .Methods}}
|
2023-04-03 10:34:24 +00:00
|
|
|
MakeCall(contract util.Uint160, method string, params ...any) (*transaction.Transaction, error)
|
2022-10-27 13:20:35 +00:00
|
|
|
MakeRun(script []byte) (*transaction.Transaction, error)
|
2023-04-03 10:34:24 +00:00
|
|
|
MakeUnsignedCall(contract util.Uint160, method string, attrs []transaction.Attribute, params ...any) (*transaction.Transaction, error)
|
2022-10-27 13:20:35 +00:00
|
|
|
MakeUnsignedRun(script []byte, attrs []transaction.Attribute) (*transaction.Transaction, error)
|
2023-04-03 10:34:24 +00:00
|
|
|
SendCall(contract util.Uint160, method string, params ...any) (util.Uint256, uint32, error)
|
2022-10-27 13:20:35 +00:00
|
|
|
SendRun(script []byte) (util.Uint256, uint32, error)
|
|
|
|
{{end -}}
|
|
|
|
}
|
|
|
|
{{end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- if .HasReader}}
|
|
|
|
// ContractReader implements safe contract methods.
|
2022-10-26 20:27:24 +00:00
|
|
|
type ContractReader struct {
|
|
|
|
{{if .IsNep11D}}nep11.DivisibleReader
|
|
|
|
{{end -}}
|
|
|
|
{{if .IsNep11ND}}nep11.NonDivisibleReader
|
|
|
|
{{end -}}
|
|
|
|
{{if .IsNep17}}nep17.TokenReader
|
|
|
|
{{end -}}
|
|
|
|
invoker Invoker
|
2023-05-11 10:25:45 +00:00
|
|
|
hash util.Uint160
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
2022-11-08 13:36:38 +00:00
|
|
|
{{end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- if .HasWriter}}
|
|
|
|
// Contract implements all contract methods.
|
2022-10-27 13:20:35 +00:00
|
|
|
type Contract struct {
|
2022-11-08 13:36:38 +00:00
|
|
|
{{if .HasReader}}ContractReader
|
|
|
|
{{end -}}
|
2022-10-27 13:20:35 +00:00
|
|
|
{{if .IsNep11D}}nep11.DivisibleWriter
|
|
|
|
{{end -}}
|
|
|
|
{{if .IsNep11ND}}nep11.BaseWriter
|
|
|
|
{{end -}}
|
|
|
|
{{if .IsNep17}}nep17.TokenWriter
|
|
|
|
{{end -}}
|
|
|
|
actor Actor
|
2023-05-11 10:25:45 +00:00
|
|
|
hash util.Uint160
|
2022-10-27 13:20:35 +00:00
|
|
|
}
|
|
|
|
{{end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- if .HasReader}}
|
2023-05-11 10:25:45 +00:00
|
|
|
// NewReader creates an instance of ContractReader using {{if len .Hash -}}Hash{{- else -}}provided contract hash{{- end}} and the given Invoker.
|
|
|
|
func NewReader(invoker Invoker{{- if not (len .Hash) -}}, hash util.Uint160{{- end -}}) *ContractReader {
|
|
|
|
{{if len .Hash -}}
|
|
|
|
var hash = Hash
|
|
|
|
{{end -}}
|
2022-10-26 20:27:24 +00:00
|
|
|
return &ContractReader{
|
2023-05-11 10:25:45 +00:00
|
|
|
{{- if .IsNep11D}}*nep11.NewDivisibleReader(invoker, hash), {{end}}
|
|
|
|
{{- if .IsNep11ND}}*nep11.NewNonDivisibleReader(invoker, hash), {{end}}
|
|
|
|
{{- if .IsNep17}}*nep17.NewReader(invoker, hash), {{end -}}
|
|
|
|
invoker, hash}
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
2022-11-08 13:36:38 +00:00
|
|
|
{{end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- if .HasWriter}}
|
2023-05-11 10:25:45 +00:00
|
|
|
// New creates an instance of Contract using {{if len .Hash -}}Hash{{- else -}}provided contract hash{{- end}} and the given Actor.
|
|
|
|
func New(actor Actor{{- if not (len .Hash) -}}, hash util.Uint160{{- end -}}) *Contract {
|
|
|
|
{{if len .Hash -}}
|
|
|
|
var hash = Hash
|
|
|
|
{{end -}}
|
|
|
|
{{if .IsNep11D}}var nep11dt = nep11.NewDivisible(actor, hash)
|
2022-10-27 13:20:35 +00:00
|
|
|
{{end -}}
|
2023-05-11 10:25:45 +00:00
|
|
|
{{if .IsNep11ND}}var nep11ndt = nep11.NewNonDivisible(actor, hash)
|
2022-10-27 13:20:35 +00:00
|
|
|
{{end -}}
|
2023-05-11 10:25:45 +00:00
|
|
|
{{if .IsNep17}}var nep17t = nep17.New(actor, hash)
|
2022-10-27 13:20:35 +00:00
|
|
|
{{end -}}
|
2022-11-08 13:36:38 +00:00
|
|
|
return &Contract{
|
|
|
|
{{- if .HasReader}}ContractReader{
|
2022-10-27 13:20:35 +00:00
|
|
|
{{- if .IsNep11D}}nep11dt.DivisibleReader, {{end -}}
|
|
|
|
{{- if .IsNep11ND}}nep11ndt.NonDivisibleReader, {{end -}}
|
|
|
|
{{- if .IsNep17}}nep17t.TokenReader, {{end -}}
|
2023-05-11 10:25:45 +00:00
|
|
|
actor, hash}, {{end -}}
|
2022-10-27 13:20:35 +00:00
|
|
|
{{- if .IsNep11D}}nep11dt.DivisibleWriter, {{end -}}
|
|
|
|
{{- if .IsNep11ND}}nep11ndt.BaseWriter, {{end -}}
|
|
|
|
{{- if .IsNep17}}nep17t.TokenWriter, {{end -}}
|
2023-05-11 10:25:45 +00:00
|
|
|
actor, hash}
|
2022-10-27 13:20:35 +00:00
|
|
|
}
|
|
|
|
{{end -}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- range $m := .SafeMethods }}{{template "SAFEMETHOD" $m }}{{ end -}}
|
|
|
|
{{- range $m := .Methods -}}{{template "METHOD" $m }}{{ end -}}
|
2023-11-21 18:46:21 +00:00
|
|
|
{{- range $index, $typ := .NamedTypes }}
|
|
|
|
// itemTo{{toTypeName $typ.Name}} converts stack item into *{{toTypeName $typ.Name}}.
|
|
|
|
func itemTo{{toTypeName $typ.Name}}(item stackitem.Item, err error) (*{{toTypeName $typ.Name}}, error) {
|
2022-11-25 10:04:24 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2023-11-21 18:46:21 +00:00
|
|
|
var res = new({{toTypeName $typ.Name}})
|
2023-05-03 11:55:02 +00:00
|
|
|
err = res.FromStackItem(item)
|
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
2023-11-21 18:46:21 +00:00
|
|
|
// FromStackItem retrieves fields of {{toTypeName $typ.Name}} from the given
|
2023-05-31 12:47:10 +00:00
|
|
|
// [stackitem.Item] or returns an error if it's not possible to do to so.
|
2023-11-21 18:46:21 +00:00
|
|
|
func (res *{{toTypeName $typ.Name}}) FromStackItem(item stackitem.Item) error {
|
2022-11-25 10:04:24 +00:00
|
|
|
arr, ok := item.Value().([]stackitem.Item)
|
|
|
|
if !ok {
|
2023-05-03 11:55:02 +00:00
|
|
|
return errors.New("not an array")
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
|
|
|
if len(arr) != {{len $typ.Fields}} {
|
2023-05-03 11:55:02 +00:00
|
|
|
return errors.New("wrong number of structure elements")
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{if len .Fields}}
|
|
|
|
var (
|
2023-05-03 11:55:02 +00:00
|
|
|
index = -1
|
2023-10-18 15:09:58 +00:00
|
|
|
err error
|
2023-05-03 11:55:02 +00:00
|
|
|
)
|
2022-11-25 10:04:24 +00:00
|
|
|
{{- range $m := $typ.Fields}}
|
|
|
|
index++
|
2023-08-10 17:23:54 +00:00
|
|
|
res.{{ upperFirst .Field}}, err = {{etTypeConverter .ExtendedType "arr[index]"}}
|
2022-11-25 10:04:24 +00:00
|
|
|
if err != nil {
|
2023-08-10 17:23:54 +00:00
|
|
|
return fmt.Errorf("field {{ upperFirst .Field}}: %w", err)
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
|
|
|
{{end}}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{- end}}
|
2023-05-03 11:55:02 +00:00
|
|
|
return nil
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
2023-05-24 08:52:14 +00:00
|
|
|
{{ end -}}
|
|
|
|
{{- range $e := .CustomEvents }}
|
2023-05-23 11:17:39 +00:00
|
|
|
// {{$e.Name}}sFromApplicationLog retrieves a set of all emitted events
|
2023-05-31 12:47:10 +00:00
|
|
|
// with "{{$e.ManifestName}}" name from the provided [result.ApplicationLog].
|
2023-05-23 11:17:39 +00:00
|
|
|
func {{$e.Name}}sFromApplicationLog(log *result.ApplicationLog) ([]*{{$e.Name}}, error) {
|
2023-05-24 08:52:14 +00:00
|
|
|
if log == nil {
|
|
|
|
return nil, errors.New("nil application log")
|
|
|
|
}
|
|
|
|
|
2023-05-23 11:17:39 +00:00
|
|
|
var res []*{{$e.Name}}
|
2023-05-24 08:52:14 +00:00
|
|
|
for i, ex := range log.Executions {
|
|
|
|
for j, e := range ex.Events {
|
|
|
|
if e.Name != "{{$e.ManifestName}}" {
|
|
|
|
continue
|
|
|
|
}
|
2023-05-23 11:17:39 +00:00
|
|
|
event := new({{$e.Name}})
|
2023-05-24 08:52:14 +00:00
|
|
|
err := event.FromStackItem(e.Item)
|
|
|
|
if err != nil {
|
2023-05-30 08:59:16 +00:00
|
|
|
return nil, fmt.Errorf("failed to deserialize {{$e.Name}} from stackitem (execution #%d, event #%d): %w", i, j, err)
|
2023-05-24 08:52:14 +00:00
|
|
|
}
|
|
|
|
res = append(res, event)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return res, nil
|
|
|
|
}
|
|
|
|
|
2023-05-31 12:47:10 +00:00
|
|
|
// FromStackItem converts provided [stackitem.Array] to {{$e.Name}} or
|
2023-05-30 14:34:51 +00:00
|
|
|
// returns an error if it's not possible to do to so.
|
2023-05-23 11:17:39 +00:00
|
|
|
func (e *{{$e.Name}}) FromStackItem(item *stackitem.Array) error {
|
2023-05-24 08:52:14 +00:00
|
|
|
if item == nil {
|
|
|
|
return errors.New("nil item")
|
|
|
|
}
|
|
|
|
arr, ok := item.Value().([]stackitem.Item)
|
|
|
|
if !ok {
|
|
|
|
return errors.New("not an array")
|
|
|
|
}
|
|
|
|
if len(arr) != {{len $e.Parameters}} {
|
|
|
|
return errors.New("wrong number of structure elements")
|
|
|
|
}
|
|
|
|
|
|
|
|
{{if len $e.Parameters}}var (
|
|
|
|
index = -1
|
2023-10-18 15:09:58 +00:00
|
|
|
err error
|
2023-05-24 08:52:14 +00:00
|
|
|
)
|
|
|
|
{{- range $p := $e.Parameters}}
|
|
|
|
index++
|
2023-08-10 17:23:54 +00:00
|
|
|
e.{{ upperFirst .Name}}, err = {{etTypeConverter .ExtType "arr[index]"}}
|
2023-05-24 08:52:14 +00:00
|
|
|
if err != nil {
|
2023-08-10 17:23:54 +00:00
|
|
|
return fmt.Errorf("field {{ upperFirst .Name}}: %w", err)
|
2023-05-24 08:52:14 +00:00
|
|
|
}
|
|
|
|
{{end}}
|
|
|
|
{{- end}}
|
|
|
|
return nil
|
|
|
|
}
|
2023-05-24 08:44:31 +00:00
|
|
|
{{end -}}`
|
|
|
|
|
|
|
|
srcTmpl = bindingDefinition +
|
2023-05-24 08:52:14 +00:00
|
|
|
eventDefinition +
|
2023-05-24 08:44:31 +00:00
|
|
|
safemethodDefinition +
|
|
|
|
methodDefinition
|
|
|
|
)
|
2022-10-26 20:27:24 +00:00
|
|
|
|
|
|
|
type (
|
|
|
|
ContractTmpl struct {
|
|
|
|
binding.ContractTmpl
|
2022-10-27 13:20:35 +00:00
|
|
|
|
2023-05-24 08:52:14 +00:00
|
|
|
SafeMethods []SafeMethodTmpl
|
|
|
|
CustomEvents []CustomEventTemplate
|
2023-11-21 18:46:21 +00:00
|
|
|
NamedTypes []binding.ExtendedType
|
2022-10-27 13:20:35 +00:00
|
|
|
|
2022-10-26 20:27:24 +00:00
|
|
|
IsNep11D bool
|
|
|
|
IsNep11ND bool
|
|
|
|
IsNep17 bool
|
2022-10-27 13:20:35 +00:00
|
|
|
|
2022-11-09 08:58:01 +00:00
|
|
|
HasReader bool
|
|
|
|
HasWriter bool
|
|
|
|
HasIterator bool
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
2022-12-02 07:20:55 +00:00
|
|
|
|
|
|
|
SafeMethodTmpl struct {
|
|
|
|
binding.MethodTmpl
|
|
|
|
Unwrapper string
|
|
|
|
ItemTo string
|
|
|
|
ExtendedReturn binding.ExtendedType
|
|
|
|
}
|
2023-05-24 08:52:14 +00:00
|
|
|
|
|
|
|
CustomEventTemplate struct {
|
|
|
|
// Name is the event's name that will be used as the event structure name in
|
|
|
|
// the resulting RPC binding. It is a valid go structure name and may differ
|
|
|
|
// from ManifestName.
|
|
|
|
Name string
|
|
|
|
// ManifestName is the event's name declared in the contract manifest.
|
|
|
|
// It may contain any UTF8 character.
|
|
|
|
ManifestName string
|
|
|
|
Parameters []EventParamTmpl
|
|
|
|
}
|
|
|
|
|
|
|
|
EventParamTmpl struct {
|
|
|
|
binding.ParamTmpl
|
|
|
|
|
|
|
|
// ExtType holds the event parameter's type information provided by Manifest,
|
|
|
|
// i.e. simple types only.
|
|
|
|
ExtType binding.ExtendedType
|
|
|
|
}
|
2022-10-26 20:27:24 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// NewConfig initializes and returns a new config instance.
|
|
|
|
func NewConfig() binding.Config {
|
|
|
|
return binding.NewConfig()
|
|
|
|
}
|
|
|
|
|
|
|
|
// Generate writes Go file containing smartcontract bindings to the `cfg.Output`.
|
2022-11-09 09:44:14 +00:00
|
|
|
// It doesn't check manifest from Config for validity, incorrect manifest can
|
|
|
|
// lead to unexpected results.
|
2022-10-26 20:27:24 +00:00
|
|
|
func Generate(cfg binding.Config) error {
|
2022-11-08 12:43:32 +00:00
|
|
|
// Avoid changing *cfg.Manifest.
|
|
|
|
mfst := *cfg.Manifest
|
|
|
|
mfst.ABI.Methods = make([]manifest.Method, len(mfst.ABI.Methods))
|
|
|
|
copy(mfst.ABI.Methods, cfg.Manifest.ABI.Methods)
|
|
|
|
cfg.Manifest = &mfst
|
|
|
|
|
|
|
|
var imports = make(map[string]struct{})
|
|
|
|
var ctr ContractTmpl
|
|
|
|
|
|
|
|
// Strip standard methods from NEP-XX packages.
|
|
|
|
for _, std := range cfg.Manifest.SupportedStandards {
|
|
|
|
if std == manifest.NEP11StandardName {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/rpcclient/nep11"] = struct{}{}
|
|
|
|
if standard.ComplyABI(cfg.Manifest, standard.Nep11Divisible) == nil {
|
|
|
|
mfst.ABI.Methods = dropStdMethods(mfst.ABI.Methods, standard.Nep11Divisible)
|
|
|
|
ctr.IsNep11D = true
|
|
|
|
} else if standard.ComplyABI(cfg.Manifest, standard.Nep11NonDivisible) == nil {
|
|
|
|
mfst.ABI.Methods = dropStdMethods(mfst.ABI.Methods, standard.Nep11NonDivisible)
|
|
|
|
ctr.IsNep11ND = true
|
|
|
|
}
|
2023-05-21 12:08:29 +00:00
|
|
|
mfst.ABI.Events = dropStdEvents(mfst.ABI.Events, standard.Nep11Base)
|
2022-11-08 12:43:32 +00:00
|
|
|
break // Can't be NEP-17 at the same time.
|
|
|
|
}
|
|
|
|
if std == manifest.NEP17StandardName && standard.ComplyABI(cfg.Manifest, standard.Nep17) == nil {
|
|
|
|
mfst.ABI.Methods = dropStdMethods(mfst.ABI.Methods, standard.Nep17)
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/rpcclient/nep17"] = struct{}{}
|
|
|
|
ctr.IsNep17 = true
|
2023-05-21 12:08:29 +00:00
|
|
|
mfst.ABI.Events = dropStdEvents(mfst.ABI.Events, standard.Nep17)
|
2022-11-08 12:43:32 +00:00
|
|
|
break // Can't be NEP-11 at the same time.
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-11-08 14:00:26 +00:00
|
|
|
// OnNepXXPayment handlers normally can't be called directly.
|
|
|
|
if standard.ComplyABI(cfg.Manifest, standard.Nep11Payable) == nil {
|
|
|
|
mfst.ABI.Methods = dropStdMethods(mfst.ABI.Methods, standard.Nep11Payable)
|
|
|
|
}
|
|
|
|
if standard.ComplyABI(cfg.Manifest, standard.Nep17Payable) == nil {
|
|
|
|
mfst.ABI.Methods = dropStdMethods(mfst.ABI.Methods, standard.Nep17Payable)
|
|
|
|
}
|
|
|
|
|
2022-11-09 09:44:14 +00:00
|
|
|
ctr.ContractTmpl = binding.TemplateFromManifest(cfg, scTypeToGo)
|
2023-05-24 08:52:14 +00:00
|
|
|
ctr = scTemplateToRPC(cfg, ctr, imports, scTypeToGo)
|
2023-11-21 18:46:21 +00:00
|
|
|
ctr.NamedTypes = make([]binding.ExtendedType, 0, len(cfg.NamedTypes))
|
|
|
|
for k := range cfg.NamedTypes {
|
|
|
|
ctr.NamedTypes = append(ctr.NamedTypes, cfg.NamedTypes[k])
|
|
|
|
}
|
|
|
|
sort.Slice(ctr.NamedTypes, func(i, j int) bool {
|
|
|
|
return strings.Compare(ctr.NamedTypes[i].Name, ctr.NamedTypes[j].Name) < 0
|
|
|
|
})
|
2022-11-25 10:04:24 +00:00
|
|
|
|
2023-08-11 13:59:31 +00:00
|
|
|
// Check resulting named types and events don't have duplicating field names.
|
|
|
|
for _, t := range ctr.NamedTypes {
|
|
|
|
fDict := make(map[string]struct{})
|
|
|
|
for _, n := range t.Fields {
|
|
|
|
name := upperFirst(n.Field)
|
|
|
|
if _, ok := fDict[name]; ok {
|
|
|
|
return fmt.Errorf("named type `%s` has two fields with identical resulting binding name `%s`", t.Name, name)
|
|
|
|
}
|
|
|
|
fDict[name] = struct{}{}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for _, e := range ctr.CustomEvents {
|
|
|
|
fDict := make(map[string]struct{})
|
|
|
|
for _, n := range e.Parameters {
|
|
|
|
name := upperFirst(n.Name)
|
|
|
|
if _, ok := fDict[name]; ok {
|
|
|
|
return fmt.Errorf("event `%s` has two fields with identical resulting binding name `%s`", e.Name, name)
|
|
|
|
}
|
|
|
|
fDict[name] = struct{}{}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-11-25 10:04:24 +00:00
|
|
|
var srcTemplate = template.Must(template.New("generate").Funcs(template.FuncMap{
|
2022-12-02 12:52:52 +00:00
|
|
|
"addIndent": addIndent,
|
2022-11-25 10:04:24 +00:00
|
|
|
"etTypeConverter": etTypeConverter,
|
|
|
|
"etTypeToStr": func(et binding.ExtendedType) string {
|
2023-11-21 18:46:21 +00:00
|
|
|
r, _ := extendedTypeToGo(et, cfg.NamedTypes)
|
2022-11-25 10:04:24 +00:00
|
|
|
return r
|
|
|
|
},
|
2023-05-23 11:17:39 +00:00
|
|
|
"toTypeName": toTypeName,
|
|
|
|
"cutPointer": cutPointer,
|
2023-08-10 17:23:54 +00:00
|
|
|
"upperFirst": upperFirst,
|
2022-11-25 10:04:24 +00:00
|
|
|
}).Parse(srcTmpl))
|
2022-10-26 20:27:24 +00:00
|
|
|
|
2023-10-18 14:58:39 +00:00
|
|
|
return binding.FExecute(srcTemplate, cfg.Output, ctr)
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
|
|
|
|
2022-11-08 12:43:32 +00:00
|
|
|
func dropManifestMethods(meths []manifest.Method, manifested []manifest.Method) []manifest.Method {
|
2022-10-26 20:27:24 +00:00
|
|
|
for _, m := range manifested {
|
|
|
|
for i := 0; i < len(meths); i++ {
|
2022-11-08 12:43:32 +00:00
|
|
|
if meths[i].Name == m.Name && len(meths[i].Parameters) == len(m.Parameters) {
|
2022-10-26 20:27:24 +00:00
|
|
|
meths = append(meths[:i], meths[i+1:]...)
|
|
|
|
i--
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return meths
|
|
|
|
}
|
|
|
|
|
2023-05-21 12:08:29 +00:00
|
|
|
func dropManifestEvents(events []manifest.Event, manifested []manifest.Event) []manifest.Event {
|
|
|
|
for _, e := range manifested {
|
|
|
|
for i := 0; i < len(events); i++ {
|
|
|
|
if events[i].Name == e.Name && len(events[i].Parameters) == len(e.Parameters) {
|
|
|
|
events = append(events[:i], events[i+1:]...)
|
|
|
|
i--
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return events
|
|
|
|
}
|
|
|
|
|
2022-11-08 12:43:32 +00:00
|
|
|
func dropStdMethods(meths []manifest.Method, std *standard.Standard) []manifest.Method {
|
2022-10-26 20:27:24 +00:00
|
|
|
meths = dropManifestMethods(meths, std.Manifest.ABI.Methods)
|
|
|
|
if std.Optional != nil {
|
|
|
|
meths = dropManifestMethods(meths, std.Optional)
|
|
|
|
}
|
|
|
|
if std.Base != nil {
|
|
|
|
return dropStdMethods(meths, std.Base)
|
|
|
|
}
|
|
|
|
return meths
|
|
|
|
}
|
|
|
|
|
2023-05-21 12:08:29 +00:00
|
|
|
func dropStdEvents(events []manifest.Event, std *standard.Standard) []manifest.Event {
|
|
|
|
events = dropManifestEvents(events, std.Manifest.ABI.Events)
|
|
|
|
if std.Base != nil {
|
|
|
|
return dropStdEvents(events, std.Base)
|
|
|
|
}
|
|
|
|
return events
|
|
|
|
}
|
|
|
|
|
2022-11-25 10:04:24 +00:00
|
|
|
func extendedTypeToGo(et binding.ExtendedType, named map[string]binding.ExtendedType) (string, string) {
|
|
|
|
switch et.Base {
|
2022-10-26 20:27:24 +00:00
|
|
|
case smartcontract.AnyType:
|
2023-04-03 10:34:24 +00:00
|
|
|
return "any", ""
|
2022-10-26 20:27:24 +00:00
|
|
|
case smartcontract.BoolType:
|
|
|
|
return "bool", ""
|
|
|
|
case smartcontract.IntegerType:
|
|
|
|
return "*big.Int", "math/big"
|
|
|
|
case smartcontract.ByteArrayType:
|
|
|
|
return "[]byte", ""
|
|
|
|
case smartcontract.StringType:
|
|
|
|
return "string", ""
|
|
|
|
case smartcontract.Hash160Type:
|
|
|
|
return "util.Uint160", "github.com/nspcc-dev/neo-go/pkg/util"
|
|
|
|
case smartcontract.Hash256Type:
|
|
|
|
return "util.Uint256", "github.com/nspcc-dev/neo-go/pkg/util"
|
|
|
|
case smartcontract.PublicKeyType:
|
|
|
|
return "*keys.PublicKey", "github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
|
|
case smartcontract.SignatureType:
|
|
|
|
return "[]byte", ""
|
|
|
|
case smartcontract.ArrayType:
|
2022-11-25 10:04:24 +00:00
|
|
|
if len(et.Name) > 0 {
|
|
|
|
return "*" + toTypeName(et.Name), "github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
|
|
} else if et.Value != nil {
|
|
|
|
if et.Value.Base == smartcontract.PublicKeyType { // Special array wrapper.
|
|
|
|
return "keys.PublicKeys", "github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
|
|
}
|
|
|
|
sub, pkg := extendedTypeToGo(*et.Value, named)
|
|
|
|
return "[]" + sub, pkg
|
|
|
|
}
|
2023-04-03 10:34:24 +00:00
|
|
|
return "[]any", ""
|
2022-11-25 10:04:24 +00:00
|
|
|
|
2022-10-26 20:27:24 +00:00
|
|
|
case smartcontract.MapType:
|
2022-12-01 19:16:37 +00:00
|
|
|
kt, _ := extendedTypeToGo(binding.ExtendedType{Base: et.Key}, named)
|
rpcbinding: support map[any]any conversion for extended types
Unfortunately, without pre-set user extended types configuration for events
and without --guess-eventtypes flag set we are allowed to rely only on manifest
information about types. Manifest can't give us a lot of information, but we
still need to be able to generate RPC binding. Arrays and structs are correctly
handled by the current code, but maps always rely on the fact that map's value
type is set. It's not true in the described case, so make the maps type convertor
handle this situation in a similar way how arrays are handled.
Without this commit the following panic occurs on attempt to generate RPC binding:
```
--- FAIL: TestAssistedRPCBindings/testdata/notifications (0.01s)
panic: runtime error: invalid memory address or nil pointer dereference [recovered]
panic: runtime error: invalid memory address or nil pointer dereference
[signal SIGSEGV: segmentation violation code=0x1 addr=0x0 pc=0x7f7c0e]
goroutine 190 [running]:
testing.tRunner.func1.2({0x109cb40, 0x1d58760})
/usr/local/go/src/testing/testing.go:1396 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1399 +0x39f
panic({0x109cb40, 0x1d58760})
/usr/local/go/src/runtime/panic.go:884 +0x212
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.extendedTypeToGo({0x22, {0x0, 0x0}, {0x0, 0x0}, 0x0, 0x0, {0x0, 0x0, 0x0}}, ...)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:515 +0x36e
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.scTypeToGo({0xc000206d92?, 0xc000206d80?}, 0x22, 0xc0005d70e0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:643 +0x138
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.scTemplateToRPC({{0xc00049bb07, 0x7}, 0xc0004c89c0, {0x33, 0x22, 0x11, 0x0, 0xff, 0xee, 0xdd, ...}, ...}, ...)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:686 +0xbc4
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.Generate({{0xc00049bb07, 0x7}, 0xc0004c89c0, {0x33, 0x22, 0x11, 0x0, 0xff, 0xee, 0xdd, ...}, ...})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:421 +0x387
github.com/nspcc-dev/neo-go/cli/smartcontract.contractGenerateSomething(0xc00043e2c0, 0x137cd00)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate.go:99 +0x855
github.com/nspcc-dev/neo-go/cli/smartcontract.contractGenerateRPCWrapper(0xc00043e2c0?)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate.go:60 +0x25
github.com/urfave/cli.HandleAction({0x1048380?, 0x137c660?}, 0x13?)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:524 +0x50
github.com/urfave/cli.Command.Run({{0x123539d, 0x13}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x12577ad, 0x2a}, {0x127ad35, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:173 +0x65b
github.com/urfave/cli.(*App).RunAsSubcommand(0xc0001f4000, 0xc00043e000)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:405 +0x91b
github.com/urfave/cli.Command.startApp({{0x12281e1, 0x8}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x1254d8a, 0x28}, {0x0, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:372 +0x6e7
github.com/urfave/cli.Command.Run({{0x12281e1, 0x8}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x1254d8a, 0x28}, {0x0, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:102 +0x825
github.com/urfave/cli.(*App).Run(0xc00024e000, {0xc0004f6420, 0xb, 0xb})
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:277 +0x8a7
github.com/nspcc-dev/neo-go/cli/smartcontract.TestAssistedRPCBindings.func1.1(0x9f8829?)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate_test.go:395 +0x5fc
testing.tRunner(0xc0006824e0, 0xc0004a3680)
/usr/local/go/src/testing/testing.go:1446 +0x10b
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1493 +0x35f
```
Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
2023-05-03 16:13:05 +00:00
|
|
|
var vt string
|
|
|
|
if et.Value != nil {
|
|
|
|
vt, _ = extendedTypeToGo(*et.Value, named)
|
|
|
|
} else {
|
|
|
|
vt = "any"
|
|
|
|
}
|
2022-12-01 19:16:37 +00:00
|
|
|
return "map[" + kt + "]" + vt, "github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
2022-10-26 20:27:24 +00:00
|
|
|
case smartcontract.InteropInterfaceType:
|
2023-04-03 10:34:24 +00:00
|
|
|
return "any", ""
|
2022-10-26 20:27:24 +00:00
|
|
|
case smartcontract.VoidType:
|
|
|
|
return "", ""
|
|
|
|
}
|
2022-11-25 10:04:24 +00:00
|
|
|
panic("unreachable")
|
|
|
|
}
|
|
|
|
|
|
|
|
func etTypeConverter(et binding.ExtendedType, v string) string {
|
|
|
|
switch et.Base {
|
|
|
|
case smartcontract.AnyType:
|
2023-05-03 16:18:43 +00:00
|
|
|
return v + ".Value(), error(nil)"
|
2022-11-25 10:04:24 +00:00
|
|
|
case smartcontract.BoolType:
|
|
|
|
return v + ".TryBool()"
|
|
|
|
case smartcontract.IntegerType:
|
|
|
|
return v + ".TryInteger()"
|
|
|
|
case smartcontract.ByteArrayType, smartcontract.SignatureType:
|
|
|
|
return v + ".TryBytes()"
|
|
|
|
case smartcontract.StringType:
|
2023-10-18 15:09:58 +00:00
|
|
|
return `func(item stackitem.Item) (string, error) {
|
2022-11-25 10:04:24 +00:00
|
|
|
b, err := item.TryBytes()
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
if !utf8.Valid(b) {
|
|
|
|
return "", errors.New("not a UTF-8 string")
|
|
|
|
}
|
|
|
|
return string(b), nil
|
2023-10-18 15:09:58 +00:00
|
|
|
}(` + v + `)`
|
2022-11-25 10:04:24 +00:00
|
|
|
case smartcontract.Hash160Type:
|
2023-10-18 15:09:58 +00:00
|
|
|
return `func(item stackitem.Item) (util.Uint160, error) {
|
2022-11-25 10:04:24 +00:00
|
|
|
b, err := item.TryBytes()
|
|
|
|
if err != nil {
|
|
|
|
return util.Uint160{}, err
|
|
|
|
}
|
|
|
|
u, err := util.Uint160DecodeBytesBE(b)
|
|
|
|
if err != nil {
|
|
|
|
return util.Uint160{}, err
|
|
|
|
}
|
|
|
|
return u, nil
|
2023-10-18 15:09:58 +00:00
|
|
|
}(` + v + `)`
|
2022-11-25 10:04:24 +00:00
|
|
|
case smartcontract.Hash256Type:
|
2023-10-18 15:09:58 +00:00
|
|
|
return `func(item stackitem.Item) (util.Uint256, error) {
|
2022-11-25 10:04:24 +00:00
|
|
|
b, err := item.TryBytes()
|
|
|
|
if err != nil {
|
|
|
|
return util.Uint256{}, err
|
|
|
|
}
|
|
|
|
u, err := util.Uint256DecodeBytesBE(b)
|
|
|
|
if err != nil {
|
|
|
|
return util.Uint256{}, err
|
|
|
|
}
|
|
|
|
return u, nil
|
2023-10-18 15:09:58 +00:00
|
|
|
}(` + v + `)`
|
2022-11-25 10:04:24 +00:00
|
|
|
case smartcontract.PublicKeyType:
|
2023-10-18 15:09:58 +00:00
|
|
|
return `func(item stackitem.Item) (*keys.PublicKey, error) {
|
2022-11-25 10:04:24 +00:00
|
|
|
b, err := item.TryBytes()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
k, err := keys.NewPublicKeyFromBytes(b, elliptic.P256())
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return k, nil
|
2023-10-18 15:09:58 +00:00
|
|
|
}(` + v + `)`
|
2022-11-25 10:04:24 +00:00
|
|
|
case smartcontract.ArrayType:
|
|
|
|
if len(et.Name) > 0 {
|
|
|
|
return "itemTo" + toTypeName(et.Name) + "(" + v + ", nil)"
|
|
|
|
} else if et.Value != nil {
|
|
|
|
at, _ := extendedTypeToGo(et, nil)
|
2023-10-18 15:09:58 +00:00
|
|
|
return `func(item stackitem.Item) (` + at + `, error) {
|
2022-11-25 10:04:24 +00:00
|
|
|
arr, ok := item.Value().([]stackitem.Item)
|
|
|
|
if !ok {
|
|
|
|
return nil, errors.New("not an array")
|
|
|
|
}
|
|
|
|
res := make(` + at + `, len(arr))
|
|
|
|
for i := range res {
|
2022-12-02 12:52:52 +00:00
|
|
|
res[i], err = ` + addIndent(etTypeConverter(*et.Value, "arr[i]"), "\t\t") + `
|
2022-11-25 10:04:24 +00:00
|
|
|
if err != nil {
|
2022-12-02 13:12:56 +00:00
|
|
|
return nil, fmt.Errorf("item %d: %w", i, err)
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return res, nil
|
2023-10-18 15:09:58 +00:00
|
|
|
}(` + v + `)`
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
|
|
|
return etTypeConverter(binding.ExtendedType{
|
|
|
|
Base: smartcontract.ArrayType,
|
|
|
|
Value: &binding.ExtendedType{
|
|
|
|
Base: smartcontract.AnyType,
|
|
|
|
},
|
|
|
|
}, v)
|
|
|
|
|
|
|
|
case smartcontract.MapType:
|
rpcbinding: support map[any]any conversion for extended types
Unfortunately, without pre-set user extended types configuration for events
and without --guess-eventtypes flag set we are allowed to rely only on manifest
information about types. Manifest can't give us a lot of information, but we
still need to be able to generate RPC binding. Arrays and structs are correctly
handled by the current code, but maps always rely on the fact that map's value
type is set. It's not true in the described case, so make the maps type convertor
handle this situation in a similar way how arrays are handled.
Without this commit the following panic occurs on attempt to generate RPC binding:
```
--- FAIL: TestAssistedRPCBindings/testdata/notifications (0.01s)
panic: runtime error: invalid memory address or nil pointer dereference [recovered]
panic: runtime error: invalid memory address or nil pointer dereference
[signal SIGSEGV: segmentation violation code=0x1 addr=0x0 pc=0x7f7c0e]
goroutine 190 [running]:
testing.tRunner.func1.2({0x109cb40, 0x1d58760})
/usr/local/go/src/testing/testing.go:1396 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1399 +0x39f
panic({0x109cb40, 0x1d58760})
/usr/local/go/src/runtime/panic.go:884 +0x212
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.extendedTypeToGo({0x22, {0x0, 0x0}, {0x0, 0x0}, 0x0, 0x0, {0x0, 0x0, 0x0}}, ...)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:515 +0x36e
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.scTypeToGo({0xc000206d92?, 0xc000206d80?}, 0x22, 0xc0005d70e0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:643 +0x138
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.scTemplateToRPC({{0xc00049bb07, 0x7}, 0xc0004c89c0, {0x33, 0x22, 0x11, 0x0, 0xff, 0xee, 0xdd, ...}, ...}, ...)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:686 +0xbc4
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.Generate({{0xc00049bb07, 0x7}, 0xc0004c89c0, {0x33, 0x22, 0x11, 0x0, 0xff, 0xee, 0xdd, ...}, ...})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:421 +0x387
github.com/nspcc-dev/neo-go/cli/smartcontract.contractGenerateSomething(0xc00043e2c0, 0x137cd00)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate.go:99 +0x855
github.com/nspcc-dev/neo-go/cli/smartcontract.contractGenerateRPCWrapper(0xc00043e2c0?)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate.go:60 +0x25
github.com/urfave/cli.HandleAction({0x1048380?, 0x137c660?}, 0x13?)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:524 +0x50
github.com/urfave/cli.Command.Run({{0x123539d, 0x13}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x12577ad, 0x2a}, {0x127ad35, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:173 +0x65b
github.com/urfave/cli.(*App).RunAsSubcommand(0xc0001f4000, 0xc00043e000)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:405 +0x91b
github.com/urfave/cli.Command.startApp({{0x12281e1, 0x8}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x1254d8a, 0x28}, {0x0, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:372 +0x6e7
github.com/urfave/cli.Command.Run({{0x12281e1, 0x8}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x1254d8a, 0x28}, {0x0, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:102 +0x825
github.com/urfave/cli.(*App).Run(0xc00024e000, {0xc0004f6420, 0xb, 0xb})
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:277 +0x8a7
github.com/nspcc-dev/neo-go/cli/smartcontract.TestAssistedRPCBindings.func1.1(0x9f8829?)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate_test.go:395 +0x5fc
testing.tRunner(0xc0006824e0, 0xc0004a3680)
/usr/local/go/src/testing/testing.go:1446 +0x10b
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1493 +0x35f
```
Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
2023-05-03 16:13:05 +00:00
|
|
|
if et.Value != nil {
|
|
|
|
at, _ := extendedTypeToGo(et, nil)
|
2023-10-18 15:09:58 +00:00
|
|
|
return `func(item stackitem.Item) (` + at + `, error) {
|
2022-12-01 19:16:37 +00:00
|
|
|
m, ok := item.Value().([]stackitem.MapElement)
|
|
|
|
if !ok {
|
|
|
|
return nil, fmt.Errorf("%s is not a map", item.Type().String())
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
2022-12-01 19:16:37 +00:00
|
|
|
res := make(` + at + `)
|
|
|
|
for i := range m {
|
2022-12-02 12:52:52 +00:00
|
|
|
k, err := ` + addIndent(etTypeConverter(binding.ExtendedType{Base: et.Key}, "m[i].Key"), "\t\t") + `
|
2022-12-01 19:16:37 +00:00
|
|
|
if err != nil {
|
2022-12-02 13:12:56 +00:00
|
|
|
return nil, fmt.Errorf("key %d: %w", i, err)
|
2022-12-01 19:16:37 +00:00
|
|
|
}
|
2022-12-02 12:52:52 +00:00
|
|
|
v, err := ` + addIndent(etTypeConverter(*et.Value, "m[i].Value"), "\t\t") + `
|
2022-12-01 19:16:37 +00:00
|
|
|
if err != nil {
|
2022-12-02 13:12:56 +00:00
|
|
|
return nil, fmt.Errorf("value %d: %w", i, err)
|
2022-12-01 19:16:37 +00:00
|
|
|
}
|
|
|
|
res[k] = v
|
|
|
|
}
|
|
|
|
return res, nil
|
2023-10-18 15:09:58 +00:00
|
|
|
}(` + v + `)`
|
rpcbinding: support map[any]any conversion for extended types
Unfortunately, without pre-set user extended types configuration for events
and without --guess-eventtypes flag set we are allowed to rely only on manifest
information about types. Manifest can't give us a lot of information, but we
still need to be able to generate RPC binding. Arrays and structs are correctly
handled by the current code, but maps always rely on the fact that map's value
type is set. It's not true in the described case, so make the maps type convertor
handle this situation in a similar way how arrays are handled.
Without this commit the following panic occurs on attempt to generate RPC binding:
```
--- FAIL: TestAssistedRPCBindings/testdata/notifications (0.01s)
panic: runtime error: invalid memory address or nil pointer dereference [recovered]
panic: runtime error: invalid memory address or nil pointer dereference
[signal SIGSEGV: segmentation violation code=0x1 addr=0x0 pc=0x7f7c0e]
goroutine 190 [running]:
testing.tRunner.func1.2({0x109cb40, 0x1d58760})
/usr/local/go/src/testing/testing.go:1396 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1399 +0x39f
panic({0x109cb40, 0x1d58760})
/usr/local/go/src/runtime/panic.go:884 +0x212
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.extendedTypeToGo({0x22, {0x0, 0x0}, {0x0, 0x0}, 0x0, 0x0, {0x0, 0x0, 0x0}}, ...)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:515 +0x36e
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.scTypeToGo({0xc000206d92?, 0xc000206d80?}, 0x22, 0xc0005d70e0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:643 +0x138
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.scTemplateToRPC({{0xc00049bb07, 0x7}, 0xc0004c89c0, {0x33, 0x22, 0x11, 0x0, 0xff, 0xee, 0xdd, ...}, ...}, ...)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:686 +0xbc4
github.com/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding.Generate({{0xc00049bb07, 0x7}, 0xc0004c89c0, {0x33, 0x22, 0x11, 0x0, 0xff, 0xee, 0xdd, ...}, ...})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/smartcontract/rpcbinding/binding.go:421 +0x387
github.com/nspcc-dev/neo-go/cli/smartcontract.contractGenerateSomething(0xc00043e2c0, 0x137cd00)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate.go:99 +0x855
github.com/nspcc-dev/neo-go/cli/smartcontract.contractGenerateRPCWrapper(0xc00043e2c0?)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate.go:60 +0x25
github.com/urfave/cli.HandleAction({0x1048380?, 0x137c660?}, 0x13?)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:524 +0x50
github.com/urfave/cli.Command.Run({{0x123539d, 0x13}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x12577ad, 0x2a}, {0x127ad35, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:173 +0x65b
github.com/urfave/cli.(*App).RunAsSubcommand(0xc0001f4000, 0xc00043e000)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:405 +0x91b
github.com/urfave/cli.Command.startApp({{0x12281e1, 0x8}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x1254d8a, 0x28}, {0x0, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:372 +0x6e7
github.com/urfave/cli.Command.Run({{0x12281e1, 0x8}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0x1254d8a, 0x28}, {0x0, ...}, ...}, ...)
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/command.go:102 +0x825
github.com/urfave/cli.(*App).Run(0xc00024e000, {0xc0004f6420, 0xb, 0xb})
/home/anna/go/pkg/mod/github.com/urfave/cli@v1.22.5/app.go:277 +0x8a7
github.com/nspcc-dev/neo-go/cli/smartcontract.TestAssistedRPCBindings.func1.1(0x9f8829?)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/cli/smartcontract/generate_test.go:395 +0x5fc
testing.tRunner(0xc0006824e0, 0xc0004a3680)
/usr/local/go/src/testing/testing.go:1446 +0x10b
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1493 +0x35f
```
Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
2023-05-03 16:13:05 +00:00
|
|
|
}
|
|
|
|
return etTypeConverter(binding.ExtendedType{
|
|
|
|
Base: smartcontract.MapType,
|
|
|
|
Key: et.Key,
|
|
|
|
Value: &binding.ExtendedType{
|
|
|
|
Base: smartcontract.AnyType,
|
|
|
|
},
|
|
|
|
}, v)
|
2022-11-25 10:04:24 +00:00
|
|
|
case smartcontract.InteropInterfaceType:
|
|
|
|
return "item.Value(), nil"
|
|
|
|
case smartcontract.VoidType:
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
panic("unreachable")
|
|
|
|
}
|
|
|
|
|
|
|
|
func scTypeToGo(name string, typ smartcontract.ParamType, cfg *binding.Config) (string, string) {
|
|
|
|
et, ok := cfg.Types[name]
|
|
|
|
if !ok {
|
|
|
|
et = binding.ExtendedType{Base: typ}
|
|
|
|
}
|
|
|
|
return extendedTypeToGo(et, cfg.NamedTypes)
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
|
|
|
|
2023-05-24 08:52:14 +00:00
|
|
|
func scTemplateToRPC(cfg binding.Config, ctr ContractTmpl, imports map[string]struct{}, scTypeConverter func(string, smartcontract.ParamType, *binding.Config) (string, string)) ContractTmpl {
|
2022-10-26 20:27:24 +00:00
|
|
|
for i := range ctr.Imports {
|
|
|
|
imports[ctr.Imports[i]] = struct{}{}
|
|
|
|
}
|
2023-05-11 10:25:45 +00:00
|
|
|
if !cfg.Hash.Equals(util.Uint160{}) {
|
|
|
|
ctr.Hash = fmt.Sprintf("%#v", cfg.Hash)
|
|
|
|
}
|
2022-10-26 20:27:24 +00:00
|
|
|
for i := 0; i < len(ctr.Methods); i++ {
|
|
|
|
abim := cfg.Manifest.ABI.GetMethod(ctr.Methods[i].NameABI, len(ctr.Methods[i].Arguments))
|
2022-10-27 13:20:35 +00:00
|
|
|
if abim.Safe {
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods = append(ctr.SafeMethods, SafeMethodTmpl{MethodTmpl: ctr.Methods[i]})
|
|
|
|
et, ok := cfg.Types[abim.Name]
|
|
|
|
if ok {
|
|
|
|
ctr.SafeMethods[len(ctr.SafeMethods)-1].ExtendedReturn = et
|
|
|
|
if abim.ReturnType == smartcontract.ArrayType && len(et.Name) > 0 {
|
|
|
|
ctr.SafeMethods[len(ctr.SafeMethods)-1].ItemTo = cutPointer(ctr.Methods[i].ReturnType)
|
|
|
|
}
|
|
|
|
}
|
2022-10-26 20:27:24 +00:00
|
|
|
ctr.Methods = append(ctr.Methods[:i], ctr.Methods[i+1:]...)
|
|
|
|
i--
|
2022-10-27 13:20:35 +00:00
|
|
|
} else {
|
|
|
|
ctr.Methods[i].Comment = fmt.Sprintf("creates a transaction invoking `%s` method of the contract.", ctr.Methods[i].NameABI)
|
2022-11-07 19:13:47 +00:00
|
|
|
if ctr.Methods[i].ReturnType == "bool" {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/smartcontract"] = struct{}{}
|
|
|
|
}
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
|
|
|
}
|
2022-11-25 10:04:24 +00:00
|
|
|
for _, et := range cfg.NamedTypes {
|
2023-11-21 18:46:21 +00:00
|
|
|
addETImports(et, cfg.NamedTypes, imports)
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
|
|
|
if len(cfg.NamedTypes) > 0 {
|
|
|
|
imports["errors"] = struct{}{}
|
|
|
|
}
|
2023-05-24 08:52:14 +00:00
|
|
|
for _, abiEvent := range cfg.Manifest.ABI.Events {
|
2023-05-23 11:17:39 +00:00
|
|
|
eBindingName := ToEventBindingName(abiEvent.Name)
|
2023-05-24 08:52:14 +00:00
|
|
|
eTmp := CustomEventTemplate{
|
2023-05-23 11:17:39 +00:00
|
|
|
Name: eBindingName,
|
2023-05-24 08:52:14 +00:00
|
|
|
ManifestName: abiEvent.Name,
|
|
|
|
}
|
|
|
|
for i := range abiEvent.Parameters {
|
2023-05-23 11:17:39 +00:00
|
|
|
pBindingName := ToParameterBindingName(abiEvent.Parameters[i].Name)
|
|
|
|
fullPName := eBindingName + "." + pBindingName
|
2023-05-24 08:52:14 +00:00
|
|
|
typeStr, pkg := scTypeConverter(fullPName, abiEvent.Parameters[i].Type, &cfg)
|
|
|
|
if pkg != "" {
|
|
|
|
imports[pkg] = struct{}{}
|
|
|
|
}
|
|
|
|
|
|
|
|
var (
|
|
|
|
extType binding.ExtendedType
|
|
|
|
ok bool
|
|
|
|
)
|
|
|
|
if extType, ok = cfg.Types[fullPName]; !ok {
|
|
|
|
extType = binding.ExtendedType{
|
2023-05-23 15:20:51 +00:00
|
|
|
Base: abiEvent.Parameters[i].Type,
|
2023-05-24 08:52:14 +00:00
|
|
|
}
|
2023-11-21 18:46:21 +00:00
|
|
|
addETImports(extType, cfg.NamedTypes, imports)
|
2023-05-24 08:52:14 +00:00
|
|
|
}
|
|
|
|
eTmp.Parameters = append(eTmp.Parameters, EventParamTmpl{
|
|
|
|
ParamTmpl: binding.ParamTmpl{
|
2023-05-23 11:17:39 +00:00
|
|
|
Name: pBindingName,
|
2023-05-24 08:52:14 +00:00
|
|
|
Type: typeStr,
|
|
|
|
},
|
|
|
|
ExtType: extType,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
ctr.CustomEvents = append(ctr.CustomEvents, eTmp)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(ctr.CustomEvents) > 0 {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/neorpc/result"] = struct{}{}
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/vm/stackitem"] = struct{}{}
|
|
|
|
imports["fmt"] = struct{}{}
|
|
|
|
imports["errors"] = struct{}{}
|
|
|
|
}
|
2022-11-25 10:04:24 +00:00
|
|
|
|
2022-10-27 13:20:35 +00:00
|
|
|
for i := range ctr.SafeMethods {
|
|
|
|
switch ctr.SafeMethods[i].ReturnType {
|
2023-04-03 10:34:24 +00:00
|
|
|
case "any":
|
2022-11-09 07:20:44 +00:00
|
|
|
abim := cfg.Manifest.ABI.GetMethod(ctr.SafeMethods[i].NameABI, len(ctr.SafeMethods[i].Arguments))
|
|
|
|
if abim.ReturnType == smartcontract.InteropInterfaceType {
|
|
|
|
imports["github.com/google/uuid"] = struct{}{}
|
2022-11-09 08:58:01 +00:00
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/vm/stackitem"] = struct{}{}
|
2022-11-09 07:20:44 +00:00
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/neorpc/result"] = struct{}{}
|
|
|
|
ctr.SafeMethods[i].ReturnType = "uuid.UUID, result.Iterator"
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "SessionIterator"
|
2022-11-09 08:58:01 +00:00
|
|
|
ctr.HasIterator = true
|
2022-11-09 07:20:44 +00:00
|
|
|
} else {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/vm/stackitem"] = struct{}{}
|
2023-04-03 10:34:24 +00:00
|
|
|
ctr.SafeMethods[i].ReturnType = "any"
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Item"
|
2022-11-09 07:20:44 +00:00
|
|
|
}
|
2022-10-26 20:27:24 +00:00
|
|
|
case "bool":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Bool"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "*big.Int":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "BigInt"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "string":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "UTF8String"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "util.Uint160":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Uint160"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "util.Uint256":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Uint256"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "*keys.PublicKey":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "PublicKey"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "[]byte":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Bytes"
|
2023-04-03 10:34:24 +00:00
|
|
|
case "[]any":
|
2022-10-26 20:27:24 +00:00
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/vm/stackitem"] = struct{}{}
|
2022-10-27 13:20:35 +00:00
|
|
|
ctr.SafeMethods[i].ReturnType = "[]stackitem.Item"
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Array"
|
2022-10-26 20:27:24 +00:00
|
|
|
case "*stackitem.Map":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Map"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "[]bool":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfBools"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "[]*big.Int":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfBigInts"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "[][]byte":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfBytes"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "[]string":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfUTF8Strings"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "[]util.Uint160":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfUint160"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "[]util.Uint256":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfUint256"
|
2022-11-11 11:59:01 +00:00
|
|
|
case "keys.PublicKeys":
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "ArrayOfPublicKeys"
|
2022-11-25 10:04:24 +00:00
|
|
|
default:
|
2023-11-21 18:46:21 +00:00
|
|
|
addETImports(ctr.SafeMethods[i].ExtendedReturn, cfg.NamedTypes, imports)
|
2022-12-02 07:20:55 +00:00
|
|
|
ctr.SafeMethods[i].Unwrapper = "Item"
|
2022-10-26 20:27:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-11-08 12:43:32 +00:00
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/util"] = struct{}{}
|
2022-10-27 13:20:35 +00:00
|
|
|
if len(ctr.SafeMethods) > 0 {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/rpcclient/unwrap"] = struct{}{}
|
2022-11-08 12:43:32 +00:00
|
|
|
if !(ctr.IsNep17 || ctr.IsNep11D || ctr.IsNep11ND) {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/neorpc/result"] = struct{}{}
|
|
|
|
}
|
2022-10-27 13:20:35 +00:00
|
|
|
}
|
|
|
|
if len(ctr.Methods) > 0 {
|
|
|
|
imports["github.com/nspcc-dev/neo-go/pkg/core/transaction"] = struct{}{}
|
|
|
|
}
|
|
|
|
if len(ctr.Methods) > 0 || ctr.IsNep17 || ctr.IsNep11D || ctr.IsNep11ND {
|
|
|
|
ctr.HasWriter = true
|
|
|
|
}
|
2022-11-08 13:36:38 +00:00
|
|
|
if len(ctr.SafeMethods) > 0 || ctr.IsNep17 || ctr.IsNep11D || ctr.IsNep11ND {
|
|
|
|
ctr.HasReader = true
|
|
|
|
}
|
2022-10-26 20:27:24 +00:00
|
|
|
ctr.Imports = ctr.Imports[:0]
|
|
|
|
for imp := range imports {
|
|
|
|
ctr.Imports = append(ctr.Imports, imp)
|
|
|
|
}
|
|
|
|
sort.Strings(ctr.Imports)
|
|
|
|
return ctr
|
|
|
|
}
|
2022-11-25 10:04:24 +00:00
|
|
|
|
2022-12-02 07:20:55 +00:00
|
|
|
func addETImports(et binding.ExtendedType, named map[string]binding.ExtendedType, imports map[string]struct{}) {
|
|
|
|
_, pkg := extendedTypeToGo(et, named)
|
|
|
|
if pkg != "" {
|
|
|
|
imports[pkg] = struct{}{}
|
|
|
|
}
|
|
|
|
// Additional packages used during decoding.
|
|
|
|
switch et.Base {
|
|
|
|
case smartcontract.StringType:
|
|
|
|
imports["unicode/utf8"] = struct{}{}
|
|
|
|
imports["errors"] = struct{}{}
|
|
|
|
case smartcontract.PublicKeyType:
|
|
|
|
imports["crypto/elliptic"] = struct{}{}
|
|
|
|
case smartcontract.MapType:
|
|
|
|
imports["fmt"] = struct{}{}
|
2022-12-02 13:12:56 +00:00
|
|
|
case smartcontract.ArrayType:
|
|
|
|
imports["errors"] = struct{}{}
|
|
|
|
imports["fmt"] = struct{}{}
|
2022-12-02 07:20:55 +00:00
|
|
|
}
|
|
|
|
if et.Value != nil {
|
|
|
|
addETImports(*et.Value, named, imports)
|
|
|
|
}
|
|
|
|
if et.Base == smartcontract.MapType {
|
|
|
|
addETImports(binding.ExtendedType{Base: et.Key}, named, imports)
|
|
|
|
}
|
|
|
|
for i := range et.Fields {
|
|
|
|
addETImports(et.Fields[i].ExtendedType, named, imports)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-11-25 10:04:24 +00:00
|
|
|
func cutPointer(s string) string {
|
|
|
|
if s[0] == '*' {
|
|
|
|
return s[1:]
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
|
|
|
func toTypeName(s string) string {
|
|
|
|
return strings.Map(func(c rune) rune {
|
|
|
|
if c == '.' {
|
|
|
|
return -1
|
|
|
|
}
|
|
|
|
return c
|
2023-08-11 12:31:18 +00:00
|
|
|
}, upperFirst(s))
|
2022-11-25 10:04:24 +00:00
|
|
|
}
|
2022-12-02 12:52:52 +00:00
|
|
|
|
|
|
|
func addIndent(str string, ind string) string {
|
|
|
|
return strings.ReplaceAll(str, "\n", "\n"+ind)
|
|
|
|
}
|
2023-05-24 08:52:14 +00:00
|
|
|
|
2023-05-23 11:17:39 +00:00
|
|
|
// ToEventBindingName converts event name specified in the contract manifest to
|
|
|
|
// a valid go exported event structure name.
|
|
|
|
func ToEventBindingName(eventName string) string {
|
|
|
|
return toPascalCase(eventName) + "Event"
|
|
|
|
}
|
|
|
|
|
|
|
|
// ToParameterBindingName converts parameter name specified in the contract
|
|
|
|
// manifest to a valid go structure's exported field name.
|
|
|
|
func ToParameterBindingName(paramName string) string {
|
|
|
|
return toPascalCase(paramName)
|
|
|
|
}
|
|
|
|
|
2023-05-24 08:52:14 +00:00
|
|
|
// toPascalCase removes all non-unicode characters from the provided string and
|
|
|
|
// converts it to pascal case using space as delimiter.
|
|
|
|
func toPascalCase(s string) string {
|
|
|
|
var res string
|
|
|
|
ss := strings.Split(s, " ")
|
|
|
|
for i := range ss { // TODO: use DecodeRuneInString instead.
|
|
|
|
var word string
|
|
|
|
for _, ch := range ss[i] {
|
|
|
|
var ok bool
|
|
|
|
if len(res) == 0 && len(word) == 0 {
|
|
|
|
ok = unicode.IsLetter(ch)
|
|
|
|
} else {
|
|
|
|
ok = unicode.IsLetter(ch) || unicode.IsDigit(ch) || ch == '_'
|
|
|
|
}
|
|
|
|
if ok {
|
|
|
|
word += string(ch)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if len(word) > 0 {
|
|
|
|
res += upperFirst(word)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return res
|
|
|
|
}
|
|
|
|
|
|
|
|
func upperFirst(s string) string {
|
|
|
|
return strings.ToUpper(s[0:1]) + s[1:]
|
|
|
|
}
|