# -*- 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 import validate
from chromite.api.controller import controller_util
from chromite.api.gen.chromite.api import binhost_pb2
from chromite.api.gen.chromiumos import common_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)

@validate.require('atom')
def GetBestVisible(input_proto, output_proto):
  """Returns the best visible PackageInfo for the indicated atom."""
  cpv = packages.get_best_visible(input_proto.atom)
  package_info = common_pb2.PackageInfo()
  controller_util.CPVToPackageInfo(cpv, package_info)
  output_proto.package_info.CopyFrom(package_info)