Merge lp:~jelmer/brz/depth into lp:brz/3.3

[Jelmer Vernooij]

Add a depth argument to ControlDir.sprout() and Repository.fetch().

Add a depth argument to ControlDir.sprout() and Repository.fetch().

[Martin Packman (gz)]

Thanks!

review: Approve

[Jelmer Vernooij (jelmer)]

review: Approve

[Martin Packman (gz)]

Restamping this one for clarity.

review: Approve

[The Breezy Bot (the-breezy-bot)]

Running landing tests failed
https://ci.breezy-vcs.org/job/brz/job/brz-land/255/

[The Breezy Bot (the-breezy-bot)]

Running landing tests failed
https://ci.breezy-vcs.org/job/brz/job/brz-land/267/

[Jelmer Vernooij (jelmer)]

This is blocked on support for depth between local repositories in Dulwich.

[The Breezy Bot (the-breezy-bot)]

Merging failed
https://ci.breezy-vcs.org/job/brz/job/brz-land/750/

[The Breezy Bot (the-breezy-bot)]

Merging failed
https://ci.breezy-vcs.org/job/brz/job/brz-land/958/

[The Breezy Bot (the-breezy-bot)]

Merging failed
https://ci.breezy-vcs.org/job/brz/job/brz-land/995/

[Jelmer Vernooij (jelmer)]

review: Approve

[The Breezy Bot (the-breezy-bot)]

The attempt to merge lp:~jelmer/brz/depth into lp:brz/3.3 failed. Command exited with 1.
Below is the output from the failed tests.

Collecting setuptools-gettext
  Downloading setuptools_gettext-0.1.1-py3-none-any.whl (10 kB)
Requirement already satisfied: setuptools>=46.1 in ./lib/python3.10/site-packages (from setuptools-gettext) (65.5.0)
Installing collected packages: setuptools-gettext
Successfully installed setuptools-gettext-0.1.1
Obtaining file:///tmp/tarmac/branch.53e2n39n
  Installing build dependencies: started
  Installing build dependencies: finished with status 'done'
  Checking if build backend supports build_editable: started
  Checking if build backend supports build_editable: finished with status 'done'
  Getting requirements to build editable: started
  Getting requirements to build editable: finished with status 'done'
  Preparing editable metadata (pyproject.toml): started
  Preparing editable metadata (pyproject.toml): finished with status 'done'
