25
24
from bzrlib.decorators import needs_read_lock
26
from bzrlib.trace import mutter
28
from bzrlib.plugins.git.foreign import ForeignBranch
30
from dulwich.objects import (
35
class GitTagDict(tag.BasicTags):
37
def __init__(self, branch):
39
self.repository = branch.repository
41
def get_tag_dict(self):
43
for k,v in self.repository._git.tags.iteritems():
44
obj = self.repository._git.get_object(v)
45
while isinstance(obj, Tag):
47
obj = self.repository._git.get_object(v)
48
if not isinstance(obj, Commit):
49
mutter("Tag %s points at object %r that is not a commit, ignoring", k, obj)
51
ret[k] = self.branch.mapping.revision_id_foreign_to_bzr(v)
54
def set_tag(self, name, revid):
55
self.repository._git.tags[name] = revid
26
from bzrlib.plugins.git import ids
58
29
class GitBranchConfig(config.BranchConfig):
63
34
# do not provide a BranchDataConfig
64
35
self.option_sources = self.option_sources[0], self.option_sources[2]
66
def set_user_option(self, name, value, store=config.STORE_BRANCH, warn_masked=False):
37
def set_user_option(self, name, value, local=False):
67
38
"""Force local to True"""
68
config.BranchConfig.set_user_option(self, name, value, store=config.STORE_LOCATION, warn_masked=warn_masked)
39
config.BranchConfig.set_user_option(self, name, value, local=True)
71
42
class GitBranchFormat(branch.BranchFormat):
73
def get_format_description(self):
44
def get_branch_description(self):
74
45
return 'Git Branch'
76
def supports_tags(self):
80
class GitBranch(ForeignBranch):
48
class GitBranch(branch.Branch):
81
49
"""An adapter to git repositories for bzr Branch objects."""
83
def __init__(self, bzrdir, repository, name, head, lockfiles):
84
self.repository = repository
85
super(GitBranch, self).__init__(repository.get_mapping())
51
def __init__(self, gitdir, lockfiles):
52
super(GitBranch, self).__init__()
53
from bzrlib.plugins.git import git_repository
86
55
self.control_files = lockfiles
90
self.base = bzrdir.transport.base
56
self.repository = git_repository.GitRepository(gitdir, lockfiles)
57
self.base = gitdir.root_transport.base
58
if '.git' not in gitdir.root_transport.list_dir('.'):
59
raise errors.NotBranchError(self.base)
91
60
self._format = GitBranchFormat()
93
62
def lock_write(self):
94
63
self.control_files.lock_write()
96
def get_stacked_on_url(self):
97
# Git doesn't do stacking (yet...)
100
def get_parent(self):
101
"""See Branch.get_parent()."""
104
def set_parent(self, url):
108
self.control_files.lock_read()
111
self.control_files.unlock()
113
def get_physical_lock_status(self):
117
class LocalGitBranch(GitBranch):
120
66
def last_revision(self):
121
67
# perhaps should escape this ?
122
if self.head is None:
68
head_git_id = self.repository._git.get_head()
69
if head_git_id is None:
123
70
return revision.NULL_REVISION
124
return self.mapping.revision_id_foreign_to_bzr(self.head)
126
def _make_tags(self):
127
return GitTagDict(self)
129
def _gen_revision_history(self):
130
if self.head is None:
71
return ids.convert_revision_id_git_to_bzr(head_git_id)
74
def revision_history(self):
75
node = self.last_revision()
76
if node == revision.NULL_REVISION:
132
ret = list(self.repository.iter_reverse_revision_history(self.last_revision()))
78
ancestors = self.repository.get_revision_graph(node)
80
while node is not None:
82
if len(ancestors[node]) > 0:
83
node = ancestors[node][0]
86
return list(reversed(history))
136
88
def get_config(self):
137
89
return GitBranchConfig(self)
92
self.control_files.lock_read()
95
self.control_files.unlock()
139
97
def get_push_location(self):
140
98
"""See Branch.get_push_location."""
141
99
push_loc = self.get_config().get_user_option('push_location')
144
102
def set_push_location(self, location):
145
103
"""See Branch.set_push_location."""
146
104
self.get_config().set_user_option('push_location', location,
147
store=config.STORE_LOCATION)
149
107
def supports_tags(self):
152
def sprout(self, to_bzrdir, revision_id=None):
153
"""See Branch.sprout()."""
154
result = to_bzrdir.create_branch()
155
self.copy_content_into(result, revision_id=revision_id)
156
result.set_parent(self.bzrdir.root_transport.base)