wormhole-explorer/tx-tracker/queue/converter.go

package queue

import (
	"encoding/json"
	"fmt"
	"strconv"
	"time"

	"github.com/wormhole-foundation/wormhole-explorer/common/events"
	sdk "github.com/wormhole-foundation/wormhole/sdk/vaa"
	"go.uber.org/zap"
)

// VaaEvent represents a vaa data to be handle by the pipeline.
type VaaEvent struct {
	ID               string      `json:"id"`
	ChainID          sdk.ChainID `json:"emitterChain"`
	EmitterAddress   string      `json:"emitterAddr"`
	Sequence         string      `json:"sequence"`
	GuardianSetIndex uint32      `json:"guardianSetIndex"`
	Vaa              []byte      `json:"vaas"`
	IndexedAt        time.Time   `json:"indexedAt"`
	Timestamp        *time.Time  `json:"timestamp"`
	UpdatedAt        *time.Time  `json:"updatedAt"`
	TxHash           string      `json:"txHash"`
	Version          uint16      `json:"version"`
	Revision         uint16      `json:"revision"`
	Overwrite        bool        `json:"overwrite"`
}

// VaaConverter converts a message from a VAAEvent.
func NewVaaConverter(log *zap.Logger) ConverterFunc {

	return func(msg string) (*Event, error) {
		// unmarshal message to vaaEvent
		var vaaEvent VaaEvent
		err := json.Unmarshal([]byte(msg), &vaaEvent)
		if err != nil {
			return nil, err
		}
		return &Event{
			Source:         "pipeline",
			TrackID:        fmt.Sprintf("pipeline-%s", vaaEvent.ID),
			Type:           SourceChainEvent,
			ID:             vaaEvent.ID,
			ChainID:        vaaEvent.ChainID,
			EmitterAddress: vaaEvent.EmitterAddress,
			Sequence:       vaaEvent.Sequence,
			Timestamp:      vaaEvent.Timestamp,
			Vaa:            vaaEvent.Vaa,
			IsVaaSigned:    true,
			TxHash:         vaaEvent.TxHash,
			Overwrite:      vaaEvent.Overwrite,
		}, nil
	}
}