Requirement already satisfied: pyyaml in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (5.4.1)
Requirement already satisfied: fastbencode in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (0.0.12)
Requirement already satisfied: urllib3>=1.24.1 in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (1.26.12)
Requirement already satisfied: dulwich>=0.20.46 in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (0.20.46)
Requirement already satisfied: merge3 in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (0.0.8)
Requirement already satisfied: configobj in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (5.0.6)
Requirement already satisfied: patiencediff in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (0.2.3)
Requirement already satisfied: gpg in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (1.18.0)
Collecting sphinx
  Downloading sphinx-5.3.0-py3-none-any.whl (3.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 3.2/3.2 MB 6.7 MB/s eta 0:00:00
Requirement already satisfied: setuptools in ./lib/python3.10/site-packages (from breezy==3.3.1.dev0) (65.5.0)
Collecting sphinx-epytext
  Downloading sphinx-epytext-0.0.4.tar.gz (3.6 kB)
  Preparing metadata (setup.py): started
  Preparing metadata (setup.py): finished with status 'done'
Collecting fastimport
  Downloading fastimport-0.9.14.tar.gz (41 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 41.8/41.8 kB 363.1 kB/s eta 0:00:00
  Preparing metadata (setup.py): started
  Preparing metadata (setup.py): finished with status 'done'
Requirement already satisfied: launchpadlib>=1.6.3 in /usr/lib/python3/dist-packages (from breezy==3.3.1.dev0) (1.10.17)
Collecting testtools
  Downloading testtools-2.5.0-py3-none-any.whl (181 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 182.0/182.0 kB 1.2 MB/s eta 0:00:00
Collecting python-subunit
  Downloading python_subunit-1.4.0-py2.py3-none-any.whl (106 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 106.1/106.1 kB 615.5 kB/s eta 0:00:00
Requirement already satisfied: httplib2 in /usr/lib/python3/dist-packages (from launchpadlib>=1.6.3->breezy==3.3.1.dev0) (0.20.4)
Collecting keyring
  Downloading keyring-23.9.3-py3-none-any....

Unmerged revisions

7158. By Jelmer Vernooij on 2022-10-27

Merge lp:brz/3.3

7157. By Jelmer Vernooij on 2022-09-16

Merge trunk.

7156. By Jelmer Vernooij on 2022-07-08

Merge lp:brz/3.2.

7155. By Jelmer Vernooij on 2020-06-01

Merge lp:brz/3.1.

7154. By Jelmer Vernooij on 2019-03-04

Fix tests.

7153. By Jelmer Vernooij on 2019-02-10

add depth argument to ControlDir.sprout.

7152. By Jelmer Vernooij on 2019-02-10

Enable depth fetching for Git repositories.

7151. By Jelmer Vernooij on 2019-02-09

For the moment, Dulwich doesn't support depth fetching locally.

7150. By Jelmer Vernooij on 2019-02-09

Add Repository.support_fetch_depth.

7149. By Jelmer Vernooij on 2019-02-09

Merge dulwich-compat.

=== modified file 'breezy/branch.py'
--- breezy/branch.py	2022-09-12 13:37:25 +0000
+++ breezy/branch.py	2022-10-27 12:03:56 +0000
@@ -680,18 +680,20 @@
             raise errors.UpgradeRequired(self.user_url)
         self.get_config_stack().set('append_revisions_only', enabled)
 
-    def fetch(self, from_branch, stop_revision=None, limit=None, lossy=False):
+    def fetch(self, from_branch, stop_revision=None, limit=None, lossy=False,
+              depth=None):
         """Copy revisions from from_branch into this branch.
 
         :param from_branch: Where to copy from.
         :param stop_revision: What revision to stop at (None for at the end
                               of the branch.
         :param limit: Optional rough limit of revisions to fetch
+        :param depth: Revision depth
         :return: None
         """
         with self.lock_write():
             return InterBranch.get(from_branch, self).fetch(
-                stop_revision, limit=limit, lossy=lossy)
+                stop_revision, limit=limit, lossy=lossy, depth=depth)
 
     def get_bound_location(self):
         """Return the URL of the branch we are bound to.
@@ -2105,11 +2107,12 @@
         """
         raise NotImplementedError(self.copy_content_into)
 
-    def fetch(self, stop_revision=None, limit=None, lossy=False):
+    def fetch(self, stop_revision=None, limit=None, lossy=False, depth=None):
         """Fetch revisions.
 
         :param stop_revision: Last revision to fetch
         :param limit: Optional rough limit of revisions to fetch
+        :param depth: Optional revision depth
         :return: FetchResult object
         """
         raise NotImplementedError(self.fetch)
@@ -2167,9 +2170,11 @@
             if self.source._push_should_merge_tags():
                 self.source.tags.merge_to(self.target.tags, selector=tag_selector)
 
-    def fetch(self, stop_revision=None, limit=None, lossy=False):
+    def fetch(self, stop_revision=None, limit=None, lossy=False, depth=None):
         if self.target.base == self.source.base:
             return (0, [])
+        if depth is not None:
+            raise errors.FetchDepthUnsupported(self)
         with self.source.lock_read(), self.target.lock_write():
             fetch_spec_factory = fetch.FetchSpecFactory()
             fetch_spec_factory.source_branch = self.source

=== modified file 'breezy/bzr/bzrdir.py'
--- breezy/bzr/bzrdir.py	2022-08-22 18:19:46 +0000
+++ breezy/bzr/bzrdir.py	2022-10-27 12:03:56 +0000
@@ -352,7 +352,7 @@
                recurse='down', possible_transports=None,
                accelerator_tree=None, hardlink=False, stacked=False,
                source_branch=None, create_tree_if_local=True,
-               lossy=False):
+               depth=None):
         """Create a copy of this controldir prepared for use as a new line of
         development.
 
@@ -376,8 +376,12 @@
             location of this control directory.
         :param create_tree_if_local: If true, a working-tree will be created
             when working locally.
+        :param depth: Optional fetch depth
         :return: The created control directory
         """
+        if depth is not None:
+            raise errors.FetchDepthUnsupported(self)
+
         with contextlib.ExitStack() as stack:
             fetch_spec_factory = fetch.FetchSpecFactory()
             if revision_id is not None:

=== modified file 'breezy/controldir.py'
--- breezy/controldir.py	2022-09-12 18:35:44 +0000
+++ breezy/controldir.py	2022-10-27 12:03:56 +0000
@@ -414,7 +414,7 @@
                recurse='down', possible_transports=None,
                accelerator_tree=None, hardlink=False, stacked=False,
                source_branch=None, create_tree_if_local=True,
-               lossy=False):
+               depth=None):
         """Create a copy of this controldir prepared for use as a new line of
         development.
 
@@ -438,6 +438,7 @@
             location of this control directory.
           create_tree_if_local: If true, a working-tree will be created
             when working locally.
+        :param depth: Possible fetch depth
         """
         raise NotImplementedError(self.sprout)
 

