chore: add btc del tx hash to delegation sent to babylon

babylonlabs-io · Nov 27, 2024 · d0a1c86 · d0a1c86
1 parent 9f1f0ab
commit d0a1c86
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/staker/stakerapp.go b/staker/stakerapp.go
@@ -1386,6 +1386,7 @@ func (app *App) handleSendDelegationRequest(
 		stakerAddress,
 		delegationData.Ud.UnbondingTransaction,
 		delegationData.Ud.UnbondingTxUnbondingTime,
+		resp.TxHash,
 	)
 	if err != nil {
 		return nil, btcDelTxHash, err

diff --git a/stakerdb/trackedtranactionstore.go b/stakerdb/trackedtranactionstore.go
@@ -677,6 +677,7 @@ func (c *TrackedTransactionStore) AddTransactionSentToBabylon(
 	stakerAddress btcutil.Address,
 	unbondingTx *wire.MsgTx,
 	unbondingTime uint16,
+	btcDelTxHash string,
 ) error {
 	txHash := btcTx.TxHash()
 	txHashBytes := txHash[:]
@@ -720,6 +721,7 @@ func (c *TrackedTransactionStore) AddTransactionSentToBabylon(
 		State:                        proto.TransactionState_SENT_TO_BABYLON,
 		Watched:                      false,
 		UnbondingTxData:              update,
+		BtcDelegationTxHash:          btcDelTxHash,
 	}
 
 	inputData, err := getInputData(btcTx)

diff --git a/stakerdb/trackedtransactionstore_test.go b/stakerdb/trackedtransactionstore_test.go
@@ -413,6 +413,7 @@ func FuzzTrackInputs(f *testing.F) {
 				stakerAddr,
 				randomUnbondingTx,
 				unbodningTime,
+				"txHashBtcDelegation",
 			)
 			require.NoError(t, err)
 		}