1 # Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file.
6 """Top-level presubmit script for Skia.
8 See http://dev.chromium.org/developers/how-tos/depottools/presubmit-scripts
9 for more details about the presubmit API built into gcl.
22 REVERT_CL_SUBJECT_PREFIX = 'Revert '
24 SKIA_TREE_STATUS_URL = 'http://skia-tree-status.appspot.com'
26 # Please add the complete email address here (and not just 'xyz@' or 'xyz').
30 'bsalomon@chromium.org',
31 'bsalomon@google.com',
32 'djsollen@chromium.org',
33 'djsollen@google.com',
38 AUTHORS_FILE_NAME = 'AUTHORS'
40 DOCS_PREVIEW_URL = 'https://skia.org/?cl='
41 GOLD_TRYBOT_URL = 'https://gold.skia.org/search?issue='
43 # Path to CQ bots feature is described in https://bug.skia.org/4364
44 PATH_PREFIX_TO_EXTRA_TRYBOTS = {
46 'skia.primary:Test-Ubuntu-GCC-GCE-CPU-AVX2-x86_64-Release-SKNX_NO_SIMD',
47 'include/private/SkAtomics.h': ('skia.primary:'
48 'Test-Ubuntu-Clang-GCE-CPU-AVX2-x86_64-Release-TSAN,'
49 'Test-Ubuntu-Clang-Golo-GPU-GT610-x86_64-Release-TSAN-Trybot'
52 # Below are examples to show what is possible with this feature.
53 # 'src/svg/': 'master1:abc;master2:def',
54 # 'src/svg/parser/': 'master3:ghi,jkl;master4:mno',
55 # 'src/image/SkImage_Base.h': 'master5:pqr,stu;master1:abc1;master2:def',
59 def _CheckChangeHasEol(input_api, output_api, source_file_filter=None):
60 """Checks that files end with atleast one \n (LF)."""
62 for f in input_api.AffectedSourceFiles(source_file_filter):
63 contents = input_api.ReadFile(f, 'rb')
64 # Check that the file ends in atleast one newline character.
65 if len(contents) > 1 and contents[-1:] != '\n':
66 eof_files.append(f.LocalPath())
69 return [output_api.PresubmitPromptWarning(
70 'These files should end in a newline character:',
75 def _PythonChecks(input_api, output_api):
76 """Run checks on any modified Python files."""
77 pylint_disabled_warnings = (
78 'F0401', # Unable to import.
79 'E0611', # No name in module.
80 'W0232', # Class has no __init__ method.
81 'E1002', # Use of super on an old style class.
82 'W0403', # Relative import used.
83 'R0201', # Method could be a function.
84 'E1003', # Using class name in super.
85 'W0613', # Unused argument.
86 'W0105', # String statement has no effect.
88 # Run Pylint on only the modified python files. Unfortunately it still runs
89 # Pylint on the whole file instead of just the modified lines.
90 affected_python_files = []
91 for affected_file in input_api.AffectedSourceFiles(None):
92 affected_file_path = affected_file.LocalPath()
93 if affected_file_path.endswith('.py'):
94 affected_python_files.append(affected_file_path)
95 return input_api.canned_checks.RunPylint(
96 input_api, output_api,
97 disabled_warnings=pylint_disabled_warnings,
98 white_list=affected_python_files)
101 def _IfDefChecks(input_api, output_api):
102 """Ensures if/ifdef are not before includes. See skbug/3362 for details."""
103 comment_block_start_pattern = re.compile('^\s*\/\*.*$')
104 comment_block_middle_pattern = re.compile('^\s+\*.*')
105 comment_block_end_pattern = re.compile('^\s+\*\/.*$')
106 single_line_comment_pattern = re.compile('^\s*//.*$')
107 def is_comment(line):
108 return (comment_block_start_pattern.match(line) or
109 comment_block_middle_pattern.match(line) or
110 comment_block_end_pattern.match(line) or
111 single_line_comment_pattern.match(line))
113 empty_line_pattern = re.compile('^\s*$')
114 def is_empty_line(line):
115 return empty_line_pattern.match(line)
118 for affected_file in input_api.AffectedSourceFiles(None):
119 affected_file_path = affected_file.LocalPath()
120 if affected_file_path.endswith('.cpp') or affected_file_path.endswith('.h'):
121 f = open(affected_file_path)
122 for line in f.xreadlines():
123 if is_comment(line) or is_empty_line(line):
125 # The below will be the first real line after comments and newlines.
126 if line.startswith('#if 0 '):
128 elif line.startswith('#if ') or line.startswith('#ifdef '):
129 failing_files.append(affected_file_path)
135 output_api.PresubmitError(
136 'The following files have #if or #ifdef before includes:\n%s\n\n'
137 'See https://bug.skia.org/3362 for why this should be fixed.' %
138 '\n'.join(failing_files)))
142 def _CopyrightChecks(input_api, output_api, source_file_filter=None):
144 year_pattern = r'\d{4}'
145 year_range_pattern = r'%s(-%s)?' % (year_pattern, year_pattern)
146 years_pattern = r'%s(,%s)*,?' % (year_range_pattern, year_range_pattern)
147 copyright_pattern = (
148 r'Copyright (\([cC]\) )?%s \w+' % years_pattern)
150 for affected_file in input_api.AffectedSourceFiles(source_file_filter):
151 if 'third_party' in affected_file.LocalPath():
153 contents = input_api.ReadFile(affected_file, 'rb')
154 if not re.search(copyright_pattern, contents):
155 results.append(output_api.PresubmitError(
156 '%s is missing a correct copyright header.' % affected_file))
160 def _ToolFlags(input_api, output_api):
161 """Make sure `{dm,nanobench}_flags.py test` passes if modified."""
163 sources = lambda x: ('dm_flags.py' in x.LocalPath() or
164 'nanobench_flags.py' in x.LocalPath())
165 for f in input_api.AffectedSourceFiles(sources):
166 if 0 != subprocess.call(['python', f.LocalPath(), 'test']):
167 results.append(output_api.PresubmitError('`python %s test` failed' % f))
171 def _InfraTests(input_api, output_api):
172 """Run the infra tests."""
174 if not any(f.LocalPath().startswith('infra')
175 for f in input_api.AffectedFiles()):
178 cmd = ['python', os.path.join('infra', 'bots', 'infra_tests.py')]
180 subprocess.check_output(cmd)
181 except subprocess.CalledProcessError as e:
182 results.append(output_api.PresubmitError(
183 '`%s` failed:\n%s' % (' '.join(cmd), e.output)))
187 def _CheckGNFormatted(input_api, output_api):
188 """Make sure any .gn files we're changing have been formatted."""
190 for f in input_api.AffectedFiles():
191 if (not f.LocalPath().endswith('.gn') and
192 not f.LocalPath().endswith('.gni')):
195 gn = 'gn.bat' if 'win32' in sys.platform else 'gn'
196 cmd = [gn, 'format', '--dry-run', f.LocalPath()]
198 subprocess.check_output(cmd)
199 except subprocess.CalledProcessError:
200 fix = 'gn format ' + f.LocalPath()
201 results.append(output_api.PresubmitError(
202 '`%s` failed, try\n\t%s' % (' '.join(cmd), fix)))
206 def _CommonChecks(input_api, output_api):
207 """Presubmit checks common to upload and commit."""
209 sources = lambda x: (x.LocalPath().endswith('.h') or
210 x.LocalPath().endswith('.py') or
211 x.LocalPath().endswith('.sh') or
212 x.LocalPath().endswith('.m') or
213 x.LocalPath().endswith('.mm') or
214 x.LocalPath().endswith('.go') or
215 x.LocalPath().endswith('.c') or
216 x.LocalPath().endswith('.cc') or
217 x.LocalPath().endswith('.cpp'))
220 input_api, output_api, source_file_filter=sources))
221 results.extend(_PythonChecks(input_api, output_api))
222 results.extend(_IfDefChecks(input_api, output_api))
223 results.extend(_CopyrightChecks(input_api, output_api,
224 source_file_filter=sources))
225 results.extend(_ToolFlags(input_api, output_api))
229 def CheckChangeOnUpload(input_api, output_api):
230 """Presubmit checks for the change on upload.
232 The following are the presubmit checks:
233 * Check change has one and only one EOL.
236 results.extend(_CommonChecks(input_api, output_api))
237 # Run on upload, not commit, since the presubmit bot apparently doesn't have
238 # coverage or Go installed.
239 results.extend(_InfraTests(input_api, output_api))
241 results.extend(_CheckGNFormatted(input_api, output_api))
245 def _CheckTreeStatus(input_api, output_api, json_url):
246 """Check whether to allow commit.
249 input_api: input related apis.
250 output_api: output related apis.
251 json_url: url to download json style status.
253 tree_status_results = input_api.canned_checks.CheckTreeIsOpen(
254 input_api, output_api, json_url=json_url)
255 if not tree_status_results:
256 # Check for caution state only if tree is not closed.
257 connection = input_api.urllib2.urlopen(json_url)
258 status = input_api.json.loads(connection.read())
260 if ('caution' in status['message'].lower() and
261 os.isatty(sys.stdout.fileno())):
262 # Display a prompt only if we are in an interactive shell. Without this
263 # check the commit queue behaves incorrectly because it considers
264 # prompts to be failures.
265 short_text = 'Tree state is: ' + status['general_state']
266 long_text = status['message'] + '\n' + json_url
267 tree_status_results.append(
268 output_api.PresubmitPromptWarning(
269 message=short_text, long_text=long_text))
271 # Tree status is closed. Put in message about contacting sheriff.
272 connection = input_api.urllib2.urlopen(
273 SKIA_TREE_STATUS_URL + '/current-sheriff')
274 sheriff_details = input_api.json.loads(connection.read())
276 tree_status_results[0]._message += (
277 '\n\nPlease contact the current Skia sheriff (%s) if you are trying '
278 'to submit a build fix\nand do not know how to submit because the '
279 'tree is closed') % sheriff_details['username']
280 return tree_status_results
283 class CodeReview(object):
284 """Abstracts which codereview tool is used for the specified issue."""
286 def __init__(self, input_api):
287 self._issue = input_api.change.issue
288 self._gerrit = input_api.gerrit
289 self._rietveld_properties = None
291 self._rietveld_properties = input_api.rietveld.get_issue_properties(
292 issue=int(self._issue), messages=True)
294 def GetOwnerEmail(self):
296 return self._gerrit.GetChangeOwner(self._issue)
298 return self._rietveld_properties['owner_email']
300 def GetSubject(self):
302 return self._gerrit.GetChangeInfo(self._issue)['subject']
304 return self._rietveld_properties['subject']
306 def GetDescription(self):
308 return self._gerrit.GetChangeDescription(self._issue)
310 return self._rietveld_properties['description']
314 return self._gerrit.GetChangeInfo(
315 self._issue)['labels']['Commit-Queue'].get('value', 0) == 1
317 return self._rietveld_properties['cq_dry_run']
319 def GetReviewers(self):
321 code_review_label = (
322 self._gerrit.GetChangeInfo(self._issue)['labels']['Code-Review'])
323 return [r['email'] for r in code_review_label.get('all', [])]
325 return self._rietveld_properties['reviewers']
327 def GetApprovers(self):
330 code_review_label = (
331 self._gerrit.GetChangeInfo(self._issue)['labels']['Code-Review'])
332 for m in code_review_label.get('all', []):
333 if m.get("value") == 1:
334 approvers.append(m["email"])
336 for m in self._rietveld_properties.get('messages', []):
337 if 'lgtm' in m['text'].lower():
338 approvers.append(m['sender'])
342 def _CheckOwnerIsInAuthorsFile(input_api, output_api):
344 if input_api.change.issue:
345 cr = CodeReview(input_api)
347 owner_email = cr.GetOwnerEmail()
350 for line in open(AUTHORS_FILE_NAME):
351 if not line.startswith('#'):
352 authors_content += line
353 email_fnmatches = re.findall('<(.*)>', authors_content)
354 for email_fnmatch in email_fnmatches:
355 if fnmatch.fnmatch(owner_email, email_fnmatch):
356 # Found a match, the user is in the AUTHORS file break out of the loop
360 output_api.PresubmitError(
361 'The email %s is not in Skia\'s AUTHORS file.\n'
362 'Issue owner, this CL must include an addition to the Skia AUTHORS '
366 # Do not fail if authors file cannot be found.
367 traceback.print_exc()
368 input_api.logging.error('AUTHORS file not found!')
373 def _CheckLGTMsForPublicAPI(input_api, output_api):
374 """Check LGTMs for public API changes.
376 For public API files make sure there is an LGTM from the list of owners in
380 requires_owner_check = False
381 for affected_file in input_api.AffectedFiles():
382 affected_file_path = affected_file.LocalPath()
383 file_path, file_ext = os.path.splitext(affected_file_path)
384 # We only care about files that end in .h and are under the top-level
385 # include dir, but not include/private.
386 if (file_ext == '.h' and
387 'include' == file_path.split(os.path.sep)[0] and
388 'private' not in file_path):
389 requires_owner_check = True
391 if not requires_owner_check:
394 lgtm_from_owner = False
395 if input_api.change.issue:
396 cr = CodeReview(input_api)
398 if re.match(REVERT_CL_SUBJECT_PREFIX, cr.GetSubject(), re.I):
399 # It is a revert CL, ignore the public api owners check.
403 # Ignore public api owners check for dry run CLs since they are not
404 # going to be committed.
408 for reviewer in cr.GetReviewers():
409 if reviewer in PUBLIC_API_OWNERS:
410 # If an owner is specified as an reviewer in Gerrit then ignore the
411 # public api owners check.
414 match = re.search(r'^TBR=(.*)$', cr.GetDescription(), re.M)
416 tbr_section = match.group(1).strip().split(' ')[0]
417 tbr_entries = tbr_section.split(',')
418 for owner in PUBLIC_API_OWNERS:
419 if owner in tbr_entries or owner.split('@')[0] in tbr_entries:
420 # If an owner is specified in the TBR= line then ignore the public
424 if cr.GetOwnerEmail() in PUBLIC_API_OWNERS:
425 # An owner created the CL that is an automatic LGTM.
426 lgtm_from_owner = True
428 for approver in cr.GetApprovers():
429 if approver in PUBLIC_API_OWNERS:
430 # Found an lgtm in a message from an owner.
431 lgtm_from_owner = True
434 if not lgtm_from_owner:
436 output_api.PresubmitError(
437 "If this CL adds to or changes Skia's public API, you need an LGTM "
438 "from any of %s. If this CL only removes from or doesn't change "
439 "Skia's public API, please add a short note to the CL saying so. "
440 "Add one of the owners as a reviewer to your CL. For Rietveld CLs "
441 "you also need to add one of those owners on a TBR= line. If you "
442 "don't know if this CL affects Skia's public API, treat it like it "
443 "does." % str(PUBLIC_API_OWNERS)))
447 def PostUploadHook(cl, change, output_api):
448 """git cl upload will call this hook after the issue is created/modified.
450 This hook does the following:
451 * Adds a link to preview docs changes if there are any docs changes in the CL.
452 * Adds 'NOTRY=true' if the CL contains only docs changes.
453 * Adds 'NOTREECHECKS=true' for non master branch changes since they do not
454 need to be gated on the master branch's tree.
455 * Adds 'NOTRY=true' for non master branch changes since trybots do not yet
457 * Adds 'NOPRESUBMIT=true' for non master branch changes since those don't
458 run the presubmit checks.
459 * Adds extra trybots for the paths defined in PATH_TO_EXTRA_TRYBOTS.
463 atleast_one_docs_change = False
464 all_docs_changes = True
465 for affected_file in change.AffectedFiles():
466 affected_file_path = affected_file.LocalPath()
467 file_path, _ = os.path.splitext(affected_file_path)
468 if 'site' == file_path.split(os.path.sep)[0]:
469 atleast_one_docs_change = True
471 all_docs_changes = False
472 if atleast_one_docs_change and not all_docs_changes:
477 original_description_lines, footers = cl.GetDescriptionFooters()
478 new_description_lines = list(original_description_lines)
480 # If the change includes only doc changes then add NOTRY=true in the
481 # CL's description if it does not exist yet.
482 if all_docs_changes and 'NOTRY=true' not in new_description_lines:
483 new_description_lines.append('NOTRY=true')
485 output_api.PresubmitNotifyResult(
486 'This change has only doc changes. Automatically added '
487 '\'NOTRY=true\' to the CL\'s description'))
489 # If there is atleast one docs change then add preview link in the CL's
490 # description if it does not already exist there.
491 docs_preview_line = 'DOCS_PREVIEW= %s%s' % (DOCS_PREVIEW_URL, issue)
492 if (atleast_one_docs_change and
493 docs_preview_line not in new_description_lines):
494 # Automatically add a link to where the docs can be previewed.
495 new_description_lines.append(docs_preview_line)
497 output_api.PresubmitNotifyResult(
498 'Automatically added a link to preview the docs changes to the '
499 'CL\'s description'))
501 # If the target ref is not master then add NOTREECHECKS=true and NOTRY=true
502 # to the CL's description if it does not already exist there.
503 target_ref = cl.GetRemoteBranch()[1]
504 if target_ref != 'refs/remotes/origin/master':
505 if 'NOTREECHECKS=true' not in new_description_lines:
506 new_description_lines.append('NOTREECHECKS=true')
508 output_api.PresubmitNotifyResult(
509 'Branch changes do not need to rely on the master branch\'s '
510 'tree status. Automatically added \'NOTREECHECKS=true\' to the '
511 'CL\'s description'))
512 if 'NOTRY=true' not in new_description_lines:
513 new_description_lines.append('NOTRY=true')
515 output_api.PresubmitNotifyResult(
516 'Trybots do not yet work for non-master branches. '
517 'Automatically added \'NOTRY=true\' to the CL\'s description'))
518 if 'NOPRESUBMIT=true' not in new_description_lines:
519 new_description_lines.append('NOPRESUBMIT=true')
521 output_api.PresubmitNotifyResult(
522 'Branch changes do not run the presubmit checks.'))
524 # Automatically set CQ_INCLUDE_TRYBOTS if any of the changed files here
525 # begin with the paths of interest.
526 cq_master_to_trybots = collections.defaultdict(set)
527 for affected_file in change.AffectedFiles():
528 affected_file_path = affected_file.LocalPath()
529 for path_prefix, extra_bots in PATH_PREFIX_TO_EXTRA_TRYBOTS.iteritems():
530 if affected_file_path.startswith(path_prefix):
532 output_api.PresubmitNotifyResult(
533 'Your CL modifies the path %s.\nAutomatically adding %s to '
534 'the CL description.' % (affected_file_path, extra_bots)))
535 _MergeCQExtraTrybotsMaps(
536 cq_master_to_trybots, _GetCQExtraTrybotsMap(extra_bots))
537 if cq_master_to_trybots:
538 _AddCQExtraTrybotsToDesc(cq_master_to_trybots, new_description_lines)
540 # If the description has changed update it.
541 if new_description_lines != original_description_lines:
542 cl.UpdateDescriptionFooters(new_description_lines, footers)
547 def _AddCQExtraTrybotsToDesc(cq_master_to_trybots, description_lines):
548 """Adds the specified master and trybots to the CQ_INCLUDE_TRYBOTS keyword.
550 If the keyword already exists in the description then it appends to it only
551 if the specified values do not already exist.
552 If the keyword does not exist then it creates a new section in the
557 for idx, line in enumerate(description_lines):
558 if line.startswith('CQ_INCLUDE_TRYBOTS'):
563 original_trybots_map = _GetCQExtraTrybotsMap(found)
564 _MergeCQExtraTrybotsMaps(cq_master_to_trybots, original_trybots_map)
565 new_line = _GetCQExtraTrybotsStr(cq_master_to_trybots)
566 if new_line != found:
567 description_lines[foundIdx] = new_line
569 description_lines.append(_GetCQExtraTrybotsStr(cq_master_to_trybots))
572 def _MergeCQExtraTrybotsMaps(dest_map, map_to_be_consumed):
573 """Merges two maps of masters to trybots into one."""
574 for master, trybots in map_to_be_consumed.iteritems():
575 dest_map[master].update(trybots)
579 def _GetCQExtraTrybotsMap(cq_extra_trybots_str):
580 """Parses CQ_INCLUDE_TRYBOTS str and returns a map of masters to trybots."""
581 cq_master_to_trybots = collections.defaultdict(set)
582 for section in cq_extra_trybots_str.split(';'):
584 master, bots = section.split(':')
585 cq_master_to_trybots[master].update(bots.split(','))
586 return cq_master_to_trybots
589 def _GetCQExtraTrybotsStr(cq_master_to_trybots):
590 """Constructs the CQ_INCLUDE_TRYBOTS str from a map of masters to trybots."""
592 for master, trybots in cq_master_to_trybots.iteritems():
593 sections.append('%s:%s' % (master, ','.join(trybots)))
594 return 'CQ_INCLUDE_TRYBOTS=%s' % ';'.join(sections)
597 def CheckChangeOnCommit(input_api, output_api):
598 """Presubmit checks for the change on commit.
600 The following are the presubmit checks:
601 * Check change has one and only one EOL.
602 * Ensures that the Skia tree is open in
603 http://skia-tree-status.appspot.com/. Shows a warning if it is in 'Caution'
604 state and an error if it is in 'Closed' state.
607 results.extend(_CommonChecks(input_api, output_api))
609 _CheckTreeStatus(input_api, output_api, json_url=(
610 SKIA_TREE_STATUS_URL + '/banner-status?format=json')))
611 results.extend(_CheckLGTMsForPublicAPI(input_api, output_api))
612 results.extend(_CheckOwnerIsInAuthorsFile(input_api, output_api))