import django
from django.http import JsonResponse
from django.middleware.common import CommonMiddleware
from django.utils.translation import ugettext_lazy as _


class HoboCommonMiddleware(CommonMiddleware):
    def __call__(self, *args, **kwargs):
        self.will_redirect_with_slash = False
        return super().__call__(*args, **kwargs)

    def should_redirect_with_slash(self, request):
        self.will_redirect_with_slash = super().should_redirect_with_slash(request)
        return self.will_redirect_with_slash

    def forbid_redirect(self, request, response):
        return bool(
            self.will_redirect_with_slash
            and isinstance(response, self.response_redirect_class)
            and request.path_info.startswith('/api/')
            and request.method in ("POST", "PUT", "PATCH")
        )

    def process_request(self, request):
        response = super().process_request(request)
        if self.forbid_redirect(request, response):
            return JsonResponse({'err': 1, 'err_desc': _('URL must end with a slash.')}, status=404)
        return response

    def process_response(self, request, response):
        response = super().process_response(request, response)
        if self.forbid_redirect(request, response):
            return JsonResponse({'err': 1, 'err_desc': _('URL must end with a slash.')}, status=404)
        return response