From 3857c43627d40037b20832e4e6a331dbcd14f86c Mon Sep 17 00:00:00 2001
From: Alexander Olofsson <alexander.olofsson@liu.se>
Date: Mon, 15 Apr 2024 10:59:08 +0200
Subject: [PATCH] Ensure discovery supports WDS boot

---
 lib/foreman_wds/engine.rb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/foreman_wds/engine.rb b/lib/foreman_wds/engine.rb
index 0dd9533..4bbedc7 100644
--- a/lib/foreman_wds/engine.rb
+++ b/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
-- 
GitLab