ruby-on-railsapicontrollerpunditrescue

How can I avoid double render in this situation?


I have an issue with this code in my controller:

class Api::V1::BaseController < ActionController::API
  include Pundit

  after_action :verify_authorized, except: :index
  after_action :verify_policy_scoped, only: :index

  rescue_from StandardError,                with: :internal_server_error
  rescue_from Pundit::NotAuthorizedError,   with: :user_not_authorized
  rescue_from ActiveRecord::RecordNotFound, with: :not_found

  private

  def user_not_authorized(exception)
    render json: {
      error: "Unauthorized #{exception.policy.class.to_s.underscore.camelize}.#{exception.query}"
    }, status: :unauthorized
  end

  def not_found(exception)
    render json: { error: exception.message }, status: :not_found
  end

  def internal_server_error(exception)
    if Rails.env.development?
      response = { type: exception.class.to_s, message: exception.message, backtrace: exception.backtrace }
    else
      response = { error: "Internal Server Error" }
    end
    render json: response, status: :internal_server_error
  end
end

The problem

The rescue_from StandardError is the root of all my troubles. This controller works nicely and rescues from Pundit whitelist checks when the pundit error is the only one happening.

But as soon as any other error happens alongside pundit, I get a DoubleRenderError since both rescues end up being triggered. I'm looking for a quick tweak that would avoid pundit from triggering when another error has already taken place, or an alternative solution for this problem.

Is there any other error class that I can use to avoid relying so heavily on StandardError?

Things I tried

Thank you very much!


Solution

  • You don't really need to have the rescue_from StandardError as this is the default behaviour of Rails. Rails has a middleware called PublicExceptions which does (mostly) what you want so you can just let the StandardError propagate.

    Instead of { error: "Internal Server Error" } it will render this

    { 
      status: status, 
      error: Rack::Utils::HTTP_STATUS_CODES.fetch(status, Rack::Utils::HTTP_STATUS_CODES[500]) 
    }
    

    which in case of an exception will render { status: 500, error: "Internal Server Error" }. This should be a reasonable compromise.

    For development you could think about adapting this middleware. You can set it with config.exceptions_app.

    https://guides.rubyonrails.org/configuring.html#rails-general-configuration

    https://api.rubyonrails.org/classes/ActionDispatch/PublicExceptions.html

    https://github.com/rails/rails/blob/master/actionpack/lib/action_dispatch/middleware/public_exceptions.rb#L14