From 10cc082560a040288654ea1731024489c5d4eeeb Mon Sep 17 00:00:00 2001
From: Andrei Prigorshnev <a.prigorshnev@gmail.com>
Date: Mon, 6 Dec 2021 12:06:35 +0100
Subject: [PATCH] FIX: when using external auth disallowed characters weren't
 removed from username (#15185)

---
 lib/auth/result.rb                            | 28 +++++---
 .../omniauth_callbacks_controller_spec.rb     | 66 +++++++++++++++++++
 2 files changed, 84 insertions(+), 10 deletions(-)

diff --git a/lib/auth/result.rb b/lib/auth/result.rb
index a5732810d15..2b0ba30c2f8 100644
--- a/lib/auth/result.rb
+++ b/lib/auth/result.rb
@@ -135,18 +135,9 @@ class Auth::Result
       return result
     end
 
-    suggested_username = UserNameSuggester.suggest(username_suggester_attributes)
-    if email_valid && email.present?
-      if username.present? && User.username_available?(username, email)
-        suggested_username = username
-      elsif staged_user = User.where(staged: true).find_by_email(email)
-        suggested_username = staged_user.username
-      end
-    end
-
     result = {
       email: email,
-      username: suggested_username,
+      username: resolve_username,
       auth_provider: authenticator_name,
       email_valid: !!email_valid,
       can_edit_username: can_edit_username,
@@ -165,7 +156,24 @@ class Auth::Result
 
   private
 
+  def staged_user
+    return @staged_user if defined?(@staged_user)
+    if email.present? && email_valid
+      @staged_user = User.where(staged: true).find_by_email(email)
+    end
+  end
+
   def username_suggester_attributes
     username || name || email
   end
+
+  def resolve_username
+    if staged_user
+      if !username.present? || UserNameSuggester.fix_username(username) == staged_user.username
+        return staged_user.username
+      end
+    end
+
+    UserNameSuggester.suggest(username_suggester_attributes)
+  end
 end
diff --git a/spec/requests/omniauth_callbacks_controller_spec.rb b/spec/requests/omniauth_callbacks_controller_spec.rb
index 5e774196c3a..47754a28a80 100644
--- a/spec/requests/omniauth_callbacks_controller_spec.rb
+++ b/spec/requests/omniauth_callbacks_controller_spec.rb
@@ -729,6 +729,18 @@ RSpec.describe Users::OmniauthCallbacksController do
 
         expect(cookies['authentication_data']).to be_nil
       end
+
+      it "removes disallowed characters from username" do
+        username = "strange_name*&^"
+        fixed_username = "strange_name"
+
+        mock_auth("user.with.strange.username@gmail.com", username)
+
+        get "/auth/google_oauth2/callback.json"
+        data = JSON.parse(cookies[:authentication_data])
+
+        expect(data["username"]).to eq(fixed_username)
+      end
     end
 
     context 'when attempting reconnect' do
@@ -881,5 +893,59 @@ RSpec.describe Users::OmniauthCallbacksController do
         expect(response['email']).to eq(new_email)
       end
     end
+
+    context "when user is staged" do
+      fab!(:staged_user) { Fabricate(
+          :user,
+          username: "staged_user",
+          email: "staged.user@gmail.com",
+          staged: true
+        )
+      }
+
+      it "should use username of the staged user if username is not present in payload" do
+        mock_auth(staged_user.email, nil)
+
+        get "/auth/google_oauth2/callback.json"
+        data = JSON.parse(cookies[:authentication_data])
+
+        expect(data["username"]).to eq(staged_user.username)
+      end
+
+      it "should use username of the staged user if username in payload is the same" do
+        mock_auth(staged_user.email, staged_user.username)
+
+        get "/auth/google_oauth2/callback.json"
+        data = JSON.parse(cookies[:authentication_data])
+
+        expect(data["username"]).to eq(staged_user.username)
+      end
+
+      it "should override username of the staged user if payload contains a new username" do
+        new_username = "new_username"
+        mock_auth(staged_user.email, new_username)
+
+        get "/auth/google_oauth2/callback.json"
+        data = JSON.parse(cookies[:authentication_data])
+
+        expect(data["username"]).to eq(new_username)
+      end
+    end
+
+    def mock_auth(email, nickname)
+      OmniAuth.config.mock_auth[:google_oauth2] = OmniAuth::AuthHash.new(
+        provider: 'google_oauth2',
+        uid: '123545',
+        info: OmniAuth::AuthHash::InfoHash.new(
+          email: email,
+          nickname: nickname
+        ),
+        extra: {
+          raw_info: OmniAuth::AuthHash.new(email_verified: true)
+        }
+      )
+
+      Rails.application.env_config["omniauth.auth"] = OmniAuth.config.mock_auth[:google_oauth2]
+    end
   end
 end