forked from AstraProtocol/astra
/
ante.go
35 lines (30 loc) · 1.13 KB
/
ante.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package ante
import (
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
feeburntype "github.com/dungtt-astra/astra/v3/x/feeburn/types"
)
// FeeBurnDecorator Run his after we already deduct the fee from the account with
// the ante.NewDeductFeeDecorator() decorator. We pull funds from the FeeCollector ModuleAccount
type FeeBurnDecorator struct {
bankKeeper feeburntype.BankKeeper
feeBurnKeeper feeburntype.FeeBurnKeeper
}
func NewFeeBurnDecorator(bk feeburntype.BankKeeper, fb feeburntype.FeeBurnKeeper) FeeBurnDecorator {
return FeeBurnDecorator{
bankKeeper: bk,
feeBurnKeeper: fb,
}
}
func (fsd FeeBurnDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (newCtx sdk.Context, err error) {
feeTx, ok := tx.(sdk.FeeTx)
if !ok {
return ctx, sdkerrors.Wrap(sdkerrors.ErrTxDecode, "Tx must be a FeeTx")
}
params := fsd.feeBurnKeeper.GetParams(ctx)
err = fsd.feeBurnKeeper.BurnFee(ctx, fsd.bankKeeper, feeTx.GetFee(), params)
if err != nil {
return ctx, sdkerrors.Wrapf(sdkerrors.ErrInsufficientFunds, err.Error())
}
return next(ctx, tx, simulate)
}