from django.utils.cache import patch_vary_headers
from django.utils import translation

class LocaleURLMiddleware:
    def get_language_from_request (self,request):
        from django.conf import settings
        import re
        supported = dict(settings.LANGUAGES)
        lang = settings.LANGUAGE_CODE[:2]
        check = re.match(r".*/intl/(\w\w)/.*", request.path)
        changed = False
        if check is not None:
            t = check.group(1)
            if t in supported:
                lang = t
                changed = True
        if not changed:
            if hasattr(request, 'session'):
                lang = request.session.get('django_language', None)
                if lang in supported and lang is not None:
                    return lang
            else:
                lang = request.COOKIES.get('django_language', None)
                if lang in supported and lang is not None:
                    return lang
            if settings.ENABLE_GEOIP:
                import GeoIP
                gi = GeoIP.new(GeoIP.GEOIP_MEMORY_CACHE)
                cc = gi.country_code_by_addr(request.META['REMOTE_ADDR'])
                if settings.CC2LANG.has_key(cc):
                    return settings.CC2LANG[cc]
        return lang
    def process_request(self, request):
        from django.conf import settings
        language = self.get_language_from_request(request)
        if language is None:
            language = settings.LANGUAGE_CODE[:2]
        translation.activate(language)
#        assert False, translation.get_language()+translation.gettext('Hello world')
        request.LANGUAGE_CODE = str(translation.get_language())
    def process_response(self, request, response):
        from django.conf import settings
        patch_vary_headers(response, ('Accept-Language',))
        language = self.get_language_from_request(request)
        if language is None:
            language = settings.LANGUAGE_CODE[:2]
#        if language == settings.LANGUAGE_CODE[:2]:
#            response.content = response.content.replace('/intl/xx','')
#        else:
#            response.content = response.content.replace('/intl/xx','/intl/'+language)
        translation.deactivate()
        return response

def get_absolute_path_without_lang(request):
    for lang in settings.LANGUAGES:
        if '/' + str(lang[0]) + '/' in request.path:
            return request.path.replace('/' + str(lang[0]) + '/', '/')
    return request.path