OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright 2015 Google Inc. | 3 # Copyright 2015 Google Inc. |
4 # | 4 # |
5 # Use of this source code is governed by a BSD-style license that can be | 5 # Use of this source code is governed by a BSD-style license that can be |
6 # found in the LICENSE file. | 6 # found in the LICENSE file. |
7 | 7 |
8 # This script does a very rough simulation of BUILD file expansion, | 8 # This script does a very rough simulation of BUILD file expansion, |
9 # mostly to see the effects of glob(). | 9 # mostly to see the effects of glob(). |
10 | 10 |
11 # We start by adding some symbols to our namespace that BUILD.public calls. | 11 # We start by adding some symbols to our namespace that BUILD.public calls. |
12 | 12 |
13 import glob | 13 import glob |
| 14 import os |
14 import pprint | 15 import pprint |
| 16 import re |
15 | 17 |
16 def noop(*args, **kwargs): | 18 def noop(*args, **kwargs): |
17 pass | 19 pass |
18 | 20 |
| 21 DOUBLE_STAR_RE = re.compile(r'/\*\*/') |
| 22 STAR_RE = re.compile(r'\*') |
| 23 DOUBLE_STAR_PLACEHOLDER = "xxxdoublestarxxx" |
| 24 STAR_PLACEHOLDER = "xxxstarxxx" |
| 25 |
| 26 # Returns a set of files that match pattern. |
| 27 def BUILD_glob_single(pattern): |
| 28 if pattern.find('**') < 0: |
| 29 # If pattern doesn't include **, glob.glob more-or-less does the right |
| 30 # thing. |
| 31 return glob.glob(pattern) |
| 32 # First transform pattern into a regexp. |
| 33 # Temporarily remove ** and *. |
| 34 pattern2 = DOUBLE_STAR_RE.sub(DOUBLE_STAR_PLACEHOLDER, pattern) |
| 35 pattern3 = STAR_RE.sub(STAR_PLACEHOLDER, pattern2) |
| 36 # Replace any regexp special characters. |
| 37 pattern4 = re.escape(pattern3) |
| 38 # Replace * with [^/]* and ** with .*. |
| 39 pattern5 = pattern4.replace(STAR_PLACEHOLDER, '[^/]*') |
| 40 pattern6 = pattern5.replace(DOUBLE_STAR_PLACEHOLDER, '.*/') |
| 41 # Anchor the match at the beginning and end. |
| 42 pattern7 = "^" + pattern6 + "$" |
| 43 pattern_re = re.compile(pattern7) |
| 44 matches = set() |
| 45 for root, _, files in os.walk('.'): |
| 46 for fname in files: |
| 47 # Remove initial "./". |
| 48 path = os.path.join(root, fname)[2:] |
| 49 if pattern_re.match(path): |
| 50 matches.add(path) |
| 51 return matches |
| 52 |
19 # Simulates BUILD file glob(). | 53 # Simulates BUILD file glob(). |
20 def BUILD_glob(include, exclude=()): | 54 def BUILD_glob(include, exclude=()): |
21 files = set() | 55 files = set() |
22 for pattern in include: | 56 for pattern in include: |
23 files.update(glob.glob(pattern)) | 57 files.update(BUILD_glob_single(pattern)) |
24 for pattern in exclude: | 58 for pattern in exclude: |
25 files.difference_update(glob.glob(pattern)) | 59 files.difference_update(BUILD_glob_single(pattern)) |
26 return list(sorted(files)) | 60 return list(sorted(files)) |
27 | 61 |
28 # With these namespaces, we can treat BUILD.public as if it were | 62 # With these namespaces, we can treat BUILD.public as if it were |
29 # Python code. This pulls its variable definitions (SRCS, HDRS, | 63 # Python code. This pulls its variable definitions (SRCS, HDRS, |
30 # DEFINES, etc.) into local_names. | 64 # DEFINES, etc.) into local_names. |
31 global_names = { | 65 global_names = { |
32 'exports_files': noop, | 66 'exports_files': noop, |
| 67 'cc_library': noop, |
| 68 'cc_test': noop, |
33 'glob': BUILD_glob, | 69 'glob': BUILD_glob, |
| 70 'EXTERNAL_DEPS': [], |
| 71 'BASE_DIR': "", |
| 72 'DM_EXTERNAL_DEPS': [], |
34 } | 73 } |
35 local_names = {} | 74 local_names = {} |
36 execfile('BUILD.public', global_names, local_names) | 75 execfile('BUILD.public', global_names, local_names) |
37 | 76 |
38 with open('tools/BUILD.public.expected', 'w') as out: | 77 with open('tools/BUILD.public.expected', 'w') as out: |
39 print >>out, "This file is auto-generated by tools/BUILD_simulator.py." | 78 print >>out, "This file is auto-generated by tools/BUILD_simulator.py." |
40 print >>out, "It expands BUILD.public to make it easy to see changes." | 79 print >>out, "It expands BUILD.public to make it easy to see changes." |
41 for name, value in sorted(local_names.items()): | 80 for name, value in sorted(local_names.items()): |
42 print >>out, name, '= ', | 81 print >>out, name, '= ', |
43 pprint.pprint(value, out) | 82 pprint.pprint(value, out) |
OLD | NEW |