# Copyright 2019 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.post_process import Filter

DEPS = [
    'depot_tools/bot_update',
    'depot_tools/depot_tools',
    'depot_tools/gclient',
    'goma',
    'recipe_engine/buildbucket',
    'recipe_engine/context',
    'recipe_engine/file',
    'recipe_engine/path',
    'recipe_engine/properties',
    'recipe_engine/python',
    'recipe_engine/runtime',
    'recipe_engine/step'
]

def RunSteps(api):
  api.gclient.set_config('emscripten_releases')
  goma_dir = api.goma.ensure_goma()
  env = {
      'GOMA_DIR': goma_dir,
  }
  api.goma.start()

  cache_dir = api.path['builder_cache']
  sync_dir = cache_dir.join('emscripten-releases')
  api.file.ensure_directory('Ensure sync dir', sync_dir)
  build_dir = cache_dir.join('emscripten-releases', 'build')
  install_dir = api.path['start_dir'].join('install')
  waterfall_build = sync_dir.join('waterfall', 'src', 'build.py')
  dir_flags = ['--sync-dir=%s' % sync_dir,
               '--build-dir=%s' % build_dir,
               '--prebuilt-dir=%s' % sync_dir,
               '--v8-dir=%s' % cache_dir.join('v8'),
               '--install-dir=%s' % install_dir]
  build_only_flags = dir_flags + ['--no-sync', '--no-test']
  test_only_flags = dir_flags + ['--no-sync', '--no-build']

  api.file.ensure_directory('Ensure install dir', install_dir)
  with api.context(cwd=cache_dir):
    api.bot_update.ensure_checkout()
    api.gclient.runhooks()

  # Depot tools on path is for ninja
  with api.depot_tools.on_path():
    with api.context(env=env):
      try:
        api.python('Build Wabt', waterfall_build,
                   build_only_flags + ['--build-include=wabt'])
        api.python('Build Binaryen', waterfall_build,
                   build_only_flags + ['--build-include=binaryen'])
        api.python('Build V8', waterfall_build,
                   build_only_flags + ['--build-include=v8'])
        api.python('Build LLVM', waterfall_build,
                   build_only_flags + ['--build-include=llvm',
                                       '--no-tool-tests'])
        api.python('Build fastcomp', waterfall_build,
                   build_only_flags + ['--build-include=fastcomp'])
        api.python('Build Emscripten', waterfall_build,
                   build_only_flags + ['--build-include=emscripten'])
      except api.step.StepFailure as e:
        # If any of these builds fail, testing won't be meaningful.
        exit_status = e.retcode
        raise
      else:
        exit_status = 0
      finally:
        api.goma.stop(build_exit_status=exit_status)

      for name, flag in (('upstream', 'emtest'), ('asm2wasm', 'emtest-asm')):
        try:
          api.python('Emscripten testsuite (%s)' % name, waterfall_build,
                     test_only_flags + ['--test-include=%s' % flag])
        except api.step.StepFailure as e:
          # Log the failure but continue.
          exit_status = e.retcode
  if exit_status != 0:
    raise api.step.StepFailure('Emscripten tests failed')

def GenTests(api):
  def test(name):
    return (
        api.test(name) +
        api.properties(path_config='kitchen') +
        api.runtime(is_luci=True, is_experimental=False)
    )

  yield test('linux')

  yield (
      test('linux_buildfail') +
      api.step_data('Build Wabt', retcode=1) +
      api.post_process(Filter('postprocess_for_goma.upload_log'))
  )

  yield (
      # Check that if the first test fails, the second runs but the
      # overall result is failure.
      test('linux_emtest_fail') +
      api.step_data('Emscripten testsuite (upstream)', retcode=1) +
      api.post_process(Filter('Emscripten testsuite (asm2wasm)',
                              '$result'))
  )