1
# Copyright (C) 2006 Canonical Ltd
3
# Authors: Robert Collins <robert.collins@canonical.com>
4
# Jelmer Vernooij <jelmer@samba.org>
5
# John Carr <john.carr@unrouted.co.uk>
7
# This program is free software; you can redistribute it and/or modify
8
# it under the terms of the GNU General Public License as published by
9
# the Free Software Foundation; either version 2 of the License, or
10
# (at your option) any later version.
12
# This program is distributed in the hope that it will be useful,
13
# but WITHOUT ANY WARRANTY; without even the implied warranty of
14
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15
# GNU General Public License for more details.
17
# You should have received a copy of the GNU General Public License
18
# along with this program; if not, write to the Free Software
19
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22
"""A GIT branch and repository format implementation for bzr."""
26
from bzrlib import bzrdir
27
from bzrlib.foreign import foreign_vcs_registry
28
from bzrlib.lockable_files import TransportLock
29
from bzrlib.transport import register_lazy_transport
30
from bzrlib.commands import Command, register_command
31
from bzrlib.option import Option
32
from bzrlib.trace import warning
34
MINIMUM_DULWICH_VERSION = (0, 1, 0)
35
COMPATIBLE_BZR_VERSIONS = [(1, 12, 0)]
37
_versions_checked = False
38
def lazy_check_versions():
39
global _versions_checked
42
_versions_checked = True
44
from dulwich import __version__ as dulwich_version
46
warning("Please install dulwich, https://launchpad.net/dulwich")
49
if dulwich_version < MINIMUM_DULWICH_VERSION:
50
warning("Dulwich is too old; at least %d.%d.%d is required" % MINIMUM_DULWICH_VERSION)
53
bzrlib.api.require_any_api(bzrlib, COMPATIBLE_BZR_VERSIONS)
55
bzrdir.format_registry.register_lazy('git',
56
"bzrlib.plugins.git.dir", "LocalGitBzrDirFormat",
57
help='GIT repository.', native=False, experimental=True,
60
class GitBzrDirFormat(bzrdir.BzrDirFormat):
61
_lock_class = TransportLock
63
def is_supported(self):
67
class LocalGitBzrDirFormat(GitBzrDirFormat):
68
"""The .git directory control format."""
71
def _known_formats(self):
72
return set([LocalGitBzrDirFormat()])
74
def open(self, transport, _found=None):
75
"""Open this directory.
79
# we dont grok readonly - git isn't integrated with transport.
81
if url.startswith('readonly+'):
82
url = url[len('readonly+'):]
85
gitrepo = git.repo.Repo(transport.local_abspath("."))
86
except errors.bzr_errors.NotLocalUrl:
87
raise errors.bzr_errors.NotBranchError(path=transport.base)
88
from bzrlib.plugins.git.dir import LocalGitDir, GitLockableFiles, GitLock
89
lockfiles = GitLockableFiles(transport, GitLock())
90
return LocalGitDir(transport, lockfiles, gitrepo, self)
93
def probe_transport(klass, transport):
94
"""Our format is present if the transport ends in '.not/'."""
95
from bzrlib.transport.local import LocalTransport
97
if not isinstance(transport, LocalTransport):
98
raise errors.bzr_errors.NotBranchError(path=transport.base)
100
# This should quickly filter out most things that are not
101
# git repositories, saving us the trouble from loading dulwich.
102
if not transport.has(".git") and not transport.has("objects"):
103
raise errors.bzr_errors.NotBranchError(path=transport.base)
105
import dulwich as git
108
format.open(transport)
110
except git.errors.NotGitRepository, e:
111
raise errors.bzr_errors.NotBranchError(path=transport.base)
112
raise errors.bzr_errors.NotBranchError(path=transport.base)
114
def get_format_description(self):
115
return "Local Git Repository"
117
def get_format_string(self):
118
return "Local Git Repository"
120
def initialize_on_transport(self, transport):
121
from bzrlib.transport.local import LocalTransport
123
if not isinstance(transport, LocalTransport):
124
raise NotImplementedError(self.initialize,
125
"Can't create Git Repositories/branches on "
126
"non-local transports")
128
from dulwich.repo import Repo
129
Repo.create(transport.local_abspath("."))
130
return self.open(transport)
132
def is_supported(self):
136
class RemoteGitBzrDirFormat(GitBzrDirFormat):
137
"""The .git directory control format."""
140
def _known_formats(self):
141
return set([RemoteGitBzrDirFormat()])
143
def open(self, transport, _found=None):
144
"""Open this directory.
147
from bzrlib.plugins.git.remote import RemoteGitDir, GitSmartTransport
148
if not isinstance(transport, GitSmartTransport):
149
raise errors.bzr_errors.NotBranchError(transport.base)
150
# we dont grok readonly - git isn't integrated with transport.
152
if url.startswith('readonly+'):
153
url = url[len('readonly+'):]
155
from bzrlib.plugins.git.dir import GitLockableFiles, GitLock
156
lockfiles = GitLockableFiles(transport, GitLock())
157
return RemoteGitDir(transport, lockfiles, self)
160
def probe_transport(klass, transport):
161
"""Our format is present if the transport ends in '.not/'."""
162
# little ugly, but works
164
from bzrlib.plugins.git.remote import GitSmartTransport
165
if not isinstance(transport, GitSmartTransport):
166
raise errors.bzr_errors.NotBranchError(transport.base)
167
# The only way to know a path exists and contains a valid repository
168
# is to do a request against it:
170
transport.fetch_pack(lambda x: [], None, lambda x: None,
171
lambda x: mutter("git: %s" % x))
172
except errors.git_errors.GitProtocolError:
173
raise errors.bzr_errors.NotBranchError(path=transport.base)
176
raise errors.bzr_errors.NotBranchError(path=transport.base)
178
def get_format_description(self):
179
return "Remote Git Repository"
181
def get_format_string(self):
182
return "Remote Git Repository"
184
def initialize_on_transport(self, transport):
185
raise errors.bzr_errors.UninitializableFormat(self)
188
bzrdir.BzrDirFormat.register_control_format(LocalGitBzrDirFormat)
189
bzrdir.BzrDirFormat.register_control_format(RemoteGitBzrDirFormat)
191
register_lazy_transport("git://", 'bzrlib.plugins.git.remote',
194
foreign_vcs_registry.register_lazy("git",
195
"bzrlib.plugins.git.mapping",
197
"Stupid content tracker")
200
class cmd_git_serve(Command):
201
"""Provide access to a Bazaar branch using the git protocol.
203
This command is experimental and doesn't do much yet.
207
help='serve contents of directory',
211
def run(self, directory=None):
212
lazy_check_versions()
213
from dulwich.server import TCPGitServer
214
from bzrlib.plugins.git.server import BzrBackend
215
from bzrlib.trace import warning
218
warning("server support in bzr-git is experimental.")
220
if directory is None:
221
directory = os.getcwd()
223
backend = BzrBackend(directory)
225
server = TCPGitServer(backend, 'localhost')
226
server.serve_forever()
228
register_command(cmd_git_serve)
231
class cmd_git_import(Command):
232
"""Import all branches from a git repository.
236
takes_args = ["src_location", "dest_location"]
238
def run(self, src_location, dest_location):
239
from bzrlib.bzrdir import BzrDir, format_registry
240
from bzrlib.errors import NoRepositoryPresent, NotBranchError
241
from bzrlib.repository import Repository
242
source_repo = Repository.open(src_location)
243
format = format_registry.make_bzrdir('rich-root-pack')
245
target_bzrdir = BzrDir.open(dest_location)
246
except NotBranchError:
247
target_bzrdir = BzrDir.create(dest_location, format=format)
249
target_repo = target_bzrdir.open_repository()
250
except NoRepositoryPresent:
251
target_repo = target_bzrdir.create_repository(shared=True)
253
target_repo.fetch(source_repo)
254
for name, ref in source_repo._git.heads().iteritems():
255
head_loc = os.path.join(dest_location, name)
257
head_bzrdir = BzrDir.open(head_loc)
258
except NotBranchError:
259
head_bzrdir = BzrDir.create(head_loc, format=format)
261
head_branch = head_bzrdir.open_branch()
262
except NotBranchError:
263
head_branch = head_bzrdir.create_branch()
264
head_branch.generate_revision_history(source_repo.get_mapping().revision_id_foreign_to_bzr(ref))
267
register_command(cmd_git_import)
271
from bzrlib.plugins.git import tests
272
return tests.test_suite()