# HG changeset patch
# User Siddharth Agarwal <sid0@fb.com>
# Date 2015-10-05 21:26:53
# Node ID df2dc5141721d10ec12325944bf8161c386aa807
# Parent  3f234db6fe8d7cc3d39083f7e7523444bc21e5a2

localrepo: add a way to get the current lock if it's held

We'll use this in upcoming patches to return a reference to the wlock if it's
held.

diff --git a/mercurial/localrepo.py b/mercurial/localrepo.py
--- a/mercurial/localrepo.py
+++ b/mercurial/localrepo.py
@@ -1300,6 +1300,15 @@ class localrepository(object):
         self._wlockref = weakref.ref(l)
         return l
 
+    def _currentlock(self, lockref):
+        """Returns the lock if it's held, or None if it's not."""
+        if lockref is None:
+            return None
+        l = lockref()
+        if l is None or not l.held:
+            return None
+        return l
+
     def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
         """
         commit an individual file as part of a larger transaction