e1f478da41
Co-authored-by: unknown unknown <unknown@unknown> Co-authored-by: Marko <marbar3778@yahoo.com> Co-authored-by: Facundo Medica <14063057+facundomedica@users.noreply.github.com>
111 lines
3.1 KiB
Go
111 lines
3.1 KiB
Go
package accounts
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"cosmossdk.io/x/accounts/internal/implementation"
|
|
v1 "cosmossdk.io/x/accounts/v1"
|
|
)
|
|
|
|
var _ v1.QueryServer = queryServer{}
|
|
|
|
func NewQueryServer(k Keeper) v1.QueryServer {
|
|
return &queryServer{k}
|
|
}
|
|
|
|
type queryServer struct {
|
|
k Keeper
|
|
}
|
|
|
|
func (q queryServer) AccountQuery(ctx context.Context, request *v1.AccountQueryRequest) (*v1.AccountQueryResponse, error) {
|
|
// get target addr
|
|
targetAddr, err := q.k.addressCodec.StringToBytes(request.Target)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// decode req into boxed concrete type
|
|
queryReq, err := implementation.UnpackAnyRaw(request.Request)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// run query
|
|
resp, err := q.k.Query(ctx, targetAddr, queryReq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// encode response
|
|
respAny, err := implementation.PackAny(resp)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &v1.AccountQueryResponse{
|
|
Response: respAny,
|
|
}, nil
|
|
}
|
|
|
|
func (q queryServer) Schema(_ context.Context, request *v1.SchemaRequest) (*v1.SchemaResponse, error) {
|
|
// TODO: maybe we should cache this, considering accounts types are not
|
|
// added on the fly as the chain is running.
|
|
schemas := v1.MakeAccountsSchemas(q.k.accounts)
|
|
schema, ok := schemas[request.AccountType]
|
|
if !ok {
|
|
return nil, fmt.Errorf("%w: %s", errAccountTypeNotFound, request.AccountType)
|
|
}
|
|
return schema, nil
|
|
}
|
|
|
|
func (q queryServer) AccountType(ctx context.Context, request *v1.AccountTypeRequest) (*v1.AccountTypeResponse, error) {
|
|
addr, err := q.k.addressCodec.StringToBytes(request.Address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
accType, err := q.k.AccountsByType.Get(ctx, addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &v1.AccountTypeResponse{
|
|
AccountType: accType,
|
|
}, nil
|
|
}
|
|
|
|
func (q queryServer) AccountNumber(ctx context.Context, request *v1.AccountNumberRequest) (*v1.AccountNumberResponse, error) {
|
|
addr, err := q.k.addressCodec.StringToBytes(request.Address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
number, err := q.k.AccountByNumber.Get(ctx, addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &v1.AccountNumberResponse{Number: number}, nil
|
|
}
|
|
|
|
const (
|
|
// TODO(tip): evaluate if the following numbers should be parametrised over state, or over the node.
|
|
SimulateAuthenticateGasLimit = 1_000_000
|
|
SimulateBundlerPaymentGasLimit = SimulateAuthenticateGasLimit
|
|
ExecuteGasLimit = SimulateAuthenticateGasLimit
|
|
)
|
|
|
|
func (q queryServer) SimulateUserOperation(ctx context.Context, request *v1.SimulateUserOperationRequest) (*v1.SimulateUserOperationResponse, error) {
|
|
_, err := q.k.addressCodec.StringToBytes(request.Bundler)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if request.UserOperation == nil {
|
|
return nil, fmt.Errorf("nil user operation")
|
|
}
|
|
|
|
request.UserOperation.AuthenticationGasLimit = SimulateAuthenticateGasLimit
|
|
request.UserOperation.BundlerPaymentGasLimit = SimulateBundlerPaymentGasLimit
|
|
request.UserOperation.ExecutionGasLimit = ExecuteGasLimit
|
|
|
|
resp := q.k.ExecuteUserOperation(ctx, request.Bundler, request.UserOperation)
|
|
return &v1.SimulateUserOperationResponse{UserOperationResponse: resp}, nil
|
|
}
|