zooniverse/Talk-Api

require 'honeybadger'

module ActionRescuing
  extend ActiveSupport::Concern

  included do
    rescue_from ActiveRecord::RecordNotFound, with: :not_found
    rescue_from ActiveRecord::RecordInvalid, with: :bad_request
    rescue_from ActiveRecord::RecordNotUnique, with: :conflict
    rescue_from ActiveRecord::UnknownAttributeError, with: :unprocessable
    rescue_from ActiveRecord::RecordNotDestroyed, with: :unauthorized

    rescue_from ActionController::UnpermittedParameters, with: :unprocessable
    rescue_from ActionController::RoutingError, with: :not_found
    rescue_from ActionController::ParameterMissing, with: :unprocessable

    rescue_from JSON::Schema::ValidationError, with: :unprocessable

    rescue_from Pundit::AuthorizationNotPerformedError, with: :not_implemented
    rescue_from Pundit::PolicyScopingNotPerformedError, with: :not_implemented
    rescue_from Pundit::NotDefinedError, with: :not_implemented
    rescue_from Pundit::NotAuthorizedError, with: :unauthorized

    rescue_from RangeError, with: :bad_request
    rescue_from RestPack::Serializer::InvalidInclude, with: :bad_request

    rescue_from TalkService::ParameterError, with: :unprocessable
    rescue_from Talk::InvalidParameterError, with: :unprocessable
    rescue_from Talk::BannedUserError, with: :forbidden
    rescue_from Talk::BlockedUserError, with: :forbidden
    rescue_from Talk::UserBlockedError, with: :forbidden
    rescue_from Talk::UserUnconfirmedError, with: :forbidden
    rescue_from OauthAccessToken::ExpiredError, with: :unauthorized
    rescue_from OauthAccessToken::RevokedError, with: :unauthorized
  end

  module ClassMethods
    def disallow(*methods)
      methods.each do |method|
        define_method method do
          not_allowed StandardError.new('Method Not Allowed')
        end
      end
    end
  end

  def require_user!
    raise Pundit::NotAuthorizedError.new('not logged in') unless current_user
  end

  def required_param(name)
    raise ActionController::ParameterMissing.new(name) unless params[name]
    params[name]
  end

  def unauthorized(exception)
    render_exception :unauthorized, exception
  end

  def forbidden(exception)
    render_exception :forbidden, exception
  end

  def not_found(exception)
    render_exception :not_found, exception
  end

  def not_allowed(exception)
    render_exception :method_not_allowed, exception
  end

  def bad_request(exception)
    render_exception :bad_request, exception
  end

  def unprocessable(exception)
    render_exception :unprocessable_entity, exception
  end

  def not_implemented(exception)
    render_exception :not_implemented, exception
  end

  def conflict(exception)
    render_exception :conflict, exception
  end

  def render_exception(status, exception)
    self.response_body = nil
    render status: status, json: { error: exception.message }
  end
end