module ApplicationController::PreventsCsrf
  extend ActiveSupport::Concern

  included do
    # disable Rails default (>= 5.2) CSRF checks
    skip_before_action :verify_authenticity_token, raise: false

    before_action :verify_csrf_token
    after_action  :set_csrf_token_headers
  end

  private

  def set_csrf_token_headers
    return true if @_auth_type.present? && @_auth_type != 'session'

    headers['CSRF-TOKEN'] = form_authenticity_token
  end

  def verify_csrf_token
    return true if !protect_against_forgery?
    return true if request.get?
    return true if request.head?
    return true if %w[token_auth basic_auth].include?(@_auth_type)
    return true if valid_authenticity_token?(session, params[:authenticity_token] || request.headers['X-CSRF-Token'])

    logger.info 'CSRF token verification failed'
    raise Exceptions::NotAuthorized, 'CSRF token verification failed!'
  end
end