1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#!/usr/bin/env python3
#
# Copyright 2020 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import argparse
import re
import sys
import zipfile
from util import build_utils # pylint: disable=unused-import
from util import java_cpp_utils
import action_helpers # build_utils adds //build to sys.path.
import zip_helpers
class FeatureParserDelegate(java_cpp_utils.CppConstantParser.Delegate):
# Ex. 'BASE_FEATURE(kConstantName, "StringNameOfTheFeature", ...);'
# would parse as:
# ExtractConstantName() -> 'ConstantName'
# ExtractValue() -> '"StringNameOfTheFeature"'
FEATURE_RE = re.compile(r'BASE_FEATURE\(k([^,]+),')
VALUE_RE = re.compile(r'\s*("(?:\"|[^"])*")\s*,')
def ExtractConstantName(self, line):
match = FeatureParserDelegate.FEATURE_RE.match(line)
return match.group(1) if match else None
def ExtractValue(self, line):
match = FeatureParserDelegate.VALUE_RE.search(line)
return match.group(1) if match else None
def CreateJavaConstant(self, name, value, comments):
return java_cpp_utils.JavaString(name, value, comments)
def _GenerateOutput(template, source_paths, features):
description_template = """
// This following string constants were inserted by
// {SCRIPT_NAME}
// From
// {SOURCE_PATHS}
"""
values = {
'SCRIPT_NAME': java_cpp_utils.GetScriptName(),
'SOURCE_PATHS': ',\n // '.join(source_paths),
}
description = description_template.format(**values)
native_features = '\n\n'.join(x.Format() for x in features)
# TODO(agrieve): Remove {{ and }} from input templates.
template = template.replace('{{', '{').replace('}}', '}')
return template.replace('{NATIVE_FEATURES}', description + native_features)
def _ParseFeatureFile(path):
with open(path, encoding='utf-8') as f:
feature_file_parser = java_cpp_utils.CppConstantParser(
FeatureParserDelegate(), f.readlines())
return feature_file_parser.Parse()
def _Generate(source_paths, template):
package, class_name = java_cpp_utils.ParseTemplateFile(template)
output_path = java_cpp_utils.GetJavaFilePath(package, class_name)
features = []
for source_path in source_paths:
features.extend(_ParseFeatureFile(source_path))
output = _GenerateOutput(template, source_paths, features)
return output, output_path
def _Main(argv):
parser = argparse.ArgumentParser()
parser.add_argument('--srcjar',
required=True,
help='The path at which to generate the .srcjar file')
parser.add_argument('--template',
help='The template file with which to generate the Java '
'class. Must have "{NATIVE_FEATURES}" somewhere in '
'the template.')
parser.add_argument('--class-name', help='FQN of Java class to generate')
parser.add_argument('inputs',
nargs='+',
help='Input file(s)',
metavar='INPUTFILE')
args = parser.parse_args(argv)
if args.template:
with open(args.template, encoding='utf-8') as f:
template = f.read()
else:
template = java_cpp_utils.CreateDefaultTemplate(args.class_name,
'{NATIVE_FEATURES}')
with action_helpers.atomic_output(args.srcjar) as f:
with zipfile.ZipFile(f, 'w', zipfile.ZIP_STORED) as srcjar:
data, path = _Generate(args.inputs, template)
zip_helpers.add_to_zip_hermetic(srcjar, path, data=data)
if __name__ == '__main__':
_Main(sys.argv[1:])
|
