From 65866dddb8ecc70381c7ff6534f2c2556fa14ea3 Mon Sep 17 00:00:00 2001
From: Alexander Olofsson <alexander.olofsson@liu.se>
Date: Tue, 5 Nov 2024 10:00:37 +0100
Subject: [PATCH] Inject passwordstate into discovery controller

---
 lib/foreman_passwordstate/engine.rb | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lib/foreman_passwordstate/engine.rb b/lib/foreman_passwordstate/engine.rb
index be04d52..e45a036 100644
--- a/lib/foreman_passwordstate/engine.rb
+++ b/lib/foreman_passwordstate/engine.rb
@@ -28,6 +28,9 @@ module ForemanPasswordstate
       HostsController.prepend ForemanPasswordstate::HostsControllerExtensions
       HostgroupsController.prepend ForemanPasswordstate::HostgroupsControllerExtensions
       Operatingsystem.prepend ForemanPasswordstate::OperatingsystemExtensions
+      if Foreman::Plugin.installed?('foreman_discovery')
+        DiscoveredHostsController.prepend ForemanPasswordstate::HostsControllerExtensions
+      end
     rescue StandardError => e
       Rails.logger.fatal "foreman_passwordstate: skipping engine hook (#{e})"
     end
-- 
GitLab