# -*- coding: utf-8 -*-

import re
from django.http import Http404, HttpResponseRedirect
from modoboa.admin.models import Extension

class ExtControlMiddleware(object):
    def process_view(self, request, view, args, kwargs):
        m = re.match("modoboa\.extensions\.(\w+)", view.__module__)
        if m is None:
            return None
        ext = Extension.objects.get(name=m.group(1))
        if ext is None:
            return None
        if ext.enabled:
            return None
        raise Http404

class AjaxLoginRedirect(object):
    def process_response(self, request, response):
        if request.is_ajax():
            if type(response) == HttpResponseRedirect:
                response.status_code = 278
        return response