update

Author: pkieltyka

receipts/token_transfers.go

@@ -10,6 +10,11 @@ import (
 	"github.com/0xsequence/go-sequence/contracts/gen/tokens"
 )
 
+var tokenTransferTopicHashes = []common.Hash{
+	common.HexToHash("0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"), // ERC20 Transfer
+	common.HexToHash("0xe6497e3ee548a3372136af2fcb0696db31fc6cf20260707645068bd3fe97f3c4"), // Polygon POL LogTransfer (custom)
+}
+
 // FetchReceiptTokenTransfers fetches the transaction receipt for the given transaction hash
 // and decodes any token transfer events (ERC20) that occurred within that transaction. TODOXXX: we
 // currently only support ERC20 token transfers, but we can extend this to support ERC721 and ERC1155 as well.
@@ -68,7 +73,7 @@ type TokenTransfer struct {
 
 type TokenTransfers []*TokenTransfer
 
-func (t TokenTransfers) FilterTokenTransfersByContractAddress(ctx context.Context, contract common.Address) TokenTransfers {
+func (t TokenTransfers) FilterByContractAddress(ctx context.Context, contract common.Address) TokenTransfers {
 	var out TokenTransfers
 	for _, transfer := range t {
 		if transfer.Raw.Address == contract {
@@ -78,7 +83,17 @@ func (t TokenTransfers) FilterTokenTransfersByContractAddress(ctx context.Contex
 	return out
 }
 
-func (t TokenTransfers) FilterTokenTransfersByFromAddress(ctx context.Context, from common.Address) TokenTransfers {
+func (t TokenTransfers) FilterByAccountAddress(ctx context.Context, account common.Address) TokenTransfers {
+	var out TokenTransfers
+	for _, transfer := range t {
+		if transfer.From == account || transfer.To == account {
+			out = append(out, transfer)
+		}
+	}
+	return out
+}
+
+func (t TokenTransfers) FilterByFromAddress(ctx context.Context, from common.Address) TokenTransfers {
 	var out TokenTransfers
 	for _, transfer := range t {
 		if transfer.From == from {
@@ -88,7 +103,7 @@ func (t TokenTransfers) FilterTokenTransfersByFromAddress(ctx context.Context, f
 	return out
 }
 
-func (t TokenTransfers) FilterTokenTransfersByToAddress(ctx context.Context, to common.Address) TokenTransfers {
+func (t TokenTransfers) FilterByToAddress(ctx context.Context, to common.Address) TokenTransfers {
 	var out TokenTransfers
 	for _, transfer := range t {
 		if transfer.To == to {

receipts/token_transfers_test.go

@@ -12,16 +12,44 @@ import (
 )
 
 func TestFetchReceiptTokenTransfers(t *testing.T) {
-	// txnHash := https://polygonscan.com/tx/0x252419983224542bfb07dab75808fa57186a7a269d0d267ae655eb7ef037fdd5
-	provider, err := ethrpc.NewProvider("https://nodes.sequence.app/polygon")
-	require.NoError(t, err)
 
-	txnHash := common.HexToHash("0x252419983224542bfb07dab75808fa57186a7a269d0d267ae655eb7ef037fdd5")
+	t.Run("ERC20 Transfer on Arbitrum", func(t *testing.T) {
+		// https://arbiscan.io/tx/0xb88cc2fea7cd26c88e169f6244fea76f590fc0797ba4c424669d1b74643f1dc9
+		provider, err := ethrpc.NewProvider("https://nodes.sequence.app/arbitrum")
+		require.NoError(t, err)
 
-	receipt, transfers, err := receipts.FetchReceiptTokenTransfers(context.Background(), provider, txnHash)
-	require.NoError(t, err)
-	require.NotNil(t, receipt)
-	require.Greater(t, len(transfers), 0)
+		txnHash := common.HexToHash("0xb88cc2fea7cd26c88e169f6244fea76f590fc0797ba4c424669d1b74643f1dc9")
 
-	spew.Dump(transfers)
+		// TODO: lets find a very simple metamask erc20 transfer, and check it in the test
+		// TODO2: find a batch of different erc20 transfers to test against.. for example
+		// its possible there can be multiple erc20 transfers in a single tx that come and go from an individual
+		// so what we want to see is the delta, the diff, etc. .. aka, the "Result" .. aka... "TokenTransferResult"
+		// and not just all of the TokenTransferLogs ..
+		// TODO3: vault bridge USDC .. lets check the token transfer event, prob just erc20 too
+
+		receipt, transfers, err := receipts.FetchReceiptTokenTransfers(context.Background(), provider, txnHash)
+		require.NoError(t, err)
+		require.NotNil(t, receipt)
+		require.Greater(t, len(transfers), 0)
+
+		spew.Dump(transfers)
+
+	})
+
+	t.Run("Polygon POL LogTransfer", func(t *testing.T) {
+		t.Skip("POL")
+
+		// txnHash := https://polygonscan.com/tx/0x252419983224542bfb07dab75808fa57186a7a269d0d267ae655eb7ef037fdd5
+		provider, err := ethrpc.NewProvider("https://nodes.sequence.app/polygon")
+		require.NoError(t, err)
+
+		txnHash := common.HexToHash("0x252419983224542bfb07dab75808fa57186a7a269d0d267ae655eb7ef037fdd5")
+
+		receipt, transfers, err := receipts.FetchReceiptTokenTransfers(context.Background(), provider, txnHash)
+		require.NoError(t, err)
+		require.NotNil(t, receipt)
+		require.Greater(t, len(transfers), 0)
+
+		spew.Dump(transfers)
+	})
 }