/brz/remove-bazaar

# Copyright (C) 2006-2009 Canonical Ltd

# Authors: Robert Collins <robert.collins@canonical.com>

#          Jelmer Vernooij <jelmer@samba.org>

#          John Carr <john.carr@unrouted.co.uk>

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""A GIT branch and repository format implementation for bzr."""

import os

import sys

import bzrlib

import bzrlib.api

from info import (

    bzr_compatible_versions,

    bzr_plugin_version as version_info,

    dulwich_minimum_version,

    )

if version_info[3] == 'final':

    version_string = '%d.%d.%d' % version_info[:3]

else:

    version_string = '%d.%d.%d%s%d' % version_info

__version__ = version_string

bzrlib.api.require_any_api(bzrlib, bzr_compatible_versions)

from bzrlib import (

    errors as bzr_errors,

    )

from bzrlib.controldir import (

    ControlDirFormat,

    Prober,

    format_registry,

    network_format_registry as controldir_network_format_registry,

    )

from bzrlib.foreign import (

    foreign_vcs_registry,

    )

from bzrlib.help_topics import (

    topic_registry,

    )

from bzrlib.transport import (

    register_lazy_transport,

    register_transport_proto,

    )

from bzrlib.commands import (

    plugin_cmds,

    )

from bzrlib.send import (

    format_registry as send_format_registry,

    )

if getattr(sys, "frozen", None):

    # allow import additional libs from ./_lib for bzr.exe only

    sys.path.append(os.path.normpath(

        os.path.join(os.path.dirname(__file__), '_lib')))

def import_dulwich():

    try:

        from dulwich import __version__ as dulwich_version

    except ImportError:

        raise bzr_errors.DependencyNotPresent("dulwich",

            "bzr-git: Please install dulwich, https://launchpad.net/dulwich")

    else:

        if dulwich_version < dulwich_minimum_version:

            raise bzr_errors.DependencyNotPresent("dulwich",

                "bzr-git: Dulwich is too old; at least %d.%d.%d is required" %

                    dulwich_minimum_version)

_versions_checked = False

def lazy_check_versions():

    global _versions_checked

    if _versions_checked:

        return

    import_dulwich()

    _versions_checked = True

format_registry.register_lazy('git',

    "bzrlib.plugins.git.dir", "LocalGitControlDirFormat",

    help='GIT repository.', native=False, experimental=False,

    )

format_registry.register_lazy('git-bare',

    "bzrlib.plugins.git.dir", "BareLocalGitControlDirFormat",

    help='Bare GIT repository (no working tree).', native=False,

    experimental=False,

    )

from bzrlib.revisionspec import revspec_registry

revspec_registry.register_lazy("git:", "bzrlib.plugins.git.revspec",

    "RevisionSpec_git")

from bzrlib.revisionspec import dwim_revspecs, RevisionSpec_dwim

if getattr(RevisionSpec_dwim, "append_possible_lazy_revspec", None):

    RevisionSpec_dwim.append_possible_lazy_revspec(

        "bzrlib.plugins.git.revspec", "RevisionSpec_git")

else: # bzr < 2.4

    from bzrlib.plugins.git.revspec import RevisionSpec_git

    dwim_revspecs.append(RevisionSpec_git)

class LocalGitProber(Prober):

    def probe_transport(self, transport):

        try:

            if not transport.has_any(['.git/HEAD', 'HEAD', 'objects', '.git/objects']):

                raise bzr_errors.NotBranchError(path=transport.base)

        except bzr_errors.NoSuchFile:

            raise bzr_errors.NotBranchError(path=transport.base)

        from bzrlib import urlutils

        if urlutils.split(transport.base)[1] == ".git":

            raise bzr_errors.NotBranchError(path=transport.base)

        lazy_check_versions()

        import dulwich

        from bzrlib.plugins.git.transportgit import TransportRepo

        try:

            gitrepo = TransportRepo(transport)

        except dulwich.errors.NotGitRepository, e:

            raise bzr_errors.NotBranchError(path=transport.base)

        else:

            from bzrlib.plugins.git.dir import (

                BareLocalGitControlDirFormat,

                LocalGitControlDirFormat,

                )

            if gitrepo.bare:

                return BareLocalGitControlDirFormat()

            else:

                return LocalGitControlDirFormat()

    @classmethod

    def known_formats(cls):

        from bzrlib.plugins.git.dir import (

            BareLocalGitControlDirFormat,

            LocalGitControlDirFormat,

            )

        return set([BareLocalGitControlDirFormat(), LocalGitControlDirFormat()])

class RemoteGitProber(Prober):

    def probe_http_transport(self, transport):

        from bzrlib import urlutils

        url = urlutils.join(transport.external_url(), "info/refs") + "?service=git-upload-pack"

        from bzrlib.transport.http._urllib import HttpTransport_urllib, Request

        if isinstance(transport, HttpTransport_urllib):

            req = Request('GET', url, accepted_errors=[200, 403, 404, 405],

                          headers={"Content-Type": "application/x-git-upload-pack-request"})

            req.follow_redirections = True

            resp = transport._perform(req)

            if resp.code == 404:

                raise bzr_errors.NotBranchError(transport.base)

            headers = resp.headers

        else:

            try:

                from bzrlib.transport.http._pycurl import PyCurlTransport

            except bzr_errors.DependencyNotPresent:

                raise bzr_errors.NotBranchError(transport.base)

            else:

                import pycurl

                from cStringIO import StringIO

                if isinstance(transport, PyCurlTransport):

                    conn = transport._get_curl()

                    conn.setopt(pycurl.URL, url)

                    transport._set_curl_options(conn)

                    conn.setopt(pycurl.HTTPGET, 1)

                    conn.setopt(pycurl.NOBODY, 1)

                    header = StringIO()

                    data = StringIO()

                    conn.setopt(pycurl.HEADERFUNCTION, header.write)

                    conn.setopt(pycurl.WRITEFUNCTION, data.write)

                    transport._curl_perform(conn, header,

                        ["Content-Type: application/x-git-upload-pack-request"])

                    code = conn.getinfo(pycurl.HTTP_CODE)

                    if code == 404:

                        raise bzr_errors.NotBranchError(transport.base)

                    headers = transport._parse_headers(header)

                else:

                    raise bzr_errors.NotBranchError(transport.base)

        ct = headers.getheader("Content-Type")

        if ct.startswith("application/x-git"):

            from bzrlib.plugins.git.remote import RemoteGitControlDirFormat

            return RemoteGitControlDirFormat()

        else:

            from bzrlib.plugins.git.dir import (

                BareLocalGitControlDirFormat,

                )

            return BareLocalGitControlDirFormat()

    def probe_transport(self, transport):

        try:

            external_url = transport.external_url()

        except bzr_errors.InProcessTransport:

            raise bzr_errors.NotBranchError(path=transport.base)

        if (external_url.startswith("http:") or

            external_url.startswith("https:")):

            return self.probe_http_transport(transport)

        if (not external_url.startswith("git://") and

            not external_url.startswith("git+")):

            raise bzr_errors.NotBranchError(transport.base)

        # little ugly, but works

        from bzrlib.plugins.git.remote import (

            GitSmartTransport,

            RemoteGitControlDirFormat,

            )

        if isinstance(transport, GitSmartTransport):

            return RemoteGitControlDirFormat()

        raise bzr_errors.NotBranchError(path=transport.base)

    @classmethod

    def known_formats(cls):

        from bzrlib.plugins.git.remote import RemoteGitControlDirFormat

        return set([RemoteGitControlDirFormat()])

if not getattr(Prober, "known_formats", None): # bzr < 2.4

    from bzrlib.plugins.git.dir import (

        LocalGitControlDirFormat, BareLocalGitControlDirFormat,

        )

    from bzrlib.plugins.git.remote import RemoteGitControlDirFormat

    ControlDirFormat.register_format(LocalGitControlDirFormat())

    ControlDirFormat.register_format(BareLocalGitControlDirFormat())

    ControlDirFormat.register_format(RemoteGitControlDirFormat())

    # Provide RevisionTree.get_file_revision, so various parts of bzr-svn

    # can avoid inventories.

    from bzrlib.revisiontree import RevisionTree

    def get_file_revision(tree, file_id, path=None):

        return tree.inventory[file_id].revision

    RevisionTree.get_file_revision = get_file_revision

ControlDirFormat.register_prober(LocalGitProber)

ControlDirFormat._server_probers.insert(0, RemoteGitProber)

register_transport_proto('git://',

        help="Access using the Git smart server protocol.")

register_transport_proto('git+ssh://',

        help="Access using the Git smart server protocol over SSH.")

register_lazy_transport("git://", 'bzrlib.plugins.git.remote',

                        'TCPGitSmartTransport')

register_lazy_transport("git+ssh://", 'bzrlib.plugins.git.remote',

                        'SSHGitSmartTransport')

foreign_vcs_registry.register_lazy("git",

    "bzrlib.plugins.git.mapping", "foreign_vcs_git", "Stupid content tracker")

plugin_cmds.register_lazy("cmd_git_import", [], "bzrlib.plugins.git.commands")

plugin_cmds.register_lazy("cmd_git_object", ["git-objects", "git-cat"],

    "bzrlib.plugins.git.commands")

plugin_cmds.register_lazy("cmd_git_refs", [], "bzrlib.plugins.git.commands")

plugin_cmds.register_lazy("cmd_git_apply", [], "bzrlib.plugins.git.commands")

def extract_git_foreign_revid(rev):

    try:

        foreign_revid = rev.foreign_revid

    except AttributeError:

        from bzrlib.plugins.git.mapping import mapping_registry

        foreign_revid, mapping = \

            mapping_registry.parse_revision_id(rev.revision_id)

        return foreign_revid

    else:

        from bzrlib.plugins.git.mapping import foreign_vcs_git

        if rev.mapping.vcs == foreign_vcs_git:

            return foreign_revid

        else:

            raise bzr_errors.InvalidRevisionId(rev.revision_id, None)

def update_stanza(rev, stanza):

    mapping = getattr(rev, "mapping", None)

    try:

        git_commit = extract_git_foreign_revid(rev)

    except bzr_errors.InvalidRevisionId:

        pass

    else:

        stanza.add("git-commit", git_commit)

try:

    from bzrlib.hooks import install_lazy_named_hook

except ImportError: # Compatibility with bzr < 2.4

    from bzrlib.version_info_formats.format_rio import (

        RioVersionInfoBuilder,

        )

    RioVersionInfoBuilder.hooks.install_named_hook('revision', update_stanza,

        "git commits")

else:

    install_lazy_named_hook("bzrlib.version_info_formats.format_rio",

        "RioVersionInfoBuilder.hooks", "revision", update_stanza,

        "git commits")

from bzrlib.transport import transport_server_registry

transport_server_registry.register_lazy('git',

    'bzrlib.plugins.git.server',

    'serve_git',

    'Git Smart server protocol over TCP. (default port: 9418)')

from bzrlib.repository import (

    format_registry as repository_format_registry,

    network_format_registry as repository_network_format_registry,

    )

repository_network_format_registry.register_lazy('git',

    'bzrlib.plugins.git.repository', 'GitRepositoryFormat')

try:

    register_extra_lazy_repository_format = getattr(repository_format_registry,

        "register_extra_lazy")

except AttributeError: # bzr < 2.4

    pass

else:

    register_extra_lazy_repository_format('bzrlib.plugins.git.repository',

        'GitRepositoryFormat')

from bzrlib.branch import (

    network_format_registry as branch_network_format_registry,

    )

branch_network_format_registry.register_lazy('git',

    'bzrlib.plugins.git.branch', 'GitBranchFormat')

try:

    from bzrlib.branch import (

        format_registry as branch_format_registry,

        )

except ImportError: # bzr < 2.4

    pass

else:

    branch_format_registry.register_extra_lazy(

        'bzrlib.plugins.git.branch',

        'GitBranchFormat',

        )

try:

    from bzrlib.workingtree import (

        format_registry as workingtree_format_registry,

        )

except ImportError: # bzr < 2.4

    pass

else:

    workingtree_format_registry.register_extra_lazy(

        'bzrlib.plugins.git.workingtree',

        'GitWorkingTreeFormat',

        )

controldir_network_format_registry.register_lazy('git',

    "bzrlib.plugins.git.dir", "GitControlDirFormat")

send_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',

                                   'send_git', 'Git am-style diff format')

topic_registry.register_lazy('git', 'bzrlib.plugins.git.help', 'help_git',

    'Using Bazaar with Git')

from bzrlib.diff import format_registry as diff_format_registry

diff_format_registry.register_lazy('git', 'bzrlib.plugins.git.send',

    'GitDiffTree', 'Git am-style diff format')

def update_git_cache(repository, revid):

    """Update the git cache after a local commit."""

    if getattr(repository, "_git", None) is not None:

        return # No need to update cache for git repositories

    from bzrlib.plugins.git.object_store import BazaarObjectStore

    if not repository.control_transport.has("git"):

        return # No existing cache, don't bother updating

    store = BazaarObjectStore(repository)

    store.lock_write()

    try:

        parent_revisions = set(repository.get_parent_map([revid])[revid])

        missing_revisions = store._missing_revisions(parent_revisions)

        if not missing_revisions:

            # Only update if the cache was up to date previously

            store._update_sha_map_revision(revid)

    finally:

        store.unlock()

def post_commit_update_cache(local_branch, master_branch, old_revno, old_revid,

        new_revno, new_revid):

    if local_branch is not None:

        update_git_cache(local_branch.repository, new_revid)

    update_git_cache(master_branch.repository, new_revid)

try:

    from bzrlib.hooks import install_lazy_named_hook

except ImportError: # Compatibility with bzr < 2.4

    pass

else:

    install_lazy_named_hook("bzrlib.branch",

        "Branch.hooks", "post_commit", post_commit_update_cache,

        "git cache")

def test_suite():

    from bzrlib.plugins.git import tests

    return tests.test_suite()