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

"""Enforces Vulkan types autogen matches script output.

See http://dev.chromium.org/developers/how-tos/depottools/presubmit-scripts
for more details on the presubmit API built into depot_tools.
"""

import os.path


def CommonChecks(input_api, output_api):
  generating_files = input_api.AffectedFiles(
      file_filter=lambda x: os.path.basename(x.LocalPath()) in [
          'generate_vulkan_types.py'])
  generated_files = input_api.AffectedFiles(
      file_filter=lambda x: os.path.basename(x.LocalPath()) in [
          'vulkan_types.mojom', 'vulkan_types_mojom_traits.h',
          'vulkan_types_mojom_traits.cc', 'generated_vulkan_type_mappings.gni'
      ])


  messages = []

  if generated_files and not generating_files:
    long_text = 'Changed files:\n'
    for generated_file in generated_files:
      long_text += generated_file.LocalPath() + '\n'
      long_text += '\n'
      messages.append(output_api.PresubmitError(
          'Vulkan types generated files changed but the generator '
          'did not.', long_text=long_text))

  temp_dir = input_api.CreateTemporaryDirectory()
  commands = []
  if generating_files:
    commands.append(input_api.Command(name='generate_vulkan_types',
                                      cmd=[input_api.python3_executable,
                                           'generate_vulkan_types.py',
                                           '--check',
                                           '--output-dir=' + temp_dir],
                                      kwargs={},
                                      message=output_api.PresubmitError))
  if commands:
    messages.extend(input_api.RunTests(commands))

  return messages

def CheckChangeOnUpload(input_api, output_api):
  return CommonChecks(input_api, output_api)

def CheckChangeOnCommit(input_api, output_api):
  return CommonChecks(input_api, output_api)