| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 | |
| 3 # Copyright (c) 2016 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 os | |
| 12 import re | |
| 13 import shutil | |
| 14 import subprocess | |
| 15 import sys | |
| 16 import tempfile | |
| 17 | |
| 18 from collections import defaultdict | |
| 19 | |
| 20 TARGET_RE = re.compile( | |
| 21 r'(?P<indentation_level>\s*)\w*\("(?P<target_name>\w*)"\) {$') | |
| 22 | |
| 23 class TemporaryDirectory(object): | |
| 24 def __init__(self): | |
| 25 self._closed = False | |
| 26 self._name = None | |
| 27 self._name = tempfile.mkdtemp() | |
| 28 | |
| 29 def __enter__(self): | |
| 30 return self._name | |
| 31 | |
| 32 def __exit__(self, exc, value, tb): | |
| 33 if self._name and not self._closed: | |
| 34 shutil.rmtree(self._name) | |
| 35 self._closed = True | |
| 36 | |
| 37 | |
| 38 def Run(cmd): | |
| 39 print 'Running:', ' '.join(cmd) | |
| 40 sub = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) | |
| 41 return sub.communicate() | |
| 42 | |
| 43 def FixErrors(filename, missing_deps, deleted_sources): | |
| 44 with open(filename) as f: | |
| 45 lines = f.readlines() | |
| 46 | |
| 47 fixed_file = '' | |
| 48 indentation_level = None | |
| 49 for line in lines: | |
| 50 match = TARGET_RE.match(line) | |
| 51 if match: | |
| 52 target = match.group('target_name') | |
| 53 if target in missing_deps: | |
| 54 indentation_level = match.group('indentation_level') | |
| 55 elif indentation_level is not None: | |
| 56 match = re.match(indentation_level + '}$', line) | |
| 57 if match: | |
| 58 line = ('deps = [\n' + | |
| 59 ''.join(' "' + dep + '",\n' for dep in missing_deps[target]) + | |
| 60 ']\n') + line | |
| 61 indentation_level = None | |
| 62 elif line.strip().startswith('deps'): | |
| 63 is_empty_deps = line.strip() == 'deps = []' | |
| 64 line = 'deps = [\n' if is_empty_deps else line | |
| 65 line += ''.join(' "' + dep + '",\n' for dep in missing_deps[target]) | |
| 66 line += ']\n' if is_empty_deps else '' | |
| 67 indentation_level = None | |
| 68 | |
| 69 if line.strip() not in deleted_sources: | |
| 70 fixed_file += line | |
| 71 | |
| 72 with open(filename, 'w') as f: | |
| 73 f.write(fixed_file) | |
| 74 | |
| 75 Run(['gn', 'format', filename]) | |
| 76 | |
| 77 def Rebase(base_path, dependency_path, dependency): | |
| 78 base_path = base_path.split(os.path.sep) | |
| 79 dependency_path = dependency_path.split(os.path.sep) | |
| 80 | |
| 81 first_difference = None | |
| 82 shortest_length = min(len(dependency_path), len(base_path)) | |
| 83 for i in range(shortest_length): | |
| 84 if dependency_path[i] != base_path[i]: | |
| 85 first_difference = i | |
| 86 break | |
| 87 | |
| 88 first_difference = first_difference or shortest_length | |
| 89 base_path = base_path[first_difference:] | |
| 90 dependency_path = dependency_path[first_difference:] | |
| 91 return (os.path.sep.join((['..'] * len(base_path)) + dependency_path) + | |
| 92 ':' + dependency) | |
| 93 | |
| 94 def main(): | |
| 95 deleted_sources = set() | |
| 96 errors_by_file = defaultdict(lambda: defaultdict(set)) | |
| 97 | |
| 98 with TemporaryDirectory() as tmp_dir: | |
| 99 mb_gen_command = ([ | |
| 100 'tools/mb/mb.py', 'gen', | |
| 101 tmp_dir, | |
| 102 '--config-file', 'webrtc/build/mb_config.pyl', | |
| 103 ] + sys.argv[1:]) | |
| 104 | |
| 105 mb_output = Run(mb_gen_command) | |
| 106 errors = mb_output[0].split('ERROR')[1:] | |
| 107 | |
| 108 if mb_output[1]: | |
| 109 print mb_output[1] | |
| 110 return 1 | |
| 111 | |
| 112 for error in errors: | |
| 113 error = error.splitlines() | |
| 114 target_msg = 'The target:' | |
| 115 if target_msg not in error: | |
| 116 target_msg = 'It is not in any dependency of' | |
| 117 if target_msg not in error: | |
| 118 print '\n'.join(error) | |
| 119 continue | |
| 120 index = error.index(target_msg) + 1 | |
| 121 path, target = error[index].strip().split(':') | |
| 122 if error[index+1] in ('is including a file from the target:', | |
| 123 'The include file is in the target(s):'): | |
| 124 dep = error[index+2].strip() | |
| 125 dep_path, dep = dep.split(':') | |
| 126 dep = Rebase(path, dep_path, dep) | |
| 127 path = os.path.join(path[2:], 'BUILD.gn') | |
| 128 errors_by_file[path][target].add(dep) | |
| 129 elif error[index+1] == 'has a source file:': | |
| 130 deleted_file = '"' + os.path.basename(error[index+2].strip()) + '",' | |
| 131 deleted_sources.add(deleted_file) | |
| 132 else: | |
| 133 print '\n'.join(error) | |
| 134 continue | |
| 135 | |
| 136 for path, missing_deps in errors_by_file.items(): | |
| 137 FixErrors(path, missing_deps, deleted_sources) | |
| 138 | |
| 139 return 0 | |
| 140 | |
| 141 if __name__ == '__main__': | |
| 142 sys.exit(main()) | |
| OLD | NEW |