feat: Gas consumption script

app/antehander.go

@@ -0,0 +1,41 @@
+package app
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
+	"github.com/cosmos/cosmos-sdk/x/auth/ante"
+)
+
+// NewAnteDecorators returns an AnteDecorators that check and increment
+// sequence numbers, checks signatures & account numbers, and deducts fees from
+// the first signer.
+func NewAnteDecorators(options ante.HandlerOptions) ([]sdk.AnteDecorator, error) {
+	if options.AccountKeeper == nil {
+		return nil, sdkerrors.Wrap(sdkerrors.ErrLogic, "account keeper is required for ante builder")
+	}
+
+	if options.BankKeeper == nil {
+		return nil, sdkerrors.Wrap(sdkerrors.ErrLogic, "bank keeper is required for ante builder")
+	}
+
+	if options.SignModeHandler == nil {
+		return nil, sdkerrors.Wrap(sdkerrors.ErrLogic, "sign mode handler is required for ante builder")
+	}
+
+	anteDecorators := []sdk.AnteDecorator{
+		ante.NewSetUpContextDecorator(), // outermost AnteDecorator. SetUpContext must be called first
+		ante.NewExtensionOptionsDecorator(options.ExtensionOptionChecker),
+		ante.NewValidateBasicDecorator(),
+		ante.NewTxTimeoutHeightDecorator(),
+		ante.NewValidateMemoDecorator(options.AccountKeeper),
+		ante.NewConsumeGasForTxSizeDecorator(options.AccountKeeper),
+		ante.NewDeductFeeDecorator(options.AccountKeeper, options.BankKeeper, options.FeegrantKeeper, options.TxFeeChecker),
+		ante.NewSetPubKeyDecorator(options.AccountKeeper), // SetPubKeyDecorator must be called before all signature verification decorators
+		ante.NewValidateSigCountDecorator(options.AccountKeeper),
+		ante.NewSigGasConsumeDecorator(options.AccountKeeper, options.SigGasConsumer),
+		ante.NewSigVerificationDecorator(options.AccountKeeper, options.SignModeHandler),
+		ante.NewIncrementSequenceDecorator(options.AccountKeeper),
+	}
+
+	return anteDecorators, nil
+}

app/app.go

@@ -97,6 +97,9 @@ const (
 	BondDenomAlias = "microtia"
 	// DisplayDenom defines the name, symbol, and display value of the Celestia token.
 	DisplayDenom = "TIA"
+	// MonitorGasOptionKey is the key used in the app options that determines if
+	// the gas monitor should be used
+	MonitorGasOptionKey = "MonitorGas"
 )
 
 // These constants are derived from the above variables.
@@ -501,7 +504,7 @@ func New(
 	app.SetInitChainer(app.InitChainer)
 	app.SetBeginBlocker(app.BeginBlocker)
 	app.SetEndBlocker(app.EndBlocker)
-	anteHandler, err := ante.NewAnteHandler(
+	anteDecorators, err := NewAnteDecorators(
 		ante.HandlerOptions{
 			AccountKeeper:   app.AccountKeeper,
 			BankKeeper:      app.BankKeeper,
@@ -513,7 +516,19 @@ func New(
 	if err != nil {
 		panic(err)
 	}
-	app.SetAnteHandler(anteHandler)
+
+	gasMonitor := appOpts.Get(MonitorGasOptionKey)
+	// insert the gas monitor into the earliest position if it is configured
+	if gasMonitor != nil {
+		decorator, ok := gasMonitor.(sdk.AnteDecorator)
+		if !ok {
+			panic("unexpected gas metor decorator")
+		}
+		anteDecorators = append(anteDecorators, nil)
+		copy(anteDecorators[2:], anteDecorators[1:])
+		anteDecorators[1] = decorator
+	}
+	app.SetAnteHandler(sdk.ChainAnteDecorators(anteDecorators...))
 	app.setPostHanders()
 
 	if loadLatest {