watchtower: add DeleteCommittedUpdate DB method

Add a new DeleteCommittedUpdate method to the wtdb In preparation for an
upcoming commit that will replay committed updates from one session to
another.

watchtower/wtdb/client_db.go

@@ -2073,6 +2073,42 @@ func (c *ClientDB) GetDBQueue(namespace []byte) Queue[*BackupID] {
 	)
 }
 
+// DeleteCommittedUpdate deletes the committed update with the given sequence
+// number from the given session.
+func (c *ClientDB) DeleteCommittedUpdate(id *SessionID, seqNum uint16) error {
+	return kvdb.Update(c.db, func(tx kvdb.RwTx) error {
+		sessions := tx.ReadWriteBucket(cSessionBkt)
+		if sessions == nil {
+			return ErrUninitializedDB
+		}
+
+		sessionBkt := sessions.NestedReadWriteBucket(id[:])
+		if sessionBkt == nil {
+			return fmt.Errorf("session bucket %s not found",
+				id.String())
+		}
+
+		// If the commits sub-bucket doesn't exist, there can't possibly
+		// be a corresponding update to remove.
+		sessionCommits := sessionBkt.NestedReadWriteBucket(
+			cSessionCommits,
+		)
+		if sessionCommits == nil {
+			return ErrCommittedUpdateNotFound
+		}
+
+		var seqNumBuf [2]byte
+		byteOrder.PutUint16(seqNumBuf[:], seqNum)
+
+		if sessionCommits.Get(seqNumBuf[:]) == nil {
+			return ErrCommittedUpdateNotFound
+		}
+
+		// Remove the corresponding committed update.
+		return sessionCommits.Delete(seqNumBuf[:])
+	}, func() {})
+}
+
 // putChannelToSessionMapping adds the given session ID to a channel's
 // cChanSessions bucket.
 func putChannelToSessionMapping(chanDetails kvdb.RwBucket,

watchtower/wtdb/client_db_test.go

@@ -195,6 +195,15 @@ func (h *clientDBHarness) ackUpdate(id *wtdb.SessionID, seqNum uint16,
 	require.ErrorIs(h.t, err, expErr)
 }
 
+func (h *clientDBHarness) deleteCommittedUpdate(id *wtdb.SessionID,
+	seqNum uint16, expErr error) {
+
+	h.t.Helper()
+
+	err := h.db.DeleteCommittedUpdate(id, seqNum)
+	require.ErrorIs(h.t, err, expErr)
+}
+
 func (h *clientDBHarness) markChannelClosed(id lnwire.ChannelID,
 	blockHeight uint32, expErr error) []wtdb.SessionID {
 
@@ -567,7 +576,8 @@ func testChanSummaries(h *clientDBHarness) {
 	h.registerChan(chanID, expPkScript, wtdb.ErrChannelAlreadyRegistered)
 }
 
-// testCommitUpdate tests the behavior of CommitUpdate, ensuring that they can
+// testCommitUpdate tests the behavior of CommitUpdate and
+// DeleteCommittedUpdate.
 func testCommitUpdate(h *clientDBHarness) {
 	const blobType = blob.TypeAltruistCommit
 
@@ -648,6 +658,22 @@ func testCommitUpdate(h *clientDBHarness) {
 		*update1,
 		*update2,
 	}, nil)
+
+	// We will now also test that the DeleteCommittedUpdates method also
+	// works.
+	// First, try to delete a committed update that does not exist.
+	h.deleteCommittedUpdate(
+		&session.ID, update4.SeqNum, wtdb.ErrCommittedUpdateNotFound,
+	)
+
+	// Now delete an existing committed update and ensure that it succeeds.
+	h.deleteCommittedUpdate(&session.ID, update1.SeqNum, nil)
+	h.assertUpdates(session.ID, []wtdb.CommittedUpdate{
+		*update2,
+	}, nil)
+
+	h.deleteCommittedUpdate(&session.ID, update2.SeqNum, nil)
+	h.assertUpdates(session.ID, []wtdb.CommittedUpdate{}, nil)
 }
 
 // testMarkChannelClosed asserts the behaviour of MarkChannelClosed.