From e9cc719f3638b780b9386c86244995ac0ff357f0 Mon Sep 17 00:00:00 2001
From: Francesco Siddi <francesco.siddi@gmail.com>
Date: Wed, 4 Nov 2015 22:12:42 +0100
Subject: [PATCH] Ensure unique username on user creation

---
 pillar/application/__init__.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/pillar/application/__init__.py b/pillar/application/__init__.py
index d15bd9c2..4b549a17 100644
--- a/pillar/application/__init__.py
+++ b/pillar/application/__init__.py
@@ -86,7 +86,22 @@ def validate_token():
             users = app.data.driver.db['users']
             email = validation['data']['user']['email']
             db_user = users.find_one({'email': email})
+            # Ensure unique username
             username = email.split('@')[0]
+            def make_unique_username(username, index=1):
+                """Ensure uniqueness of a username by appending an incremental
+                digit at the end of it.
+                """
+                user_from_username = users.find_one({'username': username})
+                if user_from_username:
+                    if index > 1:
+                        index += 1
+                        username = username[:-1]
+                    username = "{0}{1}".format(username, index)
+                    return make_unique_username(username, index=index)
+                return username
+            username = make_unique_username(username)
+
             full_name = username
             if not db_user:
                 user_data = {