80 lines
2.8 KiB
Go
80 lines
2.8 KiB
Go
/*
|
|
Package nep22 contains RPC wrappers to work with NEP-22 contracts.
|
|
|
|
Contract provides state-changing method of upgradeable contracts.
|
|
*/
|
|
package nep22
|
|
|
|
import (
|
|
"github.com/tutus-one/tutus-chain/pkg/core/transaction"
|
|
"github.com/tutus-one/tutus-chain/pkg/smartcontract"
|
|
"github.com/tutus-one/tutus-chain/pkg/util"
|
|
)
|
|
|
|
// Actor is used by Contract wrapper to create and send transactions.
|
|
type Actor interface {
|
|
MakeRun(script []byte) (*transaction.Transaction, error)
|
|
MakeUnsignedRun(script []byte, attrs []transaction.Attribute) (*transaction.Transaction, error)
|
|
SendRun(script []byte) (util.Uint256, uint32, error)
|
|
}
|
|
|
|
// Contract is an RPC wrapper that implements the full NEP-22 interface.
|
|
type Contract struct {
|
|
hash util.Uint160
|
|
actor Actor
|
|
}
|
|
|
|
// NewContract creates an instance of Contract for contract with the given hash
|
|
// using the given Actor.
|
|
func NewContract(actor Actor, hash util.Uint160) *Contract {
|
|
return &Contract{
|
|
hash: hash,
|
|
actor: actor,
|
|
}
|
|
}
|
|
|
|
// Update creates and sends a transaction that performs an `update` method
|
|
// call using the given parameters. This transaction is signed and immediately
|
|
// sent to the network. The returned values are its hash, ValidUntilBlock value
|
|
// and an error, if any.
|
|
func (t *Contract) Update(nef []byte, manifest []byte, data any) (util.Uint256, uint32, error) {
|
|
script, err := t.updateScript(nef, manifest, data)
|
|
if err != nil {
|
|
return util.Uint256{}, 0, err
|
|
}
|
|
return t.actor.SendRun(script)
|
|
}
|
|
|
|
// UpdateTransaction creates and signs a transaction invoking the `update` method
|
|
// with the provided NEF bytes, manifest and arbitrary data,
|
|
// without sending it. The returned values are the signed Transaction and an error
|
|
// if any.
|
|
func (t *Contract) UpdateTransaction(nef []byte, manifest []byte, data any) (*transaction.Transaction, error) {
|
|
script, err := t.updateScript(nef, manifest, data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return t.actor.MakeRun(script)
|
|
}
|
|
|
|
// UpdateUnsigned creates an unsigned transaction invoking the `update` method
|
|
// with the provided NEF bytes, manifest and arbitrary data payload,
|
|
// without sending it. The returned values are the unsigned Transaction and
|
|
// an error if any.
|
|
func (t *Contract) UpdateUnsigned(nef []byte, manifest []byte, data any) (*transaction.Transaction, error) {
|
|
script, err := t.updateScript(nef, manifest, data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return t.actor.MakeUnsignedRun(script, nil)
|
|
}
|
|
|
|
// updateScript builds a VM script invoking the `update` method of the contract
|
|
// using the given parameters. The returned values are the raw script
|
|
// and an error, if any.
|
|
func (t *Contract) updateScript(nef []byte, manifest []byte, data any) ([]byte, error) {
|
|
b := smartcontract.NewBuilder()
|
|
b.InvokeMethod(t.hash, "update", nef, manifest, data)
|
|
return b.Script()
|
|
}
|