--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/onsub.py	Tue Jul 05 10:23:32 2022 +0100
@@ -0,0 +1,195 @@
+# onsub.py - execute commands recursively on subrepositories
+#
+# Copyright 2010, 2011 aragost Trifork
+#
+# This software may be used and distributed according to the terms of
+# the GNU General Public License version 2 or any later version.
+
+import os
+from mercurial.i18n import _
+from mercurial import extensions, subrepo, util, registrar
+
+"""execute a command in each subrepository"""
+
+cmdtable = {}
+command = registrar.command(cmdtable)
+
+@command(b'onsub',
+    [(b'b', b'breadth-first', None,
+           _(b'use breadth-first traversal')),
+          (b'p', b'post-order', None,
+           _(b'use post-order depth-first traversal')),
+          (b'', b'root-repo', None,
+           _(b'include root repository in traversal')),
+          (b'', b'max-depth', -1,
+           _(b'limit recursion to N levels (negative for no limit)'), b'N'),
+          (b'', b'ignore-errors', None,
+           _(b'continue execution despite errors')),
+          (b't', b'type', b'',
+           _(b'the type of repo to filter'), b'TYPE'),
+          (b'0', b'print0', None,
+           _(b'end subrepository names with NUL, for use with xargs'))],
+         _(b'[-b] [-0] [-t TYPE] [--ignore-errors] CMD [POST-CMD]'),
+         helpcategory=command.CATEGORY_CHANGE_MANAGEMENT,
+         helpbasic=True
+    )
+def onsub(ui, repo, *args, **opts):
+    """execute a command in each subrepository
+
+    Executes CMD with the current working directory set to the root of
+    each subrepository. By default, execution stops if CMD returns a
+    non-zero exit code. Use --ignore-errors to override this.
+
+    If a POST-CMD is specified, this will be executed after all
+    subrepositories below the current subrepository has been visited.
+    This corresponds to a post-order traversal of the tree.
+
+    It is an error to specify a POST-CMD together with the
+    --breadth-first flag.
+
+    Use --verbose/-v to print the command being run and the subrepo
+    name for each run of CMD in a subrepo. Alternately, use
+    --print0/-0 to print just the subrepo name followed by a NUL
+    character instead of a newline. This can be useful in combination
+    with :hg:`status --print0`.
+
+    The command has access to the following environment variables:
+
+    ``HG_REPO``:
+        Absolute path to the top-level repository in which the onsub
+        command was executed.
+
+    ``HG_SUBPATH``:
+        Relative path to the current subrepository from the top-level
+        repository.
+
+    ``HG_SUBURL``:
+        URL for the current subrepository as specified in the
+        containing repository's ``.hgsub`` file.
+
+    ``HG_SUBSTATE``:
+        State of the current subrepository as specified in the
+        containing repository's ``.hgsubstate`` file.
+
+    ``HG_SUBTYPE``:
+        The type of the current subrepository (hg, git or svn).
+    """
+
+    # function level "constants" - these won't be modified by the nested functions
+    print0 = opts.get('print0')
+    if opts.get('ignore_errors'):
+        onerr = None
+    else:
+        onerr = util.Abort
+    maxdepth = opts.get('max_depth')
+    precmd = None
+    postcmd = None
+    includeroot = opts.get('root_repo')
+    repotypefilter = opts.get('type')
+
+    def execCmd(sub, cmd, kind):
+        """if sub == None, cmd is executed inside repo; else, inside sub.
+        If cmd == None, do nothing. If cmd == '', do only the print0 (if needed). 
+        Else, do either print0 or the debugging message, then execute the command.
+        kind is the type of the (sub)repo.
+        """
+        if sub == None:
+            envargdict = dict(HG_SUBPATH='.',
+                              HG_SUBURL='.',
+                              HG_SUBSTATE=repo['.'].hex(),
+                              HG_REPO=repo.root,
+                              HG_SUBTYPE=kind)
+            relpath = '.'
+            cmdwd = repo.root
+        else:
+            # subrepo.relpath was renamed to subrepo.subrelpath in
+            # 18b5b6392fcf.
+            if hasattr(subrepo, 'relpath'):
+                relpath = subrepo.relpath(sub)
+            else:
+                relpath = subrepo.subrelpath(sub)
+            envargdict = dict(HG_SUBPATH=relpath,
+                              HG_SUBURL=sub._path,
+                              HG_SUBSTATE=sub._state[1],
+                              HG_REPO=repo.root,
+                              HG_SUBTYPE=kind)
+            cmdwd = os.path.join(repo.root, relpath)
+        if cmd != None and (repotypefilter == '' or repotypefilter == kind):
+            if print0:
+                ui.write(relpath, "\0")
+            if cmd != '':
+                if not print0: ui.write(_("executing '%s' in %s\n") % (cmd, relpath))
+                util.system(cmd, environ=envargdict, cwd=cmdwd, onerr=onerr,
+                            errprefix=_('terminated onsub in %s') % relpath)
+
+    def bfs():
+        """execute precmd in repo.root and in each subrepository, breadth-first"""
+        if includeroot:
+            execCmd(None, precmd, 'hg') 
+        ctx = repo['.']
+        work = [(1, ctx.sub(subpath), ctx.substate[subpath][2]) for subpath in sorted(ctx.substate)]
+        while work:
+            (depth, sub, kind) = work.pop(0)
+            if depth > maxdepth >= 0:
+                continue
+            execCmd(sub, precmd, kind) 
+            if kind == 'hg':
+                rev = sub._state[1]
+                ctx = sub._repo[rev]
+                w = [(depth + 1, ctx.sub(subpath), ctx.substate[subpath][2]) 
+                     for subpath in sorted(ctx.substate)]
+                work.extend(w)
+    
+    def dfs():
+        """execute pre-/postcmd in repo.root and in each subrepository, depth-first"""
+
+        def dfs_rek(depth, sub, kind):
+            if depth > maxdepth >= 0:
+                return
+            execCmd(sub, precmd, kind) 
+            if kind == 'hg':
+                rev = sub._state[1]
+                ctx = sub._repo[rev]
+                for subpath in sorted(ctx.substate):
+                    dfs_rek(depth+1, ctx.sub(subpath), ctx.substate[subpath][2])
+            execCmd(sub, postcmd, kind)
+    
+        ctx = repo['.']
+        work = [(ctx.sub(subpath), ctx.substate[subpath][2]) for subpath in sorted(ctx.substate)]
+        if includeroot:
+            execCmd(None, precmd, 'hg') 
+        for (sub, kind) in work:
+            dfs_rek(1, sub, kind)
+        if includeroot:
+            execCmd(None, postcmd, 'hg') 
+        
+    ### start of main function part ###
+    if len(args) == 2:
+        precmd = args[0]
+        postcmd = args[1]
+        if opts.get('breadth_first') or opts.get('post_order'):
+            raise util.Abort(_("onsub: '-b' and '-p' imply the use of only one command"))
+    elif len(args) == 1:
+        if opts.get('post_order'):
+            precmd = None
+            postcmd = args[0]
+        else:
+            precmd = args[0]
+            postcmd = None
+    elif len(args) == 0:
+        # cmd == '' means only do print0
+        if opts.get('post_order'):
+            precmd = None
+            postcmd = ''
+        else:
+            precmd = ''
+            postcmd = None
+    else:
+        raise util.Abort(_("onsub: at most 2 command arguments required"))
+    if opts.get('post_order') and opts.get('breadth_first'):
+        raise util.Abort(_("onsub: '-b' and '-p' are mutually exclusive"))
+
+    if opts.get('breadth_first'):
+        bfs()
+    else:
+        dfs()