Adds some progress on user registration.

1 files changed, 16 insertions, 3 deletions

diff --git a/src/login/lgn_handler.erl b/src/login/lgn_handler.erl
index dc154b0..fe87bc6 100644
--- a/src/login/lgn_handler.erl
+++ b/src/login/lgn_handler.erl
@@ -17,7 +17,7 @@ ensure_player_exists (ID, Username, Password, Email) ->
    case shr_database:fetch(player_db, ID, admin) of
       {ok, _} -> ok;
       not_found ->
-         shr_database:insert
+         shr_database:insert_at
          (
             player_db,
             ID,
@@ -27,10 +27,10 @@ ensure_player_exists (ID, Username, Password, Email) ->
          )
    end,
 
-   case shr_database:fetch(login_db, ID, admin) of
+   case shr_database:fetch(login_db, Username, admin) of
       {ok, _} -> ok;
       not_found ->
-         shr_database:insert
+         shr_database:insert_at
          (
             login_db,
             Username,
@@ -40,6 +40,19 @@ ensure_player_exists (ID, Username, Password, Email) ->
          )
    end,
 
+   case shr_database:fetch(login_db, Email, admin) of
+      {ok, _} -> ok;
+      not_found ->
+         shr_database:insert_at
+         (
+            login_db,
+            Email,
+            any,
+            any,
+            ID
+         )
+   end,
+
    ok.
 
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%