foreman_wreckingball 3.0.1 → 3.4.1

data/app/views/foreman_wreckingball/hosts/_status_row_actions.html.erb

@@ -0,0 +1,22 @@
+<% actions = status_actions(host_association, owned_only, supports_remediate) %>
+
+<% if actions.any? %>
+  <div class='list-view-pf-actions'>
+    <div class='dropdown pull-right dropdown-kebab-pf'>
+      <%= content_tag :button, class: 'btn btn-link dropdown-toggle',
+                               type: 'button',
+                               id: "dropdownKebabRight#{id}",
+                               'data-toggle': 'dropdown',
+                               'aria-haspopup': true,
+                               'aria-expanded': true do %>
+        <span class='fa fa-ellipsis-v' />
+      <% end %>
+
+      <%= content_tag :ul, class: 'dropdown-menu dropdown-menu-right', 'aria-labelledby': "dropdownKebabRight#{id}" do %>
+        <% actions.each do |action| %>
+          <%= content_tag :li, action %>
+        <% end %>
+      <% end %>
+    </div>
+  </div>
+<% end %>

data/app/views/foreman_wreckingball/hosts/_status_row_hosts_table.html.erb

@@ -0,0 +1,24 @@
+<%= render :partial => 'status_row_hosts_table_actions', locals: { supports_remediate: supports_remediate } %>
+
+<%= content_tag :table, id: status,
+                        class: 'table table-striped table-fixed status-hosts',
+                        'data-hosts-url': ajax_status_dashboard_hosts_path(status, owned_only: params[:owned_only]) do %>
+  <%= content_tag :thead do %>
+    <%= content_tag :tr do %>
+      <%= content_tag :th do %>
+        <%= check_box_tag 'select-all' %>
+      <% end %>
+      <%= content_tag :th, _('Hostname') %>
+      <%= content_tag :th, _('Status') %>
+      <%= content_tag :th, _('Owner') %>
+      <%= content_tag :th, _('Puppet Environment') %>
+      <%= content_tag :th, _('Actions') %>
+    <% end %>
+  <% end %>
+  <%= content_tag(:tbody) {} %>
+<% end %>
+
+<%= render :partial => 'status_row_hosts_table_actions', locals: { supports_remediate: supports_remediate } %>
+
+<%= alert header: _("Oops, we're sorry but something went wrong"), text: '',
+          class: 'alert-danger', close: false %>

data/app/views/foreman_wreckingball/hosts/_status_row_hosts_table_actions.html.erb

@@ -0,0 +1,9 @@
+<%= content_tag :div, class: 'text-right' do %>
+  <% if supports_remediate %>
+    <%= content_tag :a, _('Remediate selected'),
+                        href: schedule_remediate_hosts_path,
+                        class: 'btn btn-default remediate-selected',
+                        onclick: 'show_modal(this); return false;',
+                        disabled: true %>
+  <% end %>
+<% end %>

data/app/views/foreman_wreckingball/hosts/schedule_remediate.html.erb

@@ -1,35 +1,53 @@
-<% title _('Remediate %s') % @host %>
+<% title _('Remediate') %>
 <% javascript 'foreman_tasks/trigger_form' %>
 <% stylesheet 'foreman_tasks/trigger_form' %>
 
+<% if @statuses.any? %>
+  <% if @statuses.group_by(&:class).map { |i| i.first}.select { |i| i.try(:dangerous_remediate?) }.any? %>
+    <%= alert(:text => _('This will cause a service interruption.'), :class => 'alert-warning', :close => false) %>
+  <% end %>
 
-<% if @status.class.respond_to?(:dangerous_remediate?) && @status.class.dangerous_remediate? -%>
-<%= alert(:text => _('This will cause a service interruption.'), :class => 'alert-warning', :close => false) %>
-<% end -%>
+  <%= n_('One host selected for remediation.', '%s hosts selected for remediation.', @statuses.count) % @statuses.count %>
 
