gaze-indexer/modules/nodesale/delegate.go

package nodesale

import (
	"context"
	"encoding/hex"
	"fmt"

	"github.com/btcsuite/btcd/btcec/v2"
	"github.com/gaze-network/indexer-network/core/types"
	"github.com/gaze-network/indexer-network/modules/nodesale/repository/postgres/gen"
	"github.com/jackc/pgx/v5/pgtype"
)

func (p *Processor) processDelegate(ctx context.Context, qtx gen.Querier, block *types.Block, event nodesaleEvent) error {
	valid := true
	delegate := event.eventMessage.Delegate
	nodeIds := make([]int32, len(delegate.NodeIDs))
	for i, id := range delegate.NodeIDs {
		nodeIds[i] = int32(id)
	}
	nodes, err := qtx.GetNodes(ctx, gen.GetNodesParams{
		SaleBlock:   int32(delegate.DeployID.Block),
		SaleTxIndex: int32(delegate.DeployID.TxIndex),
		NodeIds:     nodeIds,
	})
	if err != nil {
		return fmt.Errorf("Failed to get nodes : %w", err)
	}

	if len(nodeIds) != len(nodes) {
		valid = false
	}

	if valid {
		for _, node := range nodes {
			OwnerPublicKeyBytes, err := hex.DecodeString(node.OwnerPublicKey)
			if err != nil {
				valid = false
				break
			}
			OwnerPublicKey, err := btcec.ParsePubKey(OwnerPublicKeyBytes)
			if err != nil {
				valid = false
				break
			}
			if !event.txPubkey.IsEqual(OwnerPublicKey) {
				valid = false
				break
			}
		}
	}

	err = qtx.AddEvent(ctx, gen.AddEventParams{
		TxHash:         event.transaction.TxHash.String(),
		TxIndex:        int32(event.transaction.Index),
		Action:         int32(event.eventMessage.Action),
		RawMessage:     event.rawData,
		ParsedMessage:  event.eventJson,
		BlockTimestamp: pgtype.Timestamp{Time: block.Header.Timestamp, Valid: true},
		BlockHash:      event.transaction.BlockHash.String(),
		BlockHeight:    int32(event.transaction.BlockHeight),
		Valid:          valid,
		// WalletAddress:  event.txAddress.EncodeAddress(),
		WalletAddress: p.pubkeyToPkHashAddress(event.txPubkey).EncodeAddress(),
		Metadata:      []byte("{}"),
	})
	if err != nil {
		return fmt.Errorf("Failed to insert event : %w", err)
	}
	if valid {
		_, err = qtx.SetDelegates(ctx, gen.SetDelegatesParams{
			SaleBlock:   int32(delegate.DeployID.Block),
			SaleTxIndex: int32(delegate.DeployID.TxIndex),
			Delegatee:   delegate.DelegateePublicKey,
			NodeIds:     nodeIds,
		})
		if err != nil {
			return fmt.Errorf("Failed to set delegate : %w", err)
		}
	}

	return nil
}