133
123
revspec_registry.register_lazy("git:", "bzrlib.plugins.git.revspec",
134
124
"RevisionSpec_git")
137
from bzrlib.revisionspec import dwim_revspecs
126
from bzrlib.revisionspec import dwim_revspecs, RevisionSpec_dwim
127
if getattr(RevisionSpec_dwim, "append_possible_lazy_revspec", None):
128
RevisionSpec_dwim.append_possible_lazy_revspec(
129
"bzrlib.plugins.git.revspec", "RevisionSpec_git")
141
131
from bzrlib.plugins.git.revspec import RevisionSpec_git
142
132
dwim_revspecs.append(RevisionSpec_git)
145
class GitControlDirFormat(ControlDirFormat):
147
_lock_class = TransportLock
149
colocated_branches = True
151
def __eq__(self, other):
152
return type(self) == type(other)
154
def is_supported(self):
157
def network_name(self):
161
135
class LocalGitProber(Prober):
163
137
def probe_transport(self, transport):
165
if not transport.has_any(['info/refs', '.git/branches',
167
raise bzr_errors.NotBranchError(path=transport.base)
168
except bzr_errors.NoSuchFile:
139
external_url = transport.external_url()
140
except bzr_errors.InProcessTransport:
141
raise bzr_errors.NotBranchError(path=transport.base)
142
if (external_url.startswith("http:") or
143
external_url.startswith("https:")):
144
# Already handled by RemoteGitProber
169
145
raise bzr_errors.NotBranchError(path=transport.base)
170
146
from bzrlib import urlutils
171
147
if urlutils.split(transport.base)[1] == ".git":
172
148
raise bzr_errors.NotBranchError(path=transport.base)
149
if not transport.has_any(['objects', '.git/objects']):
150
raise bzr_errors.NotBranchError(path=transport.base)
173
151
lazy_check_versions()
175
from bzrlib.plugins.git.transportgit import TransportRepo
177
gitrepo = TransportRepo(transport)
178
except dulwich.errors.NotGitRepository, e:
179
raise bzr_errors.NotBranchError(path=transport.base)
152
from bzrlib.plugins.git.dir import (
153
BareLocalGitControlDirFormat,
154
LocalGitControlDirFormat,
156
if transport.has_any(['.git/objects']):
157
return LocalGitControlDirFormat()
158
if transport.has('info') and transport.has('objects'):
159
return BareLocalGitControlDirFormat()
160
raise bzr_errors.NotBranchError(path=transport.base)
163
def known_formats(cls):
164
from bzrlib.plugins.git.dir import (
165
BareLocalGitControlDirFormat,
166
LocalGitControlDirFormat,
168
return set([BareLocalGitControlDirFormat(), LocalGitControlDirFormat()])
171
class RemoteGitProber(Prober):
173
def probe_http_transport(self, transport):
174
from bzrlib import urlutils
175
base_url, _ = urlutils.split_segment_parameters(transport.external_url())
176
url = urlutils.join(base_url, "info/refs") + "?service=git-upload-pack"
177
from bzrlib.transport.http._urllib import HttpTransport_urllib, Request
178
if isinstance(transport, HttpTransport_urllib):
179
req = Request('GET', url, accepted_errors=[200, 403, 404, 405],
180
headers={"Content-Type": "application/x-git-upload-pack-request"})
181
req.follow_redirections = True
182
resp = transport._perform(req)
183
if resp.code in (404, 405):
184
raise bzr_errors.NotBranchError(transport.base)
185
headers = resp.headers
186
refs_text = resp.read()
182
return BareLocalGitControlDirFormat()
189
from bzrlib.transport.http._pycurl import PyCurlTransport
190
except bzr_errors.DependencyNotPresent:
191
raise bzr_errors.NotBranchError(transport.base)
184
return LocalGitControlDirFormat()
187
class LocalGitControlDirFormat(GitControlDirFormat):
188
"""The .git directory control format."""
193
def _known_formats(self):
194
return set([LocalGitControlDirFormat()])
196
def open(self, transport, _found=None):
197
"""Open this directory.
200
lazy_check_versions()
201
from bzrlib.plugins.git.transportgit import TransportRepo
202
gitrepo = TransportRepo(transport)
203
from bzrlib.plugins.git.dir import LocalGitDir, GitLockableFiles, GitLock
204
lockfiles = GitLockableFiles(transport, GitLock())
205
return LocalGitDir(transport, lockfiles, gitrepo, self)
208
def probe_transport(klass, transport):
209
prober = LocalGitProber()
210
return prober.probe_transport(transport)
212
def get_format_description(self):
213
return "Local Git Repository"
215
def initialize_on_transport(self, transport):
216
from bzrlib.transport.local import LocalTransport
218
if not isinstance(transport, LocalTransport):
219
raise NotImplementedError(self.initialize,
220
"Can't create Git Repositories/branches on "
221
"non-local transports")
222
lazy_check_versions()
223
from dulwich.repo import Repo
224
Repo.init(transport.local_abspath(".").encode(osutils._fs_enc),
226
return self.open(transport)
228
def is_supported(self):
232
class BareLocalGitControlDirFormat(LocalGitControlDirFormat):
235
supports_workingtrees = False
238
def _known_formats(self):
239
return set([RemoteGitControlDirFormat()])
241
def get_format_description(self):
242
return "Local Git Repository (bare)"
245
class RemoteGitProber(Prober):
194
from cStringIO import StringIO
195
if isinstance(transport, PyCurlTransport):
196
conn = transport._get_curl()
197
conn.setopt(pycurl.URL, url)
198
conn.setopt(pycurl.FOLLOWLOCATION, 1)
199
transport._set_curl_options(conn)
200
conn.setopt(pycurl.HTTPGET, 1)
203
conn.setopt(pycurl.HEADERFUNCTION, header.write)
204
conn.setopt(pycurl.WRITEFUNCTION, data.write)
205
transport._curl_perform(conn, header,
206
["Content-Type: application/x-git-upload-pack-request"])
207
code = conn.getinfo(pycurl.HTTP_CODE)
208
if code in (404, 405):
209
raise bzr_errors.NotBranchError(transport.base)
211
raise bzr_errors.InvalidHttpResponse(transport._path,
213
headers = transport._parse_headers(header)
215
raise bzr_errors.NotBranchError(transport.base)
216
refs_text = data.getvalue()
217
ct = headers.getheader("Content-Type")
219
raise bzr_errors.NotBranchError(transport.base)
220
if ct.startswith("application/x-git"):
221
from bzrlib.plugins.git.remote import RemoteGitControlDirFormat
222
return RemoteGitControlDirFormat()
224
from bzrlib.plugins.git.dir import (
225
BareLocalGitControlDirFormat,
227
ret = BareLocalGitControlDirFormat()
228
ret._refs_text = refs_text
247
231
def probe_transport(self, transport):
249
if url.startswith('readonly+'):
250
url = url[len('readonly+'):]
251
if (not url.startswith("git://") and not url.startswith("git+")):
233
external_url = transport.external_url()
234
except bzr_errors.InProcessTransport:
235
raise bzr_errors.NotBranchError(path=transport.base)
237
if (external_url.startswith("http:") or
238
external_url.startswith("https:")):
239
return self.probe_http_transport(transport)
241
if (not external_url.startswith("git://") and
242
not external_url.startswith("git+")):
252
243
raise bzr_errors.NotBranchError(transport.base)
253
245
# little ugly, but works
254
from bzrlib.plugins.git.remote import GitSmartTransport
255
if not isinstance(transport, GitSmartTransport):
256
raise bzr_errors.NotBranchError(transport.base)
257
return RemoteGitControlDirFormat()
261
class RemoteGitControlDirFormat(GitControlDirFormat):
262
"""The .git directory control format."""
264
supports_workingtrees = False
246
from bzrlib.plugins.git.remote import (
248
RemoteGitControlDirFormat,
250
if isinstance(transport, GitSmartTransport):
251
return RemoteGitControlDirFormat()
252
raise bzr_errors.NotBranchError(path=transport.base)
267
def _known_formats(self):
255
def known_formats(cls):
256
from bzrlib.plugins.git.remote import RemoteGitControlDirFormat
268
257
return set([RemoteGitControlDirFormat()])
270
def open(self, transport, _found=None):
271
"""Open this directory.
274
# we dont grok readonly - git isn't integrated with transport.
276
if url.startswith('readonly+'):
277
url = url[len('readonly+'):]
278
if (not url.startswith("git://") and not url.startswith("git+")):
279
raise bzr_errors.NotBranchError(transport.base)
280
from bzrlib.plugins.git.remote import RemoteGitDir, GitSmartTransport
281
if not isinstance(transport, GitSmartTransport):
282
raise bzr_errors.NotBranchError(transport.base)
283
from bzrlib.plugins.git.dir import GitLockableFiles, GitLock
284
lockfiles = GitLockableFiles(transport, GitLock())
285
return RemoteGitDir(transport, lockfiles, self)
288
def probe_transport(klass, transport):
289
"""Our format is present if the transport ends in '.not/'."""
290
prober = RemoteGitProber()
291
return prober.probe_transport(transport)
293
def get_format_description(self):
294
return "Remote Git Repository"
296
def initialize_on_transport(self, transport):
297
raise bzr_errors.UninitializableFormat(self)
260
if not getattr(Prober, "known_formats", None): # bzr < 2.4
261
from bzrlib.plugins.git.dir import (
262
LocalGitControlDirFormat, BareLocalGitControlDirFormat,
264
from bzrlib.plugins.git.remote import RemoteGitControlDirFormat
301
265
ControlDirFormat.register_format(LocalGitControlDirFormat())
302
266
ControlDirFormat.register_format(BareLocalGitControlDirFormat())
303
267
ControlDirFormat.register_format(RemoteGitControlDirFormat())
304
ControlDirFormat.register_prober(LocalGitProber)
305
ControlDirFormat.register_prober(RemoteGitProber)
307
ControlDirFormat.register_control_format(LocalGitControlDirFormat)
308
ControlDirFormat.register_control_format(BareLocalGitControlDirFormat)
309
ControlDirFormat.register_control_format(RemoteGitControlDirFormat)
268
# Provide RevisionTree.get_file_revision, so various parts of bzr-svn
269
# can avoid inventories.
270
from bzrlib.revisiontree import RevisionTree
271
def get_file_revision(tree, file_id, path=None):
272
return tree.inventory[file_id].revision
273
RevisionTree.get_file_revision = get_file_revision
275
ControlDirFormat.register_prober(LocalGitProber)
276
ControlDirFormat._server_probers.insert(0, RemoteGitProber)
311
278
register_transport_proto('git://',
312
279
help="Access using the Git smart server protocol.")
345
340
'Git Smart server protocol over TCP. (default port: 9418)')
342
transport_server_registry.register_lazy('git-receive-pack',
343
'bzrlib.plugins.git.server',
344
'serve_git_receive_pack',
345
help='Git Smart server receive pack command (inetd mode only)')
346
transport_server_registry.register_lazy('git-upload-pack',
347
'bzrlib.plugins.git.server',
348
'serve_git_upload_pack',
349
help='Git Smart server upload pack command (inetd mode only)')
348
351
from bzrlib.repository import (
352
format_registry as repository_format_registry,
349
353
network_format_registry as repository_network_format_registry,
351
355
repository_network_format_registry.register_lazy('git',
352
356
'bzrlib.plugins.git.repository', 'GitRepositoryFormat')
355
from bzrlib.controldir import (
356
network_format_registry as controldir_network_format_registry,
359
from bzrlib.bzrdir import (
360
network_format_registry as controldir_network_format_registry,
362
controldir_network_format_registry.register('git', GitControlDirFormat)
359
register_extra_lazy_repository_format = getattr(repository_format_registry,
360
"register_extra_lazy")
361
except AttributeError: # bzr < 2.4
364
register_extra_lazy_repository_format('bzrlib.plugins.git.repository',
365
'GitRepositoryFormat')
367
from bzrlib.branch import (
368
network_format_registry as branch_network_format_registry,
370
branch_network_format_registry.register_lazy('git',
371
'bzrlib.plugins.git.branch', 'GitBranchFormat')
374
from bzrlib.branch import (
375
format_registry as branch_format_registry,
377
except ImportError: # bzr < 2.4
380
branch_format_registry.register_extra_lazy(
381
'bzrlib.plugins.git.branch',
386
from bzrlib.workingtree import (
387
format_registry as workingtree_format_registry,
389
except ImportError: # bzr < 2.4
392
workingtree_format_registry.register_extra_lazy(
393
'bzrlib.plugins.git.workingtree',
394
'GitWorkingTreeFormat',
397
controldir_network_format_registry.register_lazy('git',
398
"bzrlib.plugins.git.dir", "GitControlDirFormat")
364
400
send_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',
365
401
'send_git', 'Git am-style diff format')
367
topic_registry.register_lazy('git',
368
'bzrlib.plugins.git.help',
369
'help_git', 'Using Bazaar with Git')
403
topic_registry.register_lazy('git', 'bzrlib.plugins.git.help', 'help_git',
404
'Using Bazaar with Git')
406
from bzrlib.diff import format_registry as diff_format_registry
407
diff_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',
408
'GitDiffTree', 'Git am-style diff format')
411
def update_git_cache(repository, revid):
412
"""Update the git cache after a local commit."""
413
if getattr(repository, "_git", None) is not None:
414
return # No need to update cache for git repositories
416
if not repository.control_transport.has("git"):
417
return # No existing cache, don't bother updating
419
lazy_check_versions()
420
except bzr_errors.DependencyNotPresent, e:
421
# dulwich is probably missing. silently ignore
422
trace.mutter("not updating git map for %r: %s",
425
from bzrlib.plugins.git.object_store import BazaarObjectStore
426
store = BazaarObjectStore(repository)
429
parent_revisions = set(repository.get_parent_map([revid])[revid])
430
missing_revisions = store._missing_revisions(parent_revisions)
431
if not missing_revisions:
432
# Only update if the cache was up to date previously
433
store._update_sha_map_revision(revid)
438
def post_commit_update_cache(local_branch, master_branch, old_revno, old_revid,
439
new_revno, new_revid):
440
if local_branch is not None:
441
update_git_cache(local_branch.repository, new_revid)
442
update_git_cache(master_branch.repository, new_revid)
445
def loggerhead_git_hook(branch_app, environ):
446
from bzrlib.config import GlobalConfig
447
branch = branch_app.branch
448
if GlobalConfig().get_user_option('http_git') != 'True':
450
from bzrlib.plugins.git.server import git_http_hook
451
return git_http_hook(branch, environ['REQUEST_METHOD'],
452
environ['PATH_INFO'])
372
from bzrlib.diff import format_registry as diff_format_registry
455
from bzrlib.hooks import install_lazy_named_hook
456
except ImportError: # Compatibility with bzr < 2.4
376
diff_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',
377
'GitDiffTree', 'Git am-style diff format')
459
install_lazy_named_hook("bzrlib.branch",
460
"Branch.hooks", "post_commit", post_commit_update_cache,
462
install_lazy_named_hook("bzrlib.plugins.loggerhead.apps.branch",
463
"BranchWSGIApp.hooks", "controller",
464
loggerhead_git_hook, "git support")
379
467
def test_suite():
380
468
from bzrlib.plugins.git import tests