Add DevSettings/DisableUserData for Image Builder

Author: Himani Anil Deshpande

cli/src/pcluster/config/imagebuilder_config.py

@@ -237,6 +237,7 @@ class ImagebuilderDevSettings(BaseDevSettings):
     def __init__(
         self,
         disable_pcluster_component: bool = None,
+        disable_user_data: bool = None,
         distribution_configuration: DistributionConfiguration = None,
         terminate_instance_on_failure: bool = None,
         disable_validate_and_test: bool = None,
@@ -247,6 +248,7 @@ def __init__(
     ):
         super().__init__(**kwargs)
         self.disable_pcluster_component = Resource.init_param(disable_pcluster_component, default=False)
+        self.disable_user_data = Resource.init_param(disable_user_data, default=True)
         self.distribution_configuration = distribution_configuration
         self.terminate_instance_on_failure = Resource.init_param(terminate_instance_on_failure, default=True)
         self.disable_validate_and_test = Resource.init_param(disable_validate_and_test, default=True)

cli/src/pcluster/schemas/imagebuilder_schema.py

@@ -239,6 +239,7 @@ class ImagebuilderDevSettingsSchema(BaseDevSettingsSchema):
     """Represent the schema of the ImageBuilder Dev Setting."""
 
     disable_pcluster_component = fields.Bool()
+    disable_user_data = fields.Bool()
     distribution_configuration = fields.Nested(DistributionConfigurationSchema)
     terminate_instance_on_failure = fields.Bool()
     disable_validate_and_test = fields.Bool()

cli/src/pcluster/templates/imagebuilder_stack.py

@@ -366,8 +366,26 @@ def _add_imagebuilder_distribution_configuration(self, ami_tags, build_tags, lam
 
         return distribution_configuration_resource
 
+    def _add_imagebuilder_user_data_override(self):
+        disable_user_data = (
+            self.config.dev_settings.disable_user_data
+            if self.config.dev_settings and self.config.dev_settings.disable_user_data
+            else False
+        )
+        additional_instance_configuration_property = None
+        if not disable_user_data:
+            imagebuilder_user_data = os.path.join(imagebuilder_utils.get_resources_directory(), "imagebuilder", "user_data.sh")
+            with open(imagebuilder_user_data, "r", encoding="utf-8") as user_data_file:
+                user_data_content = user_data_file.read()
+
+            additional_instance_configuration_property=imagebuilder.CfnImageRecipe.AdditionalInstanceConfigurationProperty(
+                user_data_override=Fn.base64(user_data_content)
+            )
+        return additional_instance_configuration_property
+
     def _add_imagebuilder_image_recipe(self, build_tags, components, lambda_cleanup_policy_statements):
         # ImageBuilderImageRecipe
+
         image_recipe_resource = imagebuilder.CfnImageRecipe(
             self,
             "ImageRecipe",
@@ -376,9 +394,7 @@ def _add_imagebuilder_image_recipe(self, build_tags, components, lambda_cleanup_
             tags=build_tags,
             parent_image=self.config.build.parent_image,
             components=components,
-            additional_instance_configuration=imagebuilder.CfnImageRecipe.AdditionalInstanceConfigurationProperty(
-                user_data_override=self._add_imgaebuilder_user_data_override()
-            ),
+            additional_instance_configuration=self._add_imagebuilder_user_data_override(),
             block_device_mappings=[
                 imagebuilder.CfnImageRecipe.InstanceBlockDeviceMappingProperty(
                     device_name=self._get_root_device_name(),
@@ -401,12 +417,6 @@ def _add_imagebuilder_image_recipe(self, build_tags, components, lambda_cleanup_
 
         return image_recipe_resource
 
-    def _add_imgaebuilder_user_data_override(self):
-        imagebuilder_user_data = os.path.join(imagebuilder_utils.get_resources_directory(), "imagebuilder", "user_data.sh")
-        with open(imagebuilder_user_data, "r", encoding="utf-8") as user_data_file:
-            user_data_content = user_data_file.read()
-
-        return Fn.base64(Fn.sub(user_data_content))
 
     def _add_imagebuilder_components(self, build_tags, lambda_cleanup_policy_statements):
         imagebuilder_resources_dir = os.path.join(imagebuilder_utils.get_resources_directory(), "imagebuilder")