Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #!/usr/bin/env python | |
| 2 | |
| 3 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | |
| 4 # | |
| 5 # Use of this source code is governed by a BSD-style license | |
| 6 # that can be found in the LICENSE file in the root of the source | |
| 7 # tree. An additional intellectual property rights grant can be found | |
| 8 # in the file PATENTS. All contributing project authors may | |
| 9 # be found in the AUTHORS file in the root of the source tree. | |
| 10 | |
| 11 import argparse | |
| 12 import logging | |
| 13 import os | |
| 14 import sys | |
| 15 import re | |
|
kjellander_webrtc
2017/01/16 07:26:10
sort
ehmaldonado_webrtc
2017/01/16 08:52:50
Acknowledged.
| |
| 16 | |
| 17 from collections import defaultdict | |
| 18 | |
| 19 | |
| 20 DISPLAY_LEVEL = 1 | |
| 21 IGNORE_LEVEL = 0 | |
| 22 | |
| 23 TARGET_RE = re.compile(r'\s*\w*\("(\w*)"\) {$') | |
| 24 SOURCES_RE = re.compile(r'\s*sources \+?= \[$') | |
| 25 SOURCE_FILE_RE = re.compile(r'\s*"([\w\./]*)",$') | |
| 26 INLINE_SOURCES_RE = re.compile(r'\s*sources \+?= \[ "([\w\./]*)" ]$') | |
| 27 | |
| 28 LOG_FORMAT = '%(message)s' | |
| 29 ERROR_MESSAGE = ("{}:{} in target '{}'\n" | |
| 30 "\tSource file '{}'\n" | |
| 31 "\tCrosses boundary of package '{}'.") | |
| 32 | |
| 33 | |
| 34 class Logger(object): | |
| 35 def __init__(self, messagesLeft=None): | |
| 36 self.logLevel = DISPLAY_LEVEL | |
|
kjellander_webrtc
2017/01/16 07:26:10
The style guide says local variables and class/glo
ehmaldonado_webrtc
2017/01/16 08:52:51
Acknowledged.
| |
| 37 self.messagesLeft = messagesLeft | |
| 38 | |
| 39 def log(self, message): | |
| 40 if self.messagesLeft is not None: | |
| 41 if not self.messagesLeft: | |
| 42 self.logLevel = IGNORE_LEVEL | |
| 43 else: | |
| 44 self.messagesLeft -= 1 | |
| 45 logging.log(self.logLevel, message) | |
| 46 | |
| 47 | |
| 48 class RegexAux(object): | |
|
kjellander_webrtc
2017/01/16 07:26:10
I'm a bit confused by the name Aux, is RegexUtil b
ehmaldonado_webrtc
2017/01/16 08:52:51
Acknowledged.
| |
| 49 def __init__(self): | |
| 50 self.match = None | |
| 51 | |
| 52 def matches(self, regex, target): | |
|
kjellander_webrtc
2017/01/16 07:26:10
Rename target->line to avoid confusion here since
ehmaldonado_webrtc
2017/01/16 08:52:51
Acknowledged.
| |
| 53 match_obj = regex.match(target) | |
| 54 if match_obj and match_obj.groups(): | |
| 55 self.match = match_obj.groups()[0] | |
| 56 return match_obj | |
| 57 | |
| 58 | |
| 59 def _BuildTrie(sequences): | |
| 60 trie = defaultdict(list) | |
| 61 for sequence in sequences: | |
| 62 if sequence: | |
| 63 head = sequence.pop() | |
| 64 trie[head].append(sequence) | |
| 65 return dict((k, _BuildTrie(v)) for k, v in trie.items()) | |
| 66 | |
| 67 def _LongestPrefix(trie, query): | |
|
kjellander_webrtc
2017/01/16 07:26:10
+1 blank line for top level methods
ehmaldonado_webrtc
2017/01/16 08:52:50
Acknowledged.
| |
| 68 if not query: | |
| 69 return [] | |
| 70 head = query.pop() | |
| 71 if head not in trie: | |
| 72 return [] | |
| 73 return [head] + _LongestPrefix(trie[head], query) | |
| 74 | |
| 75 def _CheckBuildFile(trie, buildFilePath, logger): | |
| 76 targetName = None | |
| 77 foundErrors = False | |
| 78 processingSourceFiles = False | |
| 79 | |
| 80 regexAux = RegexAux() | |
| 81 package = os.path.dirname(buildFilePath) | |
| 82 | |
| 83 with open(buildFilePath) as buildFile: | |
| 84 for lineNumber, line in enumerate(buildFile): | |
| 85 sourceFile = None | |
| 86 | |
| 87 if regexAux.matches(TARGET_RE, line): | |
|
kjellander_webrtc
2017/01/16 07:26:10
I find it hard to follow what's going on here sinc
ehmaldonado_webrtc
2017/01/16 08:52:51
I'll try to make this clearer.
You're right, this
| |
| 88 targetName = regexAux.match | |
| 89 elif regexAux.matches(SOURCES_RE, line): | |
| 90 processingSourceFiles = True | |
| 91 elif (regexAux.matches(INLINE_SOURCES_RE, line) or | |
| 92 (processingSourceFiles and regexAux.matches(SOURCE_FILE_RE, line))): | |
| 93 sourceFile = regexAux.match | |
| 94 else: | |
| 95 processingSourceFiles = False | |
| 96 | |
| 97 if not sourceFile: | |
| 98 continue | |
| 99 | |
| 100 query = os.path.join(package, sourceFile).split(os.path.sep)[::-1] | |
| 101 realPackage = os.path.sep.join(_LongestPrefix(trie, query)) | |
| 102 if realPackage != package: | |
| 103 foundErrors = True | |
| 104 logger.log(ERROR_MESSAGE.format( | |
| 105 buildFilePath, lineNumber, targetName, sourceFile, realPackage)) | |
| 106 | |
| 107 return foundErrors | |
| 108 | |
| 109 | |
| 110 def main(): | |
| 111 parser = argparse.ArgumentParser( | |
| 112 description='Script that checks package boundary violations in GN ' | |
| 113 'build files.') | |
| 114 | |
| 115 parser.add_argument('root_dir', metavar='ROOT_DIR', | |
| 116 help='The root directory that contains all BUILD.gn ' | |
| 117 'files to be processed.') | |
| 118 parser.add_argument('build_files', metavar='BUILD_FILE', nargs='*', | |
| 119 help='A list of BUILD.gn files to be processed. If no ' | |
| 120 'files are given, all BUILD.gn files under ROOT_DIR ' | |
| 121 'will be processed.') | |
| 122 parser.add_argument('--max_messages', type=int, default=None, | |
| 123 help='If set, the maximum number of violations to be ' | |
|
kjellander_webrtc
2017/01/16 07:26:10
Does this exist only as a performance optimization
ehmaldonado_webrtc
2017/01/16 08:52:50
No, it's not for performance optimization.
If you
| |
| 124 'displayed.') | |
| 125 | |
| 126 args = parser.parse_args() | |
| 127 | |
| 128 logging.basicConfig(format=LOG_FORMAT) | |
| 129 logging.getLogger().setLevel(DISPLAY_LEVEL) | |
| 130 logger = Logger(args.max_messages) | |
| 131 | |
| 132 packages = [root for root, _, files in os.walk(args.root_dir) | |
| 133 if 'BUILD.gn' in files] | |
| 134 trie = _BuildTrie(package.split(os.path.sep)[::-1] | |
| 135 for package in packages) | |
| 136 defaultBuildFiles = [os.path.join(package, 'BUILD.gn') | |
| 137 for package in packages] | |
| 138 | |
| 139 buildFiles = args.build_files or defaultBuildFiles | |
| 140 return any(_CheckBuildFile(trie, buildFilePath, logger) | |
| 141 for buildFilePath in buildFiles) | |
| 142 | |
| 143 if __name__ == '__main__': | |
| 144 sys.exit(main()) | |
| OLD | NEW |