feat: ics 29 packet callbacks

modules/apps/29-fee/ibc_module.go

@@ -9,7 +9,7 @@ import (
 	"github.com/cosmos/ibc-go/modules/apps/29-fee/types"
 	channeltypes "github.com/cosmos/ibc-go/modules/core/04-channel/types"
 	porttypes "github.com/cosmos/ibc-go/modules/core/05-port/types"
-	ibcexported "github.com/cosmos/ibc-go/modules/core/exported"
+	"github.com/cosmos/ibc-go/modules/core/exported"
 )
 
 // IBCModule implements the ICS26 callbacks for the fee middleware given the fee keeper and the underlying application.
@@ -175,8 +175,19 @@ func (im IBCModule) OnRecvPacket(
 	ctx sdk.Context,
 	packet channeltypes.Packet,
 	relayer sdk.AccAddress,
-) ibcexported.Acknowledgement {
-	// TODO: Implement fee specific logic if fee is enabled for the given channel
+) exported.Acknowledgement {
+	if im.keeper.IsFeeEnabled(ctx, packet.SourcePort, packet.SourceChannel) {
+		ack := im.app.OnRecvPacket(ctx, packet, relayer)
+		forwardRelayer, found := im.keeper.GetCounterpartyAddress(ctx, relayer.String())
+		if !found {
+			forwardRelayer = ""
+		}
+
+		return types.IncentivizedAcknowledgement{
+			Result:                ack.Acknowledgement(),
+			ForwardRelayerAddress: forwardRelayer,
+		}
+	}
 	return im.app.OnRecvPacket(ctx, packet, relayer)
 }
 
@@ -187,7 +198,23 @@ func (im IBCModule) OnAcknowledgementPacket(
 	acknowledgement []byte,
 	relayer sdk.AccAddress,
 ) error {
-	// TODO: Implement fee specific logic if fee is enabled for the given channel
+	if im.keeper.IsFeeEnabled(ctx, packet.SourcePort, packet.SourceChannel) {
+		ack := &types.IncentivizedAcknowledgement{}
+		if err := types.ModuleCdc.UnmarshalJSON(acknowledgement, ack); err != nil {
+			return sdkerrors.Wrapf(sdkerrors.ErrUnknownRequest, "cannot unmarshal ICS-29 incentivized packet acknowledgement: %v", err)
+		}
+
+		packetId := channeltypes.NewPacketId(packet.SourceChannel, packet.SourcePort, packet.Sequence)
+		identifiedPacketFee, found := im.keeper.GetFeeInEscrow(ctx, packetId)
+		if !found {
+			return sdkerrors.Wrapf(types.ErrFeeNotFound, "fee not found for packet id %s", packetId)
+		}
+
+		if err := im.keeper.DistributeFee(ctx, sdk.AccAddress(identifiedPacketFee.RefundAddress), sdk.AccAddress(ack.ForwardRelayerAddress), relayer, packetId); err != nil {
+			return err
+		}
+		return im.app.OnAcknowledgementPacket(ctx, packet, ack.Result, relayer)
+	}
 	return im.app.OnAcknowledgementPacket(ctx, packet, acknowledgement, relayer)
 }
 
@@ -197,6 +224,15 @@ func (im IBCModule) OnTimeoutPacket(
 	packet channeltypes.Packet,
 	relayer sdk.AccAddress,
 ) error {
-	// TODO: Implement fee specific logic if fee is enabled for the given channel
+	if im.keeper.IsFeeEnabled(ctx, packet.SourcePort, packet.SourceChannel) {
+		packetId := channeltypes.NewPacketId(packet.SourceChannel, packet.SourcePort, packet.Sequence)
+		identifiedPacketFee, exists := im.keeper.GetFeeInEscrow(ctx, packetId)
+		if !exists {
+			return sdkerrors.Wrapf(types.ErrFeeNotFound, "fee not found for packet id %s", packetId)
+		}
+		if err := im.keeper.DistributeFeeTimeout(ctx, sdk.AccAddress(identifiedPacketFee.RefundAddress), relayer, channeltypes.NewPacketId(packet.SourceChannel, packet.SourcePort, packet.Sequence)); err != nil {
+			return err
+		}
+	}
 	return im.app.OnTimeoutPacket(ctx, packet, relayer)
 }