Add basic Gerrit support for presubmit_support.

presubmit_support.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """Enables directory-specific presubmit checks to run at upload and/or commit.
 """
 
 __version__ = '1.8.0'
 
@@ skipping 18 matching lines @@
 import pickle  # Exposed through the API.
 import random
 import re  # Exposed through the API.
 import sys  # Parts exposed through API.
 import tempfile  # Exposed through the API.
 import time
 import traceback  # Exposed through the API.
 import types
 import unittest  # Exposed through the API.
 import urllib2  # Exposed through the API.
+import urlparse
 from warnings import warn
 
 # Local imports.
 import auth
 import fix_encoding
 import gclient_utils
+import gerrit_util
 import owners
 import presubmit_canned_checks
 import rietveld
 import scm
 import subprocess2 as subprocess  # Exposed through the API.
 
 
 # Ask for feedback only once in program lifetime.
 _ASKED_FOR_FEEDBACK = False
 
@@ skipping 1580 matching lines @@
                     "system directories will also be searched.")
   parser.add_option("--upstream",
                     help="Git only: the base ref or upstream branch against "
                     "which the diff should be computed.")
   parser.add_option("--default_presubmit")
   parser.add_option("--may_prompt", action='store_true', default=False)
   parser.add_option("--skip_canned", action='append', default=[],
                     help="A list of checks to skip which appear in "
                     "presubmit_canned_checks. Can be provided multiple times "
                     "to skip multiple canned checks.")
+  parser.add_option("--gerrit_url", help=optparse.SUPPRESS_HELP)
+  parser.add_option("--gerrit_fetch", help=optparse.SUPPRESS_HELP)
   parser.add_option("--rietveld_url", help=optparse.SUPPRESS_HELP)
   parser.add_option("--rietveld_email", help=optparse.SUPPRESS_HELP)
   parser.add_option("--rietveld_fetch", action='store_true', default=False,
                     help=optparse.SUPPRESS_HELP)
   # These are for OAuth2 authentication for bots. See also apply_issue.py
   parser.add_option("--rietveld_email_file", help=optparse.SUPPRESS_HELP)
   parser.add_option("--rietveld_private_key_file", help=optparse.SUPPRESS_HELP)
 
   # TODO(phajdan.jr): Update callers and remove obsolete --trybot-json .
   parser.add_option("--trybot-json",
                     help="Output trybot information to the file specified.")
   auth.add_auth_options(parser)
   options, args = parser.parse_args(argv)
   auth_config = auth.extract_auth_config_from_options(options)
 
   if options.verbose >= 2:
     logging.basicConfig(level=logging.DEBUG)
   elif options.verbose:
     logging.basicConfig(level=logging.INFO)
   else:
     logging.basicConfig(level=logging.ERROR)
 
+  if (any((options.rietveld_url, options.rietveld_email_file,
+           options.rietveld_fetch, options.rietveld_private_key_file))
+      and any((options.gerrit_url, options.gerrit_fetch))):
+    parser.error('Options for only codereview --rietveld_* or --gerrit_* '
+                 'allowed')
+
   if options.rietveld_email and options.rietveld_email_file:
     parser.error("Only one of --rietveld_email or --rietveld_email_file "
                  "can be passed to this program.")
-
   if options.rietveld_email_file:
     with open(options.rietveld_email_file, "rb") as f:
       options.rietveld_email = f.read().strip()
 
   change_class, files = load_files(options, args)
   if not change_class:
     parser.error('For unversioned directory, <files> is not optional.')
   logging.info('Found %d file(s).' % len(files))
 
   rietveld_obj = None
   if options.rietveld_url:
     # The empty password is permitted: '' is not None.
     if options.rietveld_private_key_file:
       rietveld_obj = rietveld.JwtOAuth2Rietveld(
         options.rietveld_url,
         options.rietveld_email,
         options.rietveld_private_key_file)
     else:
       rietveld_obj = rietveld.CachingRietveld(
         options.rietveld_url,
         auth_config,
         options.rietveld_email)
     if options.rietveld_fetch:
       assert options.issue
       props = rietveld_obj.get_issue_properties(options.issue, False)
       options.author = props['owner_email']
       options.description = props['description']
       logging.info('Got author: "%s"', options.author)
       logging.info('Got description: """\n%s\n"""', options.description)
+
+  if options.gerrit_url and options.gerrit_fetch:
+    rietveld_obj = None
+    assert options.issue and options.patchset
+    props = gerrit_util.GetChangeDetail(
+        urlparse.urlparse(options.gerrit_url).netloc, str(options.issue),
+        ['ALL_REVISIONS'])
+    options.author = props['owner']['email']
+    for rev, rev_info in props['revisions'].iteritems():
+      if str(rev_info['_number']) == str(options.patchset):
+        options.description = gerrit_util.GetChangeDescriptionFromGitiles(
+            rev_info['fetch']['http']['url'], rev)
+        break
+    else:
+      print >> sys.stderr, ('Patchset %d was not found in Gerrit issue %d' %
+                            options.patchset, options.issue)
+      return 2
+    logging.info('Got author: "%s"', options.author)
+    logging.info('Got description: """\n%s\n"""', options.description)
+
   try:
     with canned_check_filter(options.skip_canned):
       results = DoPresubmitChecks(
           change_class(options.name,
                       options.description,
                       options.root,
                       files,
                       options.issue,
                       options.patchset,
                       options.author,
@@ skipping 17 matching lines @@
     return 2
 
 
 if __name__ == '__main__':
   fix_encoding.fix_encoding()
   try:
     sys.exit(main())
   except KeyboardInterrupt:
     sys.stderr.write('interrupted\n')
     sys.exit(1)