← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-08-17 18:13:57 UTC
  • mfrom: (5268.7.29 transport-segments)
  • Revision ID: pqm@pqm.ubuntu.com-20110817181357-y5q5eth1hk8bl3om
(jelmer) Allow specifying the colocated branch to use in the branch URL,
 and retrieving the branch name using ControlDir._get_selected_branch.
 (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/http/wsgi.py
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""WSGI application for bzr HTTP smart server.
18
18
 
25
25
from bzrlib.smart import protocol, medium
26
26
from bzrlib.transport import chroot, get_transport
27
27
from bzrlib.urlutils import local_path_to_url
28
 
    
29
 
 
30
 
def make_app(root, prefix, path_var='REQUEST_URI', readonly=True):
 
28
 
 
29
 
 
30
def make_app(root, prefix, path_var='REQUEST_URI', readonly=True,
 
31
    load_plugins=True, enable_logging=True):
31
32
    """Convenience function to construct a WSGI bzr smart server.
32
 
    
 
33
 
33
34
    :param root: a local path that requests will be relative to.
34
35
    :param prefix: See RelpathSetter.
35
36
    :param path_var: See RelpathSetter.
39
40
        base_transport = get_transport('readonly+' + local_url)
40
41
    else:
41
42
        base_transport = get_transport(local_url)
 
43
    if load_plugins:
 
44
        from bzrlib.plugin import load_plugins
 
45
        load_plugins()
 
46
    if enable_logging:
 
47
        import bzrlib.trace
 
48
        bzrlib.trace.enable_default_logging()
42
49
    app = SmartWSGIApp(base_transport, prefix)
43
50
    app = RelpathSetter(app, '', path_var)
44
51
    return app
46
53
 
47
54
class RelpathSetter(object):
48
55
    """WSGI middleware to set 'bzrlib.relpath' in the environ.
49
 
    
 
56
 
50
57
    Different servers can invoke a SmartWSGIApp in different ways.  This
51
58
    middleware allows an adminstrator to configure how to the SmartWSGIApp will
52
59
    determine what path it should be serving for a given request for many common
58
65
    prefix="/some/prefix/" and path_var="REQUEST_URI" will set that request's
59
66
    'bzrlib.relpath' variable to "repo/branch".
60
67
    """
61
 
    
 
68
 
62
69
    def __init__(self, app, prefix='', path_var='REQUEST_URI'):
63
70
        """Constructor.
64
71
 
94
101
            backing_transport.  This is used to interpret relpaths received from
95
102
            the client.
96
103
        """
97
 
        # Use a ChrootTransportDecorator so that this web application won't
 
104
        # Use a ChrootServer so that this web application won't
98
105
        # accidentally let people access locations they shouldn't.
99
106
        # e.g. consider a smart server request for "get /etc/passwd" or
100
107
        # something.
101
108
        self.chroot_server = chroot.ChrootServer(backing_transport)
102
 
        self.chroot_server.setUp()
 
109
        self.chroot_server.start_server()
103
110
        self.backing_transport = get_transport(self.chroot_server.get_url())
104
111
        self.root_client_path = root_client_path
105
112
        # While the chroot server can technically be torn down at this point,
106
 
        # as all it does is remove the scheme registration from transport's 
107
 
        # protocol dictionary, we don't *just in case* there are parts of 
 
113
        # as all it does is remove the scheme registration from transport's
 
114
        # protocol dictionary, we don't *just in case* there are parts of
108
115
        # bzrlib that will invoke 'get_transport' on urls rather than cloning
109
116
        # around the existing transport.
110
 
        #self.chroot_server.tearDown()
 
117
        #self.chroot_server.stop_server()
111
118
 
112
119
    def __call__(self, environ, start_response):
113
120
        """WSGI application callable."""
169
176
    def make_request(self, transport, write_func, request_bytes, rcp):
170
177
        protocol_factory, unused_bytes = medium._get_protocol_factory_for_bytes(
171
178
            request_bytes)
172
 
        server_protocol = protocol_factory(transport, write_func, rcp)
 
179
        server_protocol = protocol_factory(
 
180
            transport, write_func, rcp, self.backing_transport)
173
181
        server_protocol.accept_bytes(unused_bytes)
174
182
        return server_protocol