109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
package tx
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
modulev1 "cosmossdk.io/api/cosmos/tx/module/v1"
|
|
"cosmossdk.io/core/appmodule"
|
|
"cosmossdk.io/depinject"
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/ante"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/posthandler"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/tx"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
feegrantkeeper "github.com/cosmos/cosmos-sdk/x/feegrant/keeper"
|
|
)
|
|
|
|
func init() {
|
|
appmodule.Register(&modulev1.Module{},
|
|
appmodule.Provide(provideModule),
|
|
)
|
|
}
|
|
|
|
type txInputs struct {
|
|
depinject.In
|
|
|
|
Config *modulev1.Module
|
|
ProtoCodecMarshaler codec.ProtoCodecMarshaler
|
|
|
|
AccountKeeper ante.AccountKeeper `optional:"true"`
|
|
BankKeeper authtypes.BankKeeper `optional:"true"`
|
|
FeeGrantKeeper feegrantkeeper.Keeper `optional:"true"`
|
|
}
|
|
|
|
type txOutputs struct {
|
|
depinject.Out
|
|
|
|
TxConfig client.TxConfig
|
|
BaseAppOption runtime.BaseAppOption
|
|
}
|
|
|
|
func provideModule(in txInputs) txOutputs {
|
|
txConfig := tx.NewTxConfig(in.ProtoCodecMarshaler, tx.DefaultSignModes)
|
|
|
|
baseAppOption := func(app *baseapp.BaseApp) {
|
|
// AnteHandlers
|
|
if !in.Config.SkipAnteHandler {
|
|
anteHandler, err := newAnteHandler(txConfig, in)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
app.SetAnteHandler(anteHandler)
|
|
}
|
|
|
|
// PostHandlers
|
|
if !in.Config.SkipPostHandler {
|
|
// In v0.46, the SDK introduces _postHandlers_. PostHandlers are like
|
|
// antehandlers, but are run _after_ the `runMsgs` execution. They are also
|
|
// defined as a chain, and have the same signature as antehandlers.
|
|
//
|
|
// In baseapp, postHandlers are run in the same store branch as `runMsgs`,
|
|
// meaning that both `runMsgs` and `postHandler` state will be committed if
|
|
// both are successful, and both will be reverted if any of the two fails.
|
|
//
|
|
// The SDK exposes a default empty postHandlers chain.
|
|
//
|
|
// Please note that changing any of the anteHandler or postHandler chain is
|
|
// likely to be a state-machine breaking change, which needs a coordinated
|
|
// upgrade.
|
|
postHandler, err := posthandler.NewPostHandler(
|
|
posthandler.HandlerOptions{},
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
app.SetPostHandler(postHandler)
|
|
}
|
|
|
|
// TxDecoder
|
|
app.SetTxDecoder(txConfig.TxDecoder())
|
|
}
|
|
|
|
return txOutputs{TxConfig: txConfig, BaseAppOption: baseAppOption}
|
|
}
|
|
|
|
func newAnteHandler(txConfig client.TxConfig, in txInputs) (sdk.AnteHandler, error) {
|
|
if in.BankKeeper == nil {
|
|
return nil, fmt.Errorf("both AccountKeeper and BankKeeper are required")
|
|
}
|
|
|
|
anteHandler, err := ante.NewAnteHandler(
|
|
ante.HandlerOptions{
|
|
AccountKeeper: in.AccountKeeper,
|
|
BankKeeper: in.BankKeeper,
|
|
SignModeHandler: txConfig.SignModeHandler(),
|
|
FeegrantKeeper: in.FeeGrantKeeper,
|
|
SigGasConsumer: ante.DefaultSigVerificationGasConsumer,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create ante handler: %w", err)
|
|
}
|
|
|
|
return anteHandler, nil
|
|
}
|