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
ref = branch_name_to_ref(name, None)
120
from dulwich.repo import SYMREF
121
refcontents = self._git.refs.read_ref(ref)
122
if refcontents.startswith(SYMREF):
123
ref = refcontents[len(SYMREF):]
126
def is_control_filename(self, filename):
127
return filename == '.git' or filename.startswith('.git/')
129
def get_branch_transport(self, branch_format, name):
130
if branch_format is None:
131
return self.transport
132
if isinstance(branch_format, LocalGitBzrDirFormat):
133
return self.transport
134
raise bzr_errors.IncompatibleFormat(branch_format, self._format)
136
def get_repository_transport(self, format):
138
return self.transport
139
if isinstance(format, LocalGitBzrDirFormat):
140
return self.transport
141
raise bzr_errors.IncompatibleFormat(format, self._format)
143
def get_workingtree_transport(self, format):
145
return self.transport
146
if isinstance(format, LocalGitBzrDirFormat):
147
return self.transport
148
raise bzr_errors.IncompatibleFormat(format, self._format)
150
def _open_branch(self, name=None, ignore_fallbacks=None, unsupported=False):
151
"""'create' a branch for this dir."""
152
repo = self.open_repository()
153
from bzrlib.plugins.git.branch import LocalGitBranch
154
return LocalGitBranch(self, repo, self._branch_name_to_ref(name),
157
def destroy_branch(self, name=None):
158
del self._git.refs[self._branch_name_to_ref(name)]
160
def list_branches(self):
162
for name in self._git.get_refs():
163
if name.startswith("refs/heads/"):
164
ret.append(self.open_branch(name=name))
167
def open_repository(self, shared=False):
168
"""'open' a repository for this dir."""
169
return self._gitrepository_class(self, self._lockfiles)
171
def open_workingtree(self, recommend_upgrade=True):
172
if not self._git.bare:
173
from dulwich.errors import NoIndexPresent
174
repo = self.open_repository()
176
index = repo._git.open_index()
177
except NoIndexPresent:
180
from bzrlib.plugins.git.workingtree import GitWorkingTree
182
branch = self.open_branch()
183
except bzr_errors.NotBranchError:
186
return GitWorkingTree(self, repo, branch, index)
187
loc = urlutils.unescape_for_display(self.root_transport.base, 'ascii')
188
raise bzr_errors.NoWorkingTree(loc)
190
def create_repository(self, shared=False):
191
return self.open_repository()
193
def create_branch(self, name=None):
194
refname = self._branch_name_to_ref(name)
195
from dulwich.protocol import ZERO_SHA
196
self._git.refs[refname or "HEAD"] = ZERO_SHA
197
return self.open_branch(name)
199
def backup_bzrdir(self):
201
self.root_transport.copy_tree(".git", ".git.backup")
202
return (self.root_transport.abspath(".git"),
203
self.root_transport.abspath(".git.backup"))
205
raise bzr_errors.BzrError("Unable to backup bare repositories")
207
def create_workingtree(self, revision_id=None, from_branch=None,
208
accelerator_tree=None, hardlink=False):
210
raise bzr_errors.BzrError("Can't create working tree in a bare repo")
211
from dulwich.index import write_index
212
from dulwich.pack import SHA1Writer
213
f = open(self.transport.local_abspath("index"), 'w+')
219
return self.open_workingtree()