76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cosmossdk.io/x/circuit/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/types/query"
|
|
)
|
|
|
|
var _ types.QueryServer = QueryServer{}
|
|
|
|
type QueryServer struct {
|
|
keeper Keeper
|
|
}
|
|
|
|
// NewQueryServer returns an implementation of the circuit QueryServer interface
|
|
// for the provided Keeper.
|
|
func NewQueryServer(keeper Keeper) types.QueryServer {
|
|
return &QueryServer{keeper: keeper}
|
|
}
|
|
|
|
// Account returns account permissions.
|
|
func (qs QueryServer) Account(ctx context.Context, req *types.QueryAccountRequest) (*types.AccountResponse, error) {
|
|
add, err := qs.keeper.addressCodec.StringToBytes(req.Address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
perms, err := qs.keeper.Permissions.Get(ctx, add)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &types.AccountResponse{Permission: &perms}, nil
|
|
}
|
|
|
|
// Accounts returns multiple accounts permissions.
|
|
func (qs QueryServer) Accounts(ctx context.Context, req *types.QueryAccountsRequest) (*types.AccountsResponse, error) {
|
|
results, pageRes, err := query.CollectionPaginate(
|
|
ctx,
|
|
qs.keeper.Permissions,
|
|
req.Pagination,
|
|
func(key []byte, value types.Permissions) (*types.GenesisAccountPermissions, error) {
|
|
addrStr, err := qs.keeper.addressCodec.BytesToString(key)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &types.GenesisAccountPermissions{
|
|
Address: addrStr,
|
|
Permissions: &value,
|
|
}, nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &types.AccountsResponse{Accounts: results, Pagination: pageRes}, nil
|
|
}
|
|
|
|
// DisabledList returns a list of disabled message urls
|
|
func (qs QueryServer) DisabledList(ctx context.Context, req *types.QueryDisabledListRequest) (*types.DisabledListResponse, error) {
|
|
// Iterate over disabled list and perform the callback
|
|
var msgs []string
|
|
err := qs.keeper.DisableList.Walk(ctx, nil, func(msgUrl string) (bool, error) {
|
|
msgs = append(msgs, msgUrl)
|
|
return false, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &types.DisabledListResponse{DisabledList: msgs}, nil
|
|
}
|