# -*- coding: utf-8 -*- # Copyright (C) 2012-2017 RhodeCode GmbH # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License, version 3 # (only), as published by the Free Software Foundation. # # 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 Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. # # This program is dual-licensed. If you wish to learn more about the # RhodeCode Enterprise Edition, including its added features, Support services, # and proprietary license terms, please see https://rhodecode.com/licenses/ """ Utility to call a WSGI app wrapped in a WSGIAppCaller object. """ import logging log = logging.getLogger(__name__) def _get_clean_environ(environ): """Return a copy of the WSGI environment without wsgi.* keys. It also omits any non-string values. :param environ: WSGI environment to clean :type environ: dict :returns: WSGI environment to pass to WSGIAppCaller.handle. :rtype: dict """ clean_environ = dict( (k, v) for k, v in environ.iteritems() if type(v) == str and type(k) == str and not k.startswith('wsgi.') ) return clean_environ # pylint: disable=too-few-public-methods class RemoteAppCaller(object): """Create and calls a remote WSGI app using the given factory. It first cleans the environment, so as to reduce the data transferred. """ def __init__(self, remote_wsgi, *args, **kwargs): """ :param remote_wsgi: The remote wsgi object that creates a WSGIAppCaller. This object has to have a handle method, with the signature: handle(environ, start_response, *args, **kwargs) :param args: args to be passed to the app creation :param kwargs: kwargs to be passed to the app creation """ self._remote_wsgi = remote_wsgi self._args = args self._kwargs = kwargs def __call__(self, environ, start_response): """ :param environ: WSGI environment with which the app will be run :type environ: dict :param start_response: callable of WSGI protocol :type start_response: callable :returns: an iterable with the data returned by the app :rtype: iterable<str> """ log.debug("Forwarding WSGI request via proxy %s", self._remote_wsgi) input_data = environ['wsgi.input'].read() clean_environ = _get_clean_environ(environ) data, status, headers = self._remote_wsgi.handle( clean_environ, input_data, *self._args, **self._kwargs) log.debug("Got result from proxy, returning to WSGI container") start_response(status, headers) return data