/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to __init__.py

Add convenience command for accessing virtual git refs.

Show diffs side-by-side

added added

removed removed

Lines of Context:
__init__.py
 
1
# Copyright (C) 2006-2009 Canonical Ltd
 
2
 
 
3
# Authors: Robert Collins <robert.collins@canonical.com>
 
4
#          Jelmer Vernooij <jelmer@samba.org>
 
5
#          John Carr <john.carr@unrouted.co.uk>
 
6
#
 
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.
 
11
#
 
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.
 
16
#
 
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
 
20
 
 
21
 
 
22
"""A GIT branch and repository format implementation for bzr."""
 
23
 
 
24
import os
 
25
import sys
 
26
 
 
27
import bzrlib
 
28
import bzrlib.api
 
29
 
 
30
from info import (
 
31
    bzr_compatible_versions,
 
32
    bzr_plugin_version as version_info,
 
33
    dulwich_minimum_version,
 
34
    )
 
35
 
 
36
if version_info[3] == 'final':
 
37
    version_string = '%d.%d.%d' % version_info[:3]
 
38
else:
 
39
    version_string = '%d.%d.%d%s%d' % version_info
 
40
__version__ = version_string
 
41
 
 
42
bzrlib.api.require_any_api(bzrlib, bzr_compatible_versions)
 
43
 
 
44
 
 
45
from bzrlib import (
 
46
    bzrdir,
 
47
    errors as bzr_errors,
 
48
    osutils,
 
49
    )
 
50
from bzrlib.foreign import (
 
51
    foreign_vcs_registry,
 
52
    )
 
53
from bzrlib.lockable_files import (
 
54
    TransportLock,
 
55
    )
 
56
from bzrlib.transport import (
 
57
    register_lazy_transport,
 
58
    register_transport_proto,
 
59
    )
 
60
from bzrlib.commands import (
 
61
    plugin_cmds,
 
62
    )
 
63
from bzrlib.version_info_formats.format_rio import (
 
64
    RioVersionInfoBuilder,
 
65
    )
 
66
from bzrlib.send import (
 
67
    format_registry as send_format_registry,
 
68
    )
 
69
 
 
70
 
 
71
if getattr(sys, "frozen", None):
 
72
    # allow import additional libs from ./_lib for bzr.exe only
 
73
    sys.path.append(os.path.normpath(os.path.join(os.path.dirname(__file__), '_lib')))
 
74
 
 
75
_versions_checked = False
 
76
def lazy_check_versions():
 
77
    global _versions_checked
 
78
    if _versions_checked:
 
79
        return
 
80
    _versions_checked = True
 
81
    try:
 
82
        from dulwich import __version__ as dulwich_version
 
83
    except ImportError:
 
84
        raise bzr_errors.DependencyNotPresent("dulwich",
 
85
            "bzr-git: Please install dulwich, https://launchpad.net/dulwich")
 
86
    else:
 
87
        if dulwich_version < dulwich_minimum_version:
 
88
            raise bzr_errors.DependencyNotPresent("dulwich", "bzr-git: Dulwich is too old; at least %d.%d.%d is required" % dulwich_minimum_version)
 
89
 
 
90
bzrdir.format_registry.register_lazy('git',
 
91
    "bzrlib.plugins.git.dir", "LocalGitBzrDirFormat",
 
92
    help='GIT repository.', native=False, experimental=True,
 
93
    )
 
94
 
 
95
from bzrlib.revisionspec import revspec_registry
 
96
revspec_registry.register_lazy("git:", "bzrlib.plugins.git.revspec",
 
97
    "RevisionSpec_git")
 
98
 
 
99
try:
 
100
    from bzrlib.revisionspec import dwim_revspecs
 
101
except ImportError:
 
102
    pass
 
103
else:
 
104
    from bzrlib.plugins.git.revspec import RevisionSpec_git
 
105
    dwim_revspecs.append(RevisionSpec_git)
 
106
 
 
107
 
 
108
class GitBzrDirFormat(bzrdir.BzrDirFormat):
 
109
 
 
110
    _lock_class = TransportLock
 
111
 
 
112
    def is_supported(self):
 
113
        return True
 
114
 
 
115
    def network_name(self):
 
116
        return "git"
 
117
 
 
118
 
 
119
class LocalGitBzrDirFormat(GitBzrDirFormat):
 
