# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from recipe_engine import recipe_api

# Pinned gcloud version.
GCLOUD_VERSION = 'version:266.0.0.chromium0'


class GCloudApi(recipe_api.RecipeApi):
  """GCloudApi provides support for common gcloud operations."""

  def __init__(self, *args, **kwargs):
    super(GCloudApi, self).__init__(*args, **kwargs)
    self._gcloud_executable = None

  def __call__(self, *args, **kwargs):
    """Executes specified gcloud command."""
    self._ensure()
    step_name = kwargs.pop('step_name', 'gcloud %s' % args[0])
    cmd = [self._gcloud_executable]
    return self.m.step(step_name, cmd + list(args), **kwargs)

  def _ensure(self):
    if self._gcloud_executable:
      return
    with self.m.step.nest('ensure gcloud'):
      with self.m.context(infra_steps=True):
        pkgs = self.m.cipd.EnsureFile()
        pkgs.add_package('infra/gcloud/${platform}', GCLOUD_VERSION)
        cipd_dir = self.m.path['start_dir'].join('cipd', 'gcloud')
        self.m.cipd.ensure(cipd_dir, pkgs)
        self.m.file.remove('remove gcloud wrapper',
                           cipd_dir.join('bin', 'gcloud'))
        self.m.file.copy('copy patched gcloud', self.resource('gcloud'),
                         cipd_dir.join('bin', 'gcloud'))
        self._gcloud_executable = cipd_dir.join('bin').join('gcloud')

  def container_image_exists(self, image):
    step_result = self(
        'container',
        'images',
        'describe',
        image,
        ok_ret='any',
        step_name='check existence of %s' % image)
    return step_result.retcode == 0