compiler: extend manifest generation with custom types

This commit is contained in:
Evgenii Stratonikov 2020-08-28 11:08:04 +03:00
parent cee1836183
commit 25f8545cdf
2 changed files with 41 additions and 0 deletions

View file

@ -201,6 +201,21 @@ func (c *codegen) scTypeFromExpr(typ ast.Expr) string {
if c.typeOf(typ) == nil {
return "Any"
}
if named, ok := t.(*types.Named); ok {
if isInteropPath(named.String()) {
name := named.Obj().Name()
pkg := named.Obj().Pkg().Name()
switch pkg {
case "blockchain", "contract":
return "Array" // Block, Transaction, Contract
case "interop":
if name != "Interface" {
return name
}
}
return "InteropInterface"
}
}
switch t := t.Underlying().(type) {
case *types.Basic:
info := t.Info()

View file

@ -15,6 +15,10 @@ import (
func TestCodeGen_DebugInfo(t *testing.T) {
src := `package foo
import "github.com/nspcc-dev/neo-go/pkg/interop"
import "github.com/nspcc-dev/neo-go/pkg/interop/storage"
import "github.com/nspcc-dev/neo-go/pkg/interop/blockchain"
import "github.com/nspcc-dev/neo-go/pkg/interop/contract"
func Main(op string) bool {
var s string
_ = s
@ -42,6 +46,12 @@ func MethodByteArray() []byte { return nil }
func MethodArray() []bool { return nil }
func MethodStruct() struct{} { return struct{}{} }
func unexportedMethod() int { return 1 }
func MethodParams(addr interop.Hash160, h interop.Hash256,
sig interop.Signature, pub interop.PublicKey,
inter interop.Interface, ctr contract.Contract,
ctx storage.Context, tx blockchain.Transaction) bool {
return true
}
type MyStruct struct {}
func (ms MyStruct) MethodOnStruct() { }
func (ms *MyStruct) MethodOnPointerToStruct() { }
@ -72,6 +82,7 @@ func (ms *MyStruct) MethodOnPointerToStruct() { }
"unexportedMethod": "Integer",
"MethodOnStruct": "Void",
"MethodOnPointerToStruct": "Void",
"MethodParams": "Boolean",
}
for i := range d.Methods {
name := d.Methods[i].ID
@ -201,6 +212,21 @@ func (ms *MyStruct) MethodOnPointerToStruct() { }
},
ReturnType: smartcontract.StringType,
},
{
Name: "methodParams",
Offset: 125,
Parameters: []manifest.Parameter{
manifest.NewParameter("addr", smartcontract.Hash160Type),
manifest.NewParameter("h", smartcontract.Hash256Type),
manifest.NewParameter("sig", smartcontract.SignatureType),
manifest.NewParameter("pub", smartcontract.PublicKeyType),
manifest.NewParameter("inter", smartcontract.InteropInterfaceType),
manifest.NewParameter("ctr", smartcontract.ArrayType),
manifest.NewParameter("ctx", smartcontract.InteropInterfaceType),
manifest.NewParameter("tx", smartcontract.ArrayType),
},
ReturnType: smartcontract.BoolType,
},
},
Events: []manifest.Event{},
},