ManageIQ/manageiq-ui-classic

View on GitHub
app/controllers/mixins/checked_id_mixin.rb

Summary

Maintainability
A
45 mins
Test Coverage
A
100%
module Mixins
  module CheckedIdMixin
    def checked_item_id(hash = params)
      return hash[:id] if hash[:id]

      checked_items = find_checked_items
      checked_items[0] if checked_items.length == 1
    end

    # Common routine to find checked items on a page (checkbox ids are
    # "check_xxx" where xxx is the item id or index)
    def find_checked_items(prefix = nil)
      if params[:miq_grid_checks].present?
        params[:miq_grid_checks].split(",").collect(&:to_i)
      else
        prefix = "check" if prefix.nil?

        items = []
        params.each do |var, val|
          vars = var.to_s.split("_")
          if vars[0] == prefix && val == "1"
            ids = vars[1..-1]
            items << ids.join("_")
          end
        end
        items
      end
    end

    # Find a record by model and id and test it with RBAC
    #
    # Wrapper for find_records_with_rbac method for case when only a single
    # record is required
    #
    # Params:
    #   klass   - class of accessed object
    #   id      - accessed object id
    #   options - additional options
    #
    # Returns:
    #   Instance of selected item
    #
    def find_record_with_rbac(klass, id, options = {})
      find_records_with_rbac(klass, Array.wrap(id), options).first
    end

    # Find records by model and id and test it with RBAC
    # Params:
    #   klass   - class or scope of accessed objects
    #   ids     - accessed object ids
    #   options - :named_scope :
    #
    # Returns:
    #   Array of selected class instances. If user does not have rights for it,
    #   either sets flash or raises exception
    #
    def find_records_with_rbac(klass, ids, options = {})
      raise(ActiveRecord::RecordNotFound, _("Can't access records without an id")) if ids.include?(nil) || ids.empty?

      filtered = Rbac.filtered(klass.where(:id => ids), :named_scope => options[:named_scope])
      raise(ActiveRecord::RecordNotFound, _("Can't access selected records")) unless ids.length == filtered.length

      filtered
    end

    # Tries to load checked items from params.
    # If there are none, takes the id sent in params[:id]
    #
    # Returns:
    #   Array of ids of the items as a Fixnum
    #
    def checked_or_params
      (checked = find_checked_items).blank? && params[:id].present? ? Array(params[:id]) : checked
    end
  end
end