registrations_controller.rb 2.6 KB
Newer Older
M
Marin Jankovski 已提交
1
class RegistrationsController < Devise::RegistrationsController
2
  include Recaptcha::Verify
M
Marin Jankovski 已提交
3

4 5
  before_action :whitelist_query_limiting, only: [:destroy]

6 7 8 9
  def new
    redirect_to(new_user_session_path)
  end

10
  def create
11 12 13 14 15 16
    # To avoid duplicate form fields on the login page, the registration form
    # names fields using `new_user`, but Devise still wants the params in
    # `user`.
    if params["new_#{resource_name}"].present? && params[resource_name].blank?
      params[resource_name] = params.delete(:"new_#{resource_name}")
    end
17

18
    if !Gitlab::Recaptcha.load_configurations! || verify_recaptcha
19 20
      super
    else
21
      flash[:alert] = 'There was an error with the reCAPTCHA. Please solve the reCAPTCHA again.'
22 23 24
      flash.delete :recaptcha_error
      render action: 'new'
    end
25 26
  rescue Gitlab::Access::AccessDeniedError
    redirect_to(new_user_session_path)
27 28
  end

M
Marin Jankovski 已提交
29
  def destroy
30 31 32 33 34 35
    if destroy_confirmation_valid?
      current_user.delete_async(deleted_by: current_user)
      session.try(:destroy)
      redirect_to new_user_session_path, status: 303, notice: s_('Profiles|Account scheduled for removal.')
    else
      redirect_to profile_account_path, status: 303, alert: destroy_confirmation_failure_message
M
Marin Jankovski 已提交
36 37 38
    end
  end

39 40
  protected

41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
  def destroy_confirmation_valid?
    if current_user.confirm_deletion_with_password?
      current_user.valid_password?(params[:password])
    else
      current_user.username == params[:username]
    end
  end

  def destroy_confirmation_failure_message
    if current_user.confirm_deletion_with_password?
      s_('Profiles|Invalid password')
    else
      s_('Profiles|Invalid username')
    end
  end

57
  def build_resource(hash = nil)
58 59 60
    super
  end

61
  def after_sign_up_path_for(user)
62
    Gitlab::AppLogger.info("User Created: username=#{user.username} email=#{user.email} ip=#{request.remote_ip} confirmed:#{user.confirmed?}")
63
    user.confirmed? ? dashboard_projects_path : users_almost_there_path
64 65
  end

66 67
  def after_inactive_sign_up_path_for(resource)
    Gitlab::AppLogger.info("User Created: username=#{resource.username} email=#{resource.email} ip=#{request.remote_ip} confirmed:false")
P
Phil Hughes 已提交
68
    users_almost_there_path
69 70
  end

M
Marin Jankovski 已提交
71 72
  private

73
  def sign_up_params
74
    params.require(:user).permit(:username, :email, :email_confirmation, :name, :password)
75
  end
76 77 78 79 80 81

  def resource_name
    :user
  end

  def resource
G
geoandri 已提交
82
    @resource ||= Users::BuildService.new(current_user, sign_up_params).execute
83 84 85 86 87
  end

  def devise_mapping
    @devise_mapping ||= Devise.mappings[:user]
  end
88 89 90 91

  def whitelist_query_limiting
    Gitlab::QueryLimiting.whitelist('https://gitlab.com/gitlab-org/gitlab-ce/issues/42380')
  end
92
end