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 BzrDir."""
25
version_info as bzrlib_version,
28
LockWarner = getattr(lockable_files, "_LockWarner", None)
30
from bzrlib.plugins.git import (
35
class GitLock(object):
36
"""A lock that thunks through to Git."""
38
def lock_write(self, token=None):
50
def validate_token(self, token):
57
class GitLockableFiles(lockable_files.LockableFiles):
58
"""Git specific lockable files abstraction."""
60
def __init__(self, transport, lock):
62
self._transaction = None
63
self._lock_mode = None
64
self._transport = transport
65
if LockWarner is None:
69
self._lock_warner = LockWarner(repr(self))
72
class GitDir(bzrdir.BzrDir):
73
"""An adapter to the '.git' dir used by git."""
75
def is_supported(self):
78
def can_convert_format(self):
81
def cloning_metadir(self, stacked=False):
82
return bzrdir.format_registry.make_bzrdir("default")
84
def _branch_name_to_ref(self, name):
85
raise NotImplementedError(self._branch_name_to_ref)
87
if bzrlib_version >= (2, 2):
88
def open_branch(self, name=None, unsupported=False,
89
ignore_fallbacks=None):
90
return self._open_branch(name=name,
91
ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
93
def open_branch(self, ignore_fallbacks=None, unsupported=False):
94
return self._open_branch(name=None,
95
ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
98
class LocalGitDir(GitDir):
99
"""An adapter to the '.git' dir used by git."""
101
def _get_gitrepository_class(self):
102
from bzrlib.plugins.git.repository import LocalGitRepository
103
return LocalGitRepository
105
_gitrepository_class = property(_get_gitrepository_class)
107
def __init__(self, transport, lockfiles, gitrepo, format):
108
self._format = format
109
self.root_transport = transport
112
self.transport = transport
114
self.transport = transport.clone('.git')
115
self._lockfiles = lockfiles
116
self._mode_check_done = None
118
def _branch_name_to_ref(self, name):
119
from bzrlib.plugins.git.refs import branch_name_to_ref
120
ref = branch_name_to_ref(name, None)
122
from dulwich.repo import SYMREF
123
refcontents = self._git.refs.read_ref(ref)
124
if refcontents.startswith(SYMREF):
125
ref = refcontents[len(SYMREF):]
128
def is_control_filename(self, filename):
129
return filename == '.git' or filename.startswith('.git/')
131
def get_branch_transport(self, branch_format, name=None):
132
if branch_format is None:
133
return self.transport
134
if isinstance(branch_format, LocalGitBzrDirFormat):
135
return self.transport
136
raise bzr_errors.IncompatibleFormat(branch_format, self._format)
138
def get_repository_transport(self, format):
140
return self.transport
141
if isinstance(format, LocalGitBzrDirFormat):
142
return self.transport
143
raise bzr_errors.IncompatibleFormat(format, self._format)
145
def get_workingtree_transport(self, format):
147
return self.transport
148
if isinstance(format, LocalGitBzrDirFormat):
149
return self.transport
150
raise bzr_errors.IncompatibleFormat(format, self._format)
152
def _open_branch(self, name=None, ignore_fallbacks=None, unsupported=False):
153
"""'create' a branch for this dir."""
154
repo = self.open_repository()
155
from bzrlib.plugins.git.branch import LocalGitBranch
156
return LocalGitBranch(self, repo, self._branch_name_to_ref(name),
159
def destroy_branch(self, name=None):
160
refname = self._branch_name_to_ref(name)
161
if not refname in self._git.refs:
162
raise bzr_errors.NotBranchError(self.root_transport.base,
164
del self._git.refs[refname]
166
def destroy_repository(self):
167
raise bzr_errors.UnsupportedOperation(self.destroy_repository, self)
169
def destroy_workingtree(self):
170
raise bzr_errors.UnsupportedOperation(self.destroy_workingtree, self)
172
def needs_format_conversion(self, format=None):
173
return not isinstance(self._format, format.__class__)
175
def list_branches(self):
177
for name in self._git.get_refs():
178
if name.startswith("refs/heads/"):
179
ret.append(self.open_branch(name=name))
182
def open_repository(self, shared=False):
183
"""'open' a repository for this dir."""
184
return self._gitrepository_class(self, self._lockfiles)
186
def open_workingtree(self, recommend_upgrade=True):
187
if not self._git.bare:
188
from dulwich.errors import NoIndexPresent
189
repo = self.open_repository()
191
index = repo._git.open_index()
192
except NoIndexPresent:
195
from bzrlib.plugins.git.workingtree import GitWorkingTree
197
branch = self.open_branch()
198
except bzr_errors.NotBranchError:
201
return GitWorkingTree(self, repo, branch, index)
202
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
203
raise bzr_errors.NoWorkingTree(loc)
205
def create_repository(self, shared=False):
206
return self.open_repository()
208
def create_branch(self, name=None):
209
refname = self._branch_name_to_ref(name)
210
from dulwich.protocol import ZERO_SHA
211
self._git.refs[refname or "HEAD"] = ZERO_SHA
212
return self.open_branch(name)
214
def backup_bzrdir(self):
216
self.root_transport.copy_tree(".git", ".git.backup")
217
return (self.root_transport.abspath(".git"),
218
self.root_transport.abspath(".git.backup"))
220
raise bzr_errors.BzrError("Unable to backup bare repositories")
222
def create_workingtree(self, revision_id=None, from_branch=None,
223
accelerator_tree=None, hardlink=False):
225
raise bzr_errors.BzrError("Can't create working tree in a bare repo")
226
from dulwich.index import write_index
227
from dulwich.pack import SHA1Writer
228
f = open(self.transport.local_abspath("index"), 'w+')
234
return self.open_workingtree()