diff --git a/demo/app/app.go b/demo/app/app.go index 156b232..6ba845d 100644 --- a/demo/app/app.go +++ b/demo/app/app.go @@ -12,6 +12,7 @@ import ( wasmkeeper "github.com/CosmWasm/wasmd/x/wasm/keeper" wasmtypes "github.com/CosmWasm/wasmd/x/wasm/types" appparams "github.com/babylonlabs-io/babylon-sdk/demo/app/params" + appwasm "github.com/babylonlabs-io/babylon-sdk/demo/app/wasm" abci "github.com/cometbft/cometbft/abci/types" tmos "github.com/cometbft/cometbft/libs/os" cmtproto "github.com/cometbft/cometbft/proto/tendermint/types" @@ -492,6 +493,9 @@ func NewConsumerApp( // add support for the custom queries wasmkeeper.WithQueryHandlerDecorator(bbnkeeper.NewQueryDecorator(app.BabylonKeeper)), ) + // Add grpc query support for the whitelisted grpc queries + wasmOpts = append(wasmOpts, appwasm.RegisterGrpcQueries(bApp, appCodec)...) + // The last arguments can contain custom message handlers, and custom query handlers, // if we want to allow any custom callbacks availableCapabilities := AllCapabilities() diff --git a/demo/app/wasm/grpc_whitelist.go b/demo/app/wasm/grpc_whitelist.go new file mode 100644 index 0000000..9b4b342 --- /dev/null +++ b/demo/app/wasm/grpc_whitelist.go @@ -0,0 +1,29 @@ +package wasm + +import ( + wasmkeeper "github.com/CosmWasm/wasmd/x/wasm/keeper" + "github.com/cosmos/cosmos-sdk/baseapp" + "github.com/cosmos/cosmos-sdk/codec" + minttypes "github.com/cosmos/cosmos-sdk/x/mint/types" +) + +func RegisterGrpcQueries(bApp *baseapp.BaseApp, appCodec codec.Codec) []wasmkeeper.Option { + queryRouter := bApp.GRPCQueryRouter() + queryPluginOpt := wasmkeeper.WithQueryPlugins( + &wasmkeeper.QueryPlugins{ + Stargate: wasmkeeper.AcceptListStargateQuerier(WhitelistedGrpcQuery(), queryRouter, appCodec), + Grpc: wasmkeeper.AcceptListGrpcQuerier(WhitelistedGrpcQuery(), queryRouter, appCodec), + }) + + return []wasmkeeper.Option{ + queryPluginOpt, + } +} + +// WhitelistedGrpcQuery returns the whitelisted Grpc queries +func WhitelistedGrpcQuery() wasmkeeper.AcceptedQueries { + return wasmkeeper.AcceptedQueries{ + // mint + "/cosmos.mint.v1beta1.Query/Params": &minttypes.QueryParamsResponse{}, + } +}