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

PRESUBMIT_VERSION = '2.0.0'

def CheckSqlModules(input_api, output_api):
  stdlib_dir = input_api.PresubmitLocalPath()
  chromium_src_dir = input_api.os_path.abspath(
    input_api.os_path.join(stdlib_dir, '..', '..'))
  perfetto_src_dir = input_api.os_path.join(
    chromium_src_dir, 'third_party', 'perfetto')
  tool = input_api.os_path.join(
    chromium_src_dir,
    'third_party', 'perfetto', 'tools', 'check_sql_modules.py')
  cmd = [
    input_api.python3_executable,
    tool,
    '--stdlib-sources',
    input_api.os_path.join(
    perfetto_src_dir,
    '..', '..', 'base', 'tracing', 'stdlib', 'chrome')
    ]
  test_cmd = input_api.Command(
    name='check_sql_modules',
    cmd=cmd,
    kwargs={},
    message=output_api.PresubmitNotifyResult)
  return input_api.RunTests([test_cmd])

_STDLIB_PATHS = (
  r"^base/tracing/stdlib/",
  r"^base/tracing/test/",
  r"^base/tracing/protos/"
)

def CheckPerfettoTestsTag(input_api, output_api):
  """Checks that commits to the trace processor chrome stdlib or the
  Perfetto diff tests contain a PERFETTO_TESTS tag in their commit
  message."""
  def FileFilter(affected_file):
    return input_api.FilterSourceFile(affected_file,
                                      files_to_check=_STDLIB_PATHS)

  # Only consider changes to chrome stdlib or tests paths
  if not any (input_api.AffectedFiles(file_filter=FileFilter)):
    return []

  if input_api.change.PERFETTO_TESTS:
    return []

  message = (
    'Must provide PERFETTO_TESTS='
    '`autoninja -C out/Default perfetto_diff_tests && '
    'out/Default/bin/run_perfetto_diff_tests` line in CL description.'
    '\nPlease ensure the Perfetto diff tests pass before submitting.'
  )
  return [output_api.PresubmitNotifyResult(message)]