json_rpc.go

package jsonrpc

import (
	"context"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"strconv"

	"github.com/btcsuite/btcutil/bech32"
	codectypes "github.com/cosmos/cosmos-sdk/codec/types"
	cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/cosmos/cosmos-sdk/types/tx"
	"github.com/cosmos/cosmos-sdk/types/tx/signing"
	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
	minttypes "github.com/cosmos/cosmos-sdk/x/mint/types"
	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
	"github.com/gogo/protobuf/proto"
	"github.com/pkg/errors"
	tmbytes "github.com/tendermint/tendermint/libs/bytes"
	ctypes "github.com/tendermint/tendermint/rpc/core/types"
	"github.com/tendermint/tendermint/types"
)

const DefGasLimit int64 = 200000

type jsonRPCCaller interface {
	Call(ctx context.Context, method string, params map[string]interface{}, result interface{}) (err error)
}

type NodeRPC struct {
	chainId   string
	gasPrices sdk.Coins
	ctx       context.Context
	caller    jsonRPCCaller
	height    int64
}

func NewNodeRPC(caller jsonRPCCaller) *NodeRPC {
	return &NodeRPC{caller: caller, ctx: context.Background(), height: 0}
}

func (c *NodeRPC) WithContext(ctx context.Context) *NodeRPC {
	c.ctx = ctx
	return c
}

func (c *NodeRPC) WithGasPrices(gasPrices sdk.Coins) *NodeRPC {
	c.gasPrices = gasPrices
	return c
}

func (c *NodeRPC) WithHeight(height int64) *NodeRPC {
	c.height = height
	return c
}

// Custom API

func (c *NodeRPC) GetChainId() (chain string, err error) {
	res, err := c.Genesis()
	if err != nil {
		return
	}
	return res.Genesis.ChainID, nil
}

func (c *NodeRPC) GetBlockHeight() (int64, error) {
	status, err := c.Status()
	if err != nil {
		return 0, err
	}
	if status.SyncInfo.CatchingUp {
		return 0, errors.New("the node is catching up with the new block data")
	}
	return status.SyncInfo.LatestBlockHeight, nil
}

func (c *NodeRPC) GetMintDenom() (denom string, err error) {
	genesis, err := c.Genesis()
	if err != nil {
		return
	}

	var appState map[string]json.RawMessage
	if err = json.Unmarshal(genesis.Genesis.AppState, &appState); err != nil {
		return denom, err
	}

	var mintGenesis struct {
		Params struct {
			MintDenom string `json:"mint_denom"`
		} `json:"params"`
	}
	return mintGenesis.Params.MintDenom, json.Unmarshal(appState[minttypes.ModuleName], &mintGenesis)
}

func (c *NodeRPC) GetAddressPrefix() (prefix string, err error) {
	genesis, err := c.Genesis()
	if err != nil {
		return
	}
	var appState map[string]json.RawMessage
	if err = json.Unmarshal(genesis.Genesis.AppState, &appState); err != nil {
		return
	}

	var authGen struct {
		Accounts []struct {
			Address string `json:"address"`
		} `json:"accounts"`
	}
	if err = json.Unmarshal(appState[authtypes.ModuleName], &authGen); err != nil {
		return
	}
	for _, account := range authGen.Accounts {
		prefix, _, err := bech32.Decode(account.Address)
		return prefix, err
	}
	return sdk.Bech32MainPrefix, nil
}

func (c *NodeRPC) GetStakeValidators(status stakingtypes.BondStatus) (stakingtypes.Validators, error) {
	data, err := json.Marshal(map[string]string{"Page": "1", "Limit": "200", "Status": status.String()})
	if err != nil {
		return nil, err
	}
	result, err := c.ABCIQueryIsOk("/custom/staking/validators", data)
	if err != nil {
		return nil, err
	}
	validators := make(stakingtypes.Validators, 0)
	if err := json.Unmarshal(result.Response.Value, &validators); err != nil {
		return nil, err
	}
	return validators, err
}

