Fee Middleware: Packet Callbacks

[AdityaSripal]

No description provided.

[AdityaSripal]

// OnRecvPacket implements the IBCModule interface.
func (am AppModule) 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
	ack := am.app.OnRecvPacket(ctx, packet, relayer)
        if !IsFeeEnabled() {
              return ack
        }
        return constructIncentivizedAck(ack, relayer)
}

// OnAcknowledgementPacket implements the IBCModule interface
func (am AppModule) OnAcknowledgementPacket(
	ctx sdk.Context,
	packet channeltypes.Packet,
	acknowledgement []byte,
	relayer sdk.AccAddress,
) error {
	// TODO: Implement fee specific logic if fee is enabled for the given channel
	if am.keeper.IsFeeEnabled(packet.DestPort, packet.DestChannel) {
		incentivized_ack := unmarshal(acknowledgment)
		recvRelayer = getRecvRelayer(incentivized_ack)
		recvFee := am.keeper.GetRecvFee(packet.DestPort, packet.DestChannel, packet.Sequence) // Sean creates this method
		send(recvFee, recvRelayer) // from escrow address
		// similar GetAckFee method
		ackFee := am.keeper.GetAckFee(packet.DestPort, packet.DestChannel, packet.Sequence) // Sean creates this method
		send(ackFee, relayer)
		refund(timeoutFee)
	}
	return am.app.OnAcknowledgementPacket(ctx, packet, acknowledgement, relayer)
}

// OnTimeoutPacket implements the IBCModule interface
func (am AppModule) OnTimeoutPacket(
	ctx sdk.Context,
	packet channeltypes.Packet,
	relayer sdk.AccAddress,
) error {
	// TODO: Implement fee specific logic if fee is enabled for the given channel
	if am.keeper.IsFeeEnabled() {
		timeoutFee := am.keeper.GetTimeoutFee(packet.DestPort, packet.DestChannel, packet.Sequence) // Sean creates this method
		send(timeoutFee, relayer)
		recvFee := am.keeper.GetRecvFee(packet.DestPort, packet.DestChannel, packet.Sequence) // Sean creates this method
		ackFee := am.keeper.GetAckFee(packet.DestPort, packet.DestChannel, packet.Sequence) // Sean creates this method
		refund(ackFee)
		refund(recvFee)
	}
	return am.app.OnTimeoutPacket(ctx, packet, relayer)
}