64
120
def supports_tags(self):
123
def make_tags(self, branch):
124
if getattr(branch.repository, "get_refs", None) is not None:
125
from bzrlib.plugins.git.remote import RemoteGitTagDict
126
return RemoteGitTagDict(branch)
128
return LocalGitTagDict(branch)
68
131
class GitBranch(ForeignBranch):
69
132
"""An adapter to git repositories for bzr Branch objects."""
71
def __init__(self, bzrdir, repository, head, base, lockfiles):
134
def __init__(self, bzrdir, repository, name, lockfiles):
72
135
self.repository = repository
73
super(GitBranch, self).__init__(default_mapping)
136
self._format = GitBranchFormat()
74
137
self.control_files = lockfiles
75
138
self.bzrdir = bzrdir
78
self._format = GitBranchFormat()
139
super(GitBranch, self).__init__(repository.get_mapping())
142
self.base = bzrdir.transport.base
144
def _get_nick(self, local=False, possible_master_transports=None):
145
"""Find the nick name for this branch.
151
def _set_nick(self, nick):
152
raise NotImplementedError
154
nick = property(_get_nick, _set_nick)
157
return "%s(%r, %r)" % (self.__class__.__name__, self.repository.base, self.name)
159
def generate_revision_history(self, revid, old_revid=None):
160
# FIXME: Check that old_revid is in the ancestry of revid
161
newhead, self.mapping = self.mapping.revision_id_bzr_to_foreign(revid)
162
self._set_head(newhead)
80
164
def lock_write(self):
81
165
self.control_files.lock_write()
167
def get_stacked_on_url(self):
168
# Git doesn't do stacking (yet...)
171
def get_parent(self):
172
"""See Branch.get_parent()."""
173
# FIXME: Set "origin" url from .git/config ?
176
def set_parent(self, url):
177
# FIXME: Set "origin" url in .git/config ?
181
self.control_files.lock_read()
184
return self.control_files.is_locked()
187
self.control_files.unlock()
189
def get_physical_lock_status(self):
84
193
def last_revision(self):
85
194
# perhaps should escape this ?
87
196
return revision.NULL_REVISION
88
197
return self.mapping.revision_id_foreign_to_bzr(self.head)
91
return GitTagDict(self)
94
"""See Branch.get_parent()."""
97
def get_stacked_on_url(self):
199
def _basic_push(self, target, overwrite=False, stop_revision=None):
200
return branch.InterBranch.get(self, target)._basic_push(
201
overwrite, stop_revision)
204
class LocalGitBranch(GitBranch):
205
"""A local Git branch."""
207
def _get_checkout_format(self):
208
"""Return the most suitable metadir for a checkout of this branch.
209
Weaves are used if this branch's repository uses weaves.
211
format = self.repository.bzrdir.checkout_metadir()
212
format.set_branch_format(self._format)
215
def create_checkout(self, to_location, revision_id=None, lightweight=False,
216
accelerator_tree=None, hardlink=False):
218
t = transport.get_transport(to_location)
220
format = self._get_checkout_format()
221
checkout = format.initialize_on_transport(t)
222
from_branch = branch.BranchReferenceFormat().initialize(checkout,
224
tree = checkout.create_workingtree(revision_id,
225
from_branch=from_branch, hardlink=hardlink)
228
return self._create_heavyweight_checkout(to_location, revision_id,
231
def _create_heavyweight_checkout(self, to_location, revision_id=None,
233
"""Create a new heavyweight checkout of this branch.
235
:param to_location: URL of location to create the new checkout in.
236
:param revision_id: Revision that should be the tip of the checkout.
237
:param hardlink: Whether to hardlink
238
:return: WorkingTree object of checkout.
240
checkout_branch = bzrdir.BzrDir.create_branch_convenience(
241
to_location, force_new_tree=False, format=get_rich_root_format())
242
checkout = checkout_branch.bzrdir
243
checkout_branch.bind(self)
244
# pull up to the specified revision_id to set the initial
245
# branch tip correctly, and seed it with history.
246
checkout_branch.pull(self, stop_revision=revision_id)
247
return checkout.create_workingtree(revision_id, hardlink=hardlink)
100
249
def _gen_revision_history(self):
101
250
if self.head is None:
109
cms = self.repository._git.commits(self.head, max_count=max_count, skip=skip)
113
ret.append(self.mapping.revision_id_foreign_to_bzr(cm.id))
117
nextid = cm.parents[0].id
252
ret = list(self.repository.iter_reverse_revision_history(
253
self.last_revision()))
259
return self.repository._git.ref(self.name)
263
def set_last_revision_info(self, revno, revid):
264
self.set_last_revision(revid)
266
def set_last_revision(self, revid):
267
(self.head, self.mapping) = self.mapping.revision_id_bzr_to_foreign(
270
def _set_head(self, value):
272
self.repository._git.refs[self.name] = self._head
273
self._clear_cached_state()
275
head = property(_get_head, _set_head)
121
277
def get_config(self):
122
278
return GitBranchConfig(self)
125
self.control_files.lock_read()
128
self.control_files.unlock()
130
def get_physical_lock_status(self):
133
280
def get_push_location(self):
134
281
"""See Branch.get_push_location."""
135
282
push_loc = self.get_config().get_user_option('push_location')
138
285
def set_push_location(self, location):
139
286
"""See Branch.set_push_location."""
140
287
self.get_config().set_user_option('push_location', location,
288
store=config.STORE_LOCATION)
143
290
def supports_tags(self):
146
def sprout(self, to_bzrdir, revision_id=None):
147
"""See Branch.sprout()."""
148
result = to_bzrdir.create_branch()
149
self.copy_content_into(result, revision_id=revision_id)
150
result.set_parent(self.bzrdir.root_transport.base)
294
class GitBranchPullResult(branch.PullResult):
296
def report(self, to_file):
298
if self.old_revid == self.new_revid:
299
to_file.write('No revisions to pull.\n')
301
to_file.write('Now on revision %d (git sha: %s).\n' %
302
(self.new_revno, self.new_git_head))
303
self._show_tag_conficts(to_file)
306
class GitBranchPushResult(branch.BranchPushResult):
308
def _lookup_revno(self, revid):
309
assert isinstance(revid, str), "was %r" % revid
310
# Try in source branch first, it'll be faster
312
return self.source_branch.revision_id_to_revno(revid)
313
except NoSuchRevision:
314
# FIXME: Check using graph.find_distance_to_null() ?
315
return self.target_branch.revision_id_to_revno(revid)
319
return self._lookup_revno(self.old_revid)
323
return self._lookup_revno(self.new_revid)
326
class InterFromGitBranch(branch.GenericInterBranch):
327
"""InterBranch implementation that pulls from Git into bzr."""
330
def is_compatible(self, source, target):
331
return (isinstance(source, GitBranch) and
332
not isinstance(target, GitBranch))
334
def update_revisions(self, stop_revision=None, overwrite=False,
336
"""See InterBranch.update_revisions()."""
337
interrepo = repository.InterRepository.get(self.source.repository,
338
self.target.repository)
340
self._last_revid = None
341
def determine_wants(heads):
342
if not self.source.name in heads:
343
raise NoSuchRef(self.source.name, heads.keys())
344
if stop_revision is not None:
345
self._last_revid = stop_revision
346
self._head, mapping = self.source.repository.lookup_git_revid(
349
self._head = heads[self.source.name]
351
self.source.mapping.revision_id_foreign_to_bzr(self._head)
352
if self.target.repository.has_revision(self._last_revid):
355
interrepo.fetch_objects(determine_wants, self.source.mapping)
357
prev_last_revid = None
359
prev_last_revid = self.target.last_revision()
360
self.target.generate_revision_history(self._last_revid, prev_last_revid)
362
def pull(self, overwrite=False, stop_revision=None,
363
possible_transports=None, _hook_master=None, run_hooks=True,
364
_override_hook_target=None, local=False):
367
:param _hook_master: Private parameter - set the branch to
368
be supplied as the master to pull hooks.
369
:param run_hooks: Private parameter - if false, this branch
370
is being called because it's the master of the primary branch,
371
so it should not run its hooks.
372
:param _override_hook_target: Private parameter - set the branch to be
373
supplied as the target_branch to pull hooks.
375
# This type of branch can't be bound.
377
raise errors.LocalRequiresBoundBranch()
378
result = GitBranchPullResult()
379
result.source_branch = self.source
380
if _override_hook_target is None:
381
result.target_branch = self.target
383
result.target_branch = _override_hook_target
384
self.source.lock_read()
386
# We assume that during 'pull' the target repository is closer than
388
graph = self.target.repository.get_graph(self.source.repository)
389
result.old_revno, result.old_revid = \
390
self.target.last_revision_info()
391
self.update_revisions(stop_revision, overwrite=overwrite,
393
result.new_git_head = self._head
394
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
396
result.new_revno, result.new_revid = self.target.last_revision_info()
398
result.master_branch = _hook_master
399
result.local_branch = result.target_branch
401
result.master_branch = result.target_branch
402
result.local_branch = None
404
for hook in branch.Branch.hooks['post_pull']:
410
def _basic_push(self, overwrite=False, stop_revision=None):
411
result = branch.BranchPushResult()
412
result.source_branch = self.source
413
result.target_branch = self.target
414
graph = self.target.repository.get_graph(self.source.repository)
415
result.old_revno, result.old_revid = self.target.last_revision_info()
416
self.update_revisions(stop_revision, overwrite=overwrite,
418
result.new_git_head = self._head
419
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
421
result.new_revno, result.new_revid = self.target.last_revision_info()
425
class InterGitBranch(branch.GenericInterBranch):
426
"""InterBranch implementation that pulls between Git branches."""
429
class InterGitLocalRemoteBranch(InterGitBranch):
430
"""InterBranch that copies from a local to a remote git branch."""
433
def is_compatible(self, source, target):
434
from bzrlib.plugins.git.remote import RemoteGitBranch
435
return (isinstance(source, LocalGitBranch) and
436
isinstance(target, RemoteGitBranch))
438
def _basic_push(self, overwrite=False, stop_revision=None):
439
result = GitBranchPushResult()
440
result.source_branch = self.source
441
result.target_branch = self.target
442
if stop_revision is None:
443
stop_revision = self.source.last_revision()
444
# FIXME: Check for diverged branches
445
def get_changed_refs(old_refs):
446
result.old_revid = self.target.mapping.revision_id_foreign_to_bzr(old_refs["refs/heads/master"])
447
refs = { "refs/heads/master": self.source.repository.lookup_git_revid(stop_revision)[0] }
448
result.new_revid = stop_revision
449
for name, sha in self.source.repository._git.refs.as_dict("refs/tags").iteritems():
450
refs["refs/tags/%s" % name] = sha
452
self.target.repository.send_pack(get_changed_refs,
453
self.source.repository._git.object_store.generate_pack_contents)
457
class InterGitRemoteLocalBranch(InterGitBranch):
458
"""InterBranch that copies from a remote to a local git branch."""
461
def is_compatible(self, source, target):
462
from bzrlib.plugins.git.remote import RemoteGitBranch
463
return (isinstance(source, RemoteGitBranch) and
464
isinstance(target, LocalGitBranch))
466
def _basic_push(self, overwrite=False, stop_revision=None):
467
result = branch.BranchPushResult()
468
result.source_branch = self.source
469
result.target_branch = self.target
470
result.old_revid = self.target.last_revision()
471
refs, stop_revision = self.update_refs(stop_revision)
472
self.target.generate_revision_history(stop_revision, result.old_revid)
473
self.update_tags(refs)
474
result.new_revid = self.target.last_revision()
477
def update_tags(self, refs):
478
for name, revid in extract_tags(refs, self.target.mapping).iteritems():
479
self.target.tags.set_tag(name, revid)
481
def update_refs(self, stop_revision=None):
482
interrepo = repository.InterRepository.get(self.source.repository,
483
self.target.repository)
484
if stop_revision is None:
485
refs = interrepo.fetch_refs(branches=["HEAD"])
486
stop_revision = self.target.mapping.revision_id_foreign_to_bzr(refs["HEAD"])
488
refs = interrepo.fetch_refs(revision_id=stop_revision)
489
return refs, stop_revision
491
def pull(self, stop_revision=None, overwrite=False,
492
possible_transports=None, local=False):
493
# This type of branch can't be bound.
495
raise errors.LocalRequiresBoundBranch()
496
result = GitPullResult()
497
result.source_branch = self.source
498
result.target_branch = self.target
499
result.old_revid = self.target.last_revision()
500
refs, stop_revision = self.update_refs(stop_revision)
501
self.target.generate_revision_history(stop_revision, result.old_revid)
502
self.update_tags(refs)
503
result.new_revid = self.target.last_revision()
507
class InterToGitBranch(branch.InterBranch):
508
"""InterBranch implementation that pulls from Git into bzr."""
511
def is_compatible(self, source, target):
512
return (not isinstance(source, GitBranch) and
513
isinstance(target, GitBranch))
515
def push(self, overwrite=True, stop_revision=None,
516
_override_hook_source_branch=None):
517
raise NoPushSupport()
519
def lossy_push(self, stop_revision=None):
520
result = GitBranchPushResult()
521
result.source_branch = self.source
522
result.target_branch = self.target
523
result.old_revid = self.target.last_revision()
524
if stop_revision is None:
525
stop_revision = self.source.last_revision()
526
# FIXME: Check for diverged branches
527
refs = { "refs/heads/master": stop_revision }
528
for name, revid in self.source.tags.get_tag_dict().iteritems():
529
if self.source.repository.has_revision(revid):
530
refs["refs/tags/%s" % name] = revid
531
revidmap, new_refs = self.target.repository.dfetch_refs(
532
self.source.repository, refs)
534
self.target.generate_revision_history(revidmap[stop_revision])
535
result.new_revid = revidmap[stop_revision]
537
result.new_revid = result.old_revid
538
result.revidmap = revidmap
542
branch.InterBranch.register_optimiser(InterGitRemoteLocalBranch)
543
branch.InterBranch.register_optimiser(InterFromGitBranch)
544
branch.InterBranch.register_optimiser(InterToGitBranch)
545
branch.InterBranch.register_optimiser(InterGitLocalRemoteBranch)