PackagesService: Uprev
Add uprev endpoint to uprev packages without committing them.
BUG=chromium:904936
TEST=run_tests
Change-Id: I202966be6b18174839c53afd3824c1510d72161f
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1656516
Tested-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Alex Klein <saklein@chromium.org>
Commit-Queue: Alex Klein <saklein@chromium.org>
Auto-Submit: Alex Klein <saklein@chromium.org>
diff --git a/api/controller/packages.py b/api/controller/packages.py
new file mode 100644
index 0000000..bc9ab47
--- /dev/null
+++ b/api/controller/packages.py
@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+# Copyright 2019 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Package related functionality."""
+
+from __future__ import print_function
+
+from chromite.api.controller import controller_util
+from chromite.api.gen.chromite.api import binhost_pb2
+from chromite.lib import build_target_util
+from chromite.lib import constants
+from chromite.lib import cros_build_lib
+from chromite.service import packages
+
+
+_OVERLAY_TYPE_TO_NAME = {
+ binhost_pb2.OVERLAYTYPE_PUBLIC: constants.PUBLIC_OVERLAYS,
+ binhost_pb2.OVERLAYTYPE_PRIVATE: constants.PRIVATE_OVERLAYS,
+ binhost_pb2.OVERLAYTYPE_BOTH: constants.BOTH_OVERLAYS,
+}
+
+
+def Uprev(input_proto, _output_proto):
+ """Uprev all cros workon ebuilds that have changes."""
+ if not input_proto.overlay_type:
+ cros_build_lib.Die('Overlay type is required.')
+ elif input_proto.overlay_type not in _OVERLAY_TYPE_TO_NAME:
+ cros_build_lib.Die('Overlay type must be one of: %s',
+ ', '.join(_OVERLAY_TYPE_TO_NAME.values()))
+
+ target_names = [t.name for t in input_proto.build_targets]
+ build_targets = [build_target_util.BuildTarget(t) for t in target_names]
+ overlay_type = _OVERLAY_TYPE_TO_NAME[input_proto.overlay_type]
+ chroot = controller_util.ParseChroot(input_proto.chroot)
+ output_dir = input_proto.output_dir or None
+
+ try:
+ packages.uprev_build_targets(build_targets, overlay_type, chroot,
+ output_dir)
+ except packages.Error as e:
+ # Handle module errors nicely, let everything else bubble up.
+ cros_build_lib.Die(e.message)