From 2482d7150d0870cd31cde846c2c5060fe7e2ba94 Mon Sep 17 00:00:00 2001
From: Toby <toby.junk@gmail.com>
Date: Tue, 28 Feb 2012 15:02:51 +0000
Subject: [PATCH] [xs] fix the redirect language issues on logging in/out

---
 ckan/config/middleware.py | 23 ++++++++++++++++++++---
 ckan/lib/i18n.py          | 10 ++++++++--
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/ckan/config/middleware.py b/ckan/config/middleware.py
index f28a896f378..16613bf78e1 100644
--- a/ckan/config/middleware.py
+++ b/ckan/config/middleware.py
@@ -133,6 +133,17 @@ def __init__(self, app, config):
         self.default_locale = config.get('ckan.locale_default', 'en')
         self.local_list = get_locales()
 
+    def get_cookie_lang(self, environ):
+        # get the lang from cookie if present
+        cookie = environ.get('HTTP_COOKIE')
+        if cookie:
+            cookies = [c.strip() for c in cookie.split(';')]
+            lang = [c.split('=')[1] for c in cookies \
+                    if c.startswith('ckan_lang')][0]
+            if lang in self.local_list:
+                return lang
+        return None
+
     def __call__(self, environ, start_response):
         # strip the language selector from the requested url
         # and set environ variables for the language selected
@@ -153,9 +164,15 @@ def __call__(self, environ, start_response):
                 else:
                     environ['PATH_INFO'] = '/'
             else:
-                # use default language from config
-                environ['CKAN_LANG'] = self.default_locale
-                environ['CKAN_LANG_IS_DEFAULT'] = True
+                # use cookie lang or default language from config
+                cookie_lang = self.get_cookie_lang(environ)
+                if cookie_lang:
+                    environ['CKAN_LANG'] = cookie_lang
+                    environ['CKAN_LANG_IS_DEFAULT'] = False
+                else:
+                    environ['CKAN_LANG'] = self.default_locale
+                    environ['CKAN_LANG_IS_DEFAULT'] = True
+
 
             # Current application url
             path_info = environ['PATH_INFO']
diff --git a/ckan/lib/i18n.py b/ckan/lib/i18n.py
index 743ef880aad..e71d741f076 100644
--- a/ckan/lib/i18n.py
+++ b/ckan/lib/i18n.py
@@ -3,6 +3,7 @@
 from babel import Locale, localedata
 from babel.core import LOCALE_ALIASES
 from pylons import config
+from pylons import response
 from pylons import i18n
 
 import ckan.i18n
@@ -89,11 +90,16 @@ def get_available_locales():
 
 def handle_request(request, tmpl_context):
     ''' Set the language for the request '''
-    lang = request.environ.get('CKAN_LANG',
-                               config.get('ckan.locale_default', 'en'))
+    lang = request.environ.get('CKAN_LANG') or \
+                        config.get('ckan.locale_default', 'en')
     if lang != 'en':
         i18n.set_lang(lang)
     tmpl_context.language = lang
+
+    # set ckan_lang cookie if we have changed the language. We need to
+    # remember this because repoze.who does it's own redirect.
+    if request.cookies.get('ckan_lang') != lang:
+        response.set_cookie('ckan_lang', lang, max_age=3600)
     return lang
 
 def get_lang():