=== modified file 'breezy/errors.py'
--- breezy/errors.py	2022-08-22 18:19:46 +0000
+++ breezy/errors.py	2022-10-27 12:03:56 +0000
@@ -1658,6 +1658,14 @@
         self.tname = type(method_self).__name__
 
 
+class FetchDepthUnsupported(UnsupportedOperation):
+
+    fmt = ("InterBranch %(interbranch)r does not support fetching depths.")
+
+    def __init__(self, interbranch):
+        BzrError.__init__(self, interbranch=interbranch)
+
+
 class FetchLimitUnsupported(UnsupportedOperation):
 
     fmt = ("InterBranch %(interbranch)r does not support fetching limits.")

=== modified file 'breezy/git/branch.py'
--- breezy/git/branch.py	2022-08-28 18:54:11 +0000
+++ breezy/git/branch.py	2022-10-27 12:03:56 +0000
@@ -1007,12 +1007,15 @@
             return False
         return True
 
-    def fetch(self, stop_revision=None, fetch_tags=None, limit=None, lossy=False):
+    def fetch(self, stop_revision=None, fetch_tags=None, limit=None,
+              lossy=False, depth=None):
         self.fetch_objects(
-            stop_revision, fetch_tags=fetch_tags, limit=limit, lossy=lossy)
+            stop_revision, fetch_tags=fetch_tags, limit=limit, lossy=lossy,
+            depth=depth)
         return _mod_repository.FetchResult()
 
-    def fetch_objects(self, stop_revision, fetch_tags, limit=None, lossy=False, tag_selector=None):
+    def fetch_objects(self, stop_revision, fetch_tags, limit=None, lossy=False,
+                      tag_selector=None, depth=None):
         interrepo = self._get_interrepo(self.source, self.target)
         if fetch_tags is None:
             c = self.source.get_config_stack()
@@ -1033,7 +1036,7 @@
                 [self._last_revid], include_tags=fetch_tags, tag_selector=tag_selector)
             return real(heads)
         pack_hint, head, refs = interrepo.fetch_objects(
-            determine_wants, self.source.mapping, limit=limit,
+            determine_wants, self.source.mapping, limit=limit, depth=depth,
             lossy=lossy)
         if (pack_hint is not None and
                 self.target.repository._format.pack_compresses):
@@ -1168,7 +1171,8 @@
 class InterGitBranch(branch.GenericInterBranch):
     """InterBranch implementation that pulls between Git branches."""
 
-    def fetch(self, stop_revision=None, fetch_tags=None, limit=None, lossy=False):
+    def fetch(self, stop_revision=None, fetch_tags=None, limit=None, lossy=False,
+              depth=None):
         raise NotImplementedError(self.fetch)
 
 
@@ -1248,7 +1252,8 @@
         return (isinstance(source, GitBranch) and
                 isinstance(target, LocalGitBranch))
 
-    def fetch(self, stop_revision=None, fetch_tags=None, limit=None, lossy=False):
+    def fetch(self, stop_revision=None, fetch_tags=None, limit=None, lossy=False,
+              depth=None):
         interrepo = _mod_repository.InterRepository.get(
             self.source.repository, self.target.repository)
         if stop_revision is None:
@@ -1258,7 +1263,8 @@
             fetch_tags = c.get('branch.fetch_tags')
         determine_wants = interrepo.get_determine_wants_revids(
             [stop_revision], include_tags=fetch_tags)
-        interrepo.fetch_objects(determine_wants, limit=limit, lossy=lossy)
+        interrepo.fetch_objects(
+            determine_wants, limit=limit, lossy=lossy, depth=depth)
         return _mod_repository.FetchResult()
 
     def _basic_push(self, overwrite=False, stop_revision=None, tag_selector=None):
@@ -1455,7 +1461,7 @@
         return ret
 
     def fetch(self, stop_revision=None, fetch_tags=None, lossy=False,
-              limit=None):
+              limit=None, depth=None):
         if stop_revision is None:
             stop_revision = self.source.last_revision()
         ret = []
@@ -1465,7 +1471,8 @@
         ret.append((None, stop_revision))
         if getattr(self.interrepo, 'fetch_revs', None):
             try:
-                revidmap = self.interrepo.fetch_revs(ret, lossy=lossy, limit=limit)
+                revidmap = self.interrepo.fetch_revs(
+                    ret, lossy=lossy, limit=limit, depth=depth)
             except NoPushSupport:
                 raise errors.NoRoundtrippingSupport(self.source, self.target)
             return _mod_repository.FetchResult(revidmap={
@@ -1481,7 +1488,7 @@
                 return wants
 
             self.interrepo.fetch_objects(
-                determine_wants, lossy=lossy, limit=limit)
+                determine_wants, lossy=lossy, limit=limit, depth=depth)
             return _mod_repository.FetchResult()
 
     def pull(self, overwrite=False, stop_revision=None, local=False,

=== modified file 'breezy/git/dir.py'
--- breezy/git/dir.py	2022-08-22 18:19:46 +0000
+++ breezy/git/dir.py	2022-10-27 12:03:56 +0000
@@ -157,7 +157,7 @@
     def sprout(self, url, revision_id=None, force_new_repo=False,
                recurse='down', possible_transports=None,
                accelerator_tree=None, hardlink=False, stacked=False,
-               source_branch=None, create_tree_if_local=True):
+               source_branch=None, create_tree_if_local=True, depth=None):
         from ..repository import InterRepository
         from ..transport.local import LocalTransport
         from ..transport import get_transport
@@ -187,7 +187,8 @@
         else:
             determine_wants = interrepo.determine_wants_all
         interrepo.fetch_objects(determine_wants=determine_wants,
-                                mapping=source_branch.mapping)
+                                mapping=source_branch.mapping,
+                                depth=depth)
         result_branch = source_branch.sprout(
             result, revision_id=revision_id, repository=result_repo)
         if (create_tree_if_local and

=== modified file 'breezy/git/fetch.py'
--- breezy/git/fetch.py	2022-08-22 18:19:46 +0000
+++ breezy/git/fetch.py	2022-10-27 12:03:56 +0000
@@ -453,7 +453,7 @@
 
 def import_git_objects(repo, mapping, object_iter,
                        target_git_object_retriever, heads, pb=None,
-                       limit=None):
+                       limit=None, depth=None):
     """Import a set of git objects into a bzr repository.
 
     :param repo: Target Bazaar repository
@@ -461,6 +461,8 @@
     :param object_iter: Iterator over Git objects.
     :return: Tuple with pack hints and last imported revision id
     """
+    if depth is None:
+        raise NotImplementedError
     def lookup_object(sha):
         try:
             return object_iter[sha]