func (c *NodeRPC) BuildTx(privKey cryptotypes.PrivKey, msgs []sdk.Msg) (*tx.TxRaw, error) {
	account, err := c.QueryAccount(sdk.AccAddress(privKey.PubKey().Address().Bytes()).String())
	if err != nil {
		return nil, err
	}
	if len(c.chainId) <= 0 {
		chainId, err := c.GetChainId()
		if err != nil {
			return nil, err
		}
		c.chainId = chainId
	}

	txBodyMessage := make([]*codectypes.Any, 0)
	for i := 0; i < len(msgs); i++ {
		msgAnyValue, err := codectypes.NewAnyWithValue(msgs[i])
		if err != nil {
			return nil, err
		}
		txBodyMessage = append(txBodyMessage, msgAnyValue)
	}

	txBody := &tx.TxBody{
		Messages:                    txBodyMessage,
		Memo:                        "",
		TimeoutHeight:               0,
		ExtensionOptions:            nil,
		NonCriticalExtensionOptions: nil,
	}
	txBodyBytes, err := proto.Marshal(txBody)
	if err != nil {
		return nil, err
	}

	pubAny, err := codectypes.NewAnyWithValue(privKey.PubKey())
	if err != nil {
		return nil, err
	}

	var gasPrice sdk.Coin
	if len(c.gasPrices) <= 0 {
		gasPrices, err := c.GetGasPrices()
		if err != nil {
			return nil, err
		}
		if len(gasPrices) > 0 {
			gasPrice = gasPrices[0]
		}
	} else {
		gasPrice = c.gasPrices[0]
	}

	authInfo := &tx.AuthInfo{
		SignerInfos: []*tx.SignerInfo{
			{
				PublicKey: pubAny,
				ModeInfo: &tx.ModeInfo{
					Sum: &tx.ModeInfo_Single_{
						Single: &tx.ModeInfo_Single{Mode: signing.SignMode_SIGN_MODE_DIRECT},
					},
				},
				Sequence: account.GetSequence(),
			},
		},
		Fee: &tx.Fee{
			Amount:   sdk.NewCoins(sdk.NewCoin(gasPrice.Denom, gasPrice.Amount.MulRaw(DefGasLimit))),
			GasLimit: uint64(DefGasLimit),
			Payer:    "",
			Granter:  "",
		},
	}

	txAuthInfoBytes, err := proto.Marshal(authInfo)
	if err != nil {
		return nil, err
	}
	signDoc := &tx.SignDoc{
		BodyBytes:     txBodyBytes,
		AuthInfoBytes: txAuthInfoBytes,
		ChainId:       c.chainId,
		AccountNumber: account.GetAccountNumber(),
	}
	signatures, err := proto.Marshal(signDoc)
	if err != nil {
		return nil, err
	}
	sign, err := privKey.Sign(signatures)
	if err != nil {
		return nil, err
	}
	gasInfo, err := c.EstimatingGas(&tx.TxRaw{
		BodyBytes:     txBodyBytes,
		AuthInfoBytes: signDoc.AuthInfoBytes,
		Signatures:    [][]byte{sign},
	})
	if err != nil {
		return nil, err
	}

	authInfo.Fee.GasLimit = gasInfo.GasUsed * 12 / 10
	authInfo.Fee.Amount = sdk.NewCoins(sdk.NewCoin(gasPrice.Denom, gasPrice.Amount.MulRaw(int64(authInfo.Fee.GasLimit))))

	signDoc.AuthInfoBytes, err = proto.Marshal(authInfo)
	if err != nil {
		return nil, err
	}
	signatures, err = proto.Marshal(signDoc)
	if err != nil {
		return nil, err
	}
	sign, err = privKey.Sign(signatures)
	if err != nil {
		return nil, err
	}
	return &tx.TxRaw{
		BodyBytes:     txBodyBytes,
		AuthInfoBytes: signDoc.AuthInfoBytes,
		Signatures:    [][]byte{sign},
	}, nil
}

