1
# Copyright (C) 2007 Canonical Ltd
2
# Copyright (C) 2010 Jelmer Vernooij
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
"""An adapter between a Git control dir and a Bazaar ControlDir."""
24
version_info as bzrlib_version,
27
LockWarner = getattr(lockable_files, "_LockWarner", None)
29
from bzrlib.plugins.git import (
30
LocalGitControlDirFormat,
33
from bzrlib.controldir import (
39
from bzrlib.bzrdir import (
46
class GitLock(object):
47
"""A lock that thunks through to Git."""
49
def lock_write(self, token=None):
61
def validate_token(self, token):
68
class GitLockableFiles(lockable_files.LockableFiles):
69
"""Git specific lockable files abstraction."""
71
def __init__(self, transport, lock):
73
self._transaction = None
74
self._lock_mode = None
75
self._transport = transport
76
if LockWarner is None:
80
self._lock_warner = LockWarner(repr(self))
83
class GitDir(ControlDir):
84
"""An adapter to the '.git' dir used by git."""
86
def is_supported(self):
89
def can_convert_format(self):
92
def cloning_metadir(self, stacked=False):
93
return format_registry.make_bzrdir("default")
95
def _branch_name_to_ref(self, name):
96
raise NotImplementedError(self._branch_name_to_ref)
98
if bzrlib_version >= (2, 2):
99
def open_branch(self, name=None, unsupported=False,
100
ignore_fallbacks=None):
101
return self._open_branch(name=name,
102
ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
104
def open_branch(self, ignore_fallbacks=None, unsupported=False):
105
return self._open_branch(name=None,
106
ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
109
class LocalGitDir(GitDir):
110
"""An adapter to the '.git' dir used by git."""
112
def _get_gitrepository_class(self):
113
from bzrlib.plugins.git.repository import LocalGitRepository
114
return LocalGitRepository
116
_gitrepository_class = property(_get_gitrepository_class)
118
def __init__(self, transport, lockfiles, gitrepo, format):
119
self._format = format
120
self.root_transport = transport
123
self.transport = transport
125
self.transport = transport.clone('.git')
126
self._lockfiles = lockfiles
127
self._mode_check_done = None
129
def _branch_name_to_ref(self, name):
130
from bzrlib.plugins.git.refs import branch_name_to_ref
131
ref = branch_name_to_ref(name, None)
133
from dulwich.repo import SYMREF
134
refcontents = self._git.refs.read_ref(ref)
135
if refcontents.startswith(SYMREF):
136
ref = refcontents[len(SYMREF):]
139
def is_control_filename(self, filename):
140
return filename == '.git' or filename.startswith('.git/')
142
def get_branch_transport(self, branch_format, name=None):
143
if branch_format is None:
144
return self.transport
145
if isinstance(branch_format, LocalGitControlDirFormat):
146
return self.transport
147
raise bzr_errors.IncompatibleFormat(branch_format, self._format)
149
def get_repository_transport(self, format):
151
return self.transport
152
if isinstance(format, LocalGitControlDirFormat):
153
return self.transport
154
raise bzr_errors.IncompatibleFormat(format, self._format)
156
def get_workingtree_transport(self, format):
158
return self.transport
159
if isinstance(format, LocalGitControlDirFormat):
160
return self.transport
161
raise bzr_errors.IncompatibleFormat(format, self._format)
163
def _open_branch(self, name=None, ignore_fallbacks=None, unsupported=False):
164
"""'create' a branch for this dir."""
165
repo = self.open_repository()
166
from bzrlib.plugins.git.branch import LocalGitBranch
167
return LocalGitBranch(self, repo, self._branch_name_to_ref(name),
170
def destroy_branch(self, name=None):
171
refname = self._branch_name_to_ref(name)
172
if not refname in self._git.refs:
173
raise bzr_errors.NotBranchError(self.root_transport.base,
175
del self._git.refs[refname]
177
def destroy_repository(self):
178
raise bzr_errors.UnsupportedOperation(self.destroy_repository, self)
180
def destroy_workingtree(self):
181
raise bzr_errors.UnsupportedOperation(self.destroy_workingtree, self)
183
def needs_format_conversion(self, format=None):
184
return not isinstance(self._format, format.__class__)
186
def list_branches(self):
188
for name in self._git.get_refs():
189
if name.startswith("refs/heads/"):
190
ret.append(self.open_branch(name=name))
193
def open_repository(self, shared=False):
194
"""'open' a repository for this dir."""
195
return self._gitrepository_class(self, self._lockfiles)
197
def open_workingtree(self, recommend_upgrade=True):
198
if not self._git.bare:
199
from dulwich.errors import NoIndexPresent
200
repo = self.open_repository()
202
index = repo._git.open_index()
203
except NoIndexPresent:
206
from bzrlib.plugins.git.workingtree import GitWorkingTree
208
branch = self.open_branch()
209
except bzr_errors.NotBranchError:
212
return GitWorkingTree(self, repo, branch, index)
213
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
214
raise bzr_errors.NoWorkingTree(loc)
216
def create_repository(self, shared=False):
217
return self.open_repository()
219
def create_branch(self, name=None):
220
refname = self._branch_name_to_ref(name)
221
from dulwich.protocol import ZERO_SHA
222
self._git.refs[refname or "HEAD"] = ZERO_SHA
223
return self.open_branch(name)
225
def backup_bzrdir(self):
227
self.root_transport.copy_tree(".git", ".git.backup")
228
return (self.root_transport.abspath(".git"),
229
self.root_transport.abspath(".git.backup"))
231
raise bzr_errors.BzrError("Unable to backup bare repositories")
233
def create_workingtree(self, revision_id=None, from_branch=None,
234
accelerator_tree=None, hardlink=False):
236
raise bzr_errors.BzrError("Can't create working tree in a bare repo")
237
from dulwich.index import write_index
238
from dulwich.pack import SHA1Writer
239
f = open(self.transport.local_abspath("index"), 'w+')
245
return self.open_workingtree()