feat(08-wasm): querier plugins implemented (backport #5345) (#5381)

* feat(08-wasm): querier plugins implemented (#5345)

* imp: moved gas to internal

* fix: fixed compiler complaints

* feat: added 'QueryRouter' interface

* imp: passing the queryRouter to keeper

* Revert "fix: fixed compiler complaints"

This reverts commit 208e314.

* Revert "imp: moved gas to internal"

This reverts commit b45b605.

* fix(test): fixed keeper_test.go

* imp: initial querier template

* imp: moved querier to types

* fix: compiler complaints

* imp: removed querier from keeper

* feat: including default querier

* imp: added options

* feat: querier implemented fully

* docs: improved godocs

* imp: improved the querier

* style: improved styling of querier

* fix: fixed options

* fix: fixed options not being passed with config

* style: renamed variables

* imp: added review items

* imp: review items

* imp: set and get query plugins made private

* docs: added more godocs

* fix: default plugin not set

* imp: review items

* docs: added a godoc

---------

Co-authored-by: Carlos Rodriguez <carlos@interchain.io>
(cherry picked from commit e2bcb77)

# Conflicts:
#	modules/light-clients/08-wasm/internal/ibcwasm/wasm.go
#	modules/light-clients/08-wasm/keeper/keeper.go
#	modules/light-clients/08-wasm/keeper/keeper_test.go
#	modules/light-clients/08-wasm/keeper/migrations.go
#	modules/light-clients/08-wasm/testing/simapp/app.go
#	modules/light-clients/08-wasm/types/querier_test.go

* fix conflicts.

* lint issues

---------

Co-authored-by: srdtrk <59252793+srdtrk@users.noreply.github.com>
Co-authored-by: DimitrisJim <d.f.hilliard@gmail.com>

Author: 3 people

modules/light-clients/08-wasm/internal/ibcwasm/expected_interfaces.go

@@ -3,6 +3,9 @@ package ibcwasm
 import (
 	wasmvm "github.com/CosmWasm/wasmvm"
 	wasmvmtypes "github.com/CosmWasm/wasmvm/types"
+
+	"github.com/cosmos/cosmos-sdk/baseapp"
+	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
 var _ WasmEngine = (*wasmvm.VM)(nil)
@@ -115,3 +118,14 @@ type WasmEngine interface {
 	// Unpin is idempotent.
 	Unpin(checksum wasmvm.Checksum) error
 }
+
+type QueryRouter interface {
+	// Route returns the GRPCQueryHandler for a given query route path or nil
+	// if not found
+	Route(path string) baseapp.GRPCQueryHandler
+}
+
+type QueryPluginsI interface {
+	// HandleQuery will route the query to the correct plugin and return the result
+	HandleQuery(ctx sdk.Context, caller string, request wasmvmtypes.QueryRequest) ([]byte, error)
+}

modules/light-clients/08-wasm/internal/ibcwasm/querier.go

@@ -3,13 +3,14 @@ package ibcwasm
 import (
 	"errors"
 
-	wasmvm "github.com/CosmWasm/wasmvm"
 	storetypes "github.com/cosmos/cosmos-sdk/store/types"
 )
 
 var (
-	vm      WasmEngine
-	querier wasmvm.Querier
+	vm WasmEngine
+
+	queryRouter  QueryRouter
+	queryPlugins QueryPluginsI
 
 	// wasmStoreKey stores the storeKey for the 08-wasm module. Using a global storeKey is required since
 	// the client state interface functions do not have access to the keeper.
@@ -40,17 +41,29 @@ func GetWasmStoreKey() storetypes.StoreKey {
 	return wasmStoreKey
 }
 
-// SetQuerier sets the custom wasm query handle for the 08-wasm module.
-// If wasmQuerier is nil a default querier is used that return always an error for any query.
-func SetQuerier(wasmQuerier wasmvm.Querier) {
-	if wasmQuerier == nil {
-		querier = &defaultQuerier{}
-	} else {
-		querier = wasmQuerier
+// SetQueryRouter sets the custom wasm query router for the 08-wasm module.
+// Panics if the queryRouter is nil.
+func SetQueryRouter(router QueryRouter) {
+	if router == nil {
+		panic(errors.New("query router must be not nil"))
+	}
+	queryRouter = router
+}
+
+// GetQueryRouter returns the custom wasm query router for the 08-wasm module.
+func GetQueryRouter() QueryRouter {
+	return queryRouter
+}
+
+// SetQueryPlugins sets the current query plugins
+func SetQueryPlugins(plugins QueryPluginsI) {
+	if plugins == nil {
+		panic(errors.New("query plugins must be not nil"))
 	}
+	queryPlugins = plugins
 }
 
-// GetQuerier returns the custom wasm query handler for the 08-wasm module.
-func GetQuerier() wasmvm.Querier {
-	return querier
+// GetQueryPlugins returns the current query plugins
+func GetQueryPlugins() QueryPluginsI {
+	return queryPlugins
 }

modules/light-clients/08-wasm/internal/ibcwasm/wasm.go

@@ -10,9 +10,9 @@ import (
 	wasmvm "github.com/CosmWasm/wasmvm"
 
 	errorsmod "cosmossdk.io/errors"
-	storetypes "github.com/cosmos/cosmos-sdk/store/types"
 
 	"github.com/cosmos/cosmos-sdk/codec"
+	storetypes "github.com/cosmos/cosmos-sdk/store/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
 	"github.com/cosmos/ibc-go/modules/light-clients/08-wasm/internal/ibcwasm"
@@ -41,7 +41,8 @@ func NewKeeperWithVM(
 	clientKeeper types.ClientKeeper,
 	authority string,
 	vm ibcwasm.WasmEngine,
-	querier wasmvm.Querier,
+	queryRouter ibcwasm.QueryRouter,
+	opts ...Option,
 ) Keeper {
 	if clientKeeper == nil {
 		panic(errors.New("client keeper must be not nil"))
@@ -55,15 +56,24 @@ func NewKeeperWithVM(
 		panic(errors.New("authority must be non-empty"))
 	}
 
-	ibcwasm.SetVM(vm)
-	ibcwasm.SetQuerier(querier)
-	ibcwasm.SetWasmStoreKey(storeKey)
-
-	return Keeper{
+	keeper := &Keeper{
 		cdc:          cdc,
 		clientKeeper: clientKeeper,
 		authority:    authority,
 	}
+
+	// set query plugins to ensure there is a non-nil query plugin
+	// regardless of what options the user provides
+	ibcwasm.SetQueryPlugins(types.NewDefaultQueryPlugins())
+	for _, opt := range opts {
+		opt.apply(keeper)
+	}
+
+	ibcwasm.SetVM(vm)
+	ibcwasm.SetQueryRouter(queryRouter)
+	ibcwasm.SetWasmStoreKey(storeKey)
+
+	return *keeper
 }
 
 // NewKeeperWithConfig creates a new Keeper instance with the provided Wasm configuration.
@@ -75,14 +85,15 @@ func NewKeeperWithConfig(
 	clientKeeper types.ClientKeeper,
 	authority string,
 	wasmConfig types.WasmConfig,
-	querier wasmvm.Querier,
+	queryRouter ibcwasm.QueryRouter,
+	opts ...Option,
 ) Keeper {
 	vm, err := wasmvm.NewVM(wasmConfig.DataDir, wasmConfig.SupportedCapabilities, types.ContractMemoryLimit, wasmConfig.ContractDebugMode, types.MemoryCacheSize)
 	if err != nil {
 		panic(fmt.Errorf("failed to instantiate new Wasm VM instance: %v", err))
 	}
 
-	return NewKeeperWithVM(cdc, storeKey, clientKeeper, authority, vm, querier)
+	return NewKeeperWithVM(cdc, storeKey, clientKeeper, authority, vm, queryRouter, opts...)
 }
 
 // GetAuthority returns the 08-wasm module's authority.

modules/light-clients/08-wasm/keeper/keeper.go

@@ -157,7 +157,7 @@ func (suite *KeeperTestSuite) TestNewKeeper() {
 					GetSimApp(suite.chainA).IBCKeeper.ClientKeeper,
 					GetSimApp(suite.chainA).WasmClientKeeper.GetAuthority(),
 					ibcwasm.GetVM(),
-					nil,
+					GetSimApp(suite.chainA).GRPCQueryRouter(),
 				)
 			},
 			true,
@@ -172,7 +172,7 @@ func (suite *KeeperTestSuite) TestNewKeeper() {
 					GetSimApp(suite.chainA).IBCKeeper.ClientKeeper,
 					"", // authority
 					ibcwasm.GetVM(),
-					nil,
+					GetSimApp(suite.chainA).GRPCQueryRouter(),
 				)
 			},
 			false,
@@ -187,7 +187,7 @@ func (suite *KeeperTestSuite) TestNewKeeper() {
 					nil, // client keeper,
 					GetSimApp(suite.chainA).WasmClientKeeper.GetAuthority(),
 					ibcwasm.GetVM(),
-					nil,
+					GetSimApp(suite.chainA).GRPCQueryRouter(),
 				)
 			},
 			false,
@@ -202,12 +202,27 @@ func (suite *KeeperTestSuite) TestNewKeeper() {
 					GetSimApp(suite.chainA).IBCKeeper.ClientKeeper,
 					GetSimApp(suite.chainA).WasmClientKeeper.GetAuthority(),
 					nil,
-					nil,
+					GetSimApp(suite.chainA).GRPCQueryRouter(),
 				)
 			},
 			false,
 			errors.New("wasm VM must be not nil"),
 		},
+		{
+			"failure: nil query router",
+			func() {
+				keeper.NewKeeperWithVM(
+					GetSimApp(suite.chainA).AppCodec(),
+					GetSimApp(suite.chainA).GetKey(types.StoreKey),
+					GetSimApp(suite.chainA).IBCKeeper.ClientKeeper,
+					GetSimApp(suite.chainA).WasmClientKeeper.GetAuthority(),
+					ibcwasm.GetVM(),
+					nil,
+				)
+			},
+			false,
+			errors.New("query router must be not nil"),
+		},
 	}
 
 	for _, tc := range testCases {

modules/light-clients/08-wasm/keeper/keeper_test.go

@@ -0,0 +1,32 @@
+package keeper
+
+import (
+	"errors"
+
+	"github.com/cosmos/ibc-go/modules/light-clients/08-wasm/internal/ibcwasm"
+	"github.com/cosmos/ibc-go/modules/light-clients/08-wasm/types"
+)
+
+// Option is an extension point to instantiate keeper with non default values
+type Option interface {
+	apply(*Keeper)
+}
+
+type optsFn func(*Keeper)
+
+func (f optsFn) apply(keeper *Keeper) {
+	f(keeper)
+}
+
+// WithQueryPlugins is an optional constructor parameter to pass custom query plugins for wasmVM requests.
+// Missing fields will be filled with default queriers.
+func WithQueryPlugins(plugins *types.QueryPlugins) Option {
+	return optsFn(func(_ *Keeper) {
+		currentPlugins, ok := ibcwasm.GetQueryPlugins().(*types.QueryPlugins)
+		if !ok {
+			panic(errors.New("invalid query plugins type"))
+		}
+		newPlugins := currentPlugins.Merge(plugins)
+		ibcwasm.SetQueryPlugins(&newPlugins)
+	})
+}