# Copyright 2017 The Chromium 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

UPLOAD_ATTEMPTS = 5

class GCloudAPI(recipe_api.RecipeApi):
  def __call__(self, step_name, *args):
    """Run "gcloud" with the given args.

    This assumes there exists an executable called gcloud on the PATH.
    """
    gcloud = 'gcloud'
    if self.m.platform.is_win:
      gcloud = 'gcloud.cmd'
    token = self.m.service_account.default().get_access_token(
      scopes=['https://www.googleapis.com/auth/devstorage.read_write'])
    with self.m.context(env={'CLOUDSDK_AUTH_ACCESS_TOKEN': token}):
      return self.m.step(step_name, cmd=[gcloud] + list(args))

  def cp(self, name, src, dst, extra_args=None):
    """Attempt to upload or download files to/from Google Cloud Storage (GCS).

    Args:
      name: string. Will be used to fill out the step name.
      src: string. Absolute path for a local file or gcs file (e.g. gs://...)
      dst: string. Same as src.
      extra_args: optional list of args to be passed to gcloud cp. e.g. [-Z]
        asks all files be compressed with gzip after upload and before download.

    If the operation fails, it will be retried multiple times.
    """
    cmd = ['storage', 'cp']
    if extra_args:
      cmd.extend(extra_args)
    cmd.extend([src, dst])

    name = 'upload %s' % name
    for i in range(UPLOAD_ATTEMPTS):
      step_name = name
      if i > 0:
        step_name += ' (attempt %d)' % (i+1)
      try:
        self(step_name, *cmd)
        break
      except self.m.step.StepFailure:
        if i == UPLOAD_ATTEMPTS - 1:
          raise