[build-image] Add internal parameter to distinguish official AMI build

This allows more complete AMI cleanup when build official AMIs

Signed-off-by: Hanwen <[email protected]>

Author: hanwen-cluster

cli/src/pcluster/resources/imagebuilder/parallelcluster.yaml

@@ -329,7 +329,7 @@ phases:
           commands:
             - |
               set -v
-              /usr/local/sbin/ami_cleanup.sh
+              /usr/local/sbin/ami_cleanup.sh "${CfnParamIsOfficialAmiBuild}"
 
   - name: validate
     steps:

cli/src/pcluster/templates/imagebuilder_stack.py

@@ -213,6 +213,25 @@ def _add_cfn_parameters(self):
             ),
             description="UpdateOsAndReboot",
         )
+        CfnParameter(
+            self,
+            "CfnParamIsOfficialAmiBuild",
+            type="String",
+            default=(
+                json.loads(self.config.dev_settings.cookbook.extra_chef_attributes)
+                .get("cluster")
+                .get("is_official_ami_build")
+                if self.config.dev_settings
+                and self.config.dev_settings.cookbook
+                and self.config.dev_settings.cookbook.extra_chef_attributes
+                and json.loads(self.config.dev_settings.cookbook.extra_chef_attributes).get("cluster")
+                and json.loads(self.config.dev_settings.cookbook.extra_chef_attributes)
+                .get("cluster")
+                .get("is_official_ami_build")
+                else "false"
+            ),
+            description="IsOfficialAmiBuild",
+        )
 
     # -- Resources --------------------------------------------------------------------------------------------------- #