793
815
old_repository = self.repository
794
816
if len(old_repository._fallback_repositories) != 1:
795
817
raise AssertionError("can't cope with fallback repositories "
796
"of %r" % (self.repository,))
797
# unlock it, including unlocking the fallback
818
"of %r (fallbacks: %r)" % (old_repository,
819
old_repository._fallback_repositories))
820
# Open the new repository object.
821
# Repositories don't offer an interface to remove fallback
822
# repositories today; take the conceptually simpler option and just
823
# reopen it. We reopen it starting from the URL so that we
824
# get a separate connection for RemoteRepositories and can
825
# stream from one of them to the other. This does mean doing
826
# separate SSH connection setup, but unstacking is not a
827
# common operation so it's tolerable.
828
new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
829
new_repository = new_bzrdir.find_repository()
830
if new_repository._fallback_repositories:
831
raise AssertionError("didn't expect %r to have "
832
"fallback_repositories"
833
% (self.repository,))
834
# Replace self.repository with the new repository.
835
# Do our best to transfer the lock state (i.e. lock-tokens and
836
# lock count) of self.repository to the new repository.
837
lock_token = old_repository.lock_write().repository_token
838
self.repository = new_repository
839
if isinstance(self, remote.RemoteBranch):
840
# Remote branches can have a second reference to the old
841
# repository that need to be replaced.
842
if self._real_branch is not None:
843
self._real_branch.repository = new_repository
844
self.repository.lock_write(token=lock_token)
845
if lock_token is not None:
846
old_repository.leave_lock_in_place()
798
847
old_repository.unlock()
848
if lock_token is not None:
849
# XXX: self.repository.leave_lock_in_place() before this
850
# function will not be preserved. Fortunately that doesn't
851
# affect the current default format (2a), and would be a
852
# corner-case anyway.
853
# - Andrew Bennetts, 2010/06/30
854
self.repository.dont_leave_lock_in_place()
858
old_repository.unlock()
859
except errors.LockNotHeld:
862
if old_lock_count == 0:
863
raise AssertionError(
864
'old_repository should have been locked at least once.')
865
for i in range(old_lock_count-1):
866
self.repository.lock_write()
867
# Fetch from the old repository into the new.
799
868
old_repository.lock_read()
801
# Repositories don't offer an interface to remove fallback
802
# repositories today; take the conceptually simpler option and just
803
# reopen it. We reopen it starting from the URL so that we
804
# get a separate connection for RemoteRepositories and can
805
# stream from one of them to the other. This does mean doing
806
# separate SSH connection setup, but unstacking is not a
807
# common operation so it's tolerable.
808
new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
809
new_repository = new_bzrdir.find_repository()
810
self.repository = new_repository
811
if self.repository._fallback_repositories:
812
raise AssertionError("didn't expect %r to have "
813
"fallback_repositories"
814
% (self.repository,))
815
# this is not paired with an unlock because it's just restoring
816
# the previous state; the lock's released when set_stacked_on_url
818
self.repository.lock_write()
819
870
# XXX: If you unstack a branch while it has a working tree
820
871
# with a pending merge, the pending-merged revisions will no
821
872
# longer be present. You can (probably) revert and remerge.
823
# XXX: This only fetches up to the tip of the repository; it
824
# doesn't bring across any tags. That's fairly consistent
825
# with how branch works, but perhaps not ideal.
826
self.repository.fetch(old_repository,
827
revision_id=self.last_revision(),
874
tags_to_fetch = set(self.tags.get_reverse_tag_dict())
875
except errors.TagsNotSupported:
876
tags_to_fetch = set()
877
fetch_spec = _mod_graph.NotInOtherForRevs(self.repository,
878
old_repository, required_ids=[self.last_revision()],
879
if_present_ids=tags_to_fetch, find_ghosts=True).execute()
880
self.repository.fetch(old_repository, fetch_spec=fetch_spec)
830
882
old_repository.unlock()
1576
1648
def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1577
lock_type='metadir', set_format=True):
1578
1650
"""Initialize a branch in a bzrdir, with specified files
1580
1652
:param a_bzrdir: The bzrdir to initialize the branch in
1581
1653
:param utf8_files: The files to create as a list of
1582
1654
(filename, content) tuples
1583
1655
:param name: Name of colocated branch to create, if any
1584
:param set_format: If True, set the format with
1585
self.get_format_string. (BzrBranch4 has its format set
1587
1656
:return: a branch in this format
1589
1658
mutter('creating branch %r in %s', self, a_bzrdir.user_url)
1590
1659
branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1592
'metadir': ('lock', lockdir.LockDir),
1593
'branch4': ('branch-lock', lockable_files.TransportLock),
1595
lock_name, lock_class = lock_map[lock_type]
1596
1660
control_files = lockable_files.LockableFiles(branch_transport,
1597
lock_name, lock_class)
1661
'lock', lockdir.LockDir)
1598
1662
control_files.create_lock()
1663
control_files.lock_write()
1600
control_files.lock_write()
1601
except errors.LockContention:
1602
if lock_type != 'branch4':
1608
1665
utf8_files += [('format', self.get_format_string())]
1610
1666
for (filename, content) in utf8_files:
1611
1667
branch_transport.put_bytes(
1612
1668
filename, content,
1613
1669
mode=a_bzrdir._get_file_mode())
1616
control_files.unlock()
1617
branch = self.open(a_bzrdir, name, _found=True)
1671
control_files.unlock()
1672
branch = self.open(a_bzrdir, name, _found=True,
1673
found_repository=repository)
1618
1674
self._run_post_branch_init_hooks(a_bzrdir, name, branch)
1621
def initialize(self, a_bzrdir, name=None):
1677
def initialize(self, a_bzrdir, name=None, repository=None):
1622
1678
"""Create a branch of this format in a_bzrdir.
1624
1680
:param name: Name of the colocated branch to create.
1715
1808
These are all empty initially, because by default nothing should get
1718
Hooks.__init__(self)
1719
self.create_hook(HookPoint('set_rh',
1811
Hooks.__init__(self, "bzrlib.branch", "Branch.hooks")
1812
self.add_hook('set_rh',
1720
1813
"Invoked whenever the revision history has been set via "
1721
1814
"set_revision_history. The api signature is (branch, "
1722
1815
"revision_history), and the branch will be write-locked. "
1723
1816
"The set_rh hook can be expensive for bzr to trigger, a better "
1724
"hook to use is Branch.post_change_branch_tip.", (0, 15), None))
1725
self.create_hook(HookPoint('open',
1817
"hook to use is Branch.post_change_branch_tip.", (0, 15))
1818
self.add_hook('open',
1726
1819
"Called with the Branch object that has been opened after a "
1727
"branch is opened.", (1, 8), None))
1728
self.create_hook(HookPoint('post_push',
1820
"branch is opened.", (1, 8))
1821
self.add_hook('post_push',
1729
1822
"Called after a push operation completes. post_push is called "
1730
1823
"with a bzrlib.branch.BranchPushResult object and only runs in the "
1731
"bzr client.", (0, 15), None))
1732
self.create_hook(HookPoint('post_pull',
1824
"bzr client.", (0, 15))
1825
self.add_hook('post_pull',
1733
1826
"Called after a pull operation completes. post_pull is called "
1734
1827
"with a bzrlib.branch.PullResult object and only runs in the "
1735
"bzr client.", (0, 15), None))
1736
self.create_hook(HookPoint('pre_commit',
1737
"Called after a commit is calculated but before it is is "
1828
"bzr client.", (0, 15))
1829
self.add_hook('pre_commit',
1830
"Called after a commit is calculated but before it is "
1738
1831
"completed. pre_commit is called with (local, master, old_revno, "
1739
1832
"old_revid, future_revno, future_revid, tree_delta, future_tree"
1740
1833
"). old_revid is NULL_REVISION for the first commit to a branch, "
1742
1835
"basis revision. hooks MUST NOT modify this delta. "
1743
1836
" future_tree is an in-memory tree obtained from "
1744
1837
"CommitBuilder.revision_tree() and hooks MUST NOT modify this "
1745
"tree.", (0,91), None))
1746
self.create_hook(HookPoint('post_commit',
1839
self.add_hook('post_commit',
1747
1840
"Called in the bzr client after a commit has completed. "
1748
1841
"post_commit is called with (local, master, old_revno, old_revid, "
1749
1842
"new_revno, new_revid). old_revid is NULL_REVISION for the first "
1750
"commit to a branch.", (0, 15), None))
1751
self.create_hook(HookPoint('post_uncommit',
1843
"commit to a branch.", (0, 15))
1844
self.add_hook('post_uncommit',
1752
1845
"Called in the bzr client after an uncommit completes. "
1753
1846
"post_uncommit is called with (local, master, old_revno, "
1754
1847
"old_revid, new_revno, new_revid) where local is the local branch "
1755
1848
"or None, master is the target branch, and an empty branch "
1756
"receives new_revno of 0, new_revid of None.", (0, 15), None))
1757
self.create_hook(HookPoint('pre_change_branch_tip',
1849
"receives new_revno of 0, new_revid of None.", (0, 15))
1850
self.add_hook('pre_change_branch_tip',
1758
1851
"Called in bzr client and server before a change to the tip of a "
1759
1852
"branch is made. pre_change_branch_tip is called with a "
1760
1853
"bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1761
"commit, uncommit will all trigger this hook.", (1, 6), None))
1762
self.create_hook(HookPoint('post_change_branch_tip',
1854
"commit, uncommit will all trigger this hook.", (1, 6))
1855
self.add_hook('post_change_branch_tip',
1763
1856
"Called in bzr client and server after a change to the tip of a "
1764
1857
"branch is made. post_change_branch_tip is called with a "
1765
1858
"bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1766
"commit, uncommit will all trigger this hook.", (1, 4), None))
1767
self.create_hook(HookPoint('transform_fallback_location',
1859
"commit, uncommit will all trigger this hook.", (1, 4))
1860
self.add_hook('transform_fallback_location',
1768
1861
"Called when a stacked branch is activating its fallback "
1769
1862
"locations. transform_fallback_location is called with (branch, "
1770
1863
"url), and should return a new url. Returning the same url "
1775
1868
"fallback locations have not been activated. When there are "
1776
1869
"multiple hooks installed for transform_fallback_location, "
1777
1870
"all are called with the url returned from the previous hook."
1778
"The order is however undefined.", (1, 9), None))
1779
self.create_hook(HookPoint('automatic_tag_name',
1780
"Called to determine an automatic tag name for a revision."
1871
"The order is however undefined.", (1, 9))
1872
self.add_hook('automatic_tag_name',
1873
"Called to determine an automatic tag name for a revision. "
1781
1874
"automatic_tag_name is called with (branch, revision_id) and "
1782
1875
"should return a tag name or None if no tag name could be "
1783
1876
"determined. The first non-None tag name returned will be used.",
1785
self.create_hook(HookPoint('post_branch_init',
1878
self.add_hook('post_branch_init',
1786
1879
"Called after new branch initialization completes. "
1787
1880
"post_branch_init is called with a "
1788
1881
"bzrlib.branch.BranchInitHookParams. "
1789
1882
"Note that init, branch and checkout (both heavyweight and "
1790
"lightweight) will all trigger this hook.", (2, 2), None))
1791
self.create_hook(HookPoint('post_switch',
1883
"lightweight) will all trigger this hook.", (2, 2))
1884
self.add_hook('post_switch',
1792
1885
"Called after a checkout switches branch. "
1793
1886
"post_switch is called with a "
1794
"bzrlib.branch.SwitchHookParams.", (2, 2), None))
1887
"bzrlib.branch.SwitchHookParams.", (2, 2))
3223
3310
raise NotImplementedError(self.push)
3313
def copy_content_into(self, revision_id=None):
3314
"""Copy the content of source into target
3316
revision_id: if not None, the revision history in the new branch will
3317
be truncated to end with revision_id.
3319
raise NotImplementedError(self.copy_content_into)
3322
def fetch(self, stop_revision=None, fetch_spec=None):
3325
:param stop_revision: Last revision to fetch
3326
:param fetch_spec: Fetch spec.
3328
raise NotImplementedError(self.fetch)
3226
3331
class GenericInterBranch(InterBranch):
3227
"""InterBranch implementation that uses public Branch functions.
3231
def _get_branch_formats_to_test():
3232
return BranchFormat._default_format, BranchFormat._default_format
3332
"""InterBranch implementation that uses public Branch functions."""
3335
def is_compatible(klass, source, target):
3336
# GenericBranch uses the public API, so always compatible
3340
def _get_branch_formats_to_test(klass):
3341
return [(format_registry.get_default(), format_registry.get_default())]
3344
def unwrap_format(klass, format):
3345
if isinstance(format, remote.RemoteBranchFormat):
3346
format._ensure_real()
3347
return format._custom_format
3351
def copy_content_into(self, revision_id=None):
3352
"""Copy the content of source into target
3354
revision_id: if not None, the revision history in the new branch will
3355
be truncated to end with revision_id.
3357
self.source.update_references(self.target)
3358
self.source._synchronize_history(self.target, revision_id)
3360
parent = self.source.get_parent()
3361
except errors.InaccessibleParent, e:
3362
mutter('parent was not accessible to copy: %s', e)
3365
self.target.set_parent(parent)
3366
if self.source._push_should_merge_tags():
3367
self.source.tags.merge_to(self.target.tags)
3370
def fetch(self, stop_revision=None, fetch_spec=None):
3371
if fetch_spec is not None and stop_revision is not None:
3372
raise AssertionError(
3373
"fetch_spec and last_revision are mutually exclusive.")
3374
if self.target.base == self.source.base:
3376
self.source.lock_read()
3378
if stop_revision is None and fetch_spec is None:
3379
stop_revision = self.source.last_revision()
3380
stop_revision = _mod_revision.ensure_null(stop_revision)
3381
return self.target.repository.fetch(self.source.repository,
3382
revision_id=stop_revision, fetch_spec=fetch_spec)
3384
self.source.unlock()
3234
3387
def update_revisions(self, stop_revision=None, overwrite=False,
3388
graph=None, fetch_tags=True):
3236
3389
"""See InterBranch.update_revisions()."""
3237
self.source.lock_read()
3239
other_revno, other_last_revision = self.source.last_revision_info()
3240
stop_revno = None # unknown
3241
if stop_revision is None:
3242
stop_revision = other_last_revision
3243
if _mod_revision.is_null(stop_revision):
3244
# if there are no commits, we're done.
3246
stop_revno = other_revno
3248
# what's the current last revision, before we fetch [and change it
3250
last_rev = _mod_revision.ensure_null(self.target.last_revision())
3251
# we fetch here so that we don't process data twice in the common
3252
# case of having something to pull, and so that the check for
3253
# already merged can operate on the just fetched graph, which will
3254
# be cached in memory.
3255
self.target.fetch(self.source, stop_revision)
3256
# Check to see if one is an ancestor of the other
3259
graph = self.target.repository.get_graph()
3260
if self.target._check_if_descendant_or_diverged(
3261
stop_revision, last_rev, graph, self.source):
3262
# stop_revision is a descendant of last_rev, but we aren't
3263
# overwriting, so we're done.
3265
if stop_revno is None:
3267
graph = self.target.repository.get_graph()
3268
this_revno, this_last_revision = \
3269
self.target.last_revision_info()
3270
stop_revno = graph.find_distance_to_null(stop_revision,
3271
[(other_last_revision, other_revno),
3272
(this_last_revision, this_revno)])
3273
self.target.set_last_revision_info(stop_revno, stop_revision)
3275
self.source.unlock()
3390
other_revno, other_last_revision = self.source.last_revision_info()
3391
stop_revno = None # unknown
3392
if stop_revision is None:
3393
stop_revision = other_last_revision
3394
if _mod_revision.is_null(stop_revision):
3395
# if there are no commits, we're done.
3397
stop_revno = other_revno
3399
# what's the current last revision, before we fetch [and change it
3401
last_rev = _mod_revision.ensure_null(self.target.last_revision())
3402
# we fetch here so that we don't process data twice in the common
3403
# case of having something to pull, and so that the check for
3404
# already merged can operate on the just fetched graph, which will
3405
# be cached in memory.
3407
fetch_spec_factory = fetch.FetchSpecFactory()
3408
fetch_spec_factory.source_branch = self.source
3409
fetch_spec_factory.source_branch_stop_revision_id = stop_revision
3410
fetch_spec_factory.source_repo = self.source.repository
3411
fetch_spec_factory.target_repo = self.target.repository
3412
fetch_spec_factory.target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
3413
fetch_spec = fetch_spec_factory.make_fetch_spec()
3415
fetch_spec = _mod_graph.NotInOtherForRevs(self.target.repository,
3416
self.source.repository, revision_ids=[stop_revision]).execute()
3417
self.target.fetch(self.source, fetch_spec=fetch_spec)
3418
# Check to see if one is an ancestor of the other
3421
graph = self.target.repository.get_graph()
3422
if self.target._check_if_descendant_or_diverged(
3423
stop_revision, last_rev, graph, self.source):
3424
# stop_revision is a descendant of last_rev, but we aren't
3425
# overwriting, so we're done.
3427
if stop_revno is None:
3429
graph = self.target.repository.get_graph()
3430
this_revno, this_last_revision = \
3431
self.target.last_revision_info()
3432
stop_revno = graph.find_distance_to_null(stop_revision,
3433
[(other_last_revision, other_revno),
3434
(this_last_revision, this_revno)])
3435
self.target.set_last_revision_info(stop_revno, stop_revision)
3277
3438
def pull(self, overwrite=False, stop_revision=None,
3278
possible_transports=None, _hook_master=None, run_hooks=True,
3439
possible_transports=None, run_hooks=True,
3279
3440
_override_hook_target=None, local=False):
3441
"""Pull from source into self, updating my master if any.
3282
:param _hook_master: Private parameter - set the branch to
3283
be supplied as the master to pull hooks.
3284
3443
:param run_hooks: Private parameter - if false, this branch
3285
3444
is being called because it's the master of the primary branch,
3286
3445
so it should not run its hooks.
3287
:param _override_hook_target: Private parameter - set the branch to be
3288
supplied as the target_branch to pull hooks.
3289
:param local: Only update the local branch, and not the bound branch.
3291
# This type of branch can't be bound.
3447
bound_location = self.target.get_bound_location()
3448
if local and not bound_location:
3293
3449
raise errors.LocalRequiresBoundBranch()
3294
result = PullResult()
3295
result.source_branch = self.source
3296
if _override_hook_target is None:
3297
result.target_branch = self.target
3299
result.target_branch = _override_hook_target
3300
self.source.lock_read()
3450
master_branch = None
3451
source_is_master = (self.source.user_url == bound_location)
3452
if not local and bound_location and not source_is_master:
3453
# not pulling from master, so we need to update master.
3454
master_branch = self.target.get_master_branch(possible_transports)
3455
master_branch.lock_write()
3302
# We assume that during 'pull' the target repository is closer than
3304
self.source.update_references(self.target)
3305
graph = self.target.repository.get_graph(self.source.repository)
3306
# TODO: Branch formats should have a flag that indicates
3307
# that revno's are expensive, and pull() should honor that flag.
3309
result.old_revno, result.old_revid = \
3310
self.target.last_revision_info()
3311
self.target.update_revisions(self.source, stop_revision,
3312
overwrite=overwrite, graph=graph)
3313
# TODO: The old revid should be specified when merging tags,
3314
# so a tags implementation that versions tags can only
3315
# pull in the most recent changes. -- JRV20090506
3316
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3318
result.new_revno, result.new_revid = self.target.last_revision_info()
3320
result.master_branch = _hook_master
3321
result.local_branch = result.target_branch
3323
result.master_branch = result.target_branch
3324
result.local_branch = None
3326
for hook in Branch.hooks['post_pull']:
3458
# pull from source into master.
3459
master_branch.pull(self.source, overwrite, stop_revision,
3461
return self._pull(overwrite,
3462
stop_revision, _hook_master=master_branch,
3463
run_hooks=run_hooks,
3464
_override_hook_target=_override_hook_target,
3465
merge_tags_to_master=not source_is_master)
3329
self.source.unlock()
3468
master_branch.unlock()
3332
3470
def push(self, overwrite=False, stop_revision=None,
3333
3471
_override_hook_source_branch=None):
3399
def is_compatible(self, source, target):
3400
# GenericBranch uses the public API, so always compatible
3404
class InterToBranch5(GenericInterBranch):
3407
def _get_branch_formats_to_test():
3408
return BranchFormat._default_format, BzrBranchFormat5()
3410
def pull(self, overwrite=False, stop_revision=None,
3411
possible_transports=None, run_hooks=True,
3412
_override_hook_target=None, local=False):
3413
"""Pull from source into self, updating my master if any.
3536
def _pull(self, overwrite=False, stop_revision=None,
3537
possible_transports=None, _hook_master=None, run_hooks=True,
3538
_override_hook_target=None, local=False,
3539
merge_tags_to_master=True):
3542
This function is the core worker, used by GenericInterBranch.pull to
3543
avoid duplication when pulling source->master and source->local.
3545
:param _hook_master: Private parameter - set the branch to
3546
be supplied as the master to pull hooks.
3415
3547
:param run_hooks: Private parameter - if false, this branch
3416
3548
is being called because it's the master of the primary branch,
3417
3549
so it should not run its hooks.
3550
is being called because it's the master of the primary branch,
3551
so it should not run its hooks.
3552
:param _override_hook_target: Private parameter - set the branch to be
3553
supplied as the target_branch to pull hooks.
3554
:param local: Only update the local branch, and not the bound branch.
3419
bound_location = self.target.get_bound_location()
3420
if local and not bound_location:
3556
# This type of branch can't be bound.
3421
3558
raise errors.LocalRequiresBoundBranch()
3422
master_branch = None
3423
if not local and bound_location and self.source.user_url != bound_location:
3424
# not pulling from master, so we need to update master.
3425
master_branch = self.target.get_master_branch(possible_transports)
3426
master_branch.lock_write()
3559
result = PullResult()
3560
result.source_branch = self.source
3561
if _override_hook_target is None:
3562
result.target_branch = self.target
3564
result.target_branch = _override_hook_target
3565
self.source.lock_read()
3429
# pull from source into master.
3430
master_branch.pull(self.source, overwrite, stop_revision,
3432
return super(InterToBranch5, self).pull(overwrite,
3433
stop_revision, _hook_master=master_branch,
3434
run_hooks=run_hooks,
3435
_override_hook_target=_override_hook_target)
3567
# We assume that during 'pull' the target repository is closer than
3569
self.source.update_references(self.target)
3570
graph = self.target.repository.get_graph(self.source.repository)
3571
# TODO: Branch formats should have a flag that indicates
3572
# that revno's are expensive, and pull() should honor that flag.
3574
result.old_revno, result.old_revid = \
3575
self.target.last_revision_info()
3576
self.target.update_revisions(self.source, stop_revision,
3577
overwrite=overwrite, graph=graph)
3578
# TODO: The old revid should be specified when merging tags,
3579
# so a tags implementation that versions tags can only
3580
# pull in the most recent changes. -- JRV20090506
3581
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3582
overwrite, ignore_master=not merge_tags_to_master)
3583
result.new_revno, result.new_revid = self.target.last_revision_info()
3585
result.master_branch = _hook_master
3586
result.local_branch = result.target_branch
3588
result.master_branch = result.target_branch
3589
result.local_branch = None
3591
for hook in Branch.hooks['post_pull']:
3438
master_branch.unlock()
3594
self.source.unlock()
3441
3598
InterBranch.register_optimiser(GenericInterBranch)
3442
InterBranch.register_optimiser(InterToBranch5)