Fixes #36207 - Setup puppet during Host Registration

app/controllers/concerns/foreman_puppet/extensions/api_v2_registration_controller_extensions.rb

@@ -0,0 +1,21 @@
+module ForemanPuppet
+  module Extensions
+    module ApiV2RegistrationControllerExtensions
+      extend ActiveSupport::Concern
+
+      def host_setup_extension
+        if @host.hostgroup.present?
+          @host.puppet_proxy_id = @host.hostgroup.puppet_proxy_id if @host.puppet_proxy_id.nil? && @host.hostgroup.puppet_proxy_id.present?
+          @host.puppet_ca_proxy_id = @host.hostgroup.puppet_ca_proxy_id if @host.puppet_ca_proxy_id.nil? && @host.hostgroup.puppet_ca_proxy_id.present?
+
+          if @host.puppet.nil? && @host.hostgroup.puppet&.environment.present?
+            puppet = @host.puppet || @host.build_puppet
+            puppet.environment = @host.hostgroup.puppet&.environment
+          end
+        end
+
+        super
+      end
+    end
+  end
+end

lib/foreman_puppet/engine.rb

@@ -54,6 +54,7 @@ class Engine < ::Rails::Engine
 
       ::Api::V2::BaseController.include ForemanPuppet::Extensions::ApiBaseController
       ::Api::V2::HostsController.include ForemanPuppet::Extensions::ApiV2HostsController
+      ::Api::V2::RegistrationController.prepend ForemanPuppet::Extensions::ApiV2RegistrationControllerExtensions
       ::Api::V2::HostgroupsController.include ForemanPuppet::Extensions::ApiHostgroupsController
       ::Api::V2::TemplateCombinationsController.include ForemanPuppet::Extensions::ApiTemplateCombinationsController
       ::Api::V2::HostsController.include ForemanPuppet::Extensions::ParametersHost