=== modified file 'breezy/git/interrepo.py'
--- breezy/git/interrepo.py	2022-01-16 17:15:16 +0000
+++ breezy/git/interrepo.py	2022-10-27 12:03:56 +0000
@@ -37,6 +37,7 @@
 
 from ..errors import (
     DivergedBranches,
+    FetchDepthUnsupported,
     FetchLimitUnsupported,
     InvalidRevisionId,
     LossyPushToSameVCS,
@@ -121,9 +122,9 @@
         """
         raise NotImplementedError(self.fetch_refs)
 
-    def search_missing_revision_ids(self,
-                                    find_ghosts=True, revision_ids=None,
-                                    if_present_ids=None, limit=None):
+    def search_missing_revision_ids(
+            self, find_ghosts=True, revision_ids=None, if_present_ids=None,
+            limit=None, depth=None):
         if limit is not None:
             raise FetchLimitUnsupported(self)
         git_shas = []
@@ -187,7 +188,7 @@
                 return False
         return self._commit_needs_fetching(sha_id)
 
-    def missing_revisions(self, stop_revisions):
+    def missing_revisions(self, stop_revisions, depth=None):
         """Find the revisions that are missing from the target repository.
 
         :param stop_revisions: Revisions to check for (tuples with
@@ -201,31 +202,34 @@
         for (sha1, revid) in stop_revisions:
             if sha1 is not None and revid is not None:
                 revid_sha_map[revid] = sha1
-                stop_revids.append(revid)
+                stop_revids.append((revid, 1))
             elif sha1 is not None:
                 if self._commit_needs_fetching(sha1):
                     for (kind, (revid, tree_sha, verifiers)) in self.source_store.lookup_git_sha(sha1):
                         revid_sha_map[revid] = sha1
-                        stop_revids.append(revid)
+                        stop_revids.append((revid, 1))
             else:
                 if revid is None:
                     raise AssertionError
-                stop_revids.append(revid)
+                stop_revids.append((revid, 1))
         missing = set()
         graph = self.source.get_graph()
         with ui.ui_factory.nested_progress_bar() as pb:
             while stop_revids:
-                new_stop_revids = []
-                for revid in stop_revids:
+                new_stop_revids = {}
+                for revid, revid_depth in stop_revids:
                     sha1 = revid_sha_map.get(revid)
                     if (revid not in missing and
                             self._revision_needs_fetching(sha1, revid)):
                         missing.add(revid)
-                        new_stop_revids.append(revid)
+                        if depth is None or revid_depth < depth:
+                            new_stop_revids[revid] = revid_depth
                 stop_revids = set()
-                parent_map = graph.get_parent_map(new_stop_revids)
-                for parent_revids in parent_map.values():
-                    stop_revids.update(parent_revids)
+                parent_map = graph.get_parent_map(new_stop_revids.keys())
+                for revid, parent_revids in parent_map.items():
+                    stop_revids.update(
+                        [(parent_revid, new_stop_revids[revid] + 1)
+                         for parent_revid in parent_revids])
                 pb.update("determining revisions to fetch", len(missing))
         return graph.iter_topo_order(missing)
 
@@ -286,7 +290,7 @@
                     result_refs[name] = (gitid, revid if not lossy else self.mapping.revision_id_foreign_to_bzr(gitid))
         return revidmap, old_refs, result_refs
 
-    def fetch_revs(self, revs, lossy, limit=None):
+    def fetch_revs(self, revs, lossy, limit=None, depth=None):
         if not lossy and not self.mapping.roundtripping:
             for git_sha, bzr_revid in revs:
                 if (bzr_revid is not None and
@@ -294,7 +298,7 @@
                     raise NoPushSupport(self.source, self.target, self.mapping,
                                         bzr_revid)
         with self.source_store.lock_read():
-            todo = list(self.missing_revisions(revs))[:limit]
+            todo = list(self.missing_revisions(revs, depth=depth))[:limit]
             revidmap = {}
             with ui.ui_factory.nested_progress_bar() as pb:
                 object_generator = MissingObjectsIterator(
@@ -315,7 +319,7 @@
                 return revidmap
 
     def fetch(self, revision_id=None, pb=None, find_ghosts=False,
-              fetch_spec=None, mapped_refs=None, lossy=False):
+              fetch_spec=None, mapped_refs=None, lossy=False, depth=None):
         if mapped_refs is not None:
             stop_revisions = mapped_refs
         elif revision_id is not None:
@@ -332,7 +336,7 @@
                               for revid in self.source.all_revision_ids()]
         self._warn_slow()
         try:
-            revidmap = self.fetch_revs(stop_revisions, lossy=lossy)
+            revidmap = self.fetch_revs(stop_revisions, lossy=lossy, depth=depth)
         except NoPushSupport:
             raise NoRoundtrippingSupport(self.source, self.target)
         return FetchResult(revidmap)
@@ -501,7 +505,8 @@
                 'Fetching from Git to Bazaar repository. '
                 'For better performance, fetch into a Git repository.')
 
-    def fetch_objects(self, determine_wants, mapping, limit=None, lossy=False):
+    def fetch_objects(self, determine_wants, mapping, limit=None, lossy=False,
+                      depth=None):
         """Fetch objects from a remote server.
 
         :param determine_wants: determine_wants callback
@@ -523,7 +528,8 @@
             wants, include_tags=include_tags, tag_selector=tag_selector)
 
     def fetch(self, revision_id=None, find_ghosts=False,
-              mapping=None, fetch_spec=None, include_tags=False, lossy=False):
+              mapping=None, fetch_spec=None, include_tags=False,
+              lossy=False, depth=None):
         if mapping is None:
             mapping = self.source.get_mapping()
         if revision_id is not None:
@@ -545,7 +551,7 @@
             determine_wants = self.determine_wants_all
 
         (pack_hint, _, remote_refs) = self.fetch_objects(
-            determine_wants, mapping, lossy=lossy)
+            determine_wants, mapping, lossy=lossy, depth=depth)
         if pack_hint is not None and self.target._format.pack_compresses:
             self.target.pack(hint=pack_hint)
         result = FetchResult()
@@ -566,7 +572,8 @@
             all_parents.update(values)
         return set(all_revs) - all_parents
 
-    def fetch_objects(self, determine_wants, mapping, limit=None, lossy=False):
+    def fetch_objects(self, determine_wants, mapping, limit=None, lossy=False,
+                      depth=None):
         """See `InterGitNonGitRepository`."""
         self._warn_slow()
         store = get_object_store(self.target, mapping)
@@ -579,7 +586,8 @@
 
             with ui.ui_factory.nested_progress_bar() as pb:
                 objects_iter = self.source.fetch_objects(
-                    wants_recorder, graph_walker, store.get_raw)
+                    wants_recorder, graph_walker, store.get_raw,
+                    depth=depth)
                 trace.mutter("Importing %d new revisions",
                              len(wants_recorder.wants))
                 (pack_hint, last_rev) = import_git_objects(
@@ -605,21 +613,22 @@
     """InterRepository that copies revisions from a local Git into a non-Git
     repository."""
 
-    def fetch_objects(self, determine_wants, mapping, limit=None, lossy=False):
+    def fetch_objects(self, determine_wants, mapping, limit=None, lossy=False,
+                      depth=None):
         """See `InterGitNonGitRepository`."""
+        if depth is not None:
+            raise FetchDepthUnsupported(self)
         self._warn_slow()
         remote_refs = self.source.controldir.get_refs_container().as_dict()
         wants = determine_wants(remote_refs)
         target_git_object_retriever = get_object_store(self.target, mapping)
-        with ui.ui_factory.nested_progress_bar() as pb:
-            target_git_object_retriever.lock_write()
-            try:
-                (pack_hint, last_rev) = import_git_objects(
-                    self.target, mapping, self.source._git.object_store,
-                    target_git_object_retriever, wants, pb, limit)
-                return (pack_hint, last_rev, remote_refs)
-            finally:
-                target_git_object_retriever.unlock()
+        with ui.ui_factory.nested_progress_bar() as pb, \
+                target_git_object_retriever.lock_write():
+            (pack_hint, last_rev) = import_git_objects(
+                self.target, mapping, self.source._git.object_store,
+                target_git_object_retriever, wants, pb, limit=limit,
+                depth=depth)
+            return (pack_hint, last_rev, remote_refs)
 
     @staticmethod
     def is_compatible(source, target):
@@ -657,7 +666,7 @@
         return None, old_refs, new_refs
 
     def fetch_objects(self, determine_wants, mapping=None, limit=None,
-                      lossy=False):
+                      lossy=False, depth=None):
         raise NotImplementedError(self.fetch_objects)
 
     def _target_has_shas(self, shas):
@@ -666,7 +675,7 @@
 
     def fetch(self, revision_id=None, find_ghosts=False,
               mapping=None, fetch_spec=None, branches=None, limit=None,
-              include_tags=False, lossy=False):
+              include_tags=False, lossy=False, depth=None):
         if mapping is None:
             mapping = self.source.get_mapping()
         if revision_id is not None:
@@ -688,7 +697,8 @@
             determine_wants = self.get_determine_wants_revids(
                 args, include_tags=include_tags)
         wants_recorder = DetermineWantsRecorder(determine_wants)
-        self.fetch_objects(wants_recorder, mapping, limit=limit, lossy=lossy)
+        self.fetch_objects(
+            wants_recorder, mapping, limit=limit, lossy=lossy, depth=depth)
         result = FetchResult()
         result.refs = wants_recorder.remote_refs
         return result
@@ -727,7 +737,7 @@
 class InterLocalGitLocalGitRepository(InterGitGitRepository):
 
     def fetch_objects(self, determine_wants, mapping=None, limit=None,
-                      lossy=False):
+                      lossy=False, depth=None):
         if lossy:
             raise LossyPushToSameVCS(self.source, self.target)
         if limit is not None:
@@ -737,7 +747,7 @@
             progress = DefaultProgressReporter(pb).progress
             refs = self.source._git.fetch(
                 self.target._git, determine_wants,
-                progress=progress)
+                progress=progress, depth=depth)
         return (None, None, refs)
 
     @staticmethod
@@ -750,7 +760,7 @@
 class InterRemoteGitLocalGitRepository(InterGitGitRepository):
 
     def fetch_objects(self, determine_wants, mapping=None, limit=None,
-                      lossy=False):
+                      lossy=False, depth=None):
         if lossy:
             raise LossyPushToSameVCS(self.source, self.target)
         if limit is not None:
@@ -772,10 +782,12 @@
         else:
             f, commit, abort = self.target._git.object_store.add_pack()
         try:
-            refs = self.source.controldir.fetch_pack(
-                determine_wants, graphwalker, f.write)
+            fetch_result = self.source.controldir.fetch_pack(
+                determine_wants, graphwalker, f.write, depth=depth)
             commit()
-            return (None, None, refs)
+            self.target._git.update_shallow(
+                fetch_result.new_shallow, fetch_result.new_unshallow)
+            return (None, None, fetch_result.refs)
         except BaseException:
             abort()
             raise

=== modified file 'breezy/git/remote.py'
--- breezy/git/remote.py	2022-09-14 19:01:01 +0000
+++ breezy/git/remote.py	2022-10-27 12:03:56 +0000
@@ -475,8 +475,8 @@
             if pb is not None:
                 pb.finished()
 
-    def fetch_pack(self, determine_wants, graph_walker, pack_data,
-                   progress=None):
+    def fetch_pack(self, determine_wants, graph_walker, pack_data, progress=None,
+                   depth=None):
         if progress is None:
             pb = ui.ui_factory.nested_progress_bar()
             progress = DefaultProgressReporter(pb).progress
@@ -485,7 +485,7 @@
         try:
             result = self._client.fetch_pack(
                 self._client_path, determine_wants, graph_walker, pack_data,
-                progress)
+                progress, depth=depth)
             if result.refs is None:
                 result.refs = {}
             self._refs = remote_refs_dict_to_container(
@@ -631,9 +631,10 @@
     def get_refs_container(self):
         if self._refs is not None:
             return self._refs
-        result = self.fetch_pack(lambda x: None, None,
-                                 lambda x: None,
-                                 lambda x: trace.mutter("git: %s" % x))
+        result = self.fetch_pack(
+            lambda x: None, None,
+            lambda x: None, lambda x: trace.mutter("git: %s" % x),
+            depth=None)
         self._refs = remote_refs_dict_to_container(
             result.refs, result.symrefs)
         return self._refs
@@ -995,20 +996,21 @@
         return self.controldir.archive(*args, **kwargs)
 
     def fetch_pack(self, determine_wants, graph_walker, pack_data,
-                   progress=None):
+                   progress=None, depth=None):
         return self.controldir.fetch_pack(
-            determine_wants, graph_walker, pack_data, progress)
+            determine_wants, graph_walker, pack_data, progress, depth=depth)
 
     def send_pack(self, get_changed_refs, generate_pack_data):
         return self.controldir.send_pack(get_changed_refs, generate_pack_data)
 
     def fetch_objects(self, determine_wants, graph_walker, resolve_ext_ref,
-                      progress=None):
+                      progress=None, depth=None):
         import tempfile
         fd, path = tempfile.mkstemp(suffix=".pack")
         try:
-            self.fetch_pack(determine_wants, graph_walker,
-                            lambda x: os.write(fd, x), progress)
+            self.fetch_pack(
+                determine_wants, graph_walker,
+                lambda x: os.write(fd, x), progress, depth=depth)
         finally:
             os.close(fd)
         if os.path.getsize(path) == 0:

=== modified file 'breezy/git/repository.py'
--- breezy/git/repository.py	2022-01-24 20:22:05 +0000
+++ breezy/git/repository.py	2022-10-27 12:03:56 +0000
@@ -118,6 +118,7 @@
     _serializer = None
     vcs = foreign_vcs_git
     chk_bytes = None
+    supports_fetch_depth = True
 
     def __init__(self, gitdir):
         self._transport = gitdir.root_transport

=== modified file 'breezy/plugins/weave_fmt/bzrdir.py'
--- breezy/plugins/weave_fmt/bzrdir.py	2022-08-22 18:19:46 +0000
+++ breezy/plugins/weave_fmt/bzrdir.py	2022-10-27 12:03:56 +0000
@@ -890,8 +890,10 @@
     def sprout(self, url, revision_id=None, force_new_repo=False,
                recurse=None, possible_transports=None, accelerator_tree=None,
                hardlink=False, stacked=False, create_tree_if_local=True,
-               source_branch=None):
+               source_branch=None, depth=None):
         """See ControlDir.sprout()."""
+        if depth is not None:
+            raise errors.FetchDepthUnsupported(self)
         if source_branch is not None:
             my_branch = self.open_branch()
             if source_branch.base != my_branch.base:

=== modified file 'breezy/repository.py'
--- breezy/repository.py	2022-07-02 13:44:01 +0000
+++ breezy/repository.py	2022-10-27 12:03:56 +0000
@@ -281,6 +281,10 @@
     # items in the tree, or just bulk fetching/pushing of data?
     supports_random_access = True
 
+    # Does this repository implementation support fetching with
+    # a certain graph depth?
+    supports_fetch_depth = False
+
     def abort_write_group(self, suppress_errors=False):
         """Commit the contents accrued within the current write group.
 

=== modified file 'breezy/tests/per_controldir/test_controldir.py'
--- breezy/tests/per_controldir/test_controldir.py	2022-08-22 18:19:46 +0000
+++ breezy/tests/per_controldir/test_controldir.py	2022-10-27 12:03:56 +0000
@@ -87,7 +87,7 @@
 
     def sproutOrSkip(self, from_bzrdir, to_url, revision_id=None,
                      force_new_repo=False, accelerator_tree=None,
-                     create_tree_if_local=True):
+                     create_tree_if_local=True, depth=None):
         """Sprout from_bzrdir into to_url, or raise TestSkipped.
 
         A simple wrapper for from_bzrdir.sprout that translates NotLocalUrl into
@@ -100,7 +100,8 @@
                                     force_new_repo=force_new_repo,
                                     possible_transports=[to_transport],
                                     accelerator_tree=accelerator_tree,
-                                    create_tree_if_local=create_tree_if_local)
+                                    create_tree_if_local=create_tree_if_local,
+                                    depth=depth)
         return target
 
     def test_uninitializable(self):
@@ -1065,6 +1066,20 @@
         self.addCleanup(repo.lock_read().unlock)
         self.assertEqual(None, repo.get_parent_map([rev1]).get(rev1))
 
+    def test_sprout_with_depth(self):
+        tree = self.make_branch_and_tree('source')
+        self.build_tree(['source/foo'])
+        tree.add('foo')
+        tree.commit('revision 1')
+        rev2 = tree.commit('revision 2', allow_pointless=True)
+        dir = tree.controldir
+        try:
+            target = self.sproutOrSkip(dir, self.get_url('target'), depth=1)
+        except errors.FetchDepthUnsupported:
+            self.assertFalse(tree.branch.repository.supports_fetch_depth)
+        else:
+            self.assertEqual({rev2}, target.open_repository().all_revision_ids())
+
     def test_format_initialize_find_open(self):
         # loopback test to check the current format initializes to itself.
         if not self.bzrdir_format.is_initializable():

=== modified file 'breezy/tests/per_interbranch/test_fetch.py'
--- breezy/tests/per_interbranch/test_fetch.py	2018-11-11 04:08:32 +0000
+++ breezy/tests/per_interbranch/test_fetch.py	2022-10-27 12:03:56 +0000
@@ -16,7 +16,11 @@
 
 """Tests for InterBranch.fetch."""
 
-from breezy.errors import FetchLimitUnsupported, NoRoundtrippingSupport
+from breezy.errors import (
+    FetchDepthUnsupported,
+    FetchLimitUnsupported,
+    NoRoundtrippingSupport,
+    )
 from breezy.revision import NULL_REVISION
 from breezy.tests import TestNotApplicable
 from breezy.tests.per_interbranch import (
@@ -113,3 +117,39 @@
         self.assertEqual(
             {rev1, rev2},
             b2.repository.has_revisions([rev1, rev2, rev3]))
+
+    def test_fetch_revisions_depth(self):
+        """Test fetch-revision operation."""
+        builder = self.make_branch_builder(
+            'b1', format=self.branch_format_from._matchingcontroldir)
+        builder.start_series()
+        rev1 = builder.build_commit()
+        rev2 = builder.build_commit()
+        rev3 = builder.build_commit()
+        builder.finish_series()
+        b1 = builder.get_branch()
+        b2 = self.make_to_branch('b2')
+        try:
+            if b2.repository.supports_fetch_depth:
+                b2.fetch(b1, depth=1)
+            else:
+                self.assertRaises(FetchDepthUnsupported, b2.fetch, b1, depth=1)
+                raise TestNotApplicable(
+                    'interbranch does not support fetch depths')
+        except NoRoundtrippingSupport:
+            raise TestNotApplicable(
+                'lossless cross-vcs fetch %r to %r not supported' %
+                (b1, b2))
+
+        self.assertEqual(
+            {rev3},
+            b2.repository.has_revisions([rev1, rev2, rev3]))
+
+        # fetch does not update the last revision
+        self.assertEqual(NULL_REVISION, b2.last_revision())
+
+        # Incrementally fetch one more
+        b2.fetch(b1, depth=2)
+
+        self.assertEqual(
+            {rev2, rev3}, b2.repository.has_revisions([rev1, rev2, rev3]))

=== modified file 'breezy/tests/per_repository/test_repository.py'
--- breezy/tests/per_repository/test_repository.py	2022-07-02 13:44:01 +0000
+++ breezy/tests/per_repository/test_repository.py	2022-10-27 12:03:56 +0000
@@ -126,6 +126,9 @@
     def test_attribute_format_supports_random_access(self):
         self.assertRepositoryAttribute('supports_random_access', (True, False))
 
+    def test_attribute_format_supports_fetch_depth(self):
+        self.assertRepositoryAttribute('supports_fetch_depth', (True, False))
+
     def test_attribute_format_supports_setting_revision_ids(self):
         self.assertFormatAttribute('supports_setting_revision_ids',
                                    (True, False))

=== modified file 'breezy/tests/test_foreign.py'
--- breezy/tests/test_foreign.py	2022-08-22 18:19:46 +0000
+++ breezy/tests/test_foreign.py	2022-10-27 12:03:56 +0000
@@ -340,14 +340,15 @@
     def sprout(self, url, revision_id=None, force_new_repo=False,
                recurse='down', possible_transports=None,
                accelerator_tree=None, hardlink=False, stacked=False,
-               source_branch=None):
+               source_branch=None, depth=None):
         # dirstate doesn't cope with accelerator_trees well
         # that have a different control dir
         return super(DummyForeignVcsDir, self).sprout(
             url=url,
             revision_id=revision_id, force_new_repo=force_new_repo,
             recurse=recurse, possible_transports=possible_transports,
-            hardlink=hardlink, stacked=stacked, source_branch=source_branch)
+            hardlink=hardlink, stacked=stacked, source_branch=source_branch,
+            depth=depth)
 
 
 def register_dummy_foreign_for_test(testcase):