commit 91846002a4f84d7c3ae483b6a7a3a8ebdba91d0b
author Jack Neus <jackneus@google.com> Thu Dec 01 23:49:22 2022 +0000
committer Chromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com> Mon Dec 05 23:08:27 2022 +0000
tree 3341a694c86e3346c5af7afa3723b678a3e63fd0
parent 01094ffa9e2788d60b913e81dd459a5f7e361055

add support for IMAGE_SCRIPTS archive

BUG=b:254115543
TEST=run_tests

Change-Id: I7d8cb64fefac13ea4aa09bc9eaff7c3ddfc874ab
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/4072687
Tested-by: Jack Neus <jackneus@google.com>
Reviewed-by: Alex Klein <saklein@chromium.org>
Commit-Queue: Jack Neus <jackneus@google.com>
Reviewed-by: Madeleine Hardt <hardtmad@google.com>

api/controller/image.py

diff --git a/api/controller/image.py b/api/controller/image.py
index a510434..26539d4 100644
--- a/api/controller/image.py
+++ b/api/controller/image.py
@@ -204,11 +204,16 @@
         chroot,
         build_target,
     )
+    image_scripts_func = functools.partial(
+        image.create_image_scripts_archive, build_target
+    )
+
     artifact_types = {
         in_proto.ArtifactType.DLC_IMAGE: dlc_func,
         in_proto.ArtifactType.LICENSE_CREDITS: license_func,
         in_proto.ArtifactType.FACTORY_IMAGE: factory_image_func,
         in_proto.ArtifactType.STRIPPED_PACKAGES: stripped_packags_func,
+        in_proto.ArtifactType.IMAGE_SCRIPTS: image_scripts_func,
     }
 
     for output_artifact in in_proto.output_artifacts: