792
812
old_repository = self.repository
793
813
if len(old_repository._fallback_repositories) != 1:
794
814
raise AssertionError("can't cope with fallback repositories "
795
"of %r" % (self.repository,))
796
# unlock it, including unlocking the fallback
815
"of %r (fallbacks: %r)" % (old_repository,
816
old_repository._fallback_repositories))
817
# Open the new repository object.
818
# Repositories don't offer an interface to remove fallback
819
# repositories today; take the conceptually simpler option and just
820
# reopen it. We reopen it starting from the URL so that we
821
# get a separate connection for RemoteRepositories and can
822
# stream from one of them to the other. This does mean doing
823
# separate SSH connection setup, but unstacking is not a
824
# common operation so it's tolerable.
825
new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
826
new_repository = new_bzrdir.find_repository()
827
if new_repository._fallback_repositories:
828
raise AssertionError("didn't expect %r to have "
829
"fallback_repositories"
830
% (self.repository,))
831
# Replace self.repository with the new repository.
832
# Do our best to transfer the lock state (i.e. lock-tokens and
833
# lock count) of self.repository to the new repository.
834
lock_token = old_repository.lock_write().repository_token
835
self.repository = new_repository
836
if isinstance(self, remote.RemoteBranch):
837
# Remote branches can have a second reference to the old
838
# repository that need to be replaced.
839
if self._real_branch is not None:
840
self._real_branch.repository = new_repository
841
self.repository.lock_write(token=lock_token)
842
if lock_token is not None:
843
old_repository.leave_lock_in_place()
797
844
old_repository.unlock()
845
if lock_token is not None:
846
# XXX: self.repository.leave_lock_in_place() before this
847
# function will not be preserved. Fortunately that doesn't
848
# affect the current default format (2a), and would be a
849
# corner-case anyway.
850
# - Andrew Bennetts, 2010/06/30
851
self.repository.dont_leave_lock_in_place()
855
old_repository.unlock()
856
except errors.LockNotHeld:
859
if old_lock_count == 0:
860
raise AssertionError(
861
'old_repository should have been locked at least once.')
862
for i in range(old_lock_count-1):
863
self.repository.lock_write()
864
# Fetch from the old repository into the new.
798
865
old_repository.lock_read()
800
# Repositories don't offer an interface to remove fallback
801
# repositories today; take the conceptually simpler option and just
802
# reopen it. We reopen it starting from the URL so that we
803
# get a separate connection for RemoteRepositories and can
804
# stream from one of them to the other. This does mean doing
805
# separate SSH connection setup, but unstacking is not a
806
# common operation so it's tolerable.
807
new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
808
new_repository = new_bzrdir.find_repository()
809
self.repository = new_repository
810
if self.repository._fallback_repositories:
811
raise AssertionError("didn't expect %r to have "
812
"fallback_repositories"
813
% (self.repository,))
814
# this is not paired with an unlock because it's just restoring
815
# the previous state; the lock's released when set_stacked_on_url
817
self.repository.lock_write()
818
867
# XXX: If you unstack a branch while it has a working tree
819
868
# with a pending merge, the pending-merged revisions will no
820
869
# longer be present. You can (probably) revert and remerge.
822
# XXX: This only fetches up to the tip of the repository; it
823
# doesn't bring across any tags. That's fairly consistent
824
# with how branch works, but perhaps not ideal.
825
self.repository.fetch(old_repository,
826
revision_id=self.last_revision(),
871
tags_to_fetch = set(self.tags.get_reverse_tag_dict())
872
except errors.TagsNotSupported:
873
tags_to_fetch = set()
874
fetch_spec = _mod_graph.NotInOtherForRevs(self.repository,
875
old_repository, required_ids=[self.last_revision()],
876
if_present_ids=tags_to_fetch, find_ghosts=True).execute()
877
self.repository.fetch(old_repository, fetch_spec=fetch_spec)
829
879
old_repository.unlock()
1576
1643
for hook in hooks:
1579
def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1580
lock_type='metadir', set_format=True):
1581
"""Initialize a branch in a bzrdir, with specified files
1583
:param a_bzrdir: The bzrdir to initialize the branch in
1584
:param utf8_files: The files to create as a list of
1585
(filename, content) tuples
1586
:param name: Name of colocated branch to create, if any
1587
:param set_format: If True, set the format with
1588
self.get_format_string. (BzrBranch4 has its format set
1590
:return: a branch in this format
1592
mutter('creating branch %r in %s', self, a_bzrdir.user_url)
1593
branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1595
'metadir': ('lock', lockdir.LockDir),
1596
'branch4': ('branch-lock', lockable_files.TransportLock),
1598
lock_name, lock_class = lock_map[lock_type]
1599
control_files = lockable_files.LockableFiles(branch_transport,
1600
lock_name, lock_class)
1601
control_files.create_lock()
1603
control_files.lock_write()
1604
except errors.LockContention:
1605
if lock_type != 'branch4':
1611
utf8_files += [('format', self.get_format_string())]
1613
for (filename, content) in utf8_files:
1614
branch_transport.put_bytes(
1616
mode=a_bzrdir._get_file_mode())
1619
control_files.unlock()
1620
branch = self.open(a_bzrdir, name, _found=True)
1621
self._run_post_branch_init_hooks(a_bzrdir, name, branch)
1624
def initialize(self, a_bzrdir, name=None):
1646
def initialize(self, a_bzrdir, name=None, repository=None):
1625
1647
"""Create a branch of this format in a_bzrdir.
1627
1649
:param name: Name of the colocated branch to create.
1718
1777
These are all empty initially, because by default nothing should get
1721
Hooks.__init__(self)
1722
self.create_hook(HookPoint('set_rh',
1780
Hooks.__init__(self, "bzrlib.branch", "Branch.hooks")
1781
self.add_hook('set_rh',
1723
1782
"Invoked whenever the revision history has been set via "
1724
1783
"set_revision_history. The api signature is (branch, "
1725
1784
"revision_history), and the branch will be write-locked. "
1726
1785
"The set_rh hook can be expensive for bzr to trigger, a better "
1727
"hook to use is Branch.post_change_branch_tip.", (0, 15), None))
1728
self.create_hook(HookPoint('open',
1786
"hook to use is Branch.post_change_branch_tip.", (0, 15))
1787
self.add_hook('open',
1729
1788
"Called with the Branch object that has been opened after a "
1730
"branch is opened.", (1, 8), None))
1731
self.create_hook(HookPoint('post_push',
1789
"branch is opened.", (1, 8))
1790
self.add_hook('post_push',
1732
1791
"Called after a push operation completes. post_push is called "
1733
1792
"with a bzrlib.branch.BranchPushResult object and only runs in the "
1734
"bzr client.", (0, 15), None))
1735
self.create_hook(HookPoint('post_pull',
1793
"bzr client.", (0, 15))
1794
self.add_hook('post_pull',
1736
1795
"Called after a pull operation completes. post_pull is called "
1737
1796
"with a bzrlib.branch.PullResult object and only runs in the "
1738
"bzr client.", (0, 15), None))
1739
self.create_hook(HookPoint('pre_commit',
1740
"Called after a commit is calculated but before it is is "
1797
"bzr client.", (0, 15))
1798
self.add_hook('pre_commit',
1799
"Called after a commit is calculated but before it is "
1741
1800
"completed. pre_commit is called with (local, master, old_revno, "
1742
1801
"old_revid, future_revno, future_revid, tree_delta, future_tree"
1743
1802
"). old_revid is NULL_REVISION for the first commit to a branch, "
1745
1804
"basis revision. hooks MUST NOT modify this delta. "
1746
1805
" future_tree is an in-memory tree obtained from "
1747
1806
"CommitBuilder.revision_tree() and hooks MUST NOT modify this "
1748
"tree.", (0,91), None))
1749
self.create_hook(HookPoint('post_commit',
1808
self.add_hook('post_commit',
1750
1809
"Called in the bzr client after a commit has completed. "
1751
1810
"post_commit is called with (local, master, old_revno, old_revid, "
1752
1811
"new_revno, new_revid). old_revid is NULL_REVISION for the first "
1753
"commit to a branch.", (0, 15), None))
1754
self.create_hook(HookPoint('post_uncommit',
1812
"commit to a branch.", (0, 15))
1813
self.add_hook('post_uncommit',
1755
1814
"Called in the bzr client after an uncommit completes. "
1756
1815
"post_uncommit is called with (local, master, old_revno, "
1757
1816
"old_revid, new_revno, new_revid) where local is the local branch "
1758
1817
"or None, master is the target branch, and an empty branch "
1759
"receives new_revno of 0, new_revid of None.", (0, 15), None))
1760
self.create_hook(HookPoint('pre_change_branch_tip',
1818
"receives new_revno of 0, new_revid of None.", (0, 15))
1819
self.add_hook('pre_change_branch_tip',
1761
1820
"Called in bzr client and server before a change to the tip of a "
1762
1821
"branch is made. pre_change_branch_tip is called with a "
1763
1822
"bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1764
"commit, uncommit will all trigger this hook.", (1, 6), None))
1765
self.create_hook(HookPoint('post_change_branch_tip',
1823
"commit, uncommit will all trigger this hook.", (1, 6))
1824
self.add_hook('post_change_branch_tip',
1766
1825
"Called in bzr client and server after a change to the tip of a "
1767
1826
"branch is made. post_change_branch_tip is called with a "
1768
1827
"bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1769
"commit, uncommit will all trigger this hook.", (1, 4), None))
1770
self.create_hook(HookPoint('transform_fallback_location',
1828
"commit, uncommit will all trigger this hook.", (1, 4))
1829
self.add_hook('transform_fallback_location',
1771
1830
"Called when a stacked branch is activating its fallback "
1772
1831
"locations. transform_fallback_location is called with (branch, "
1773
1832
"url), and should return a new url. Returning the same url "
1778
1837
"fallback locations have not been activated. When there are "
1779
1838
"multiple hooks installed for transform_fallback_location, "
1780
1839
"all are called with the url returned from the previous hook."
1781
"The order is however undefined.", (1, 9), None))
1782
self.create_hook(HookPoint('automatic_tag_name',
1783
"Called to determine an automatic tag name for a revision."
1840
"The order is however undefined.", (1, 9))
1841
self.add_hook('automatic_tag_name',
1842
"Called to determine an automatic tag name for a revision. "
1784
1843
"automatic_tag_name is called with (branch, revision_id) and "
1785
1844
"should return a tag name or None if no tag name could be "
1786
1845
"determined. The first non-None tag name returned will be used.",
1788
self.create_hook(HookPoint('post_branch_init',
1847
self.add_hook('post_branch_init',
1789
1848
"Called after new branch initialization completes. "
1790
1849
"post_branch_init is called with a "
1791
1850
"bzrlib.branch.BranchInitHookParams. "
1792
1851
"Note that init, branch and checkout (both heavyweight and "
1793
"lightweight) will all trigger this hook.", (2, 2), None))
1794
self.create_hook(HookPoint('post_switch',
1852
"lightweight) will all trigger this hook.", (2, 2))
1853
self.add_hook('post_switch',
1795
1854
"Called after a checkout switches branch. "
1796
1855
"post_switch is called with a "
1797
"bzrlib.branch.SwitchHookParams.", (2, 2), None))
1856
"bzrlib.branch.SwitchHookParams.", (2, 2))
3225
3314
raise NotImplementedError(self.push)
3317
def copy_content_into(self, revision_id=None):
3318
"""Copy the content of source into target
3320
revision_id: if not None, the revision history in the new branch will
3321
be truncated to end with revision_id.
3323
raise NotImplementedError(self.copy_content_into)
3326
def fetch(self, stop_revision=None):
3329
:param stop_revision: Last revision to fetch
3331
raise NotImplementedError(self.fetch)
3228
3334
class GenericInterBranch(InterBranch):
3229
"""InterBranch implementation that uses public Branch functions.
3233
def _get_branch_formats_to_test():
3234
return BranchFormat._default_format, BranchFormat._default_format
3335
"""InterBranch implementation that uses public Branch functions."""
3338
def is_compatible(klass, source, target):
3339
# GenericBranch uses the public API, so always compatible
3343
def _get_branch_formats_to_test(klass):
3344
return [(format_registry.get_default(), format_registry.get_default())]
3347
def unwrap_format(klass, format):
3348
if isinstance(format, remote.RemoteBranchFormat):
3349
format._ensure_real()
3350
return format._custom_format
3354
def copy_content_into(self, revision_id=None):
3355
"""Copy the content of source into target
3357
revision_id: if not None, the revision history in the new branch will
3358
be truncated to end with revision_id.
3360
self.source.update_references(self.target)
3361
self.source._synchronize_history(self.target, revision_id)
3363
parent = self.source.get_parent()
3364
except errors.InaccessibleParent, e:
3365
mutter('parent was not accessible to copy: %s', e)
3368
self.target.set_parent(parent)
3369
if self.source._push_should_merge_tags():
3370
self.source.tags.merge_to(self.target.tags)
3373
def fetch(self, stop_revision=None):
3374
if self.target.base == self.source.base:
3376
self.source.lock_read()
3378
fetch_spec_factory = fetch.FetchSpecFactory()
3379
fetch_spec_factory.source_branch = self.source
3380
fetch_spec_factory.source_branch_stop_revision_id = stop_revision
3381
fetch_spec_factory.source_repo = self.source.repository
3382
fetch_spec_factory.target_repo = self.target.repository
3383
fetch_spec_factory.target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
3384
fetch_spec = fetch_spec_factory.make_fetch_spec()
3385
return self.target.repository.fetch(self.source.repository,
3386
fetch_spec=fetch_spec)
3388
self.source.unlock()
3236
3391
def update_revisions(self, stop_revision=None, overwrite=False,
3238
3393
"""See InterBranch.update_revisions()."""
3239
self.source.lock_read()
3241
other_revno, other_last_revision = self.source.last_revision_info()
3242
stop_revno = None # unknown
3243
if stop_revision is None:
3244
stop_revision = other_last_revision
3245
if _mod_revision.is_null(stop_revision):
3246
# if there are no commits, we're done.
3248
stop_revno = other_revno
3250
# what's the current last revision, before we fetch [and change it
3252
last_rev = _mod_revision.ensure_null(self.target.last_revision())
3253
# we fetch here so that we don't process data twice in the common
3254
# case of having something to pull, and so that the check for
3255
# already merged can operate on the just fetched graph, which will
3256
# be cached in memory.
3257
self.target.fetch(self.source, stop_revision)
3258
# Check to see if one is an ancestor of the other
3261
graph = self.target.repository.get_graph()
3262
if self.target._check_if_descendant_or_diverged(
3263
stop_revision, last_rev, graph, self.source):
3264
# stop_revision is a descendant of last_rev, but we aren't
3265
# overwriting, so we're done.
3267
if stop_revno is None:
3269
graph = self.target.repository.get_graph()
3270
this_revno, this_last_revision = \
3271
self.target.last_revision_info()
3272
stop_revno = graph.find_distance_to_null(stop_revision,
3273
[(other_last_revision, other_revno),
3274
(this_last_revision, this_revno)])
3275
self.target.set_last_revision_info(stop_revno, stop_revision)
3277
self.source.unlock()
3394
other_revno, other_last_revision = self.source.last_revision_info()
3395
stop_revno = None # unknown
3396
if stop_revision is None:
3397
stop_revision = other_last_revision
3398
if _mod_revision.is_null(stop_revision):
3399
# if there are no commits, we're done.
3401
stop_revno = other_revno
3403
# what's the current last revision, before we fetch [and change it
3405
last_rev = _mod_revision.ensure_null(self.target.last_revision())
3406
# we fetch here so that we don't process data twice in the common
3407
# case of having something to pull, and so that the check for
3408
# already merged can operate on the just fetched graph, which will
3409
# be cached in memory.
3410
self.fetch(stop_revision=stop_revision)
3411
# Check to see if one is an ancestor of the other
3414
graph = self.target.repository.get_graph()
3415
if self.target._check_if_descendant_or_diverged(
3416
stop_revision, last_rev, graph, self.source):
3417
# stop_revision is a descendant of last_rev, but we aren't
3418
# overwriting, so we're done.
3420
if stop_revno is None:
3422
graph = self.target.repository.get_graph()
3423
this_revno, this_last_revision = \
3424
self.target.last_revision_info()
3425
stop_revno = graph.find_distance_to_null(stop_revision,
3426
[(other_last_revision, other_revno),
3427
(this_last_revision, this_revno)])
3428
self.target.set_last_revision_info(stop_revno, stop_revision)
3279
3431
def pull(self, overwrite=False, stop_revision=None,
3280
possible_transports=None, _hook_master=None, run_hooks=True,
3432
possible_transports=None, run_hooks=True,
3281
3433
_override_hook_target=None, local=False):
3434
"""Pull from source into self, updating my master if any.
3284
:param _hook_master: Private parameter - set the branch to
3285
be supplied as the master to pull hooks.
3286
3436
:param run_hooks: Private parameter - if false, this branch
3287
3437
is being called because it's the master of the primary branch,
3288
3438
so it should not run its hooks.
3289
:param _override_hook_target: Private parameter - set the branch to be
3290
supplied as the target_branch to pull hooks.
3291
:param local: Only update the local branch, and not the bound branch.
3293
# This type of branch can't be bound.
3440
bound_location = self.target.get_bound_location()
3441
if local and not bound_location:
3295
3442
raise errors.LocalRequiresBoundBranch()
3296
result = PullResult()
3297
result.source_branch = self.source
3298
if _override_hook_target is None:
3299
result.target_branch = self.target
3301
result.target_branch = _override_hook_target
3302
self.source.lock_read()
3443
master_branch = None
3444
source_is_master = (self.source.user_url == bound_location)
3445
if not local and bound_location and not source_is_master:
3446
# not pulling from master, so we need to update master.
3447
master_branch = self.target.get_master_branch(possible_transports)
3448
master_branch.lock_write()
3304
# We assume that during 'pull' the target repository is closer than
3306
self.source.update_references(self.target)
3307
graph = self.target.repository.get_graph(self.source.repository)
3308
# TODO: Branch formats should have a flag that indicates
3309
# that revno's are expensive, and pull() should honor that flag.
3311
result.old_revno, result.old_revid = \
3312
self.target.last_revision_info()
3313
self.target.update_revisions(self.source, stop_revision,
3314
overwrite=overwrite, graph=graph)
3315
# TODO: The old revid should be specified when merging tags,
3316
# so a tags implementation that versions tags can only
3317
# pull in the most recent changes. -- JRV20090506
3318
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3320
result.new_revno, result.new_revid = self.target.last_revision_info()
3322
result.master_branch = _hook_master
3323
result.local_branch = result.target_branch
3325
result.master_branch = result.target_branch
3326
result.local_branch = None
3328
for hook in Branch.hooks['post_pull']:
3451
# pull from source into master.
3452
master_branch.pull(self.source, overwrite, stop_revision,
3454
return self._pull(overwrite,
3455
stop_revision, _hook_master=master_branch,
3456
run_hooks=run_hooks,
3457
_override_hook_target=_override_hook_target,
3458
merge_tags_to_master=not source_is_master)
3331
self.source.unlock()
3461
master_branch.unlock()
3334
3463
def push(self, overwrite=False, stop_revision=None,
3335
3464
_override_hook_source_branch=None):
3401
def is_compatible(self, source, target):
3402
# GenericBranch uses the public API, so always compatible
3406
class InterToBranch5(GenericInterBranch):
3409
def _get_branch_formats_to_test():
3410
return BranchFormat._default_format, BzrBranchFormat5()
3412
def pull(self, overwrite=False, stop_revision=None,
3413
possible_transports=None, run_hooks=True,
3414
_override_hook_target=None, local=False):
3415
"""Pull from source into self, updating my master if any.
3529
def _pull(self, overwrite=False, stop_revision=None,
3530
possible_transports=None, _hook_master=None, run_hooks=True,
3531
_override_hook_target=None, local=False,
3532
merge_tags_to_master=True):
3535
This function is the core worker, used by GenericInterBranch.pull to
3536
avoid duplication when pulling source->master and source->local.
3538
:param _hook_master: Private parameter - set the branch to
3539
be supplied as the master to pull hooks.
3417
3540
:param run_hooks: Private parameter - if false, this branch
3418
3541
is being called because it's the master of the primary branch,
3419
3542
so it should not run its hooks.
3543
is being called because it's the master of the primary branch,
3544
so it should not run its hooks.
3545
:param _override_hook_target: Private parameter - set the branch to be
3546
supplied as the target_branch to pull hooks.
3547
:param local: Only update the local branch, and not the bound branch.
3421
bound_location = self.target.get_bound_location()
3422
if local and not bound_location:
3549
# This type of branch can't be bound.
3423
3551
raise errors.LocalRequiresBoundBranch()
3424
master_branch = None
3425
if not local and bound_location and self.source.user_url != bound_location:
3426
# not pulling from master, so we need to update master.
3427
master_branch = self.target.get_master_branch(possible_transports)
3428
master_branch.lock_write()
3552
result = PullResult()
3553
result.source_branch = self.source
3554
if _override_hook_target is None:
3555
result.target_branch = self.target
3557
result.target_branch = _override_hook_target
3558
self.source.lock_read()
3431
# pull from source into master.
3432
master_branch.pull(self.source, overwrite, stop_revision,
3434
return super(InterToBranch5, self).pull(overwrite,
3435
stop_revision, _hook_master=master_branch,
3436
run_hooks=run_hooks,
3437
_override_hook_target=_override_hook_target)
3560
# We assume that during 'pull' the target repository is closer than
3562
self.source.update_references(self.target)
3563
graph = self.target.repository.get_graph(self.source.repository)
3564
# TODO: Branch formats should have a flag that indicates
3565
# that revno's are expensive, and pull() should honor that flag.
3567
result.old_revno, result.old_revid = \
3568
self.target.last_revision_info()
3569
self.target.update_revisions(self.source, stop_revision,
3570
overwrite=overwrite, graph=graph)
3571
# TODO: The old revid should be specified when merging tags,
3572
# so a tags implementation that versions tags can only
3573
# pull in the most recent changes. -- JRV20090506
3574
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3575
overwrite, ignore_master=not merge_tags_to_master)
3576
result.new_revno, result.new_revid = self.target.last_revision_info()
3578
result.master_branch = _hook_master
3579
result.local_branch = result.target_branch
3581
result.master_branch = result.target_branch
3582
result.local_branch = None
3584
for hook in Branch.hooks['post_pull']:
3440
master_branch.unlock()
3587
self.source.unlock()
3443
3591
InterBranch.register_optimiser(GenericInterBranch)
3444
InterBranch.register_optimiser(InterToBranch5)