#!/usr/bin/env python
# Copyright 2015 The LUCI Authors. All rights reserved.
# Use of this source code is governed under the Apache License, Version 2.0
# that can be found in the LICENSE file.

import os
import re
import subprocess
import sys


def header(input_api):
  """Returns the expected license header regexp for this project."""
  current_year = int(input_api.time.strftime('%Y'))
  allowed_years = (str(s) for s in reversed(xrange(2011, current_year + 1)))
  years_re = '(' + '|'.join(allowed_years) + ')'
  license_header = (
    r'.*? Copyright %(year)s The LUCI Authors\. '
      r'All rights reserved\.\n'
    r'.*? Use of this source code is governed under the Apache License, '
      r'Version 2\.0\n'
    r'.*? that can be found in the LICENSE file\.(?: \*/)?\n'
  ) % {
    'year': years_re,
  }
  return license_header


def CommonChecks(input_api, output_api):
  def tests(*path):
    return input_api.canned_checks.GetUnitTestsInDirectory(
        input_api,
        output_api,
        input_api.os_path.join(*path),
        whitelist=[r'.+_test\.py'])

  results = []

  results.extend(input_api.canned_checks.PanProjectChecks(
      input_api, output_api, license_header=header(input_api),
      excluded_paths=[
          r'recipe_engine/.+_pb2\.py',
      ],
  ))

  input_api.subprocess.check_output([
    input_api.python_executable,
    'bootstrap/bootstrap_vpython.py', '--presubmit'])

  results.extend(input_api.RunTests(
      tests('recipe_engine', 'unittests') +
      tests('unittests')))

  return results


CheckChangeOnUpload = CommonChecks
CheckChangeOnCommit = CommonChecks