forked from TrueCloudLab/neoneo-go
fad061f3d9
Create a set of functions that are able to work with both session-based iterators, default unpacked iterators and client-side unpacked iterators.
305 lines
11 KiB
Go
305 lines
11 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpc/response/result"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
|
|
"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"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
)
|
|
|
|
// NEP11Decimals invokes `decimals` NEP-11 method on the specified contract.
|
|
func (c *Client) NEP11Decimals(tokenHash util.Uint160) (int64, error) {
|
|
return c.nepDecimals(tokenHash)
|
|
}
|
|
|
|
// NEP11Symbol invokes `symbol` NEP-11 method on the specified contract.
|
|
func (c *Client) NEP11Symbol(tokenHash util.Uint160) (string, error) {
|
|
return c.nepSymbol(tokenHash)
|
|
}
|
|
|
|
// NEP11TotalSupply invokes `totalSupply` NEP-11 method on the specified contract.
|
|
func (c *Client) NEP11TotalSupply(tokenHash util.Uint160) (int64, error) {
|
|
return c.nepTotalSupply(tokenHash)
|
|
}
|
|
|
|
// NEP11BalanceOf invokes `balanceOf` NEP-11 method on the specified contract.
|
|
func (c *Client) NEP11BalanceOf(tokenHash, owner util.Uint160) (int64, error) {
|
|
return c.nepBalanceOf(tokenHash, owner, nil)
|
|
}
|
|
|
|
// NEP11TokenInfo returns full NEP-11 token info.
|
|
func (c *Client) NEP11TokenInfo(tokenHash util.Uint160) (*wallet.Token, error) {
|
|
return c.nepTokenInfo(tokenHash, manifest.NEP11StandardName)
|
|
}
|
|
|
|
// TransferNEP11 creates an invocation transaction that invokes 'transfer' method
|
|
// on the given token to move the whole NEP-11 token with the specified token ID to
|
|
// the given account and sends it to the network returning just a hash of it.
|
|
func (c *Client) TransferNEP11(acc *wallet.Account, to util.Uint160,
|
|
tokenHash util.Uint160, tokenID string, data interface{}, gas int64, cosigners []SignerAccount) (util.Uint256, error) {
|
|
tx, err := c.CreateNEP11TransferTx(acc, tokenHash, gas, cosigners, to, tokenID, data)
|
|
if err != nil {
|
|
return util.Uint256{}, err
|
|
}
|
|
|
|
return c.SignAndPushTx(tx, acc, cosigners)
|
|
}
|
|
|
|
// CreateNEP11TransferTx creates an invocation transaction for the 'transfer'
|
|
// method of the given contract (token) to move the whole (or the specified amount
|
|
// of) NEP-11 token with the specified token ID to the given account and returns it.
|
|
// The returned transaction is not signed. CreateNEP11TransferTx is also a
|
|
// helper for TransferNEP11 and TransferNEP11D.
|
|
// `args` for TransferNEP11: to util.Uint160, tokenID string, data interface{};
|
|
// `args` for TransferNEP11D: from, to util.Uint160, amount int64, tokenID string, data interface{}.
|
|
func (c *Client) CreateNEP11TransferTx(acc *wallet.Account, tokenHash util.Uint160,
|
|
gas int64, cosigners []SignerAccount, args ...interface{}) (*transaction.Transaction, error) {
|
|
w := io.NewBufBinWriter()
|
|
emit.AppCall(w.BinWriter, tokenHash, "transfer", callflag.All, args...)
|
|
emit.Opcodes(w.BinWriter, opcode.ASSERT)
|
|
if w.Err != nil {
|
|
return nil, fmt.Errorf("failed to create NEP-11 transfer script: %w", w.Err)
|
|
}
|
|
from, err := address.StringToUint160(acc.Address)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("bad account address: %w", err)
|
|
}
|
|
return c.CreateTxFromScript(w.Bytes(), acc, -1, gas, append([]SignerAccount{{
|
|
Signer: transaction.Signer{
|
|
Account: from,
|
|
Scopes: transaction.CalledByEntry,
|
|
},
|
|
Account: acc,
|
|
}}, cosigners...))
|
|
}
|
|
|
|
// NEP11TokensOf returns iterator over token IDs for the specified owner of the
|
|
// specified NFT token. First return value is the session ID, the second one is
|
|
// Iterator itself, the third one is an error. Use TraverseIterator method to
|
|
// traverse iterator values or TerminateSession to terminate opened iterator
|
|
// session. See TraverseIterator and TerminateSession documentation for more details.
|
|
func (c *Client) NEP11TokensOf(tokenHash util.Uint160, owner util.Uint160) (uuid.UUID, result.Iterator, error) {
|
|
res, err := c.InvokeFunction(tokenHash, "tokensOf", []smartcontract.Parameter{
|
|
{
|
|
Type: smartcontract.Hash160Type,
|
|
Value: owner,
|
|
},
|
|
}, nil)
|
|
if err != nil {
|
|
return uuid.UUID{}, result.Iterator{}, err
|
|
}
|
|
err = getInvocationError(res)
|
|
if err != nil {
|
|
return uuid.UUID{}, result.Iterator{}, err
|
|
}
|
|
iter, err := topIteratorFromStack(res.Stack)
|
|
return res.Session, iter, err
|
|
}
|
|
|
|
// NEP11UnpackedTokensOf returns an array of token IDs for the specified owner of the specified NFT token.
|
|
// It differs from NEP11TokensOf in that no iterator session is used to retrieve values from iterator.
|
|
// Instead, unpacking VM script is created and invoked via `invokescript` JSON-RPC call.
|
|
func (c *Client) NEP11UnpackedTokensOf(tokenHash util.Uint160, owner util.Uint160) ([][]byte, error) {
|
|
result, err := c.InvokeAndPackIteratorResults(tokenHash, "tokensOf", []smartcontract.Parameter{
|
|
{
|
|
Type: smartcontract.Hash160Type,
|
|
Value: owner,
|
|
},
|
|
}, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = getInvocationError(result)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
arr, err := topIterableFromStack(result.Stack, []byte{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get token IDs from stack: %w", err)
|
|
}
|
|
ids := make([][]byte, len(arr))
|
|
for i := range ids {
|
|
ids[i] = arr[i].([]byte)
|
|
}
|
|
return ids, nil
|
|
}
|
|
|
|
// Non-divisible NFT methods section start.
|
|
|
|
// NEP11NDOwnerOf invokes `ownerOf` non-divisible NEP-11 method with the
|
|
// specified token ID on the specified contract.
|
|
func (c *Client) NEP11NDOwnerOf(tokenHash util.Uint160, tokenID []byte) (util.Uint160, error) {
|
|
result, err := c.InvokeFunction(tokenHash, "ownerOf", []smartcontract.Parameter{
|
|
{
|
|
Type: smartcontract.ByteArrayType,
|
|
Value: tokenID,
|
|
},
|
|
}, nil)
|
|
if err != nil {
|
|
return util.Uint160{}, err
|
|
}
|
|
err = getInvocationError(result)
|
|
if err != nil {
|
|
return util.Uint160{}, err
|
|
}
|
|
|
|
return topUint160FromStack(result.Stack)
|
|
}
|
|
|
|
// Non-divisible NFT methods section end.
|
|
|
|
// Divisible NFT methods section start.
|
|
|
|
// TransferNEP11D creates an invocation transaction that invokes 'transfer'
|
|
// method on the given token to move the specified amount of divisible NEP-11 assets
|
|
// (in FixedN format using contract's number of decimals) to the given account and
|
|
// sends it to the network returning just a hash of it.
|
|
func (c *Client) TransferNEP11D(acc *wallet.Account, to util.Uint160,
|
|
tokenHash util.Uint160, amount int64, tokenID []byte, data interface{}, gas int64, cosigners []SignerAccount) (util.Uint256, error) {
|
|
from, err := address.StringToUint160(acc.Address)
|
|
if err != nil {
|
|
return util.Uint256{}, fmt.Errorf("bad account address: %w", err)
|
|
}
|
|
tx, err := c.CreateNEP11TransferTx(acc, tokenHash, gas, cosigners, from, to, amount, tokenID, data)
|
|
if err != nil {
|
|
return util.Uint256{}, err
|
|
}
|
|
|
|
return c.SignAndPushTx(tx, acc, cosigners)
|
|
}
|
|
|
|
// NEP11DBalanceOf invokes `balanceOf` divisible NEP-11 method on a
|
|
// specified contract.
|
|
func (c *Client) NEP11DBalanceOf(tokenHash, owner util.Uint160, tokenID []byte) (int64, error) {
|
|
return c.nepBalanceOf(tokenHash, owner, tokenID)
|
|
}
|
|
|
|
// NEP11DOwnerOf returns iterator over the specified NEP-11 divisible token owners. First return value
|
|
// is the session ID, the second one is Iterator itself, the third one is an error. Use TraverseIterator
|
|
// method to traverse iterator values or TerminateSession to terminate opened iterator session. See
|
|
// TraverseIterator and TerminateSession documentation for more details.
|
|
func (c *Client) NEP11DOwnerOf(tokenHash util.Uint160, tokenID []byte) (uuid.UUID, result.Iterator, error) {
|
|
res, err := c.InvokeFunction(tokenHash, "ownerOf", []smartcontract.Parameter{
|
|
{
|
|
Type: smartcontract.ByteArrayType,
|
|
Value: tokenID,
|
|
},
|
|
}, nil)
|
|
sessID := res.Session
|
|
if err != nil {
|
|
return sessID, result.Iterator{}, err
|
|
}
|
|
err = getInvocationError(res)
|
|
if err != nil {
|
|
return sessID, result.Iterator{}, err
|
|
}
|
|
arr, err := topIteratorFromStack(res.Stack)
|
|
return sessID, arr, err
|
|
}
|
|
|
|
// NEP11DUnpackedOwnerOf returns list of the specified NEP-11 divisible token owners. It differs from
|
|
// NEP11DOwnerOf in that no iterator session is used to retrieve values from iterator. Instead,
|
|
// unpacking VM script is created and invoked via `invokescript` JSON-RPC call.
|
|
func (c *Client) NEP11DUnpackedOwnerOf(tokenHash util.Uint160, tokenID []byte) ([]util.Uint160, error) {
|
|
result, err := c.InvokeAndPackIteratorResults(tokenHash, "ownerOf", []smartcontract.Parameter{
|
|
{
|
|
Type: smartcontract.ByteArrayType,
|
|
Value: tokenID,
|
|
},
|
|
}, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = getInvocationError(result)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
arr, err := topIterableFromStack(result.Stack, util.Uint160{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get token IDs from stack: %w", err)
|
|
}
|
|
owners := make([]util.Uint160, len(arr))
|
|
for i := range owners {
|
|
owners[i] = arr[i].(util.Uint160)
|
|
}
|
|
return owners, nil
|
|
}
|
|
|
|
// Divisible NFT methods section end.
|
|
|
|
// Optional NFT methods section start.
|
|
|
|
// NEP11Properties invokes `properties` optional NEP-11 method on the
|
|
// specified contract.
|
|
func (c *Client) NEP11Properties(tokenHash util.Uint160, tokenID []byte) (*stackitem.Map, error) {
|
|
result, err := c.InvokeFunction(tokenHash, "properties", []smartcontract.Parameter{{
|
|
Type: smartcontract.ByteArrayType,
|
|
Value: tokenID,
|
|
}}, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = getInvocationError(result)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return topMapFromStack(result.Stack)
|
|
}
|
|
|
|
// NEP11Tokens returns iterator over the tokens minted by the contract. First return
|
|
// value is the session ID, the second one is Iterator itself, the third one is an
|
|
// error. Use TraverseIterator method to traverse iterator values or
|
|
// TerminateSession to terminate opened iterator session. See TraverseIterator and
|
|
// TerminateSession documentation for more details.
|
|
func (c *Client) NEP11Tokens(tokenHash util.Uint160) (uuid.UUID, result.Iterator, error) {
|
|
res, err := c.InvokeFunction(tokenHash, "tokens", []smartcontract.Parameter{}, nil)
|
|
if err != nil {
|
|
return uuid.UUID{}, result.Iterator{}, err
|
|
}
|
|
err = getInvocationError(res)
|
|
if err != nil {
|
|
return uuid.UUID{}, result.Iterator{}, err
|
|
}
|
|
iter, err := topIteratorFromStack(res.Stack)
|
|
return res.Session, iter, err
|
|
}
|
|
|
|
// NEP11UnpackedTokens returns list of the tokens minted by the contract. It differs from
|
|
// NEP11Tokens in that no iterator session is used to retrieve values from iterator. Instead,
|
|
// unpacking VM script is created and invoked via `invokescript` JSON-RPC call.
|
|
func (c *Client) NEP11UnpackedTokens(tokenHash util.Uint160) ([][]byte, error) {
|
|
result, err := c.InvokeAndPackIteratorResults(tokenHash, "tokens", []smartcontract.Parameter{}, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = getInvocationError(result)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
arr, err := topIterableFromStack(result.Stack, []byte{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get token IDs from stack: %w", err)
|
|
}
|
|
tokens := make([][]byte, len(arr))
|
|
for i := range tokens {
|
|
tokens[i] = arr[i].([]byte)
|
|
}
|
|
return tokens, nil
|
|
}
|
|
|
|
// Optional NFT methods section end.
|