func (c *NodeRPC) BroadcastTx(txRaw *tx.TxRaw, mode ...tx.BroadcastMode) (*sdk.TxResponse, error) {
	txBytes, err := proto.Marshal(txRaw)
	if err != nil {
		return nil, err
	}
	defaultMode := tx.BroadcastMode_BROADCAST_MODE_BLOCK
	if len(mode) > 0 {
		defaultMode = mode[0]
	}
	switch defaultMode {
	case tx.BroadcastMode_BROADCAST_MODE_SYNC:
		res, err := c.BroadcastTxSync(txBytes)
		if err != nil {
			return nil, err
		}
		return sdk.NewResponseFormatBroadcastTx(res), nil
	case tx.BroadcastMode_BROADCAST_MODE_ASYNC:
		res, err := c.BroadcastTxAsync(txBytes)
		if err != nil {
			return nil, err
		}
		return sdk.NewResponseFormatBroadcastTx(res), nil
	case tx.BroadcastMode_BROADCAST_MODE_BLOCK:
		res, err := c.BroadcastTxCommit(txBytes)
		if err != nil {
			return nil, err
		}
		return sdk.NewResponseFormatBroadcastTxCommit(res), nil
	default:
		return nil, fmt.Errorf("unsupported return type %s; supported types: sync, async, block", defaultMode)
	}
}

func (c *NodeRPC) BroadcastTxRawCommit(txRaw *tx.TxRaw) (*ctypes.ResultBroadcastTxCommit, error) {
	txBytes, err := proto.Marshal(txRaw)
	if err != nil {
		return nil, err
	}
	return c.BroadcastTxCommit(txBytes)
}

func (c *NodeRPC) TxByHash(txHash string) (*sdk.TxResponse, error) {
	hash, err := hex.DecodeString(txHash)
	if err != nil {
		return nil, err
	}
	resultTx, err := c.Tx(hash)
	if err != nil {
		return nil, err
	}
	return sdk.NewResponseResultTx(resultTx, nil, ""), nil
}

func (c *NodeRPC) EstimatingGas(raw *tx.TxRaw) (*sdk.GasInfo, error) {
	txBytes, err := proto.Marshal(raw)
	if err != nil {
		return nil, err
	}
	result, err := c.ABCIQueryIsOk("/app/simulate", txBytes)
	if err != nil {
		return nil, err
	}
	resp := struct {
		GasInfo struct {
			GasWanted string `json:"gas_wanted"`
			GasUsed   string `json:"gas_used"`
		} `json:"gas_info"`
	}{}
	if err = json.Unmarshal(result.Response.Value, &resp); err != nil {
		return nil, err
	}
	gasWanted, err := strconv.ParseUint(resp.GasInfo.GasWanted, 10, 64)
	if err != nil && len(resp.GasInfo.GasWanted) > 0 {
		return nil, err
	}
	gasUsed, err := strconv.ParseUint(resp.GasInfo.GasUsed, 10, 64)
	if err != nil && len(resp.GasInfo.GasUsed) > 0 {
		return nil, err
	}
	return &sdk.GasInfo{
		GasWanted: gasWanted,
		GasUsed:   gasUsed,
	}, nil
}

func (c *NodeRPC) AppVersion() (string, error) {
	result, err := c.ABCIQueryIsOk("/app/version", nil)
	if err != nil {
		return "", err
	}
	return string(result.Response.Value), nil
}

func (c *NodeRPC) ABCIQueryIsOk(path string, data tmbytes.HexBytes) (*ctypes.ResultABCIQuery, error) {
	result := new(ctypes.ResultABCIQuery)
	params := map[string]interface{}{"path": path, "data": data, "height": strconv.FormatInt(c.height, 10), "prove": false}
	err := c.caller.Call(c.ctx, "abci_query", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "ABCIQueryIsOk")
	}
	if result.Response.Code != 0 {
		return nil, fmt.Errorf("abci query response, space: %s, code: %d, log: %s",
			result.Response.Codespace, result.Response.Code, result.Response.Log)
	}
	return result, nil
}

// Tendermint API

