# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@octobus.net>
# Date 2021-06-08 23:10:26
# Node ID e33c6dd346e73609a38c52eb71cf5cf2e3ec4c19
# Parent  e43e68624dfb17d132ebbb687ec4668008ac54df

localrepo: introduce a clone_requirements function

This function take a source repository and return a relevant set of
requirements that should be used by a copy clone.

This will help make the creation of the destination repository during copy
clone simpler.

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

diff --git a/mercurial/localrepo.py b/mercurial/localrepo.py
--- a/mercurial/localrepo.py
+++ b/mercurial/localrepo.py
@@ -3485,6 +3485,24 @@ def defaultcreateopts(ui, createopts=Non
     return createopts
 
 
+def clone_requirements(ui, createopts, srcrepo):
+    """clone the requirements of a local repo for a local clone
+
+    The store requirements are unchanged while the working copy requirements
+    depends on the configuration
+    """
+    target_requirements = set()
+    createopts = defaultcreateopts(ui, createopts=createopts)
+    for r in newreporequirements(ui, createopts):
+        if r in requirementsmod.WORKING_DIR_REQUIREMENTS:
+            target_requirements.add(r)
+
+    for r in srcrepo.requirements:
+        if r not in requirementsmod.WORKING_DIR_REQUIREMENTS:
+            target_requirements.add(r)
+    return target_requirements
+
+
 def newreporequirements(ui, createopts):
     """Determine the set of requirements for a new local repository.