120
    """The .git directory control format."""
 
121
 
 
122
    @classmethod
 
123
    def _known_formats(self):
 
124
        return set([LocalGitBzrDirFormat()])
 
125
 
 
126
    def open(self, transport, _found=None):
 
127
        """Open this directory.
 
128
 
 
129
        """
 
130
        lazy_check_versions()
 
131
        # we dont grok readonly - git isn't integrated with transport.
 
132
        from bzrlib.transport.local import LocalTransport
 
133
        if isinstance(transport, LocalTransport):
 
134
            import dulwich
 
135
            gitrepo = dulwich.repo.Repo(transport.local_abspath(".").encode(osutils._fs_enc))
 
136
        else:
 
137
            from bzrlib.plugins.git.transportgit import TransportRepo
 
138
            gitrepo = TransportRepo(transport)
 
139
        from bzrlib.plugins.git.dir import LocalGitDir, GitLockableFiles, GitLock
 
140
        lockfiles = GitLockableFiles(transport, GitLock())
 
141
        return LocalGitDir(transport, lockfiles, gitrepo, self)
 
142
 
 
143
    @classmethod
 
144
    def probe_transport(klass, transport):
 
145
        try:
 
146
            if not (transport.has('info/refs') or 
 
147
                    transport.has('.git/branches') or 
 
148
                    transport.has('branches')):
 
149
                raise bzr_errors.NotBranchError(path=transport.base)
 
150
        except bzr_errors.NoSuchFile:
 
151
            raise bzr_errors.NotBranchError(path=transport.base)
 
152
        from bzrlib import urlutils
 
153
        if urlutils.split(transport.base)[1] == ".git":
 
154
            raise bzr_errors.NotBranchError(path=transport.base)
 
155
        lazy_check_versions()
 
156
        import dulwich
 
157
        format = klass()
 
158
        try:
 
159
            format.open(transport)
 
160
            return format
 
161
        except dulwich.errors.NotGitRepository, e:
 
162
            raise bzr_errors.NotBranchError(path=transport.base)
 
163
        raise bzr_errors.NotBranchError(path=transport.base)
 
164
 
 
165
    def get_format_description(self):
 
166
        return "Local Git Repository"
 
167
 
 
168
    def get_format_string(self):
 
169
        return "Local Git Repository"
 
170
 
 
171
    def initialize_on_transport(self, transport):
 
172
        from bzrlib.transport.local import LocalTransport
 
173
 
 
174
        if not isinstance(transport, LocalTransport):
 
175
            raise NotImplementedError(self.initialize,
 
176
                "Can't create Git Repositories/branches on "
 
177
                "non-local transports")
 
178
        lazy_check_versions()
 
179
        from dulwich.repo import Repo
 
180
        Repo.init(transport.local_abspath(".").encode(osutils._fs_enc))
 
181
        return self.open(transport)
 
182
 
 
183
    def is_supported(self):
 
184
        return True
 
185
 
 
186
 
 
187
class RemoteGitBzrDirFormat(GitBzrDirFormat):
 
188
    """The .git directory control format."""
 
189
 
 
190
    @classmethod
 
191
    def _known_formats(self):
 
192
        return set([RemoteGitBzrDirFormat()])
 
193
 
 
194
    def open(self, transport, _found=None):
 
195
        """Open this directory.
 
196
 
 
197
        """
 
198
        # we dont grok readonly - git isn't integrated with transport.
 
199
        url = transport.base
 
200
        if url.startswith('readonly+'):
 
201
            url = url[len('readonly+'):]
 
202
        if (not url.startswith("git://") and not url.startswith("git+")):
 
203
            raise bzr_errors.NotBranchError(transport.base)
 
204
        from bzrlib.plugins.git.remote import RemoteGitDir, GitSmartTransport
 
205
        if not isinstance(transport, GitSmartTransport):
 
206
            raise bzr_errors.NotBranchError(transport.base)
 
207
        from bzrlib.plugins.git.dir import GitLockableFiles, GitLock
 
208
        lockfiles = GitLockableFiles(transport, GitLock())
 
209
        return RemoteGitDir(transport, lockfiles, self)
 
210
 
 
211
    @classmethod
 
212
    def probe_transport(klass, transport):
 
213
        """Our format is present if the transport ends in '.not/'."""
 
214
        url = transport.base
 
