/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

  • Committer: Robert Collins
  • Date: 2010-05-06 07:48:22 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506074822-0bsgf2j4h8jx0xkk
Added ``bzrlib.tests.matchers`` as a place to put matchers, along with
our first in-tree matcher. See the module docstring for details.
(Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
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", "bzr-git: Please install dulwich, https://launchpad.net/dulwich")
85
 
    else:
86
 
        if dulwich_version < dulwich_minimum_version:
87
 
            raise bzr_errors.DependencyNotPresent("dulwich", "bzr-git: Dulwich is too old; at least %d.%d.%d is required" % dulwich_minimum_version)
88
 
 
89
 
bzrdir.format_registry.register_lazy('git', 
90
 
    "bzrlib.plugins.git.dir", "LocalGitBzrDirFormat",
91
 
    help='GIT repository.', native=False, experimental=True,
92
 
    )
93
 
 
94
 
from bzrlib.revisionspec import revspec_registry
95
 
revspec_registry.register_lazy("git:", "bzrlib.plugins.git.revspec", 
96
 
    "RevisionSpec_git")
97
 
 
98
 
try:
99
 
    from bzrlib.revisionspec import dwim_revspecs
100
 
except ImportError:
101
 
    pass
102
 
else:
103
 
    from bzrlib.plugins.git.revspec import RevisionSpec_git
104
 
    dwim_revspecs.append(RevisionSpec_git)
105
 
 
106
 
 
107
 
class GitBzrDirFormat(bzrdir.BzrDirFormat):
108
 
 
109
 
    _lock_class = TransportLock
110
 
 
111
 
    def is_supported(self):
112
 
        return True
113
 
 
114
 
    def network_name(self):
115
 
        return "git"
116
 
 
117
 
 
118
 
class LocalGitBzrDirFormat(GitBzrDirFormat):
119
 
    """The .git directory control format."""
120
 
 
121
 
    @classmethod
122
 
    def _known_formats(self):
123
 
        return set([LocalGitBzrDirFormat()])
124
 
 
125
 
    def open(self, transport, _found=None):
126
 
        """Open this directory.
127
 
 
128
 
        """
129
 
        lazy_check_versions()
130
 
        import dulwich
131
 
        # we dont grok readonly - git isn't integrated with transport.
132
 
        url = transport.base
133
 
        if url.startswith('readonly+'):
134
 
            url = url[len('readonly+'):]
135
 
 
136
 
        try:
137
 
            gitrepo = dulwich.repo.Repo(transport.local_abspath(".").encode(osutils._fs_enc))
138
 
        except bzr_errors.NotLocalUrl:
139
 
            raise bzr_errors.NotBranchError(path=transport.base)
140
 
        from bzrlib.plugins.git.dir import LocalGitDir, GitLockableFiles, GitLock
141
 
        lockfiles = GitLockableFiles(transport, GitLock())
142
 
        return LocalGitDir(transport, lockfiles, gitrepo, self)
143
 
 
144
 
    @classmethod
145
 
    def probe_transport(klass, transport):
146
 
        """Our format is present if the transport ends in '.not/'."""
147
 
        from bzrlib.transport.local import LocalTransport
148
 
 
149
 
        if not isinstance(transport, LocalTransport):
150
 
            raise bzr_errors.NotBranchError(path=transport.base)
151
 
 
152
 
        # This should quickly filter out most things that are not 
153
 
        # git repositories, saving us the trouble from loading dulwich.
154
 
        if not transport.has(".git") and not transport.has("objects"):
155
 
            raise bzr_errors.NotBranchError(path=transport.base)
156
 
 
157
 
        lazy_check_versions()
158
 
        import dulwich
159
 
        format = klass()
160
 
        try:
161
 
            format.open(transport)
162
 
            return format
163
 
        except dulwich.errors.NotGitRepository, e:
164
 
            raise bzr_errors.NotBranchError(path=transport.base)
165
 
        raise bzr_errors.NotBranchError(path=transport.base)
166
 
 
167
 
    def get_format_description(self):
168
 
        return "Local Git Repository"
169
 
 
170
 
    def get_format_string(self):
171
 
        return "Local Git Repository"
172
 
 
173
 
    def initialize_on_transport(self, transport):
174
 
        from bzrlib.transport.local import LocalTransport
175
 
 
176
 
        if not isinstance(transport, LocalTransport):
177
 
            raise NotImplementedError(self.initialize, 
178
 
                "Can't create Git Repositories/branches on "
179
 
                "non-local transports")
180
 
        lazy_check_versions()
181
 
        from dulwich.repo import Repo
182
 
        Repo.create(transport.local_abspath(".").encode(osutils._fs_enc)) 
183
 
        return self.open(transport)
184
 
 
185
 
    def is_supported(self):
186
 
        return True
187
 
 
188
 
 
189
 
class RemoteGitBzrDirFormat(GitBzrDirFormat):
190
 
    """The .git directory control format."""
191
 
 
192
 
    @classmethod
193
 
    def _known_formats(self):
194
 
        return set([RemoteGitBzrDirFormat()])
195
 
 
196
 
    def open(self, transport, _found=None):
197
 
        """Open this directory.
198
 
 
199
 
        """
200
 
        # we dont grok readonly - git isn't integrated with transport.
201
 
        url = transport.base
202
 
        if url.startswith('readonly+'):
203
 
            url = url[len('readonly+'):]
204
 
        if (not url.startswith("git://") and 
205
 
            not url.startswith("git+")):
206
 
            raise bzr_errors.NotBranchError(transport.base)
207
 
        from bzrlib.plugins.git.remote import RemoteGitDir, GitSmartTransport
208
 
        if not isinstance(transport, GitSmartTransport):
209
 
            raise bzr_errors.NotBranchError(transport.base)
210
 
        from bzrlib.plugins.git.dir import GitLockableFiles, GitLock
211
 
        lockfiles = GitLockableFiles(transport, GitLock())
212
 
        return RemoteGitDir(transport, lockfiles, self)
213
 
 
214
 
    @classmethod
215
 
    def probe_transport(klass, transport):
216
 
        """Our format is present if the transport ends in '.not/'."""
217
 
        url = transport.base
218
 
        if url.startswith('readonly+'):
219
 
            url = url[len('readonly+'):]
220
 
        if (not url.startswith("git://") and 
221
 
            not url.startswith("git+")):
222
 
            raise bzr_errors.NotBranchError(transport.base)
223
 
        # little ugly, but works
224
 
        format = klass()
225
 
        from bzrlib.plugins.git.remote import GitSmartTransport
226
 
        if not isinstance(transport, GitSmartTransport):
227
 
            raise bzr_errors.NotBranchError(transport.base)
228
 
        return format
229
 
 
230
 
    def get_format_description(self):
231
 
        return "Remote Git Repository"
232
 
 
233
 
    def get_format_string(self):
234
 
        return "Remote Git Repository"
235
 
 
236
 
    def initialize_on_transport(self, transport):
237
 
        raise bzr_errors.UninitializableFormat(self)
238
 
 
239
 
 
240
 
bzrdir.BzrDirFormat.register_control_format(LocalGitBzrDirFormat)
241
 
bzrdir.BzrDirFormat.register_control_format(RemoteGitBzrDirFormat)
242
 
 
243
 
register_transport_proto('git://', 
244
 
        help="Access using the Git smart server protocol.")
245
 
register_transport_proto('git+ssh://', 
246
 
        help="Access using the Git smart server protocol over SSH.")
247
 
 
248
 
register_lazy_transport("git://", 'bzrlib.plugins.git.remote',
249
 
                        'TCPGitSmartTransport')
250
 
register_lazy_transport("git+ssh://", 'bzrlib.plugins.git.remote',
251
 
                        'SSHGitSmartTransport')
252
 
 
253
 
foreign_vcs_registry.register_lazy("git", 
254
 
    "bzrlib.plugins.git.mapping", "foreign_git", "Stupid content tracker")
255
 
 
256
 
plugin_cmds.register_lazy("cmd_git_import", [], "bzrlib.plugins.git.commands")
257
 
plugin_cmds.register_lazy("cmd_git_object", ["git-objects", "git-cat"], 
258
 
    "bzrlib.plugins.git.commands")
259
 
 
260
 
def update_stanza(rev, stanza):
261
 
    mapping = getattr(rev, "mapping", None)
262
 
    if mapping is not None and mapping.revid_prefix.startswith("git-"):
263
 
        stanza.add("git-commit", rev.foreign_revid)
264
 
 
265
 
 
266
 
rio_hooks = getattr(RioVersionInfoBuilder, "hooks", None)
267
 
if rio_hooks is not None:
268
 
    rio_hooks.install_named_hook('revision', update_stanza, None)
269
 
 
270
 
 
271
 
from bzrlib.transport import transport_server_registry
272
 
transport_server_registry.register_lazy('git',
273
 
    'bzrlib.plugins.git.server', 
274
 
    'serve_git',
275
 
    'Git Smart server protocol over TCP. (default port: 9418)')
276
 
 
277
 
 
278
 
from bzrlib.repository import network_format_registry as repository_network_format_registry
279
 
repository_network_format_registry.register_lazy('git', 
280
 
    'bzrlib.plugins.git.repository', 'GitRepositoryFormat')
281
 
 
282
 
from bzrlib.bzrdir import network_format_registry as bzrdir_network_format_registry
283
 
bzrdir_network_format_registry.register('git', GitBzrDirFormat)
284
 
 
285
 
 
286
 
def get_rich_root_format(stacked=False):
287
 
    if stacked:
288
 
        return bzrdir.format_registry.make_bzrdir("1.9-rich-root")
289
 
    else:
290
 
        return bzrdir.format_registry.make_bzrdir("default-rich-root")
291
 
 
292
 
send_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',
293
 
                                   'send_git', 'Git am-style diff format')
294
 
 
295
 
def test_suite():
296
 
    from bzrlib.plugins.git import tests
297
 
    return tests.test_suite()