-<%= form_for @triggering, :url => submit_remediate_host_path(:id => @host.id, :status_id => @status), :html => { :class => 'form-horizontal', :id => 'schedule_remediate_form' } do |f| %>
-  <%= javascript_tag do %>
-    $(function() { trigger_form_selector_binds('<%= f.options[:html][:id] %>','<%= f.object_name %>') });
-  <% end %>
-  <div class="form-group">
-    <label class="col-md-2 control-label"><%= _('Schedule') %></label>
-    <div class="col-md-8">
-      <%= fields_for :triggering, @triggering do |trigger_fields| %>
-        <%= radio_button_f trigger_fields, :mode, :class => 'trigger_mode_selector', :value => 'immediate', :text => _("Execute now") %>
-        <%= radio_button_f trigger_fields, :mode, :class => 'trigger_mode_selector', :value => 'future',    :text => _("Schedule future execution") %>
+  <ul class='hosts-list' style='max-height: 100px; overflow-y: scroll;'>
+    <% @statuses.each do |status| %>
+      <li>
+        <%= status.host.name %>
+        <%= content_tag(:span, nil, class: 'glyphicon glyphicon-warning-sign text-warning', title: _('This will cause a service interruption.')) if status.class.try(:dangerous_remediate?) %>
+      </li>
+    <% end %>
+  </ul>
+
+  <%= form_for @triggering, html: { class: 'form-horizontal', id: 'schedule_remediate_form' },
+                            url: submit_remediate_hosts_path(host_association: @statuses_params[:host_association],
+                                                             owned_only: @statuses_params[:owned_only],
+                                                             status_ids: @statuses_params[:status_ids]) do |f| %>
+    <%= javascript_tag do %>
+      $(function() { trigger_form_selector_binds('<%= f.options[:html][:id] %>','<%= f.object_name %>') });
+    <% end %>
+    <div class="form-group">
+      <label class="col-md-2 control-label"><%= _('Schedule') %></label>
+      <div class="col-md-8">
+        <%= fields_for :triggering, @triggering do |trigger_fields| %>
+          <%= radio_button_f trigger_fields, :mode, :class => 'trigger_mode_selector', :value => 'immediate', :text => _('Execute now') %>
+          <%= radio_button_f trigger_fields, :mode, :class => 'trigger_mode_selector', :value => 'future',    :text => _('Schedule future execution') %>
+      </div>
     </div>
-  </div>
 
-  <div class="trigger_fields">
-    <%= content_tag(:fieldset, nil, :id => 'trigger_mode_future', :class => "trigger_mode_form #{'hidden' unless @triggering.future?}") do
-      safe_join([
-        text_f(f, :start_at_raw, :label => _('Start at'), :placeholder => 'YYYY-mm-dd HH:MM'),
-        text_f(f, :start_before_raw, :label => _('Start before'), :placeholder => 'YYYY-mm-dd HH:MM',
-                                     :label_help => _('Indicates that the action should be cancelled if it cannot be started before this time.'))
-      ])
-    end %>
-  </div>
-  <% end %>
+    <div class="trigger_fields">
+      <%= content_tag(:fieldset, nil, id: 'trigger_mode_future', class: "trigger_mode_form #{'hidden' unless @triggering.future?}") do
+        safe_join([
+          text_f(f, :start_at_raw, label: _('Start at'), placeholder: 'YYYY-mm-dd HH:MM'),
+          text_f(f, :start_before_raw, label: _('Start before'), placeholder: 'YYYY-mm-dd HH:MM',
+                                       label_help: _('Indicates that the action should be cancelled if it cannot be started before this time.'))
+        ])
+      end %>
+    </div>
+    <% end %>
 
-  <%= submit_or_cancel f, false, :cancel_path => { :controller => :'foreman_wreckingball/hosts', :action => :status_dashboard } %>
+    <%= submit_or_cancel f, false, :cancel_path => { controller: :'foreman_wreckingball/hosts', action: :status_dashboard } %>
+  <% end %>
+<% else %>
+  <%= content_tag :h3, _('No hosts selected') %>
+  <%= content_tag :p, _('Please select some hosts and try again') %>
 <% end %>

data/app/views/foreman_wreckingball/hosts/status_dashboard.html.erb

@@ -1,5 +1,8 @@
 <% title _('VMware Status Overview') %>
 <% javascript 'foreman_wreckingball/modal' %>
+<% javascript 'foreman_wreckingball/status_hosts_table' %>
+<% stylesheet 'foreman_wreckingball/status_hosts_table' %>
+<% javascript 'foreman_wreckingball/status_row' %>
 <% javascript 'foreman_tasks/trigger_form' %>
 <% stylesheet 'foreman_tasks/trigger_form' %>
 
@@ -10,11 +13,11 @@
     render 'status_dashboard_empty'
   end
 %>
-<%= render :partial => 'common/modal', :locals => {
-  :id => 'confirmation-modal',
-  :title => _('Please Confirm'),
-  :buttons => [
-    button_tag(_('Cancel'), :class => 'btn btn-default', :data => { :dismiss => 'modal' }, :type => 'button'),
-    button_tag(_('Submit'), :class => 'btn btn-primary', :data => { :action => 'submit' }, :onclick => 'submit_modal_form()')
+<%= render partial: 'common/modal', locals: {
+  id: 'confirmation-modal',
+  title: _('Please Confirm'),
+  buttons: [
+    button_tag(_('Cancel'), class: 'btn btn-default', data: { dismiss: 'modal' }, type: 'button'),
+    button_tag(_('Submit'), class: 'btn btn-primary', data: { action: 'submit' }, onclick: 'submit_modal_form()')
   ]
 } %>

data/app/views/foreman_wreckingball/hosts/status_hosts.json.rabl

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+object false
+
+node(:recordsTotal) { @count }
+node(:recordsFiltered) { @count }
+node(:data) do
+  partial 'foreman_wreckingball/hosts/_hosts', object: @hosts,
+                                               locals: {
+                                                 host_association: @status.host_association,
+                                                 supports_remediate: @status.supports_remediate?
+                                               }
+end

data/app/views/foreman_wreckingball/hosts/status_managed_hosts_dashboard.html.erb

@@ -0,0 +1,142 @@
+<% title _('VMware Managed Hosts Overview') %>
+<% javascript 'foreman_wreckingball/status_managed_hosts_dashboard' %>
+<% stylesheet 'foreman_wreckingball/status_managed_hosts_dashboard' %>
+
+<%= title_actions(
+  button_group(
+    if params[:owned_only]
+      link_to _('Show all hosts'), status_managed_hosts_dashboard_hosts_path, class: 'btn btn-default'
+    else
+      link_to _('Show only owned hosts'), status_managed_hosts_dashboard_hosts_path(owned_only: true), class: 'btn btn-default'
+    end
+  )
+) %>
+
+<%= render partial: 'status_managed_hosts_dashboard_cards', locals: {
+  missing_hosts_count: @missing_hosts.count,
+  wrong_hosts_count: @wrong_hosts.count,
+  more_than_one_hosts_count: @more_than_one_hosts.count
+} %>
+
+<% if @errors.any? %>
+  <div class="alert alert-warning alert-block base alert-dismissable">
+    <%= alert_close %>
+    <h3>
+      <%= icon_text('warning-triangle-o', '', kind: 'pficon') %>
+      <%= _('Failed to load VM data.') %>
+    </h3>
+    <ul>
+    <% @errors.each do |cr, message| %>
+      <li><strong><%= cr %>:</strong> <%= message %></li>
+    <% end %>
+    </ul>
+  </div>
+<% end %>
+
+<ul class='nav nav-tabs' data-tabs='tabs'>
+  <li class='active'>
+    <%= content_tag :a, _('List of Hosts not found in vSphere'), href: '#missing_vms_tab', 'data-toggle': 'tab' %>
+  </li>
+  <li>
+    <%= content_tag :a, _('List of Hosts associated to wrong Compute Resource'), href: '#wrong_hosts_tab', 'data-toggle': 'tab' %>
+  </li>
+  <li>
+    <%= content_tag :a, _('List of Hosts found on more than one Compute Resource'), href: '#more_than_one_hosts_tab', 'data-toggle': 'tab' %>
+  </li>
+</ul>
+
+<div class='tab-content'>
+  <div class='tab-pane active' id='missing_vms_tab'>
+    <% if @missing_hosts.empty? %>
+      <%= content_tag :p, _('No Hosts to show'), class: 'ca' %>
+    <% else %>
+      <%= content_tag :table, id: 'missing_vms', class: table_css_classes do %>
+        <thead>
+          <tr>
+            <%= content_tag :th, _('Name') %>
+            <%= content_tag :th, nil, class: 'no-sort' %>
+          </tr>
+        </thead>
+        <tbody>
+          <% @missing_hosts.each do |host| %>
+            <tr>
+              <%= content_tag :td, link_to_if_authorized(host.name, hash_for_host_path(id: host).merge(permission: 'view_hosts', auth_object: host, authorizer: @host_authorizer)) %>
+              <td>
+                <%= action_buttons(display_delete_if_authorized(hash_for_host_path(id: host).merge(permission: 'destroy_hosts', auth_object: host, authorizer: @host_authorizer),
+                                                                :data => { :confirm => _('Are you sure you want to delete host %s? This action is irreversible.') % host.name })) %>
+              </td>
+            </tr>
+          <% end %>
+        </tbody>
+      <% end %>
+    <% end %>
+  </div>
+  <div class='tab-pane' id='wrong_hosts_tab'>
+    <% if @wrong_hosts.empty? %>
+      <%= content_tag :p, _('No Hosts to show'), class: 'ca' %>
+    <% else %>
+      <%= content_tag :table, id: 'wrong_hosts', class: table_css_classes do %>
+        <thead>
+          <tr>
+            <%= content_tag :th, _('Name') %>
+            <%= content_tag :th, _('Associated to') %>
+            <%= content_tag :th, _('Found on') %>
+            <%= content_tag :th, nil, class: 'no-sort' %>
+          </tr>
+        </thead>
+        <tbody>
+          <% @wrong_hosts.each do |host| %>
+            <tr>
+              <%= content_tag :td, link_to_if_authorized(host.name, hash_for_host_path(id: host).merge(permission: 'view_hosts', auth_object: host, authorizer: @host_authorizer)) %>
+              <%= content_tag :td, link_to_if_authorized(host.compute_resource.name, hash_for_compute_resource_path(id: host.compute_resource).merge(permission: 'view_compute_resources', auth_object: host.compute_resource, authorizer: @compute_resource_authorizer)) %>
+              <%= content_tag :td, link_to_if_authorized(@vm_compute_resource_mapping[host.uuid].first.name, hash_for_compute_resource_path(id: @vm_compute_resource_mapping[host.uuid].first).merge(permission: 'view_compute_resources', auth_object: @vm_compute_resource_mapping[host.uuid].first, authorizer: @compute_resource_authorizer)) %>
+              <td>
+                <%= action_buttons(
+                  display_link_if_authorized(
+                    _('Fix Association'),
+                    hash_for_associate_compute_resource_vm_path(
+                      compute_resource_id: @vm_compute_resource_mapping[host.uuid].first,
+                      id: host.uuid
+                    ).merge(
+                      auth_object: @vm_compute_resource_mapping[host.uuid].first,
+                      authorizer: @compute_resource_authorizer,
+                      permission: 'edit_compute_resources'
+                    ),
+                    title: _('Associate Host to correct Compute Resouce'),
+                    method: :put
+                  )
+                ) %>
+              </td>
+            </tr>
+          <% end %>
+        </tbody>
+      <% end %>
+    <% end %>
+  </div>
+  <div class='tab-pane' id='more_than_one_hosts_tab'>
+    <% if @more_than_one_hosts.empty? %>
+      <%= content_tag :p, _('No Hosts to show'), class: 'ca' %>
+    <% else %>
+      <%= content_tag :table, id: 'more_than_one_hosts', class: table_css_classes do %>
+        <thead>
+          <tr>
+            <%= content_tag :th, _('Name') %>
+            <%= content_tag :th, _('Associated to') %>
+            <%= content_tag :th, _('Found on') %>
+          </tr>
+        </thead>
+        <tbody>
+          <% @more_than_one_hosts.each do |host| %>
+            <tr>
+              <%= content_tag :td, link_to_if_authorized(host.name, hash_for_host_path(id: host).merge(permission: 'view_hosts', auth_object: host, authorizer: authorizer)) %>
+              <%= content_tag :td, (link_to_if_authorized(host.compute_resource.name, hash_for_compute_resource_path(id: host.compute_resource).merge(permission: 'view_compute_resources', auth_object: host.compute_resource, authorizer: @compute_resource_authorizer)) if host.compute_resource) %>
+              <td>
+                <%= safe_join(@vm_compute_resource_mapping[host.uuid].map { |compute_resource| link_to_if_authorized(compute_resource.name, hash_for_compute_resource_path(id: compute_resource).merge(permission: 'view_compute_resources', auth_object: compute_resource, authorizer: @compute_resource_authorizer))}, ', ') %>
+              </td>
+            </tr>
+          <% end %>
+        </tbody>
+      <% end %>
+    <% end %>
+  </div>
+</div>

data/config/environments/production.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require Rails.root.join('config', 'environments', 'production.rb')
+
+Foreman::Application.configure do
+  config.assets.js_compressor = Uglifier.new(harmony: true) if defined?(Uglifier)
+end

data/config/routes.rb

@@ -4,13 +4,13 @@ Rails.application.routes.draw do
   scope '/wreckingball' do
     constraints(:id => /[^\/]+/) do
       resources :hosts, controller: 'foreman_wreckingball/hosts', only: [] do
-        member do
-          get :schedule_remediate
-          post :submit_remediate
-        end
         collection do
           get :status_dashboard
+          get :status_managed_hosts_dashboard
+          get 'status_dashboard/hosts(/:status)', as: :ajax_status_dashboard, action: :status_hosts, defaults: { format: :json }
           put :refresh_status_dashboard
+          get :schedule_remediate
+          post :submit_remediate
         end
       end
     end

data/db/migrate/20181020174609_add_power_state_to_vmware_facets.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddPowerStateToVmwareFacets < ActiveRecord::Migration[5.1]
+  def change
+    add_column :vmware_facets, :power_state, :integer, default: 1, index: true
+  end
+end

data/db/migrate/20181021111543_add_indexes_to_vmware_hypervisor_facets.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class AddIndexesToVmwareHypervisorFacets < ActiveRecord::Migration[5.1]
+  def change
+    add_index :vmware_hypervisor_facets, :uuid
+    add_index :vmware_hypervisor_facets, [:vmware_cluster_id, :uuid]
+  end
+end

data/lib/foreman_wreckingball/engine.rb

@@ -7,6 +7,14 @@ module ForemanWreckingball
   class Engine < ::Rails::Engine
     engine_name 'foreman_wreckingball'
 
+    WRECKINGBALL_STATUSES = [
+      'ForemanWreckingball::ToolsStatus',
+      'ForemanWreckingball::OperatingsystemStatus',
+      'ForemanWreckingball::CpuHotAddStatus',
+      'ForemanWreckingball::SpectreV2Status',
+      'ForemanWreckingball::HardwareVersionStatus'
+    ].freeze
+
     config.autoload_paths += Dir["#{config.root}/app/lib"]
     config.autoload_paths += Dir["#{config.root}/app/services"]
     config.autoload_paths += Dir["#{config.root}/app/models/concerns"]
@@ -23,9 +31,30 @@ module ForemanWreckingball
       end
     end
 
+    initializer 'foreman_wreckingball.load_default_settings', before: :load_config_initializers do
+      table_exists = begin
+                       Setting.table_exists?
+                     rescue StandardError
+                       false
+                     end
+      require_dependency File.expand_path('../../app/models/setting/wreckingball.rb', __dir__) if table_exists
+    end
+
     initializer 'foreman_wreckingball.register_plugin', :before => :finisher_hook do |_app|
       Foreman::Plugin.register :foreman_wreckingball do
-        requires_foreman '>= 1.18'
+        requires_foreman '>= 1.21'
+
+        automatic_assets(false)
+        precompile_assets(
+          [
+            'foreman_wreckingball/modal.js',
+            'foreman_wreckingball/status_hosts_table.js',
+            'foreman_wreckingball/status_managed_hosts_dashboard.js',
+            'foreman_wreckingball/status_row.js',
+            'foreman_wreckingball/status_hosts_table.css',
+            'foreman_wreckingball/status_managed_hosts_dashboard.css'
+          ]
+        )
 
         security_block :foreman_wreckingball do
           permission :refresh_vmware_status_hosts, {
@@ -38,7 +67,9 @@ module ForemanWreckingball
 
         # Extend built in permissions
         Foreman::AccessControl.permission(:view_hosts).actions.concat [
-          'foreman_wreckingball/hosts/status_dashboard'
+          'foreman_wreckingball/hosts/status_dashboard',
+          'foreman_wreckingball/hosts/status_managed_hosts_dashboard',
+          'foreman_wreckingball/hosts/status_hosts'
         ]
 
         menu :top_menu, :wreckingball_status_dashboard, :url_hash => { :controller => :'foreman_wreckingball/hosts', :action => :status_dashboard },
@@ -46,15 +77,18 @@ module ForemanWreckingball
                                                         :parent => :hosts_menu,
                                                         :after => :hosts
 
-        register_custom_status(ForemanWreckingball::ToolsStatus)
-        register_custom_status(ForemanWreckingball::OperatingsystemStatus)
-        register_custom_status(ForemanWreckingball::CpuHotAddStatus)
+        WRECKINGBALL_STATUSES.each { |status| register_custom_status(status.constantize) }
+
+        menu :top_menu, :wreckingball_status_managed_hosts_dashboard, url_hash: { :controller => :'foreman_wreckingball/hosts', :action => :status_managed_hosts_dashboard },
+                                                                      caption: N_('VMware Managed Status'),
+                                                                      parent: :monitor_menu,
+                                                                      after: :audits
 
         register_facet(ForemanWreckingball::VmwareFacet, :vmware_facet)
 
         register_facet(ForemanWreckingball::VmwareHypervisorFacet, :vmware_hypervisor_facet)
 
-        add_controller_action_scope(HostsController, :index) { |base_scope| base_scope.includes(:vmware_facet) }
+        add_controller_action_scope('HostsController', :index) { |base_scope| base_scope.includes(:vmware_facet) }
 
         # extend host show page
         extend_page('compute_resources/show') do |context|
@@ -63,6 +97,9 @@ module ForemanWreckingball
                               :partial => 'compute_resources/hypervisors_tab',
                               :onlyif => proc { |cr| cr.provider_friendly_name == 'VMware' && cr.vmware_hypervisor_facets.any? }
         end
+
+        # add custom logger
+        logger :import, enabled: true
       end
     end
 
@@ -72,23 +109,25 @@ module ForemanWreckingball
 
     # Include concerns in this config.to_prepare block
     config.to_prepare do
-      begin
-        ::ComputeResource.send(:include, ForemanWreckingball::ComputeResourceExtensions)
-        ::Foreman::Model::Vmware.send(:include, ForemanWreckingball::VmwareExtensions)
-        ::Host::Managed.send(:include, ForemanWreckingball::HostExtensions)
-        ::Host::Managed.send(:include, ForemanWreckingball::VmwareFacetHostExtensions)
-        ::Host::Managed.send(:include, ForemanWreckingball::VmwareHypervisorFacetHostExtensions)
-        ::HostsHelper.send(:include, ForemanWreckingball::HostsHelperExtensions)
-
-        if ForemanWreckingball.fog_patches_required?
-          Fog::Compute::Vsphere::Host.send(:include, FogExtensions::ForemanWreckingball::Vsphere::Host)
-          Fog::Compute::Vsphere::Real.send(:include, FogExtensions::ForemanWreckingball::Vsphere::Real)
-        end
-      rescue StandardError => e
-        Rails.logger.warn "ForemanWreckingball: skipping engine hook (#{e})\n#{e.backtrace.join("\n")}"
+      ::HostStatus.extend(ForemanWreckingball::HostStatusExtensions)
+
+      ::ComputeResource.send(:include, ForemanWreckingball::ComputeResourceExtensions)
+      ::Foreman::Model::Vmware.send(:include, ForemanWreckingball::VmwareExtensions)
+      ::Host::Managed.send(:include, ForemanWreckingball::HostExtensions)
+      ::Host::Managed.send(:include, ForemanWreckingball::VmwareFacetHostExtensions)
+      ::Host::Managed.send(:include, ForemanWreckingball::VmwareHypervisorFacetHostExtensions)
+      ::HostsHelper.send(:include, ForemanWreckingball::HostsHelperExtensions)
+      ::User.send(:include, ForemanWreckingball::UserExtensions)
+      ::Usergroup.send(:include, ForemanWreckingball::UsergroupExtensions)
+
+      if ForemanWreckingball.fog_patches_required?
+        ForemanWreckingball.fog_vsphere_namespace::Host.send(:include, FogExtensions::ForemanWreckingball::Vsphere::Host)
+        ForemanWreckingball.fog_vsphere_namespace::Server.send(:include, FogExtensions::ForemanWreckingball::Vsphere::Server)
+        ForemanWreckingball.fog_vsphere_namespace::Real.send(:include, FogExtensions::ForemanWreckingball::Vsphere::Real)
+        ForemanWreckingball.fog_vsphere_namespace::Mock.send(:include, FogExtensions::ForemanWreckingball::Vsphere::Mock)
       end
-
-      # load 'foreman_wreckingball/scheduled_jobs.rb'
+    rescue StandardError => e
+      Rails.logger.warn "ForemanWreckingball: skipping engine hook (#{e})\n#{e.backtrace.join("\n")}"
     end
 
     initializer 'foreman_wreckingball.register_gettext', after: :load_config_initializers do |_app|
@@ -103,8 +142,17 @@ module ForemanWreckingball
     require 'fog/vsphere'
     require 'fog/vsphere/compute'
     require 'fog/vsphere/models/compute/host'
-    !::Fog::Compute::Vsphere::Host.instance_methods.include?(:feature_capabilities)
+    true
   rescue LoadError
     false
   end
+
+  def self.fog_vsphere_namespace
+    require 'fog/vsphere/version'
+    @fog_vsphere_namespace ||= if Gem::Version.new(Fog::Vsphere::VERSION) >= Gem::Version.new('3.0.0')
+                                 Fog::Vsphere::Compute
+                               else
+                                 Fog::Compute::Vsphere
+                               end
+  end
 end