215
        if url.startswith('readonly+'):
 
216
            url = url[len('readonly+'):]
 
217
        if (not url.startswith("git://") and not url.startswith("git+")):
 
218
            raise bzr_errors.NotBranchError(transport.base)
 
219
        # little ugly, but works
 
220
        format = klass()
 
221
        from bzrlib.plugins.git.remote import GitSmartTransport
 
222
        if not isinstance(transport, GitSmartTransport):
 
223
            raise bzr_errors.NotBranchError(transport.base)
 
224
        return format
 
225
 
 
226
    def get_format_description(self):
 
227
        return "Remote Git Repository"
 
228
 
 
229
    def get_format_string(self):
 
230
        return "Remote Git Repository"
 
231
 
 
232
    def initialize_on_transport(self, transport):
 
233
        raise bzr_errors.UninitializableFormat(self)
 
234
 
 
235
 
 
236
bzrdir.BzrDirFormat.register_control_format(LocalGitBzrDirFormat)
 
237
bzrdir.BzrDirFormat.register_control_format(RemoteGitBzrDirFormat)
 
238
 
 
239
register_transport_proto('git://',
 
240
        help="Access using the Git smart server protocol.")
 
241
register_transport_proto('git+ssh://',
 
242
        help="Access using the Git smart server protocol over SSH.")
 
243
 
 
244
register_lazy_transport("git://", 'bzrlib.plugins.git.remote',
 
245
                        'TCPGitSmartTransport')
 
246
register_lazy_transport("git+ssh://", 'bzrlib.plugins.git.remote',
 
247
                        'SSHGitSmartTransport')
 
248
 
 
249
foreign_vcs_registry.register_lazy("git",
 
250
    "bzrlib.plugins.git.mapping", "foreign_git", "Stupid content tracker")
 
251
 
 
252
plugin_cmds.register_lazy("cmd_git_import", [], "bzrlib.plugins.git.commands")
 
253
plugin_cmds.register_lazy("cmd_git_object", ["git-objects", "git-cat"],
 
254
    "bzrlib.plugins.git.commands")
 
255
plugin_cmds.register_lazy("cmd_git_refs", [], "bzrlib.plugins.git.commands")
 
256
 
 
257
def update_stanza(rev, stanza):
 
258
    mapping = getattr(rev, "mapping", None)
 
259
    if mapping is not None and mapping.revid_prefix.startswith("git-"):
 
260
        stanza.add("git-commit", rev.foreign_revid)
 
261
 
 
262
 
 
263
rio_hooks = getattr(RioVersionInfoBuilder, "hooks", None)
 
264
if rio_hooks is not None:
 
265
    rio_hooks.install_named_hook('revision', update_stanza, None)
 
266
 
 
267
 
 
268
from bzrlib.transport import transport_server_registry
 
269
transport_server_registry.register_lazy('git',
 
270
    'bzrlib.plugins.git.server',
 
271
    'serve_git',
 
272
    'Git Smart server protocol over TCP. (default port: 9418)')
 
273
 
 
274
 
 
275
from bzrlib.repository import network_format_registry as repository_network_format_registry
 
276
repository_network_format_registry.register_lazy('git',
 
277
    'bzrlib.plugins.git.repository', 'GitRepositoryFormat')
 
278
 
 
279
from bzrlib.bzrdir import network_format_registry as bzrdir_network_format_registry
 
280
bzrdir_network_format_registry.register('git', GitBzrDirFormat)
 
281
 
 
282
 
 
283
def get_rich_root_format(stacked=False):
 
284
    if stacked:
 
285
        return bzrdir.format_registry.make_bzrdir("1.9-rich-root")
 
286
    else:
 
287
        return bzrdir.format_registry.make_bzrdir("default-rich-root")
 
288
 
 
289
send_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',
 
290
                                   'send_git', 'Git am-style diff format')
 
291
 
 
292
try:
 
293
    from bzrlib.diff import format_registry as diff_format_registry
 
294
except ImportError:
 
295
    pass
 
296
else:
 
297
    diff_format_registry.register_lazy('git-am', 'bzrlib.plugins.git.send',
 
298
        'GitDiffTree', 'Git am-style diff format')
 
299
 
 
300
def test_suite():
 
301
    from bzrlib.plugins.git import tests
 
302
    return tests.test_suite()