func NewNotificationEvent(log *zap.Logger) ConverterFunc {

	return func(msg string) (*Event, error) {
		// unmarshal message to NotificationEvent
		var notification events.NotificationEvent
		err := json.Unmarshal([]byte(msg), &notification)
		if err != nil {
			return nil, err
		}

		switch notification.Event {
		case events.SignedVaaType,
			events.LogMessagePublishedType,
			events.EvmTransactionFoundType,
			events.TransferRedeemedType:
			//message is valid
		default:
			log.Debug("Skip event type", zap.String("trackId", notification.TrackID), zap.String("type", notification.Event))
			return nil, nil
		}

		switch notification.Event {
		case events.SignedVaaType:
			signedVaa, err := events.GetEventData[events.SignedVaa](&notification)
			if err != nil {
				log.Error("Error decoding signedVAA from notification event", zap.String("trackId", notification.TrackID), zap.Error(err))
				return nil, nil
			}

			return &Event{
				Source:         "chain-event",
				TrackID:        notification.TrackID,
				Type:           SourceChainEvent,
				ID:             signedVaa.ID,
				ChainID:        sdk.ChainID(signedVaa.EmitterChain),
				EmitterAddress: signedVaa.EmitterAddress,
				Sequence:       strconv.FormatUint(signedVaa.Sequence, 10),
				Timestamp:      &signedVaa.Timestamp,
				IsVaaSigned:    false,
				TxHash:         signedVaa.TxHash,
			}, nil

		case events.LogMessagePublishedType:
			plm, err := events.GetEventData[events.LogMessagePublished](&notification)
			if err != nil {
				log.Error("Error decoding publishedLogMessage from notification event", zap.String("trackId", notification.TrackID), zap.Error(err))
				return nil, nil
			}

			vaa, err := events.CreateUnsignedVAA(&plm)
			if err != nil {
				log.Error("Error creating unsigned vaa", zap.String("trackId", notification.TrackID), zap.Error(err))
				return nil, err
			}

			return &Event{
				Source:         "chain-event",
				TrackID:        notification.TrackID,
				Type:           SourceChainEvent,
				ID:             vaa.MessageID(),
				ChainID:        sdk.ChainID(plm.ChainID),
				EmitterAddress: plm.Attributes.Sender,
				Sequence:       strconv.FormatUint(plm.Attributes.Sequence, 10),
				Timestamp:      &plm.BlockTime,
				IsVaaSigned:    false,
				TxHash:         plm.TxHash,
			}, nil

		case events.EvmTransactionFoundType:
			tr, err := events.GetEventData[events.EvmTransactionFound](&notification)
			if err != nil {
				log.Error("Error decoding transferRedeemed from notification event", zap.String("trackId", notification.TrackID), zap.Error(err))
				return nil, nil
			}
			address, err := sdk.StringToAddress(tr.Attributes.EmitterAddress)
			if err != nil {
				return nil, fmt.Errorf("error converting emitter address [%s]: %w", tr.Attributes.EmitterAddress, err)
			}
			vaa := sdk.VAA{
				EmitterChain:   sdk.ChainID(tr.Attributes.EmitterChain),
				EmitterAddress: address,
				Sequence:       tr.Attributes.Sequence,
			}

			if tr.Attributes.Name != events.EvmTransferRedeemedName {
				log.Warn("Skip event because it is not transfer-redeemed ", zap.String("trackId", notification.TrackID), zap.String("name", tr.Attributes.Name))
				return nil, nil
			}

			return &Event{
				Source:         "chain-event",
				TrackID:        notification.TrackID,
				Type:           TargetChainEvent,
				ID:             vaa.MessageID(),
				ChainID:        sdk.ChainID(tr.ChainID),
				EmitterAddress: tr.Attributes.EmitterAddress,
				Sequence:       strconv.FormatUint(tr.Attributes.Sequence, 10),
				Timestamp:      &tr.BlockTime,
				TxHash:         tr.TxHash,
				Attributes: &TargetChainAttributes{
					Emitter:     tr.Emitter,
					BlockHeight: tr.BlockHeight,
					TxHash:      tr.TxHash,
					From:        tr.Attributes.From,
					To:          tr.Attributes.To,
					Method:      tr.Attributes.Method,
					Status:      tr.Attributes.Status,
				},
			}, nil
		case events.TransferRedeemedType:
			tr, err := events.GetEventData[events.TransferRedeemed](&notification)
			if err != nil {
				log.Error("Error decoding transferRedeemed from notification event", zap.String("trackId", notification.TrackID), zap.Error(err))
				return nil, nil
			}
			address, err := sdk.StringToAddress(tr.Attributes.EmitterAddress)
			if err != nil {
				return nil, fmt.Errorf("error converting emitter address [%s]: %w", tr.Attributes.EmitterAddress, err)
			}
			vaa := sdk.VAA{
				EmitterChain:   sdk.ChainID(tr.Attributes.EmitterChain),
				EmitterAddress: address,
				Sequence:       tr.Attributes.Sequence,
			}

			return &Event{
				Source:         "chain-event",
				TrackID:        notification.TrackID,
				Type:           TargetChainEvent,
				ID:             vaa.MessageID(),
				ChainID:        sdk.ChainID(tr.ChainID),
				EmitterAddress: tr.Attributes.EmitterAddress,
				Sequence:       strconv.FormatUint(tr.Attributes.Sequence, 10),
				Timestamp:      &tr.BlockTime,
				TxHash:         tr.TxHash,
				Attributes: &TargetChainAttributes{
					Emitter:     tr.Emitter,
					BlockHeight: tr.BlockHeight,
					TxHash:      tr.TxHash,
					From:        tr.Attributes.From,
					To:          tr.Attributes.To,
					Method:      tr.Attributes.Method,
					Status:      tr.Attributes.Status,
				},
			}, nil
		}

		return nil, nil
	}
}