func (c *NodeRPC) Status() (*ctypes.ResultStatus, error) {
	result := new(ctypes.ResultStatus)
	err := c.caller.Call(c.ctx, "status", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "Status")
	}
	return result, nil
}

func (c *NodeRPC) ABCIInfo() (*ctypes.ResultABCIInfo, error) {
	result := new(ctypes.ResultABCIInfo)
	err := c.caller.Call(c.ctx, "abci_info", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "ABCIInfo")
	}
	return result, nil
}

func (c *NodeRPC) ABCIQuery(path string, data tmbytes.HexBytes) (*ctypes.ResultABCIQuery, error) {
	result := new(ctypes.ResultABCIQuery)
	params := map[string]interface{}{"path": path, "data": data, "height": strconv.FormatInt(c.height, 10), "prove": false}
	err := c.caller.Call(c.ctx, "abci_query", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "ABCIQuery")
	}
	return result, nil
}

func (c *NodeRPC) BroadcastTxCommit(tx types.Tx) (*ctypes.ResultBroadcastTxCommit, error) {
	result := new(ctypes.ResultBroadcastTxCommit)
	err := c.caller.Call(c.ctx, "broadcast_tx_commit", map[string]interface{}{"tx": tx}, result)
	if err != nil {
		return nil, errors.Wrap(err, "broadcast_tx_commit")
	}
	return result, nil
}

func (c *NodeRPC) BroadcastTxAsync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
	return c.broadcastTX("broadcast_tx_async", tx)
}

func (c *NodeRPC) BroadcastTxSync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
	return c.broadcastTX("broadcast_tx_sync", tx)
}

func (c *NodeRPC) broadcastTX(route string, tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
	result := new(ctypes.ResultBroadcastTx)
	err := c.caller.Call(c.ctx, route, map[string]interface{}{"tx": tx}, result)
	if err != nil {
		return nil, errors.Wrap(err, route)
	}
	return result, nil
}

func (c *NodeRPC) UnconfirmedTxs(limit int) (*ctypes.ResultUnconfirmedTxs, error) {
	result := new(ctypes.ResultUnconfirmedTxs)
	params := map[string]interface{}{"limit": strconv.Itoa(limit)}
	err := c.caller.Call(c.ctx, "unconfirmed_txs", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "unconfirmed_txs")
	}
	if len(result.Txs) <= 0 {
		result.Txs = make([]types.Tx, 0)
	}
	return result, nil
}

func (c *NodeRPC) NumUnconfirmedTxs() (*ctypes.ResultUnconfirmedTxs, error) {
	result := new(ctypes.ResultUnconfirmedTxs)
	err := c.caller.Call(c.ctx, "num_unconfirmed_txs", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "num_unconfirmed_txs")
	}
	return result, nil
}

func (c *NodeRPC) NetInfo() (*ctypes.ResultNetInfo, error) {
	result := new(ctypes.ResultNetInfo)
	err := c.caller.Call(c.ctx, "net_info", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "NetInfo")
	}
	if len(result.Peers) <= 0 {
		result.Peers = make([]ctypes.Peer, 0)
	}
	return result, nil
}

func (c *NodeRPC) DumpConsensusState() (*ctypes.ResultDumpConsensusState, error) {
	result := new(ctypes.ResultDumpConsensusState)
	err := c.caller.Call(c.ctx, "dump_consensus_state", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "DumpConsensusState")
	}
	if len(result.Peers) <= 0 {
		result.Peers = make([]ctypes.PeerStateInfo, 0)
	}
	return result, nil
}

func (c *NodeRPC) ConsensusState() (*ctypes.ResultConsensusState, error) {
	result := new(ctypes.ResultConsensusState)
	err := c.caller.Call(c.ctx, "consensus_state", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "ConsensusState")
	}
	return result, nil
}

func (c *NodeRPC) ConsensusParams(height int64) (*ctypes.ResultConsensusParams, error) {
	result := new(ctypes.ResultConsensusParams)
	params := map[string]interface{}{"height": strconv.FormatInt(height, 10)}
	if height <= 0 {
		params = map[string]interface{}{}
	}
	err := c.caller.Call(c.ctx, "consensus_params", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "ConsensusParams")
	}
	return result, nil
}

