/breezy/unstable

To get this branch, use:
bzr branch https://code.breezy-vcs.org/breezy/unstable

« back to all changes in this revision

Viewing changes to breezy/transport/http/wsgi.py

  • Committer: Jelmer Vernooij
  • Date: 2017-05-24 01:39:33 UTC
  • mfrom: (3815.3776.6)
  • Revision ID: jelmer@jelmer.uk-20170524013933-ir4y4tqtrsiz2ka2
New upstream snapshot.

Show diffs side-by-side

added added

removed removed

Lines of Context:
22
22
 
23
23
from __future__ import absolute_import
24
24
 
25
 
from cStringIO import StringIO
26
 
 
27
 
from bzrlib.smart import medium
28
 
from bzrlib.transport import chroot, get_transport
29
 
from bzrlib.urlutils import local_path_to_url
 
25
from ...sixish import (
 
26
    BytesIO,
 
27
    )
 
28
from ...smart import medium
 
29
from ...transport import chroot, get_transport
 
30
from ...urlutils import local_path_to_url
30
31
 
31
32
 
32
33
def make_app(root, prefix, path_var='REQUEST_URI', readonly=True,
43
44
    else:
44
45
        base_transport = get_transport(local_url)
45
46
    if load_plugins:
46
 
        from bzrlib.plugin import load_plugins
 
47
        from ...plugin import load_plugins
47
48
        load_plugins()
48
49
    if enable_logging:
49
 
        import bzrlib.trace
50
 
        bzrlib.trace.enable_default_logging()
 
50
        import breezy.trace
 
51
        breezy.trace.enable_default_logging()
51
52
    app = SmartWSGIApp(base_transport, prefix)
52
53
    app = RelpathSetter(app, '', path_var)
53
54
    return app
54
55
 
55
56
 
56
57
class RelpathSetter(object):
57
 
    """WSGI middleware to set 'bzrlib.relpath' in the environ.
 
58
    """WSGI middleware to set 'breezy.relpath' in the environ.
58
59
 
59
60
    Different servers can invoke a SmartWSGIApp in different ways.  This
60
61
    middleware allows an adminstrator to configure how to the SmartWSGIApp will
65
66
    a typical Apache and mod_fastcgi configuration will set `REQUEST_URI` to
66
67
    "/some/prefix/repo/branch/.bzr/smart".  A RelpathSetter with
67
68
    prefix="/some/prefix/" and path_var="REQUEST_URI" will set that request's
68
 
    'bzrlib.relpath' variable to "repo/branch".
 
69
    'breezy.relpath' variable to "repo/branch".
69
70
    """
70
71
 
71
72
    def __init__(self, app, prefix='', path_var='REQUEST_URI'):
73
74
 
74
75
        :param app: WSGI app to wrap, e.g. a SmartWSGIApp instance.
75
76
        :param path_var: the variable in the WSGI environ to calculate the
76
 
            'bzrlib.relpath' variable from.
 
77
            'breezy.relpath' variable from.
77
78
        :param prefix: a prefix to strip from the variable specified in
78
 
            path_var before setting 'bzrlib.relpath'.
 
79
            path_var before setting 'breezy.relpath'.
79
80
        """
80
81
        self.app = app
81
82
        self.prefix = prefix
87
88
        if not (path.startswith(self.prefix) and path.endswith(suffix)):
88
89
            start_response('404 Not Found', [])
89
90
            return []
90
 
        environ['bzrlib.relpath'] = path[len(self.prefix):-len(suffix)]
 
91
        environ['breezy.relpath'] = path[len(self.prefix):-len(suffix)]
91
92
        return self.app(environ, start_response)
92
93
 
93
94
 
114
115
        # While the chroot server can technically be torn down at this point,
115
116
        # as all it does is remove the scheme registration from transport's
116
117
        # protocol dictionary, we don't *just in case* there are parts of
117
 
        # bzrlib that will invoke 'get_transport' on urls rather than cloning
 
118
        # breezy that will invoke 'get_transport' on urls rather than cloning
118
119
        # around the existing transport.
119
120
        #self.chroot_server.stop_server()
120
121
 
124
125
            start_response('405 Method not allowed', [('Allow', 'POST')])
125
126
            return []
126
127
 
127
 
        relpath = environ['bzrlib.relpath']
 
128
        relpath = environ['breezy.relpath']
128
129
 
129
130
        if not relpath.startswith('/'):
130
131
            relpath = '/' + relpath
157
158
            raise AssertionError(adjusted_relpath)
158
159
 
159
160
        transport = self.backing_transport.clone(adjusted_relpath)
160
 
        out_buffer = StringIO()
 
161
        out_buffer = BytesIO()
161
162
        request_data_length = int(environ['CONTENT_LENGTH'])
162
163
        request_data_bytes = environ['wsgi.input'].read(request_data_length)
163
164
        smart_protocol_request = self.make_request(