From patchwork Tue Jul 11 04:57:06 2017
Content-Type: text/plain; charset="utf-8"
MIME-Version: 1.0
Content-Transfer-Encoding: 7bit
Subject: [7, of, 8, sparse,
 V2] dirstate: move customizations to rebuild() from sparse extension
From: Gregory Szorc <gregory.szorc@gmail.com>
X-Patchwork-Id: 22219
Message-Id: <f0ef4aff7559a680e216.1499749026@ubuntu-vm-main>
To: mercurial-devel@mercurial-scm.org
Date: Mon, 10 Jul 2017 21:57:06 -0700

# HG changeset patch
# User Gregory Szorc <gregory.szorc@gmail.com>
# Date 1499537680 25200
#      Sat Jul 08 11:14:40 2017 -0700
# Node ID f0ef4aff7559a680e216407e5822e3e281ef1a5f
# Parent  7b09c7c8f8457dd96f45334dad13a0ceaae92a18
dirstate: move customizations to rebuild() from sparse extension

This is a pretty straightforward port of the monkeypatch from
sparse directly into dirstate.

Unless the sparse feature is enabled via the sparse extension,
the sparse matcher will be an alwaysmatcher and the added
branch won't be taken.

diff --git a/hgext/sparse.py b/hgext/sparse.py
--- a/hgext/sparse.py
+++ b/hgext/sparse.py
@@ -223,23 +223,6 @@ def _setupdirstate(ui):
 
     replacefilecache(dirstate.dirstate, '_ignore', ignorewrapper)
 
-    # dirstate.rebuild should not add non-matching files
-    def _rebuild(orig, self, parent, allfiles, changedfiles=None):
-        matcher = self._sparsematcher
-        if not matcher.always():
-            allfiles = allfiles.matches(matcher)
-            if changedfiles:
-                changedfiles = [f for f in changedfiles if matcher(f)]
-
-            if changedfiles is not None:
-                # In _rebuild, these files will be deleted from the dirstate
-                # when they are not found to be in allfiles
-                dirstatefilestoremove = set(f for f in self if not matcher(f))
-                changedfiles = dirstatefilestoremove.union(changedfiles)
-
-        return orig(self, parent, allfiles, changedfiles)
-    extensions.wrapfunction(dirstate.dirstate, 'rebuild', _rebuild)
-
 @command('^debugsparse', [
     ('I', 'include', False, _('include files in the sparse checkout')),
     ('X', 'exclude', False, _('exclude files in the sparse checkout')),
diff --git a/mercurial/dirstate.py b/mercurial/dirstate.py
--- a/mercurial/dirstate.py
+++ b/mercurial/dirstate.py
@@ -774,6 +774,19 @@ class dirstate(object):
         self._dirty = True
 
     def rebuild(self, parent, allfiles, changedfiles=None):
+        # Filter allfiles and changedfiles through a sparse matcher if it is
+        # active.
+        matcher = self._sparsematcher
+        if not matcher.always():
+            allfiles = allfiles.matches(matcher)
+
+            if changedfiles:
+                changedfiles = [f for f in changedfiles if matcher(f)]
+
+            if changedfiles is not None:
+                toremove = set(f for f in self if not matcher(f))
+                changedfiles = toremove.union(changedfiles)
+
         if changedfiles is None:
             # Rebuild entire dirstate
             changedfiles = allfiles