feat: add MarkDelegationAsTransitioned method

internal/v1/db/client/delegation.go

@@ -156,6 +156,23 @@ func (v1dbclient *V1Database) ScanDelegationsPaginated(
 	)
 }
 
+// MarkDelegationAsTransformed marks an existing delegation as transformed
+func (v1dbclient *V1Database) MarkDelegationAsTransformed(ctx context.Context, stakingTxHashHex string) error {
+	client := v1dbclient.Client.Database(v1dbclient.DbName).Collection(dbmodel.V1DelegationCollection)
+	update := bson.M{"$set": bson.M{"transformed": true}}
+	result, err := client.UpdateOne(ctx, bson.M{"_id": stakingTxHashHex}, update)
+	if err != nil {
+		return err
+	}
+	if result.MatchedCount == 0 {
+		return &db.NotFoundError{
+			Key:     stakingTxHashHex,
+			Message: "Delegation not found",
+		}
+	}
+	return nil
+}
+
 // TransitionState updates the state of a staking transaction to a new state
 // It returns an NotFoundError if the staking transaction is not found or not in the eligible state to transition
 func (v1dbclient *V1Database) transitionState(

internal/v1/db/client/interface.go

@@ -30,6 +30,7 @@ type V1DBClient interface {
 		ctx context.Context, stakingTxHashHex, unbondingTxHashHex, txHex, signatureHex string,
 	) error
 	FindDelegationByTxHashHex(ctx context.Context, txHashHex string) (*v1dbmodel.DelegationDocument, error)
+	MarkDelegationAsTransformed(ctx context.Context, stakingTxHashHex string) error
 	SaveTimeLockExpireCheck(ctx context.Context, stakingTxHashHex string, expireHeight uint64, txType string) error
 	TransitionToUnbondedState(
 		ctx context.Context, stakingTxHashHex string, eligiblePreviousState []types.DelegationState,

internal/v1/db/model/delegation.go

@@ -22,6 +22,7 @@ type DelegationDocument struct {
 	StakingTx             *TimelockTransaction  `bson:"staking_tx"` // Always exist
 	UnbondingTx           *TimelockTransaction  `bson:"unbonding_tx,omitempty"`
 	IsOverflow            bool                  `bson:"is_overflow"`
+	Transformed           bool                  `bson:"transformed"`
 }
 
 type DelegationByStakerPagination struct {

internal/v1/service/delegation.go

@@ -32,6 +32,7 @@ type DelegationPublic struct {
 	IsOverflow              bool               `json:"is_overflow"`
 	IsEligibleForTransition bool               `json:"is_eligible_for_transition"`
 	IsSlashed               bool               `json:"is_slashed"`
+	Transformed             bool               `json:"transformed"`
 }
 
 func (s *V1Service) DelegationsByStakerPk(
@@ -212,6 +213,7 @@ func (s *V1Service) FromDelegationDocument(
 		IsOverflow:              d.IsOverflow,
 		IsEligibleForTransition: isFpTransitioned && !isSlashed && s.isEligibleForTransition(d, bbnHeight),
 		IsSlashed:               isSlashed,
+		Transformed:             d.Transformed,
 	}
 
 	// Add unbonding transaction if it exists

internal/v2/queue/handler/staking.go

@@ -20,6 +20,11 @@ func (h *V2QueueHandler) ActiveStakingHandler(ctx context.Context, messageBody s
 		return types.NewError(http.StatusBadRequest, types.BadRequest, err)
 	}
 
+	// Mark as v1 delegation as transformed if it exists
+	if err := h.Service.MarkV1DelegationAsTransformed(ctx, activeStakingEvent.StakingTxHashHex); err != nil {
+		log.Ctx(ctx).Error().Err(err).Msg("Failed to mark v1 delegation as transformed")
+	}
+
 	// Perform the address lookup conversion
 	addressLookupErr := h.performAddressLookupConversion(ctx, activeStakingEvent.StakerBtcPkHex, types.Active)
 	if addressLookupErr != nil {

internal/v2/service/delegation.go

@@ -170,3 +170,21 @@ func (s *V2Service) SaveUnprocessableMessages(ctx context.Context, messageBody,
 	}
 	return nil
 }
+
+func (s *V2Service) MarkV1DelegationAsTransformed(ctx context.Context, stakingTxHashHex string) *types.Error {
+	// check if v1 delegation exists
+	delegation, err := s.DbClients.V1DBClient.FindDelegationByTxHashHex(ctx, stakingTxHashHex)
+	if err != nil {
+		return types.NewInternalServiceError(err)
+	}
+	if delegation == nil {
+		return nil
+	}
+
+	// mark as transformed
+	err = s.DbClients.V1DBClient.MarkDelegationAsTransformed(ctx, stakingTxHashHex)
+	if err != nil {
+		return types.NewInternalServiceError(err)
+	}
+	return nil
+}

internal/v2/service/interface.go

@@ -14,6 +14,7 @@ type V2ServiceProvider interface {
 	IsDelegationPresent(ctx context.Context, txHashHex string) (bool, *types.Error)
 	GetDelegation(ctx context.Context, stakingTxHashHex string) (*StakerDelegationPublic, *types.Error)
 	GetDelegations(ctx context.Context, stakerPKHex string, paginationKey string) ([]*StakerDelegationPublic, string, *types.Error)
+	MarkV1DelegationAsTransformed(ctx context.Context, stakingTxHashHex string) *types.Error
 	GetOverallStats(ctx context.Context) (*OverallStatsPublic, *types.Error)
 	GetStakerStats(ctx context.Context, stakerPKHex string) (*StakerStatsPublic, *types.Error)
 	ProcessAndSaveBtcAddresses(ctx context.Context, stakerPkHex string) *types.Error