1
# Copyright (C) 2007-2009 Jelmer Vernooij <jelmer@samba.org>
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25
from bzrlib.errors import (
32
from bzrlib.transport import (
36
from bzrlib.plugins.git import (
41
from bzrlib.plugins.git.branch import (
45
from bzrlib.plugins.git.errors import (
46
GitSmartRemoteNotSupported,
49
from bzrlib.plugins.git.dir import (
52
from bzrlib.plugins.git.mapping import (
55
from bzrlib.plugins.git.repository import (
60
from dulwich.errors import (
63
from dulwich.pack import (
71
urlparse.uses_netloc.extend(['git', 'git+ssh'])
73
from dulwich.pack import load_pack_index
76
# Don't run any tests on GitSmartTransport as it is not intended to be
77
# a full implementation of Transport
78
def get_test_permutations():
82
def split_git_url(url):
86
:return: Tuple with host, port, username, path.
88
(scheme, netloc, loc, _, _) = urlparse.urlsplit(url)
89
path = urllib.unquote(loc)
90
if path.startswith("/~"):
92
(username, hostport) = urllib.splituser(netloc)
93
(host, port) = urllib.splitnport(hostport, None)
94
return (host, port, username, path)
97
class GitSmartTransport(Transport):
99
def __init__(self, url, _client=None):
100
Transport.__init__(self, url)
101
(self._host, self._port, self._username, self._path) = \
103
if 'transport' in debug.debug_flags:
104
trace.mutter('host: %r, user: %r, port: %r, path: %r',
105
self._host, self._username, self._port, self._path)
106
self._client = _client
108
def external_url(self):
111
def has(self, relpath):
114
def _get_client(self, thin_packs):
115
raise NotImplementedError(self._get_client)
120
def fetch_pack(self, determine_wants, graph_walker, pack_data, progress=None):
123
trace.info("git: %s" % text)
124
client = self._get_client(thin_packs=False)
126
return client.fetch_pack(self._get_path(), determine_wants,
127
graph_walker, pack_data, progress)
128
except GitProtocolError, e:
131
def send_pack(self, get_changed_refs, generate_pack_contents):
132
client = self._get_client(thin_packs=False)
134
return client.send_pack(self._get_path(), get_changed_refs,
135
generate_pack_contents)
136
except GitProtocolError, e:
140
raise NoSuchFile(path)
142
def abspath(self, relpath):
143
return urlutils.join(self.base, relpath)
145
def clone(self, offset=None):
146
"""See Transport.clone()."""
150
newurl = urlutils.join(self.base, offset)
152
return self.__class__(newurl, self._client)
155
class TCPGitSmartTransport(GitSmartTransport):
159
def _get_client(self, thin_packs):
160
if self._client is not None:
164
return git.client.TCPGitClient(self._host, self._port, thin_packs=thin_packs,
165
report_activity=self._report_activity)
168
class SSHGitSmartTransport(GitSmartTransport):
173
if self._path.startswith("/~/"):
174
return self._path[3:]
177
def _get_client(self, thin_packs):
178
if self._client is not None:
182
return git.client.SSHGitClient(self._host, self._port, self._username,
183
thin_packs=thin_packs, report_activity=self._report_activity)
186
class RemoteGitDir(GitDir):
188
def __init__(self, transport, lockfiles, format):
189
self._format = format
190
self.root_transport = transport
191
self.transport = transport
192
self._lockfiles = lockfiles
193
self._mode_check_done = None
195
def _branch_name_to_ref(self, name):
196
from bzrlib.plugins.git.branch import branch_name_to_ref
197
return branch_name_to_ref(name, default="refs/heads/master")
199
def open_repository(self):
200
return RemoteGitRepository(self, self._lockfiles)
202
def _open_branch(self, name=None, ignore_fallbacks=False,
204
repo = self.open_repository()
205
refname = self._branch_name_to_ref(name)
206
return RemoteGitBranch(self, repo, refname, self._lockfiles)
208
def open_workingtree(self, recommend_upgrade=False):
209
raise NotLocalUrl(self.transport.base)
212
class EmptyObjectStoreIterator(dict):
214
def iterobjects(self):
218
class TemporaryPackIterator(Pack):
220
def __init__(self, path, resolve_ext_ref):
221
super(TemporaryPackIterator, self).__init__(path)
222
self.resolve_ext_ref = resolve_ext_ref
226
if self._data is None:
227
self._data = PackData(self._data_path)
232
if self._idx is None:
233
if not os.path.exists(self._idx_path):
234
pb = ui.ui_factory.nested_progress_bar()
236
def report_progress(cur, total):
237
pb.update("generating index", cur, total)
238
self.data.create_index(self._idx_path, self.resolve_ext_ref,
239
progress=report_progress)
242
self._idx = load_pack_index(self._idx_path)
246
if self._idx is not None:
248
os.remove(self._idx_path)
249
if self._data is not None:
251
os.remove(self._data_path)
254
class RemoteGitRepository(GitRepository):
256
def __init__(self, gitdir, lockfiles):
257
GitRepository.__init__(self, gitdir, lockfiles)
261
def inventories(self):
262
raise GitSmartRemoteNotSupported()
266
raise GitSmartRemoteNotSupported()
270
raise GitSmartRemoteNotSupported()
273
if self._refs is not None:
275
self._refs = self.bzrdir.root_transport.fetch_pack(lambda x: [], None,
276
lambda x: None, lambda x: trace.mutter("git: %s" % x))
279
def fetch_pack(self, determine_wants, graph_walker, pack_data,
281
return self._transport.fetch_pack(determine_wants, graph_walker,
284
def send_pack(self, get_changed_refs, generate_pack_contents):
285
return self._transport.send_pack(get_changed_refs, generate_pack_contents)
287
def fetch_objects(self, determine_wants, graph_walker, resolve_ext_ref,
289
fd, path = tempfile.mkstemp(suffix=".pack")
290
self.fetch_pack(determine_wants, graph_walker,
291
lambda x: os.write(fd, x), progress)
293
if os.path.getsize(path) == 0:
294
return EmptyObjectStoreIterator()
295
return TemporaryPackIterator(path[:-len(".pack")], resolve_ext_ref)
297
def lookup_bzr_revision_id(self, bzr_revid):
298
# This won't work for any round-tripped bzr revisions, but it's a start..
300
return mapping_registry.revision_id_bzr_to_foreign(bzr_revid)
301
except InvalidRevisionId:
302
raise NoSuchRevision(self, bzr_revid)
305
class RemoteGitTagDict(tag.BasicTags):
307
def __init__(self, branch):
309
self.repository = branch.repository
311
def get_tag_dict(self):
313
for k, v in extract_tags(self.repository.get_refs()).iteritems():
314
tags[k] = self.branch.mapping.revision_id_foreign_to_bzr(v)
317
def set_tag(self, name, revid):
318
# FIXME: Not supported yet, should do a push of a new ref
319
raise NotImplementedError(self.set_tag)
322
class RemoteGitBranch(GitBranch):
324
def __init__(self, bzrdir, repository, name, lockfiles):
326
super(RemoteGitBranch, self).__init__(bzrdir, repository, name,
329
def revision_history(self):
330
raise GitSmartRemoteNotSupported()
332
def last_revision(self):
333
return self.mapping.revision_id_foreign_to_bzr(self.head)
335
def _get_config(self):
336
class EmptyConfig(object):
338
def _get_configobj(self):
339
return config.ConfigObj()
345
if self._ref is not None:
347
heads = self.repository.get_refs()
348
if self.name in heads:
349
self._ref = heads[self.name]
350
elif ("refs/heads/" + self.name) in heads:
351
self._ref = heads["refs/heads/" + self.name]
353
raise NoSuchRef(self.name)
356
def _synchronize_history(self, destination, revision_id):
357
"""See Branch._synchronize_history()."""
358
destination.generate_revision_history(self.last_revision())
360
def get_push_location(self):
363
def set_push_location(self, url):
added
removed
remote.py
