walk: use match.dir in statwalk

hgext/purge.py

@@ -86,11 +86,10 b' def purge(ui, repo, *dirs, **opts):'
     files = []
     missing = []
     match = cmdutil.match(repo, dirs, opts)
+    match.dir = directories.append
     for src, f, st in repo.dirstate.statwalk(match.files(), match,
-                                             ignored=ignored, directories=True):
-        if src == 'd':
-            directories.append(f)
-        elif src == 'm':
+                                             ignored=ignored):
+        if src == 'm':
             missing.append(f)
         elif src == 'f' and f not in repo.dirstate:
             files.append(f)

mercurial/dirstate.py

@@ -422,7 +422,7 b' class dirstate(object):'
             yield f
 
     def statwalk(self, files, match, unknown=True,
-                 ignored=False, badfn=None, directories=False):
+                 ignored=False, badfn=None):
         '''
         walk recursively through the directory tree, finding all files
         matched by the match function
@@ -430,7 +430,6 b' class dirstate(object):'
         results are yielded in a tuple (src, filename, st), where src
         is one of:
         'f' the file was found in the directory tree
-        'd' the file is a directory of the tree
         'm' the file was only in the dirstate and not in the tree
 
         and st is the stat result if the file was found in the directory.
@@ -485,8 +484,8 b' class dirstate(object):'
             wadd = work.append
             found = []
             add = found.append
-            if directories:
-                add((normpath(s[common_prefix_len:]), 'd', lstat(s)))
+            if hasattr(match, 'dir'):
+                match.dir(normpath(s[common_prefix_len:]))
             while work:
                 top = work.pop()
                 entries = listdir(top, stat=True)
@@ -513,8 +512,8 b' class dirstate(object):'
                     if kind == stat.S_IFDIR:
                         if not ignore(np):
                             wadd(p)
-                            if directories:
-                                add((np, 'd', st))
+                            if hasattr(match, 'dir'):
+                                match.dir(np)
                         if np in dc and match(np):
                             add((np, 'm', st))
                     elif imatch(np):