# Copyright 2018 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.
"""Recipe for building and publishing infra tools.

This recipe builds one or more Go binaries in the specified project and
publishes them all to CIPD.  If one or more tests for any package in the
project fail, or one or more packages fail to build, execution stops and no
packages are uploaded.

This recipe uses golang/dep to manage dependencies, so the given project is
expected to have a Gopkg.toml file specifying its dependency restrictions.
"""

from recipe_engine.recipe_api import Property
from recipe_engine.config import List
from recipe_engine import config

import os

DEPS = [
    'infra/checkout',
    'infra/jiri',
    'infra/git',
    'infra/go',
    'recipe_engine/buildbucket',
    'recipe_engine/cipd',
    'recipe_engine/context',
    'recipe_engine/json',
    'recipe_engine/path',
    'recipe_engine/url',
    'recipe_engine/platform',
    'recipe_engine/properties',
    'recipe_engine/raw_io',
    'recipe_engine/step',
]

PROPERTIES = {
    'project':
        Property(kind=str, help='Jiri remote manifest project', default=None),
    'manifest':
        Property(kind=str, help='Jiri manifest to use'),
    'remote':
        Property(kind=str, help='Remote manifest repository'),
    'packages':
        Property(kind=List(str), help='The list of packages to build'),
    'cipd_path':
        Property(kind=str, help='The CIPD base path to use for package uploads', default='fuchsia/infra'),
}

# The tag referencing the golang/dep CIPD package used to install dependencies.
#
# Do not confuse this with the version of https://github.com/golang/dep itself.
# The CIPD package referenced by this version should provide the same version
# of https://github.com/golang/dep that is used for local development
DEP_VERSION = 'version:0.3.2'

# The Go import path for Fuchsia's infra tools project.
PKG_NAME = 'fuchsia.googlesource.com/infra/infra'


def UploadPackage(api, bin_name, bin_dir, cipd_path, revision, remote, platform):
  """
  Creates and uploads a CIPD package containing the tool at bin_dir/bin_name.

  The tool is published to CIPD under the path '$cipd_path/$bin_name/$platform'

  Args:
    bin_dir: The absolute path to the parent directory of bin_name.
    bin_name: The name of the tool binary
    cipd_path: The CIPD base path to use for package uploads
  """

  cipd_pkg_name = '%s/%s/%s' % (cipd_path, bin_name, platform)
  pins = api.cipd.search(cipd_pkg_name, 'git_revision:%s' % revision)
  if len(pins) > 0:
    api.step('Package is up-to-date', cmd=None)
    return

  pkg_def = api.cipd.PackageDefinition(
      package_name=cipd_pkg_name,
      package_root=bin_dir,
  )
  pkg_def.add_file(bin_dir.join(bin_name))

  api.cipd.create_from_pkg(
      pkg_def,
      refs=['latest'],
      tags={
          'git_repository': remote,
          'git_revision': revision,
      },
  )


def RunSteps(api, cipd_path, project, manifest, remote, packages):
  api.jiri.ensure_jiri()
  api.go.ensure_go()

  gopath = api.path['start_dir'].join('go')

  build_input = api.buildbucket.build.input

  # Checkout the project at the specified patch.
  with api.context(infra_steps=True):
    api.checkout(manifest=manifest,
                 remote=remote,
                 project=project,
                 build_input=build_input)

  # Install golang/dep for dependencies.
  with api.step.nest('ensure_packages'):
    with api.context(infra_steps=True):
      cipd_dir = api.path['start_dir'].join('cipd')
      pkgs = api.cipd.EnsureFile()
      pkgs.add_package('go/cmd/github.com/golang/dep/${platform}', DEP_VERSION)
      api.cipd.ensure(cipd_dir, pkgs)

  with api.context(env={'GOPATH': gopath}):
    # A string representing the absolute path where Jiri places the project.
    abs_project_path = api.jiri.project([project]).json.output[0]['path']

    with api.context(cwd=api.path.abs_to_path(abs_project_path)):
      # Download dependencies.
      dep_exe = cipd_dir.join('dep')
      api.step('dep_ensure', [dep_exe, 'ensure', '-v'])

    # Run all tests in the project.
    api.go('test', api.url.join(PKG_NAME, '...'))

    bin_dir = api.path.mkdtemp('infra')


    # Build all tools for both x64 and arm64.
    for arch in ['amd64', 'arm64']:
      with api.context(env={'GOOS': 'linux', 'GOARCH': arch}):
        for pkg in packages:
          bin_name = pkg.split('/')[-1]
          api.go('build', '-o', bin_dir.join(bin_name), pkg)

          # Upload to CIPD.
          if not api.properties.get('tryjob', False):
            revision = str(build_input.gitiles_commit.id)
            assert revision
            UploadPackage(api, bin_name, bin_dir, cipd_path, revision, remote, 'linux-%s' % arch)


def GenTests(api):
  revision = 'a1b2c3'

  ci_build = api.buildbucket.ci_build(
    git_repo='https://fuchsia.googlesource.com/infra/infra',
    revision=revision,
  )
  try_build = api.buildbucket.try_build(
    git_repo='https://fuchsia.googlesource.com/infra/infra',
  )

  # Tests execution in the event that CIPD doesn't yet have this tool at the
  # specified revision.
  yield (api.test('cipd_is_missing_revision') +
    ci_build +
    api.properties(
      project='infra/infra',
      manifest='infra/infra',
      remote='https://fuchsia.googlesource.com/infra/infra',
      packages=['fuchsia.googlesource.com/infra/infra/cmd/catapult']
    ) + api.step_data(
        'cipd search fuchsia/infra/catapult/linux-amd64 git_revision:%s' % revision,
        api.json.output({'result': []}),
    )
  )

  # Tests execution in the event that CIPD already has this tool at the
  # specified revision.
  yield (api.test('cipd_has_revision') +
    ci_build +
    api.properties(
      project='infra/infra',
      manifest='infra/infra',
      remote='https://fuchsia.googlesource.com/infra/infra',
      packages=['fuchsia.googlesource.com/infra/infra/cmd/catapult']
    ) + api.override_step_data(
        'cipd search fuchsia/infra/catapult/linux-amd64 git_revision:%s' % revision,
        api.cipd.example_search('go/cmd/github.com/golang/dep/linux-amd64:%s' % revision),
    )
  )

  # Tests execution in the event that this is a tryjob.
  yield (api.test('cq_and_cipd_has_revision') +
    try_build +
    api.properties(
      project='infra/infra',
      manifest='infra/infra',
      tryjob=True,
      remote='https://fuchsia.googlesource.com/infra/infra',
      packages=['fuchsia.googlesource.com/infra/infra/cmd/catapult'])
  )