OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # Copyright 2014 The Chromium Authors. All rights reserved. | |
3 # Use of this source code is governed by a BSD-style license that can be | |
4 # found in the LICENSE file. | |
5 | |
6 """Transforms direct Java class references in Android layout .xml files | |
7 according to the specified JarJar rules.""" | |
8 | |
9 import optparse | |
10 import os | |
11 import shutil | |
12 import sys | |
13 from xml.dom import minidom | |
14 | |
15 from util import build_utils | |
16 | |
17 | |
18 class JarJarRules(object): | |
19 def __init__(self, jarjar_rules): | |
20 self._rules = [] | |
21 for line in jarjar_rules.splitlines(): | |
22 rule = line.split() | |
23 if rule[0] != 'rule': | |
24 continue | |
25 _, src, dest = rule | |
26 if src.endswith('**'): | |
27 src_real_name = src[:-2] | |
28 else: | |
29 assert not '*' in src | |
30 src_real_name = src | |
31 | |
32 if dest.endswith('@0'): | |
33 self._rules.append((src, dest[:-2] + src_real_name)) | |
34 elif dest.endswith('@1'): | |
35 assert '**' in src | |
36 self._rules.append((src, dest[:-2])) | |
37 else: | |
38 assert not '@' in dest | |
39 self._rules.append((src, dest)) | |
40 | |
41 def RenameClass(self, class_name): | |
42 for old, new in self._rules: | |
43 if old.endswith('**') and old[:-2] in class_name: | |
44 return class_name.replace(old[:-2], new, 1) | |
45 if '*' not in old and class_name.endswith(old): | |
46 return class_name.replace(old, new, 1) | |
47 return class_name | |
48 | |
49 | |
50 def RenameNodes(node, rules): | |
51 if node.nodeType == node.ELEMENT_NODE: | |
52 if node.tagName.lower() == 'view' and node.attributes.has_key('class'): | |
53 node.attributes['class'] = rules.RenameClass(node.attributes['class']) | |
54 else: | |
55 node.tagName = rules.RenameClass(node.tagName) | |
56 for child in node.childNodes: | |
57 RenameNodes(child, rules) | |
58 | |
59 | |
60 def ProcessLayoutFile(path, rules): | |
61 xmldoc = minidom.parse(path) | |
62 RenameNodes(xmldoc.documentElement, rules) | |
63 with open(path, 'w') as f: | |
64 xmldoc.writexml(f) | |
65 | |
66 | |
67 def LayoutFilesFilter(src, names): | |
68 if os.path.basename(src).lower() != 'layout': | |
69 return [] | |
70 else: | |
71 return filter(lambda n: n.endswith('.xml'), names) | |
72 | |
73 | |
74 def ProcessResources(options): | |
75 with open(options.rules_path) as f: | |
76 rules = JarJarRules(f.read()) | |
77 | |
78 build_utils.DeleteDirectory(options.output_dir) | |
79 for input_dir in options.input_dir: | |
80 shutil.copytree(input_dir, options.output_dir) | |
81 | |
82 for root, _dirnames, filenames in os.walk(options.output_dir): | |
83 layout_files = LayoutFilesFilter(root, filenames) | |
84 for layout_file in layout_files: | |
85 ProcessLayoutFile(os.path.join(root, layout_file), rules) | |
86 | |
87 | |
88 def ParseArgs(): | |
89 parser = optparse.OptionParser() | |
90 parser.add_option('--input-dir', action='append', | |
91 help='Path to the resources folder to process.') | |
92 parser.add_option('--output-dir', | |
93 help=('Directory to hold processed resources. Note: the ' + | |
94 'directory will be clobbered on every invocation.')) | |
95 parser.add_option('--rules-path', | |
96 help='Path to the jarjar rules file.') | |
97 parser.add_option('--stamp', help='Path to touch on success.') | |
98 | |
99 options, args = parser.parse_args() | |
100 | |
101 if args: | |
102 parser.error('No positional arguments should be given.') | |
103 | |
104 # Check that required options have been provided. | |
105 required_options = ('input_dir', 'output_dir', 'rules_path') | |
106 build_utils.CheckOptions(options, parser, required=required_options) | |
107 | |
108 return options | |
109 | |
110 | |
111 def main(): | |
112 options = ParseArgs() | |
113 | |
114 ProcessResources(options) | |
115 | |
116 if options.stamp: | |
117 build_utils.Touch(options.stamp) | |
118 | |
119 | |
120 if __name__ == '__main__': | |
121 sys.exit(main()) | |
OLD | NEW |