Merge pull request #78 from bandprotocol/emitter-ibc-transfer

Emitter: handle packet send fungible token

Author: Benzbeeb

hooks/emitter/channel.go

@@ -1,32 +1,64 @@
 package emitter
 
 import (
-	sdk "github.com/cosmos/cosmos-sdk/types"
-
 	"github.com/bandprotocol/chain/hooks/common"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	ibcxfertypes "github.com/cosmos/cosmos-sdk/x/ibc/applications/transfer/types"
 	"github.com/cosmos/cosmos-sdk/x/ibc/core/04-channel/types"
 	channeltypes "github.com/cosmos/cosmos-sdk/x/ibc/core/04-channel/types"
 
 	oracletypes "github.com/bandprotocol/chain/x/oracle/types"
 )
 
-// handleMsgRequestData implements emitter handler for MsgRequestData.
-func (h *Hook) handleMsgRecvPacket(
-	ctx sdk.Context, txHash []byte, msg *types.MsgRecvPacket, evMap common.EvMap, extra common.JsDict,
-) {
-	packet := common.JsDict{
-		"is_incoming":  true,
+// TODO: update transfer acknowledgement for fungible token packet
+
+func newPacket(ctx sdk.Context, srcPort string, srcChannel string, sequence uint64, dstPort string, dstChannel string, isIncoming bool) common.JsDict {
+	return common.JsDict{
+		"is_incoming":  isIncoming,
 		"block_height": ctx.BlockHeight(),
-		"src_channel":  msg.Packet.SourceChannel,
-		"src_port":     msg.Packet.SourcePort,
-		"sequence":     msg.Packet.Sequence,
-		"dst_channel":  msg.Packet.DestinationChannel,
-		"dst_port":     msg.Packet.DestinationPort,
+		"src_channel":  srcChannel,
+		"src_port":     srcPort,
+		"sequence":     sequence,
+		"dst_channel":  dstChannel,
+		"dst_port":     dstPort,
 	}
+}
+
+func (h *Hook) extractFungibleTokenPacket(
+	ctx sdk.Context, dataOfPacket []byte, evMap common.EvMap, packet common.JsDict,
+) bool {
+	var data ibcxfertypes.FungibleTokenPacketData
+	err := ibcxfertypes.ModuleCdc.UnmarshalJSON(dataOfPacket, &data)
+	if err == nil {
+		packet["type"] = "fungible token"
+		packet["data"] = common.JsDict{
+			"denom":    data.Denom,
+			"amount":   data.Amount,
+			"sender":   data.Sender,
+			"receiver": data.Receiver,
+		}
+		if events, ok := evMap[ibcxfertypes.EventTypePacket+"."+ibcxfertypes.AttributeKeyAckSuccess]; ok {
+			// TODO: patch this line when cosmos-sdk fix AttributeKeyAckSuccess value
+			if events[0] == "false" {
+				packet["acknowledgement"] = common.JsDict{
+					"success": true,
+				}
+			} else {
+				packet["acknowledgement"] = common.JsDict{
+					"success": false,
+				}
+			}
+		}
+		return true
+	}
+	return false
+}
 
-	// TODO: Check on other packet
+func (h *Hook) extractOracleRequestPacket(
+	ctx sdk.Context, txHash []byte, signer string, dataOfPacket []byte, evMap common.EvMap, extra common.JsDict, packet common.JsDict,
+) bool {
 	var data oracletypes.OracleRequestPacketData
-	err := oracletypes.ModuleCdc.UnmarshalJSON(msg.Packet.Data, &data)
+	err := oracletypes.ModuleCdc.UnmarshalJSON(dataOfPacket, &data)
 	if err == nil {
 		if events, ok := evMap[oracletypes.EventTypeRequest+"."+oracletypes.AttributeKeyID]; ok {
 			id := oracletypes.RequestID(common.Atoi(events[0]))
@@ -38,7 +70,7 @@ func (h *Hook) handleMsgRecvPacket(
 				"calldata":         parseBytes(data.Calldata),
 				"ask_count":        data.AskCount,
 				"min_count":        data.MinCount,
-				"sender":           msg.Signer,
+				"sender":           signer,
 				"client_id":        data.ClientID,
 				"resolve_status":   oracletypes.RESOLVE_STATUS_OPEN,
 				"timestamp":        ctx.BlockTime().UnixNano(),
@@ -89,26 +121,38 @@ func (h *Hook) handleMsgRecvPacket(
 				"reason":  evMap[channeltypes.EventTypeWriteAck+"."+channeltypes.AttributeKeyAck][0],
 			}
 		}
-		h.Write("NEW_PACKET", packet)
+		return true
 	}
-
+	return false
 }
 
-func (h *Hook) handleEventSendPacket(
-	ctx sdk.Context, evMap common.EvMap,
+// handleMsgRequestData implements emitter handler for MsgRequestData.
+func (h *Hook) handleMsgRecvPacket(
+	ctx sdk.Context, txHash []byte, msg *types.MsgRecvPacket, evMap common.EvMap, extra common.JsDict,
 ) {
-	packet := common.JsDict{
-		"is_incoming":  false,
-		"block_height": ctx.BlockHeight(),
-		"src_channel":  evMap[types.EventTypeSendPacket+"."+types.AttributeKeySrcChannel][0],
-		"src_port":     evMap[types.EventTypeSendPacket+"."+types.AttributeKeySrcPort][0],
-		"sequence":     common.Atoui(evMap[types.EventTypeSendPacket+"."+types.AttributeKeySequence][0]),
-		"dst_channel":  evMap[types.EventTypeSendPacket+"."+types.AttributeKeyDstChannel][0],
-		"dst_port":     evMap[types.EventTypeSendPacket+"."+types.AttributeKeyDstPort][0],
+	packet := newPacket(
+		ctx,
+		msg.Packet.SourcePort,
+		msg.Packet.SourceChannel,
+		msg.Packet.Sequence,
+		msg.Packet.DestinationPort,
+		msg.Packet.DestinationChannel,
+		true,
+	)
+	h.Write("NEW_PACKET", packet)
+	if ok := h.extractOracleRequestPacket(ctx, txHash, msg.Signer, msg.Packet.Data, evMap, extra, packet); ok {
+		return
 	}
-	// TODO: Check on other packet
+	if ok := h.extractFungibleTokenPacket(ctx, msg.Packet.Data, evMap, packet); ok {
+		return
+	}
+}
+
+func (h *Hook) extractOracleResponsePacket(
+	ctx sdk.Context, packet common.JsDict, evMap common.EvMap,
+) bool {
 	var data oracletypes.OracleResponsePacketData
-	err := h.cdc.UnmarshalJSON([]byte(evMap[types.EventTypeSendPacket+"."+types.AttributeKeyData][0]), &data)
+	err := oracletypes.ModuleCdc.UnmarshalJSON([]byte(evMap[types.EventTypeSendPacket+"."+types.AttributeKeyData][0]), &data)
 	if err == nil {
 		req := h.oracleKeeper.MustGetRequest(ctx, data.RequestID)
 		os := h.oracleKeeper.MustGetOracleScript(ctx, req.OracleScriptID)
@@ -121,7 +165,25 @@ func (h *Hook) handleEventSendPacket(
 			"resolve_status":       data.ResolveStatus,
 			"result":               parseBytes(data.Result),
 		}
+		return true
 	}
+	return false
+}
 
+func (h *Hook) handleEventSendPacket(
+	ctx sdk.Context, evMap common.EvMap,
+) {
+	packet := newPacket(
+		ctx,
+		evMap[types.EventTypeSendPacket+"."+types.AttributeKeySrcPort][0],
+		evMap[types.EventTypeSendPacket+"."+types.AttributeKeySrcChannel][0],
+		common.Atoui(evMap[types.EventTypeSendPacket+"."+types.AttributeKeySequence][0]),
+		evMap[types.EventTypeSendPacket+"."+types.AttributeKeyDstPort][0],
+		evMap[types.EventTypeSendPacket+"."+types.AttributeKeyDstChannel][0],
+		false,
+	)
 	h.Write("NEW_PACKET", packet)
+	if ok := h.extractOracleResponsePacket(ctx, packet, evMap); ok {
+		return
+	}
 }

hooks/emitter/handler.go

@@ -5,6 +5,7 @@ import (
 	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
 	distrtypes "github.com/cosmos/cosmos-sdk/x/distribution/types"
 	govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
+	transfertypes "github.com/cosmos/cosmos-sdk/x/ibc/applications/transfer/types"
 	channeltypes "github.com/cosmos/cosmos-sdk/x/ibc/core/04-channel/types"
 	slashingtypes "github.com/cosmos/cosmos-sdk/x/slashing/types"
 	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
@@ -78,6 +79,8 @@ func (h *Hook) handleMsg(ctx sdk.Context, txHash []byte, msg sdk.Msg, log sdk.AB
 		h.handleMsgDeposit(ctx, txHash, msg, extra)
 	case *channeltypes.MsgRecvPacket:
 		h.handleMsgRecvPacket(ctx, txHash, msg, evMap, extra)
+	case *transfertypes.MsgTransfer:
+		h.handleMsgTransfer(ctx, msg, evMap)
 	}
 }
 

hooks/emitter/transfer.go

@@ -0,0 +1,26 @@
+package emitter
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/x/ibc/applications/transfer/types"
+	channeltypes "github.com/cosmos/cosmos-sdk/x/ibc/core/04-channel/types"
+
+	"github.com/bandprotocol/chain/hooks/common"
+)
+
+// handleMsgTransfer implements emitter handler for msgTransfer.
+func (h *Hook) handleMsgTransfer(ctx sdk.Context, msg *types.MsgTransfer, evMap common.EvMap) {
+	if events, ok := evMap[channeltypes.EventTypeSendPacket+"."+channeltypes.AttributeKeyData]; ok {
+		packet := newPacket(
+			ctx,
+			evMap[channeltypes.EventTypeSendPacket+"."+channeltypes.AttributeKeySrcPort][0],
+			evMap[channeltypes.EventTypeSendPacket+"."+channeltypes.AttributeKeySrcChannel][0],
+			common.Atoui(evMap[channeltypes.EventTypeSendPacket+"."+channeltypes.AttributeKeySequence][0]),
+			evMap[channeltypes.EventTypeSendPacket+"."+channeltypes.AttributeKeyDstPort][0],
+			evMap[channeltypes.EventTypeSendPacket+"."+channeltypes.AttributeKeyDstChannel][0],
+			false,
+		)
+		h.extractFungibleTokenPacket(ctx, []byte(events[0]), evMap, packet)
+		h.Write("NEW_PACKET", packet)
+	}
+}