/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 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

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",
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()