"""
HTTP middleware for the Deis REST API.

See https://docs.djangoproject.com/en/1.10/topics/http/middleware/
"""

from api import __version__


class APIVersionMiddleware(object):
    """
    Include that REST API version with each response.
    """
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        """
        Include the controller's REST API major and minor version in
        a response header.
        """
        response = self.get_response(request)
        # clients shouldn't care about the patch release
        version = __version__.rsplit('.', 1)[0]
        response['DEIS_API_VERSION'] = version
        response['DEIS_PLATFORM_VERSION'] = __version__
        return response