77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
package types
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
grpc "google.golang.org/grpc"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
grpctypes "github.com/cosmos/cosmos-sdk/types/grpc"
|
|
)
|
|
|
|
// AccountRetriever defines the properties of a type that can be used to
|
|
// retrieve accounts.
|
|
type AccountRetriever struct{}
|
|
|
|
// GetAccount queries for an account given an address and a block height. An
|
|
// error is returned if the query or decoding fails.
|
|
func (ar AccountRetriever) GetAccount(clientCtx client.Context, addr sdk.AccAddress) (AccountI, error) {
|
|
account, _, err := ar.GetAccountWithHeight(clientCtx, addr)
|
|
return account, err
|
|
}
|
|
|
|
// GetAccountWithHeight queries for an account given an address. Returns the
|
|
// height of the query with the account. An error is returned if the query
|
|
// or decoding fails.
|
|
//nolint:interfacer
|
|
func (ar AccountRetriever) GetAccountWithHeight(clientCtx client.Context, addr sdk.AccAddress) (AccountI, int64, error) {
|
|
var header metadata.MD
|
|
|
|
queryClient := NewQueryClient(clientCtx)
|
|
res, err := queryClient.Account(context.Background(), &QueryAccountRequest{Address: addr.String()}, grpc.Header(&header))
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
blockHeight := header.Get(grpctypes.GRPCBlockHeightHeader)
|
|
if l := len(blockHeight); l != 1 {
|
|
return nil, 0, fmt.Errorf("unexpected '%s' header length; got %d, expected: %d", grpctypes.GRPCBlockHeightHeader, l, 1)
|
|
}
|
|
|
|
nBlockHeight, err := strconv.Atoi(blockHeight[0])
|
|
if err != nil {
|
|
return nil, 0, fmt.Errorf("failed to parse block height: %w", err)
|
|
}
|
|
|
|
var acc AccountI
|
|
if err := clientCtx.InterfaceRegistry.UnpackAny(res.Account, &acc); err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return acc, int64(nBlockHeight), nil
|
|
}
|
|
|
|
// EnsureExists returns an error if no account exists for the given address else nil.
|
|
func (ar AccountRetriever) EnsureExists(clientCtx client.Context, addr sdk.AccAddress) error {
|
|
if _, err := ar.GetAccount(clientCtx, addr); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetAccountNumberSequence returns sequence and account number for the given address.
|
|
// It returns an error if the account couldn't be retrieved from the state.
|
|
func (ar AccountRetriever) GetAccountNumberSequence(clientCtx client.Context, addr sdk.AccAddress) (uint64, uint64, error) {
|
|
acc, err := ar.GetAccount(clientCtx, addr)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
|
|
return acc.GetAccountNumber(), acc.GetSequence(), nil
|
|
}
|