smartcontract: add Builder, method invocation helpers and doc
Move the last remaining script-related things out of the rpcclient.
This commit is contained in:
parent
1b6f4051d8
commit
32ebb4a90d
6 changed files with 145 additions and 20 deletions
68
pkg/smartcontract/builder.go
Normal file
68
pkg/smartcontract/builder.go
Normal file
|
@ -0,0 +1,68 @@
|
|||
package smartcontract
|
||||
|
||||
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/opcode"
|
||||
)
|
||||
|
||||
// Builder is used to create arbitrary scripts from the set of methods it provides.
|
||||
// Each method emits some set of opcodes performing an action and (in most cases)
|
||||
// returning a result. These chunks of code can be composed together to perform
|
||||
// several actions in the same script (and therefore in the same transaction), but
|
||||
// the end result (in terms of state changes and/or resulting items) of the script
|
||||
// totally depends on what it contains and that's the responsibility of the Builder
|
||||
// user. Builder is mostly used to create transaction scripts (also known as
|
||||
// "entry scripts"), so the set of methods it exposes is tailored to this model
|
||||
// of use and any calls emitted don't limit flags in any way (always use
|
||||
// callflag.All).
|
||||
type Builder struct {
|
||||
bw *io.BufBinWriter
|
||||
}
|
||||
|
||||
// NewBuilder creates a new Builder instance.
|
||||
func NewBuilder() *Builder {
|
||||
return &Builder{bw: io.NewBufBinWriter()}
|
||||
}
|
||||
|
||||
// InvokeMethod is the most generic contract method invoker, the code it produces
|
||||
// packs all of the arguments given into an array and calls some method of the
|
||||
// contract. The correctness of this invocation (number and type of parameters) is
|
||||
// out of scope of this method, as well as return value, if contract's method returns
|
||||
// something this value just remains on the execution stack.
|
||||
func (b *Builder) InvokeMethod(contract util.Uint160, method string, params ...interface{}) {
|
||||
emit.AppCall(b.bw.BinWriter, contract, method, callflag.All, params...)
|
||||
}
|
||||
|
||||
// Assert emits an ASSERT opcode that expects a Boolean value to be on the stack,
|
||||
// checks if it's true and aborts the transaction if it's not.
|
||||
func (b *Builder) Assert() {
|
||||
emit.Opcodes(b.bw.BinWriter, opcode.ASSERT)
|
||||
}
|
||||
|
||||
// InvokeWithAssert emits an invocation of the method (see InvokeMethod) with
|
||||
// an ASSERT after the invocation. The presumption is that the method called
|
||||
// returns a Boolean value signalling the success or failure of the operation.
|
||||
// This pattern is pretty common, NEP-11 or NEP-17 'transfer' methods do exactly
|
||||
// that as well as NEO's 'vote'. The ASSERT then allow to simplify transaction
|
||||
// status checking, if action is successful then transaction is successful as
|
||||
// well, if it went wrong than whole transaction fails (ends with vmstate.FAULT).
|
||||
func (b *Builder) InvokeWithAssert(contract util.Uint160, method string, params ...interface{}) {
|
||||
b.InvokeMethod(contract, method, params...)
|
||||
b.Assert()
|
||||
}
|
||||
|
||||
// Script return current script, you can't use Builder after invoking this method
|
||||
// unless you Reset it.
|
||||
func (b *Builder) Script() ([]byte, error) {
|
||||
err := b.bw.Err
|
||||
return b.bw.Bytes(), err
|
||||
}
|
||||
|
||||
// Reset resets the Builder, allowing to reuse the same script buffer (but
|
||||
// previous script will be overwritten there).
|
||||
func (b *Builder) Reset() {
|
||||
b.bw.Reset()
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue