15f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)# Copyright 2014 The Chromium Authors. All rights reserved.
25f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)# Use of this source code is governed by a BSD-style license that can be
35f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)# found in the LICENSE file.
45f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
55f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
65f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)def RegexCheck(re, line_number, line, regex, msg):
75f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  """Searches for |regex| in |line| to check for a particular style
85f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)     violation, returning a message like the one below if the regex matches.
95f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)     The |regex| must have exactly one capturing group so that the relevant
105f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)     part of |line| can be highlighted. If more groups are needed, use
115f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)     "(?:...)" to make a non-capturing group. Sample message:
125f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
135f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)       line 6: Use var instead of const.
145f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)           const foo = bar();
155f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)           ^^^^^
165f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  """
175f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
185f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  def _highlight(match):
195f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)    """Takes a start position and a length, and produces a row of '^'s to
205f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)       highlight the corresponding part of a string.
215f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)    """
225f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)    return match.start(1) * ' ' + (match.end(1) - match.start(1)) * '^'
235f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)
245f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  match = re.search(regex, line)
255f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  if match:
265f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)    assert len(match.groups()) == 1
275f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)    return '  line %d: %s\n%s\n%s' % (line_number, msg, line, _highlight(match))
285f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles)  return ''
29