Add presubmit_canned_checks.CheckLicense()

presubmit_canned_checks.py

 #!/usr/bin/env python
 # Copyright (c) 2006-2009 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.
 
 """Generic presubmit checks that can be reused by other presubmit checks."""
 
 ### Description checks
 
 def CheckChangeHasTestField(input_api, output_api):
@@ skipping 249 matching lines @@
       if len(bad) == 5:  # Just show the first 5 errors.
         break
 
   if bad:
     msg = "Found lines longer than %s characters (first 5 shown)." % maxlen
     return [output_api.PresubmitPromptWarning(msg, items=bad)]
   else:
     return []
 
 
+def CheckLicense(input_api, output_api, license, source_file_filter=None):
+  """Verifies the license header.
+  """
+  license_re = input_api.re.compile(license, input_api.re.MULTILINE)
+  bad_files = []
+  for f in input_api.AffectedSourceFiles(source_file_filter):
+    contents = input_api.ReadFile(f, 'rb')
+    if not license_re.search(contents):
+      bad_files.append(f.LocalPath())
+  if bad_files:
+    if input_api.is_committing:
+      res_type = output_api.PresubmitPromptWarning
+    else:
+      res_type = output_api.PresubmitNotifyResult
+    return [res_type(
+        "Found a bad license header in these files:", items=bad_files)]
+  return []
+
+
 def CheckChangeSvnEolStyle(input_api, output_api, source_file_filter=None):
   """Checks that the source files have svn:eol-style=LF."""
   return CheckSvnProperty(input_api, output_api,
                           'svn:eol-style', 'LF',
                           input_api.AffectedSourceFiles(source_file_filter))
 
 
 def CheckSvnForCommonMimeTypes(input_api, output_api):
   """Checks that common binary file types have the correct svn:mime-type."""
   output = []
@@ skipping 106 matching lines @@
         stderr=input_api.subprocess.PIPE)
     stdoutdata, stderrdata = subproc.communicate()
     # Discard the output if returncode == 0
     if subproc.returncode:
       outputs.append("Test '%s' failed with code %d\n%s\n%s\n" % (
           unit_test_name, subproc.returncode, stdoutdata, stderrdata))
   if outputs:
     return [message_type("%d unit tests failed." % len(outputs),
                                 long_text='\n'.join(outputs))]
   return []