ManageIQ/manageiq-ui-classic

class TreeBuilderProvisioningDialogs < TreeBuilderAeCustomization
  private

  def tree_init_options
    {:open_all => true}
  end

  def root_options
    {
      :text    => t = _("All Dialogs"),
      :tooltip => t
    }
  end

  # Get root nodes count/array for explorer tree
  def x_get_tree_roots
    objects = MiqDialog::DIALOG_TYPES.map { |label, model| [_(label), model] }.sort.collect do |typ|
      {
        :id   => "MiqDialog_#{typ[1]}",
        :text => typ[0],
        :icon => "pficon pficon-folder-close",
        :tip  => typ[0]
      }
    end
    count_only_or_objects(false, objects)
  end

  def x_get_tree_custom_kids(object, count_only)
    objects = MiqDialog.where(:dialog_type => object[:id].split('_').last).sort_by { |a| a.description.downcase }
    count_only_or_objects(count_only, objects)
  end
end