comparison mercurial/httppeer.py @ 37609:01bfe5ad0c53

httppeer: implement ipeerconnection

This is low hanging fruit. We might as well start somewhere.

Differential Revision: https://phab.mercurial-scm.org/D3254

37608:fa4b39bb0a07

from .i18n import _
from .thirdparty import (
    cbor,
)
from . import (
    bundle2,
    error,
    httpconnection,
    pycompat,
    statichttprepo,
    url as urlmod,
    util,
    wireproto,
    wireprotoframing,

    def _abort(self, exception):
        raise exception

# TODO implement interface for version 2 peers
class httpv2peer(object):
    def __init__(self, ui, repourl, apipath, opener, requestbuilder,
                 apidescriptor):
        self.ui = ui

        if repourl.endswith('/'):
            repourl = repourl[:-1]

        self.url = repourl
        self._apipath = apipath
        self._opener = opener
        self._requestbuilder = requestbuilder
        self._descriptor = apidescriptor

    def close(self):
        pass

    # TODO require to be part of a batched primitive, use futures.
    def _call(self, name, **args):
        """Call a wire protocol command with arguments."""

        permission = {
            'push': 'rw',
            'pull': 'ro',
        }[permission]

        url = '%s/%s/%s/%s' % (self.url, self._apipath, permission, name)

        # TODO this should be part of a generic peer for the frame-based
        # protocol.
        reactor = wireprotoframing.clientreactor(hasmultiplesend=False,
                                                 buffersends=True)

37609:01bfe5ad0c53

from .i18n import _
from .thirdparty import (
    cbor,
)
from .thirdparty.zope import (
    interface as zi,
)
from . import (
    bundle2,
    error,
    httpconnection,
    pycompat,
    repository,
    statichttprepo,
    url as urlmod,
    util,
    wireproto,
    wireprotoframing,

    def _abort(self, exception):
        raise exception

# TODO implement interface for version 2 peers
@zi.implementer(repository.ipeerconnection)
class httpv2peer(object):
    def __init__(self, ui, repourl, apipath, opener, requestbuilder,
                 apidescriptor):
        self.ui = ui

        if repourl.endswith('/'):
            repourl = repourl[:-1]

        self._url = repourl
        self._apipath = apipath
        self._opener = opener
        self._requestbuilder = requestbuilder
        self._descriptor = apidescriptor

    # Start of ipeerconnection.

    def url(self):
        return self._url

    def local(self):
        return None

    def peer(self):
        return self

    def canpush(self):
        # TODO change once implemented.
        return False

    def close(self):
        pass

    # End of ipeerconnection.

    # TODO require to be part of a batched primitive, use futures.
    def _call(self, name, **args):
        """Call a wire protocol command with arguments."""

        permission = {
            'push': 'rw',
            'pull': 'ro',
        }[permission]

        url = '%s/%s/%s/%s' % (self._url, self._apipath, permission, name)

        # TODO this should be part of a generic peer for the frame-based
        # protocol.
        reactor = wireprotoframing.clientreactor(hasmultiplesend=False,
                                                 buffersends=True)