Get proper record class for host controller

Author: Hilda Stastna

app/controllers/host_controller.rb

@@ -17,6 +17,17 @@ def self.display_methods
     ]
   end
 
+  def record_class
+    case params[:display] || @display
+    when 'storages'
+      Storage
+    when 'vms', 'miq_templates'
+      VmOrTemplate
+    else
+      Host
+    end
+  end
+
   def display_config_info
     @showtype = "config"
     title = case @display