forked from treehouse/mastodon
98 lines
2.4 KiB
Ruby
98 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class Auth::ConfirmationsController < Devise::ConfirmationsController
|
|
include CaptchaConcern
|
|
|
|
layout 'auth'
|
|
|
|
before_action :set_body_classes
|
|
before_action :set_pack
|
|
before_action :set_confirmation_user!, only: [:show, :confirm_captcha]
|
|
before_action :require_unconfirmed!
|
|
|
|
before_action :extend_csp_for_captcha!, only: [:show, :confirm_captcha]
|
|
before_action :require_captcha_if_needed!, only: [:show]
|
|
|
|
skip_before_action :require_functional!
|
|
|
|
def new
|
|
super
|
|
|
|
resource.email = current_user.unconfirmed_email || current_user.email if user_signed_in?
|
|
end
|
|
|
|
def show
|
|
old_session_values = session.to_hash
|
|
reset_session
|
|
session.update old_session_values.except('session_id')
|
|
|
|
super
|
|
end
|
|
|
|
def confirm_captcha
|
|
check_captcha! do |message|
|
|
flash.now[:alert] = message
|
|
render :captcha
|
|
return
|
|
end
|
|
|
|
show
|
|
end
|
|
|
|
private
|
|
|
|
def require_captcha_if_needed!
|
|
render :captcha if captcha_required?
|
|
end
|
|
|
|
def set_confirmation_user!
|
|
# We need to reimplement looking up the user because
|
|
# Devise::ConfirmationsController#show looks up and confirms in one
|
|
# step.
|
|
confirmation_token = params[:confirmation_token]
|
|
return if confirmation_token.nil?
|
|
@confirmation_user = User.find_first_by_auth_conditions(confirmation_token: confirmation_token)
|
|
end
|
|
|
|
def captcha_user_bypass?
|
|
return true if @confirmation_user.nil? || @confirmation_user.confirmed?
|
|
|
|
invite = Invite.find(@confirmation_user.invite_id) if @confirmation_user.invite_id.present?
|
|
invite.present? && !invite.max_uses.nil?
|
|
end
|
|
|
|
def set_pack
|
|
use_pack 'auth'
|
|
end
|
|
|
|
def require_unconfirmed!
|
|
if user_signed_in? && current_user.confirmed? && current_user.unconfirmed_email.blank?
|
|
redirect_to(current_user.approved? ? root_path : edit_user_registration_path)
|
|
end
|
|
end
|
|
|
|
def set_body_classes
|
|
@body_classes = 'lighter'
|
|
end
|
|
|
|
def after_resending_confirmation_instructions_path_for(_resource_name)
|
|
if user_signed_in?
|
|
if current_user.confirmed? && current_user.approved?
|
|
edit_user_registration_path
|
|
else
|
|
auth_setup_path
|
|
end
|
|
else
|
|
new_user_session_path
|
|
end
|
|
end
|
|
|
|
def after_confirmation_path_for(_resource_name, user)
|
|
if user.created_by_application && truthy_param?(:redirect_to_app)
|
|
user.created_by_application.redirect_uri
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|