# HG changeset patch
# User Pulkit Goyal <7895pulkit@gmail.com>
# Date 1626648134 -19800
# Node ID 1e98f9b5bc71d212b9efcf9404855fd29c6a7aca
# Parent  df3021c1f09338d2df0e4175cabb258936ab811a
largefiles: add tr backup for largefilesdirstate

This will help us in automatically restoring the largefilesdirstate if a
transaction is rolled back.

Differential Revision: https://phab.mercurial-scm.org/D11611

diff -r df3021c1f093 -r 1e98f9b5bc71 hgext/largefiles/lfutil.py
--- a/hgext/largefiles/lfutil.py	Mon Jul 19 04:11:08 2021 +0530
+++ b/hgext/largefiles/lfutil.py	Mon Jul 19 04:12:14 2021 +0530
@@ -195,6 +195,8 @@
         # (1) disable PENDING mode always
         #     (lfdirstate isn't yet managed as a part of the transaction)
         # (2) avoid develwarn 'use dirstate.write with ....'
+        if tr:
+            tr.addbackup(b'largefiles/dirstate', location=b'plain')
         super(largefilesdirstate, self).write(None)
 
 
diff -r df3021c1f093 -r 1e98f9b5bc71 tests/test-largefiles-cache.t
--- a/tests/test-largefiles-cache.t	Mon Jul 19 04:11:08 2021 +0530
+++ b/tests/test-largefiles-cache.t	Mon Jul 19 04:12:14 2021 +0530
@@ -185,10 +185,12 @@
 
   $ find share_dst/.hg/largefiles/* | sort
   share_dst/.hg/largefiles/dirstate
+  share_dst/.hg/largefiles/undo.backup.dirstate
 
   $ find src/.hg/largefiles/* | egrep "(dirstate|$hash)" | sort
   src/.hg/largefiles/dirstate
   src/.hg/largefiles/e2fb5f2139d086ded2cb600d5a91a196e76bf020
+  src/.hg/largefiles/undo.backup.dirstate
 
 Verify that backwards compatibility is maintained for old storage layout
   $ mv src/.hg/largefiles/$hash share_dst/.hg/largefiles