1
# Copyright (C) 2004, 2005 by Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Make a copy of an entire branch and all its history.
19
This is the underlying function for the branch/get/clone commands."""
21
# TODO: This could be done *much* more efficiently by just copying
22
# all the whole weaves and revisions, rather than getting one
25
# TODO: Optionally, after copying, discard any irrelevant information from
26
# the destination, such as revisions committed after the last one we're interested
27
# in. This needs to apply a weave prune operation (not written yet) to each
30
# Copying must be done in a way that supports http transports, where we
31
# can't list a directory, and therefore have to rely on information
32
# retrieved from top-level objects whose names we do know.
34
# In practice this means we first fetch the revision history and ancestry.
35
# These give us a list of all the revisions that need to be fetched. We
36
# also get the inventory weave. We then just need to get a list of all
37
# file-ids ever referenced by this tree. (It might be nice to keep a list
38
# of them directly.) This is done by walking over the inventories of all
39
# copied revisions and accumulating a list of file ids.
41
# For local branches it is possible to optimize this considerably in two
42
# ways. One is to hardlink the files (if possible and requested), rather
43
# than copying them. Another is to simply list the directory rather than
44
# walking through the inventories to find out what files are present -- but
45
# there it may be better to just be consistent with remote branches.
51
from bzrlib.merge import build_working_dir
52
from bzrlib.branch import Branch
53
from bzrlib.trace import mutter, note
54
from bzrlib.store import copy_all
55
from bzrlib.errors import InvalidRevisionId
57
def copy_branch(branch_from, to_location, revision=None, basis_branch=None):
58
"""Copy branch_from into the existing directory to_location.
60
Returns the newly created branch object.
63
If not None, only revisions up to this point will be copied.
64
The head of the new branch will be that revision. Must be a
67
to_location -- The destination directory; must either exist and be
68
empty, or not exist, in which case it is created.
71
A local branch to copy revisions from, related to branch_from.
72
This is used when branching from a remote (slow) branch, and we have
73
a local branch that might contain some relevant revisions.
75
assert isinstance(branch_from, Branch)
76
assert isinstance(to_location, basestring)
77
if basis_branch is not None:
78
note("basis_branch is not supported for fast weave copy yet.")
79
branch_from.lock_read()
81
if not (branch_from.weave_store.listable()
82
and branch_from.revision_store.listable()):
83
return copy_branch_slower(branch_from, to_location, revision,
85
history = _get_truncated_history(branch_from, revision)
86
if not bzrlib.osutils.lexists(to_location):
88
branch_to = Branch.initialize(to_location)
89
mutter("copy branch from %s to %s", branch_from, branch_to)
90
branch_to.working_tree().set_root_id(branch_from.get_root_id())
91
_copy_control_weaves(branch_from, branch_to)
92
_copy_text_weaves(branch_from, branch_to)
93
_copy_revision_store(branch_from, branch_to)
94
branch_to.set_parent(branch_from.base)
95
# must be done *after* history is copied across
96
branch_to.append_revision(*history)
97
build_working_dir(to_location)
104
def _get_truncated_history(branch_from, revision_id):
105
history = branch_from.revision_history()
106
if revision_id is None:
109
idx = history.index(revision_id)
111
raise InvalidRevisionId(revision_id=revision, branch=branch_from)
112
return history[:idx+1]
114
def _copy_text_weaves(branch_from, branch_to):
115
copy_all(branch_from.weave_store, branch_to.weave_store)
118
def _copy_revision_store(branch_from, branch_to):
119
copy_all(branch_from.revision_store, branch_to.revision_store)
122
def _copy_control_weaves(branch_from, branch_to):
123
to_control = branch_to.control_weaves
124
from_control = branch_from.control_weaves
125
to_control.copy_multi(from_control, ['inventory'])
128
def copy_branch_slower(branch_from, to_location, revision=None, basis_branch=None):
129
"""Copy branch_from into the existing directory to_location.
132
If not None, only revisions up to this point will be copied.
133
The head of the new branch will be that revision. Must be a
136
to_location -- The destination directory; must either exist and be
137
empty, or not exist, in which case it is created.
140
The revision to copy up to
143
A local branch to copy revisions from, related to branch_from.
144
This is used when branching from a remote (slow) branch, and we have
145
a local branch that might contain some relevant revisions.
147
assert isinstance(branch_from, Branch)
148
assert isinstance(to_location, basestring)
149
if not bzrlib.osutils.lexists(to_location):
150
os.mkdir(to_location)
151
br_to = Branch.initialize(to_location)
152
mutter("copy branch from %s to %s", branch_from, br_to)
153
if basis_branch is not None:
154
basis_branch.push_stores(br_to)
155
br_to.working_tree().set_root_id(branch_from.get_root_id())
157
revision = branch_from.last_revision()
158
br_to.update_revisions(branch_from, stop_revision=revision)
159
build_working_dir(to_location)
160
br_to.set_parent(branch_from.base)
added
removed
bzrlib/clone.py
