From 5056b223b2494b8a3a21b15991e3dd4ea25ec2a6 Mon Sep 17 00:00:00 2001
From: str4d <str4d@mail.i2p>
Date: Tue, 7 Jan 2014 22:21:40 +0000
Subject: [PATCH] Fixed language codes in sitemap URLs

---
 i2p2www/pages/global/sitemap.xml |  2 +-
 i2p2www/sitemap.py               | 19 ++++++++++++++++---
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/i2p2www/pages/global/sitemap.xml b/i2p2www/pages/global/sitemap.xml
index 020461fb..22d4daf0 100644
--- a/i2p2www/pages/global/sitemap.xml
+++ b/i2p2www/pages/global/sitemap.xml
@@ -3,7 +3,7 @@
   xmlns:xhtml="http://www.w3.org/1999/xhtml">
 {%- for url in urls %}
   <url>
-    <loc>{{ url_root }}{{ g.lang }}{{ url.path }}</loc>
+    <loc>{{ url_root }}{{ curlang }}{{ url.path }}</loc>
     {%- for altlang in langs %}
     <xhtml:link
       rel="alternate"
diff --git a/i2p2www/sitemap.py b/i2p2www/sitemap.py
index 98e44b3f..e67865da 100644
--- a/i2p2www/sitemap.py
+++ b/i2p2www/sitemap.py
@@ -1,4 +1,4 @@
-from flask import make_response, render_template, request, safe_join
+from flask import g, make_response, render_template, request, safe_join
 import os.path
 
 from i2p2www import SITE_DIR, SUPPORTED_LANGS, cache
@@ -6,6 +6,18 @@ from i2p2www.blog.helpers import get_blog_slugs
 from i2p2www.meetings.helpers import get_meetings_ids
 
 
+def to_url(value):
+    parts = value.split('_')
+    if len(parts) == 2:
+        return parts[0] + '-' + parts[1].lower()
+    return value
+
+
+LANG_FRAGS = []
+for lang in SUPPORTED_LANGS:
+    LANG_FRAGS.append(to_url(lang))
+
+
 ##########
 # Sitemaps
 
@@ -20,7 +32,7 @@ def render_sitemap_index():
     url_root = request.url_root
 
     # Render and return the sitemap index
-    response = make_response(render_template('global/sitemap_index.xml', url_root=url_root, langs=SUPPORTED_LANGS))
+    response = make_response(render_template('global/sitemap_index.xml', url_root=url_root, langs=LANG_FRAGS))
     response.headers['Content-Type'] = 'application/xml'
     return response
 
@@ -99,6 +111,7 @@ def render_sitemap():
         })
 
     # Render and return the sitemap
-    response = make_response(render_template('global/sitemap.xml', url_root=url_root, langs=SUPPORTED_LANGS, urls=urls))
+    response = make_response(render_template('global/sitemap.xml', url_root=url_root, langs=LANG_FRAGS,
+                                             curlang=to_url(g.lang), urls=urls))
     response.headers['Content-Type'] = 'application/xml'
     return response