feat: add slashing/unbonding slashing tx in db (#74)

internal/db/delegation.go

@@ -208,3 +208,54 @@ func (db *Database) UpdateDelegationsStateByFinalityProvider(
 	)
 	return nil
 }
+
+func (db *Database) SaveBTCDelegationSlashingTxHex(
+	ctx context.Context, stakingTxHash string, slashingTxHex string,
+) error {
+	filter := bson.M{"_id": stakingTxHash}
+	update := bson.M{
+		"$set": bson.M{
+			"slashing_tx_hex": slashingTxHex,
+		},
+	}
+	result, err := db.client.Database(db.dbName).
+		Collection(model.BTCDelegationDetailsCollection).
+		UpdateOne(ctx, filter, update)
+	if err != nil {
+		return err
+	}
+
+	if result.MatchedCount == 0 {
+		return &NotFoundError{
+			Key:     stakingTxHash,
+			Message: "BTC delegation not found when updating slashing tx hex",
+		}
+	}
+
+	return nil
+}
+func (db *Database) SaveBTCDelegationUnbondingSlashingTxHex(
+	ctx context.Context, stakingTxHash string, unbondingSlashingTxHex string,
+) error {
+	filter := bson.M{"_id": stakingTxHash}
+	update := bson.M{
+		"$set": bson.M{
+			"unbonding_slashing_tx_hex": unbondingSlashingTxHex,
+		},
+	}
+	result, err := db.client.Database(db.dbName).
+		Collection(model.BTCDelegationDetailsCollection).
+		UpdateOne(ctx, filter, update)
+	if err != nil {
+		return err
+	}
+
+	if result.MatchedCount == 0 {
+		return &NotFoundError{
+			Key:     stakingTxHash,
+			Message: "BTC delegation not found when updating unbonding slashing tx hex",
+		}
+	}
+
+	return nil
+}

internal/db/interface.go

@@ -196,4 +196,20 @@ type DbInterface interface {
 	 * @return An error if the operation failed
 	 */
 	UpdateLastProcessedBbnHeight(ctx context.Context, height uint64) error
+	/**
+	 * SaveBTCDelegationSlashingTxHex saves the BTC delegation slashing tx hex.
+	 * @param ctx The context
+	 * @param stakingTxHashHex The staking tx hash hex
+	 * @param slashingTxHex The slashing tx hex
+	 * @return An error if the operation failed
+	 */
+	SaveBTCDelegationSlashingTxHex(ctx context.Context, stakingTxHashHex string, slashingTxHex string) error
+	/**
+	 * SaveBTCDelegationUnbondingSlashingTxHex saves the BTC delegation unbonding slashing tx hex.
+	 * @param ctx The context
+	 * @param stakingTxHashHex The staking tx hash hex
+	 * @param unbondingSlashingTxHex The unbonding slashing tx hex
+	 * @return An error if the operation failed
+	 */
+	SaveBTCDelegationUnbondingSlashingTxHex(ctx context.Context, stakingTxHashHex string, unbondingSlashingTxHex string) error
 }

internal/db/model/delegation.go

@@ -38,6 +38,8 @@ type BTCDelegationDetails struct {
 	UnbondingTx                 string                       `bson:"unbonding_tx"`
 	CovenantUnbondingSignatures []CovenantSignature          `bson:"covenant_unbonding_signatures"`
 	BTCDelegationCreatedBlock   BTCDelegationCreatedBbnBlock `bson:"btc_delegation_created_bbn_block"`
+	SlashingTxHex               string                       `bson:"slashing_tx_hex"`           // Will be "" if not slashed
+	UnbondingSlashingTxHex      string                       `bson:"unbonding_slashing_tx_hex"` // Will be "" if not slashed
 }
 
 func FromEventBTCDelegationCreated(

internal/services/watch_btc_events.go

@@ -12,6 +12,7 @@ import (
 	"github.com/babylonlabs-io/babylon-staking-indexer/internal/utils"
 	"github.com/babylonlabs-io/babylon/btcstaking"
 	bbn "github.com/babylonlabs-io/babylon/types"
+	bstypes "github.com/babylonlabs-io/babylon/x/btcstaking/types"
 	"github.com/btcsuite/btcd/btcec/v2"
 	"github.com/btcsuite/btcd/btcutil"
 	"github.com/btcsuite/btcd/wire"
@@ -203,6 +204,16 @@ func (s *Service) handleSpendingStakingTransaction(
 		return fmt.Errorf("failed to validate slashing tx: %w", err)
 	}
 
+	// Save slashing tx hex
+	slashingTx, err := bstypes.NewBTCSlashingTxFromMsgTx(spendingTx)
+	if err != nil {
+		return fmt.Errorf("failed to convert slashing tx to bytes: %w", err)
+	}
+	slashingTxHex := slashingTx.ToHexStr()
+	if err := s.db.SaveBTCDelegationSlashingTxHex(ctx, delegation.StakingTxHashHex, slashingTxHex); err != nil {
+		return fmt.Errorf("failed to save slashing tx hex: %w", err)
+	}
+
 	// It's a valid slashing tx, watch for spending change output
 	return s.startWatchingSlashingChange(
 		ctx,
@@ -250,6 +261,16 @@ func (s *Service) handleSpendingUnbondingTransaction(
 		return fmt.Errorf("failed to validate slashing tx: %w", err)
 	}
 
+	// Save unbonding slashing tx hex
+	unbondingSlashingTx, err := bstypes.NewBTCSlashingTxFromMsgTx(spendingTx)
+	if err != nil {
+		return fmt.Errorf("failed to convert unbonding slashing tx to bytes: %w", err)
+	}
+	unbondingSlashingTxHex := unbondingSlashingTx.ToHexStr()
+	if err := s.db.SaveBTCDelegationUnbondingSlashingTxHex(ctx, delegation.StakingTxHashHex, unbondingSlashingTxHex); err != nil {
+		return fmt.Errorf("failed to save unbonding slashing tx hex: %w", err)
+	}
+
 	// It's a valid slashing tx, watch for spending change output
 	return s.startWatchingSlashingChange(
 		ctx,