package rpcclient import ( "fmt" "github.com/nspcc-dev/neo-go/pkg/smartcontract" "github.com/nspcc-dev/neo-go/pkg/util" "github.com/nspcc-dev/neo-go/pkg/wallet" ) // nepDecimals invokes `decimals` NEP* method on the specified contract. func (c *Client) nepDecimals(tokenHash util.Uint160) (int64, error) { result, err := c.InvokeFunction(tokenHash, "decimals", []smartcontract.Parameter{}, nil) if err != nil { return 0, err } err = getInvocationError(result) if err != nil { return 0, err } return topIntFromStack(result.Stack) } // nepSymbol invokes `symbol` NEP* method on the specified contract. func (c *Client) nepSymbol(tokenHash util.Uint160) (string, error) { result, err := c.InvokeFunction(tokenHash, "symbol", []smartcontract.Parameter{}, nil) if err != nil { return "", err } err = getInvocationError(result) if err != nil { return "", err } return topStringFromStack(result.Stack) } // nepTotalSupply invokes `totalSupply` NEP* method on the specified contract. func (c *Client) nepTotalSupply(tokenHash util.Uint160) (int64, error) { result, err := c.InvokeFunction(tokenHash, "totalSupply", []smartcontract.Parameter{}, nil) if err != nil { return 0, err } err = getInvocationError(result) if err != nil { return 0, err } return topIntFromStack(result.Stack) } // nepBalanceOf invokes `balanceOf` NEP* method on the specified contract. func (c *Client) nepBalanceOf(tokenHash, acc util.Uint160, tokenID []byte) (int64, error) { params := []smartcontract.Parameter{{ Type: smartcontract.Hash160Type, Value: acc, }} if tokenID != nil { params = append(params, smartcontract.Parameter{ Type: smartcontract.ByteArrayType, Value: tokenID, }) } result, err := c.InvokeFunction(tokenHash, "balanceOf", params, nil) if err != nil { return 0, err } err = getInvocationError(result) if err != nil { return 0, err } return topIntFromStack(result.Stack) } // nepTokenInfo returns full NEP* token info. func (c *Client) nepTokenInfo(tokenHash util.Uint160, standard string) (*wallet.Token, error) { cs, err := c.GetContractStateByHash(tokenHash) if err != nil { return nil, err } var isStandardOK bool for _, st := range cs.Manifest.SupportedStandards { if st == standard { isStandardOK = true break } } if !isStandardOK { return nil, fmt.Errorf("token %s does not support %s standard", tokenHash.StringLE(), standard) } symbol, err := c.nepSymbol(tokenHash) if err != nil { return nil, err } decimals, err := c.nepDecimals(tokenHash) if err != nil { return nil, err } return wallet.NewToken(tokenHash, cs.Manifest.Name, symbol, decimals, standard), nil }