theforeman/foreman

# This mixin is used by both ApplicationController and Api::BaseController
# Searches for an object based on its id, name, label, etc and assign it to an instance variable
# friendly_id performs the logic if params[:id] is 'id' or 'id-name' or 'name'

module FindCommon
  def find_resource
    instance_variable_set("@#{resource_name}",
      resource_finder(resource_scope, params[:id]))
  end

  def resource_finder(scope, id)
    raise ActiveRecord::RecordNotFound if scope.empty?
    result = scope.from_param(id) if scope.respond_to?(:from_param)
    begin
      result ||= scope.friendly.find(id) if scope.respond_to?(:friendly)
    rescue ActiveRecord::RecordNotFound
    end
    result || scope.find(id)
  end

  def controller_permission
    controller_name
  end

  def resource_name(resource = controller_name)
    resource.singularize
  end

  def resource_class
    @resource_class ||= resource_class_for(resource_name)
    raise NameError, "Could not find resource class for resource #{resource_name}" if @resource_class.nil?
    @resource_class
  end

  def resource_scope(*args, **kwargs)
    @resource_scope ||= scope_for(resource_class, *args, **kwargs)
  end

  def scope_for(resource, *args, **kwargs)
    controller = kwargs.delete(:controller) { controller_permission }
    # don't call the #action_permission method here, we are not sure if the resource is authorized at this point
    # calling #action_permission here can cause an exception, in order to avoid this, ensure :authorized beforehand
    permission = kwargs.delete(:permission)

    if resource.respond_to?(:authorized)
      permission ||= "#{action_permission}_#{controller}"
      resource = resource.authorized(permission, resource)
    end

    # Callers rely on a plain array
    if args.empty? && kwargs.empty?
      resource.all
    else
      resource.where(*args, **kwargs)
    end
  end

  def resource_class_for(resource)
    klass = resource.classify.constantize
    return Host::Managed if klass == Host
    klass
  rescue NameError
    nil
  end
end