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 (
36
class GitLock(object):
37
"""A lock that thunks through to Git."""
39
def lock_write(self, token=None):
51
def validate_token(self, token):
58
class GitLockableFiles(lockable_files.LockableFiles):
59
"""Git specific lockable files abstraction."""
61
def __init__(self, transport, lock):
63
self._transaction = None
64
self._lock_mode = None
65
self._transport = transport
66
if LockWarner is None:
70
self._lock_warner = LockWarner(repr(self))
73
class GitDir(bzrdir.BzrDir):
74
"""An adapter to the '.git' dir used by git."""
76
def is_supported(self):
79
def cloning_metadir(self, stacked=False):
80
return get_rich_root_format(stacked)
82
def _branch_name_to_ref(self, name):
83
raise NotImplementedError(self._branch_name_to_ref)
85
if bzrlib_version >= (2, 2):
86
def open_branch(self, name=None, ignore_fallbacks=None,
88
return self._open_branch(name=name,
89
ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
91
def open_branch(self, ignore_fallbacks=None, unsupported=False):
92
return self._open_branch(name=None,
93
ignore_fallbacks=ignore_fallbacks, unsupported=unsupported)
96
class LocalGitDir(GitDir):
97
"""An adapter to the '.git' dir used by git."""
99
def _get_gitrepository_class(self):
100
from bzrlib.plugins.git.repository import LocalGitRepository
101
return LocalGitRepository
103
_gitrepository_class = property(_get_gitrepository_class)
105
def __init__(self, transport, lockfiles, gitrepo, format):
106
self._format = format
107
self.root_transport = transport
110
self.transport = transport
112
self.transport = transport.clone('.git')
113
self._lockfiles = lockfiles
114
self._mode_check_done = None
116
def _branch_name_to_ref(self, name):
117
from bzrlib.plugins.git.refs import branch_name_to_ref
118
if name in (None, "HEAD"):
119
from dulwich.repo import SYMREF
120
refcontents = self._git.refs.read_ref("HEAD")
121
if refcontents.startswith(SYMREF):
122
name = refcontents[len(SYMREF):]
125
return branch_name_to_ref(name, "HEAD")
127
def is_control_filename(self, filename):
128
return filename == '.git' or filename.startswith('.git/')
130
def get_branch_transport(self, branch_format, name):
131
if branch_format is None:
132
return self.transport
133
if isinstance(branch_format, LocalGitBzrDirFormat):
134
return self.transport
135
raise bzr_errors.IncompatibleFormat(branch_format, self._format)
137
def get_repository_transport(self, format):
139
return self.transport
140
if isinstance(format, LocalGitBzrDirFormat):
141
return self.transport
142
raise bzr_errors.IncompatibleFormat(format, self._format)
144
def get_workingtree_transport(self, format):
146
return self.transport
147
if isinstance(format, LocalGitBzrDirFormat):
148
return self.transport
149
raise bzr_errors.IncompatibleFormat(format, self._format)
151
def _open_branch(self, name=None, ignore_fallbacks=None, unsupported=False):
152
"""'create' a branch for this dir."""
153
repo = self.open_repository()
154
from bzrlib.plugins.git.branch import LocalGitBranch
155
return LocalGitBranch(self, repo, self._branch_name_to_ref(name),
158
def destroy_branch(self, name=None):
159
del self._git.refs[self._branch_name_to_ref(name)]
161
def list_branches(self):
163
for name in self._git.get_refs():
164
if name.startswith("refs/heads/"):
165
ret.append(self.open_branch(name=name))
168
def open_repository(self, shared=False):
169
"""'open' a repository for this dir."""
170
return self._gitrepository_class(self, self._lockfiles)
172
def open_workingtree(self, recommend_upgrade=True):
173
if not self._git.bare:
174
from dulwich.errors import NoIndexPresent
175
repo = self.open_repository()
177
index = repo._git.open_index()
178
except NoIndexPresent:
181
from bzrlib.plugins.git.workingtree import GitWorkingTree
182
return GitWorkingTree(self, repo, self.open_branch(), index)
183
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
184
raise bzr_errors.NoWorkingTree(loc)
186
def create_repository(self, shared=False):
187
return self.open_repository()
189
def create_branch(self, name=None):
190
refname = self._branch_name_to_ref(name)
191
from dulwich.protocol import ZERO_SHA
192
self._git.refs[refname] = ZERO_SHA
193
return self.open_branch(name)
195
def backup_bzrdir(self):
197
self.root_transport.copy_tree(".git", ".git.backup")
198
return (self.root_transport.abspath(".git"),
199
self.root_transport.abspath(".git.backup"))
201
raise bzr_errors.BzrError("Unable to backup bare repositories")
203
def create_workingtree(self, revision_id=None, from_branch=None,
204
accelerator_tree=None, hardlink=False):
206
raise bzr_errors.BzrError("Can't create working tree in a bare repo")
207
from dulwich.index import write_index
208
from dulwich.pack import SHA1Writer
209
f = open(self.transport.local_abspath("index"), 'w+')
215
return self.open_workingtree()