batch claim wasmbinding addition

wasmbinding/bindings/msg.go

@@ -1,9 +1,11 @@
 package bindings
 
 import (
+	batch "github.com/Team-Kujira/core/x/batch/wasm"
 	denom "github.com/Team-Kujira/core/x/denom/wasm"
 )
 
 type CosmosMsg struct {
 	Denom *denom.DenomMsg
+	Batch *batch.BatchMsg
 }

wasmbinding/message_plugin.go

@@ -11,9 +11,10 @@ import (
 
 	"github.com/Team-Kujira/core/wasmbinding/bindings"
 
-	denom "github.com/Team-Kujira/core/x/denom/wasm"
-
+	batchkeeper "github.com/Team-Kujira/core/x/batch/keeper"
+	batch "github.com/Team-Kujira/core/x/batch/wasm"
 	denomkeeper "github.com/Team-Kujira/core/x/denom/keeper"
+	denom "github.com/Team-Kujira/core/x/denom/wasm"
 )
 
 // CustomMessageDecorator returns decorator for custom CosmWasm bindings messages
@@ -34,6 +35,7 @@ type CustomMessenger struct {
 	wrapped wasmkeeper.Messenger
 	bank    bankkeeper.Keeper
 	denom   denomkeeper.Keeper
+	batch   batchkeeper.Keeper
 }
 
 var _ wasmkeeper.Messenger = (*CustomMessenger)(nil)
@@ -57,6 +59,10 @@ func (m *CustomMessenger) DispatchMsg(
 			return denom.HandleMsg(m.denom, m.bank, contractAddr, ctx, contractMsg.Denom)
 		}
 
+		if contractMsg.Batch != nil {
+			return batch.HandleMsg(m.batch, contractAddr, ctx, contractMsg.Batch)
+		}
+
 		return nil, nil, wasmvmtypes.UnsupportedRequest{Kind: "unknown Custom variant"}
 	}
 	return m.wrapped.DispatchMsg(ctx, contractAddr, contractIBCPortID, msg)

x/batch/wasm/interface_msg.go

@@ -0,0 +1,57 @@
+package wasm
+
+import (
+	"cosmossdk.io/errors"
+	wasmvmtypes "github.com/CosmWasm/wasmvm/types"
+	batchkeeper "github.com/Team-Kujira/core/x/batch/keeper"
+	batchtypes "github.com/Team-Kujira/core/x/batch/types"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+)
+
+type BatchMsg struct {
+	// Contracts can withdraw all rewards for the delegations from the contract.
+	WithdrawAllDelegatorRewards *WithdrawAllDelegatorRewards `json:"withdrawAllDelegatorRewards,omitempty"`
+}
+
+type WithdrawAllDelegatorRewards struct{}
+
+// withdrawAllDelegatorRewards withdraw all delegation rewards for the delegations from the contract address
+func withdrawAllDelegatorRewards(ctx sdk.Context, contractAddr sdk.AccAddress, withdrawAllDelegatorRewards *WithdrawAllDelegatorRewards, bk batchkeeper.Keeper) ([]sdk.Event, [][]byte, error) {
+	err := PerformWithdrawAllDelegatorRewards(bk, ctx, contractAddr, withdrawAllDelegatorRewards)
+	if err != nil {
+		return nil, nil, errors.Wrap(err, "perform withdrawAllDelegatorRewards denom")
+	}
+	return nil, nil, nil
+}
+
+// PerformWithdrawAllDelegatorRewards is used to perform delegation rewards from the contract delegations
+func PerformWithdrawAllDelegatorRewards(bk batchkeeper.Keeper, ctx sdk.Context, contractAddr sdk.AccAddress, withdrawAllDelegatorRewards *WithdrawAllDelegatorRewards) error {
+	if withdrawAllDelegatorRewards == nil {
+		return wasmvmtypes.InvalidRequest{Err: "withdrawAllDelegatorRewards denom null withdrawAllDelegatorRewards denom"}
+	}
+
+	msgServer := batchkeeper.NewMsgServerImpl(bk)
+	msgWithdrawAllDelegatorRewards := batchtypes.NewMsgWithdrawAllDelegatorRewards(contractAddr)
+
+	if err := msgWithdrawAllDelegatorRewards.ValidateBasic(); err != nil {
+		return errors.Wrap(err, "failed validating MsgWithdrawAllDelegatorRewards")
+	}
+
+	_, err := msgServer.WithdrawAllDelegatorRewards(
+		sdk.WrapSDKContext(ctx),
+		msgWithdrawAllDelegatorRewards,
+	)
+	if err != nil {
+		return errors.Wrap(err, "creating denom")
+	}
+	return nil
+}
+
+// QueryCustom implements custom msg interface
+func HandleMsg(dk batchkeeper.Keeper, contractAddr sdk.AccAddress, ctx sdk.Context, q *BatchMsg) ([]sdk.Event, [][]byte, error) {
+	if q.WithdrawAllDelegatorRewards != nil {
+		return withdrawAllDelegatorRewards(ctx, contractAddr, q.WithdrawAllDelegatorRewards, dk)
+	}
+
+	return nil, nil, wasmvmtypes.UnsupportedRequest{Kind: "unknown Custom variant"}
+}