commit 7bf0ecb40e23d6b6eec2aab01dfcec9fb5bb9a1f
author Alex Klein <saklein@chromium.org> Tue Jun 25 10:04:15 2019 -0600
committer Commit Bot <commit-bot@chromium.org> Wed Jun 26 16:51:22 2019 +0000
tree 926dc2350c3560f39d15a6fb33e53dc4236d4596
parent effb0bcb66db74f394cb2f91607623cedb6f6e77

ArtifactsService: Create replacement for BundlePinnedGuestImages.

The old method executed GS downloads. Instead, gather the GS/file info
to return from the new endpoint.

BUG=chromium:954299
TEST=run_tests

Change-Id: I053138c4a379098f7c7e46450bdc605820b5d6a6
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1675926
Tested-by: Alex Klein <saklein@chromium.org>
Auto-Submit: Alex Klein <saklein@chromium.org>
Reviewed-by: Evan Hernandez <evanhernandez@chromium.org>
Commit-Queue: Alex Klein <saklein@chromium.org>

api/controller/artifacts.py

diff --git a/api/controller/artifacts.py b/api/controller/artifacts.py
index 3e056a1..2181173 100644
--- a/api/controller/artifacts.py
+++ b/api/controller/artifacts.py
@@ -195,6 +195,30 @@
   output_proto.artifacts.add().path = os.path.join(output_dir, archive)
 
 
+def FetchPinnedGuestImages(input_proto, output_proto):
+  """Get the pinned guest image information."""
+  sysroot_path = input_proto.sysroot.path
+  if not sysroot_path:
+    cros_build_lib.Die('sysroot.path is required.')
+
+  chroot = controller_util.ParseChroot(input_proto.chroot)
+  sysroot = sysroot_lib.Sysroot(sysroot_path)
+
+  if not chroot.exists():
+    cros_build_lib.Die('Chroot does not exist: %s', chroot.path)
+
+  if not sysroot.Exists(chroot=chroot):
+    cros_build_lib.Die('Sysroot does not exist: %s',
+                       chroot.full_path(sysroot.path))
+
+  pins = artifacts.FetchPinnedGuestImages(chroot, sysroot)
+
+  for pin in pins:
+    pinned_image = output_proto.pinned_images.add()
+    pinned_image.filename = pin.filename
+    pinned_image.uri = pin.uri
+
+
 def BundleFirmware(input_proto, output_proto):
   """Tar the firmware images for a build target.