app/assets/javascripts/foreman_wds/host_edit_extensions.js

@@ -32,29 +32,32 @@ function wds_os_selected() {
   }
 };
 
-function wds_content_loaded() {
-  $('#wds_provisioning').detach().insertBefore('#media_select');
-  $('#host_provision_method_build').prop('disabled', false);
-  $('#host_provision_method_wds').prop('disabled', false);
-};
-
-
 function wds_provision_method_selected() {
   $('div[id*=_provisioning]').hide();
   $('#network_provisioning').show();
   $('#wds_provisioning').show();
 
   if ($('#wds_image_select select').val() === '') {
-    $('#wds_image_select select').attr('disabled', true);
+    $('#wds_image_select select').prop('disabled', true);
   }
 }
-$(document)
-  .on('change', '#host_provision_method_wds', wds_provision_method_selected)
-  .on('change', '.host-architecture-os-select', wds_os_selected)
-  .on('ContentLoad', wds_content_loaded);
 
-$(function() {
+function wds_enable_provision_methods() {
+  $('#provision_method').show();
+  $('#host_provision_method_build').prop('disabled', false);
+  $('#host_provision_method_wds').prop('disabled', false);
+
   if($('#host_provision_method_wds').is(':checked')) {
     wds_provision_method_selected();
   }
+};
+
+$(document)
+  .on('change', '#host_provision_method_wds', wds_provision_method_selected)
+  .on('change', '.host-architecture-os-select', wds_os_selected);
+
+$(function() {
+  $('#wds_provisioning').detach().insertBefore('#media_select');
+
+  wds_enable_provision_methods();
 });

app/controllers/concerns/foreman_wds/hosts_controller_extensions.rb

@@ -2,6 +2,8 @@
 
 module ForemanWds
   module HostsControllerExtensions
+    extend ActiveSupport::Concern
+
     included do
       before_action :cleanup_wds_params
 
@@ -17,8 +19,10 @@ module ForemanWds
     end
 
     def cleanup_wds_params
-      # Don't create a WDS facet unless provisioning with it
-      params[:host].delete :wds_facet_attributes if params[:host] && params[:host][:provision_method] != 'wds'
+      return unless params.dig(:host, :provision_method)
+
+      # Don't create a WDS facet if not provisioning with it
+      params[:host].delete :wds_facet_attributes if params.dig(:host, :provision_method) != 'wds'
     end
   end
 end

lib/foreman_wds/engine.rb

@@ -83,9 +83,13 @@ module ForemanWds
       HostsController.include ForemanWds::HostsControllerExtensions
       UnattendedController.prepend ForemanWds::UnattendedControllerExtensions
 
-      ComputeResource.providers.each do |_k, const|
+      ComputeResource.providers.each_value do |const|
         Kernel.const_get(const).send(:prepend, ForemanWds::ComputeResourceExtensions)
       end
+
+      if Foreman::Plugin.installed?('foreman_discovery')
+        DiscoveredHostsController.include ForemanWds::HostsControllerExtensions
+      end
     rescue StandardError => e
       Rails.logger.fatal "foreman_wds: skipping engine hook (#{e})"
     end