func (c *NodeRPC) Health() (*ctypes.ResultHealth, error) {
	result := new(ctypes.ResultHealth)
	err := c.caller.Call(c.ctx, "health", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "Health")
	}
	return result, nil
}

func (c *NodeRPC) BlockchainInfo(minHeight, maxHeight int64) (*ctypes.ResultBlockchainInfo, error) {
	result := new(ctypes.ResultBlockchainInfo)
	params := map[string]interface{}{
		"minHeight": strconv.FormatInt(minHeight, 10),
		"maxHeight": strconv.FormatInt(maxHeight, 10),
	}
	err := c.caller.Call(c.ctx, "blockchain", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "BlockchainInfo")
	}
	return result, nil
}

func (c *NodeRPC) Genesis() (*ctypes.ResultGenesis, error) {
	result := new(ctypes.ResultGenesis)
	err := c.caller.Call(c.ctx, "genesis", map[string]interface{}{}, result)
	if err != nil {
		return nil, errors.Wrap(err, "Genesis")
	}
	return result, nil
}

func (c *NodeRPC) Block(height int64) (*ctypes.ResultBlock, error) {
	result := new(ctypes.ResultBlock)
	params := map[string]interface{}{}
	if height > 0 {
		params = map[string]interface{}{"height": strconv.FormatInt(height, 10)}
	}
	err := c.caller.Call(c.ctx, "block", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "Block")
	}
	return result, nil
}

func (c *NodeRPC) BlockResults(height int64) (*ctypes.ResultBlockResults, error) {
	result := new(ctypes.ResultBlockResults)
	params := map[string]interface{}{}
	if height > 0 {
		params = map[string]interface{}{"height": strconv.FormatInt(height, 10)}
	}
	err := c.caller.Call(c.ctx, "block_results", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "Block Result")
	}
	return result, nil
}

func (c *NodeRPC) Commit(height int64) (*ctypes.ResultCommit, error) {
	result := new(ctypes.ResultCommit)
	params := map[string]interface{}{}
	if height > 0 {
		params = map[string]interface{}{"height": height}
	}
	err := c.caller.Call(c.ctx, "commit", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "Commit")
	}
	return result, nil
}

func (c *NodeRPC) Validators(height int64, page, perPage int) (*ctypes.ResultValidators, error) {
	result := new(ctypes.ResultValidators)
	params := map[string]interface{}{}
	if height > 0 {
		params = map[string]interface{}{"height": height, "page": page, "per_page": perPage}
	}
	err := c.caller.Call(c.ctx, "validators", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "Validators")
	}
	return result, nil
}

func (c *NodeRPC) Tx(hash []byte) (*ctypes.ResultTx, error) {
	result := new(ctypes.ResultTx)
	params := map[string]interface{}{"hash": hash, "prove": false}
	err := c.caller.Call(c.ctx, "tx", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "Tx")
	}
	return result, nil
}

func (c *NodeRPC) TxSearch(query string, page, perPage int, orderBy string) (
	*ctypes.ResultTxSearch, error,
) {
	result := new(ctypes.ResultTxSearch)
	params := map[string]interface{}{"query": query, "prove": false, "page": strconv.Itoa(page), "per_page": strconv.Itoa(perPage), "order_by": orderBy}
	err := c.caller.Call(c.ctx, "tx_search", params, result)
	if err != nil {
		return nil, errors.Wrap(err, "TxSearch")
	}
	return result, nil
}

func (c *NodeRPC) BroadcastEvidence(ev types.Evidence) (*ctypes.ResultBroadcastEvidence, error) {
	result := new(ctypes.ResultBroadcastEvidence)
	err := c.caller.Call(c.ctx, "broadcast_evidence", map[string]interface{}{"evidence": ev}, result)
	if err != nil {
		return nil, errors.Wrap(err, "BroadcastEvidence")
	}
	return result, nil
}