1*333d2b36SAndroid Build Coastguard Worker#!/usr/bin/env python3 2*333d2b36SAndroid Build Coastguard Worker# 3*333d2b36SAndroid Build Coastguard Worker# Copyright (C) 2018 The Android Open Source Project 4*333d2b36SAndroid Build Coastguard Worker# 5*333d2b36SAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License"); 6*333d2b36SAndroid Build Coastguard Worker# you may not use this file except in compliance with the License. 7*333d2b36SAndroid Build Coastguard Worker# You may obtain a copy of the License at 8*333d2b36SAndroid Build Coastguard Worker# 9*333d2b36SAndroid Build Coastguard Worker# http://www.apache.org/licenses/LICENSE-2.0 10*333d2b36SAndroid Build Coastguard Worker# 11*333d2b36SAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software 12*333d2b36SAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS, 13*333d2b36SAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14*333d2b36SAndroid Build Coastguard Worker# See the License for the specific language governing permissions and 15*333d2b36SAndroid Build Coastguard Worker# limitations under the License. 16*333d2b36SAndroid Build Coastguard Worker# 17*333d2b36SAndroid Build Coastguard Worker 18*333d2b36SAndroid Build Coastguard Worker"""This file generates project.xml and lint.xml files used to drive the Android Lint CLI tool.""" 19*333d2b36SAndroid Build Coastguard Worker 20*333d2b36SAndroid Build Coastguard Workerimport argparse 21*333d2b36SAndroid Build Coastguard Workerimport sys 22*333d2b36SAndroid Build Coastguard Workerfrom xml.dom import minidom 23*333d2b36SAndroid Build Coastguard Worker 24*333d2b36SAndroid Build Coastguard Workerfrom ninja_rsp import NinjaRspFileReader 25*333d2b36SAndroid Build Coastguard Worker 26*333d2b36SAndroid Build Coastguard Worker 27*333d2b36SAndroid Build Coastguard Workerdef check_action(check_type): 28*333d2b36SAndroid Build Coastguard Worker """ 29*333d2b36SAndroid Build Coastguard Worker Returns an action that appends a tuple of check_type and the argument to the dest. 30*333d2b36SAndroid Build Coastguard Worker """ 31*333d2b36SAndroid Build Coastguard Worker class CheckAction(argparse.Action): 32*333d2b36SAndroid Build Coastguard Worker def __init__(self, option_strings, dest, nargs=None, **kwargs): 33*333d2b36SAndroid Build Coastguard Worker if nargs is not None: 34*333d2b36SAndroid Build Coastguard Worker raise ValueError("nargs must be None, was %s" % nargs) 35*333d2b36SAndroid Build Coastguard Worker super(CheckAction, self).__init__(option_strings, dest, **kwargs) 36*333d2b36SAndroid Build Coastguard Worker def __call__(self, parser, namespace, values, option_string=None): 37*333d2b36SAndroid Build Coastguard Worker checks = getattr(namespace, self.dest, []) 38*333d2b36SAndroid Build Coastguard Worker checks.append((check_type, values)) 39*333d2b36SAndroid Build Coastguard Worker setattr(namespace, self.dest, checks) 40*333d2b36SAndroid Build Coastguard Worker return CheckAction 41*333d2b36SAndroid Build Coastguard Worker 42*333d2b36SAndroid Build Coastguard Worker 43*333d2b36SAndroid Build Coastguard Workerdef parse_args(): 44*333d2b36SAndroid Build Coastguard Worker """Parse commandline arguments.""" 45*333d2b36SAndroid Build Coastguard Worker 46*333d2b36SAndroid Build Coastguard Worker def convert_arg_line_to_args(arg_line): 47*333d2b36SAndroid Build Coastguard Worker for arg in arg_line.split(): 48*333d2b36SAndroid Build Coastguard Worker if arg.startswith('#'): 49*333d2b36SAndroid Build Coastguard Worker return 50*333d2b36SAndroid Build Coastguard Worker if not arg.strip(): 51*333d2b36SAndroid Build Coastguard Worker continue 52*333d2b36SAndroid Build Coastguard Worker yield arg 53*333d2b36SAndroid Build Coastguard Worker 54*333d2b36SAndroid Build Coastguard Worker parser = argparse.ArgumentParser(fromfile_prefix_chars='@') 55*333d2b36SAndroid Build Coastguard Worker parser.convert_arg_line_to_args = convert_arg_line_to_args 56*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--project_out', dest='project_out', 57*333d2b36SAndroid Build Coastguard Worker help='file to which the project.xml contents will be written.') 58*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--config_out', dest='config_out', 59*333d2b36SAndroid Build Coastguard Worker help='file to which the lint.xml contents will be written.') 60*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--name', dest='name', 61*333d2b36SAndroid Build Coastguard Worker help='name of the module.') 62*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--srcs', dest='srcs', action='append', default=[], 63*333d2b36SAndroid Build Coastguard Worker help='file containing whitespace separated list of source files.') 64*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--generated_srcs', dest='generated_srcs', action='append', default=[], 65*333d2b36SAndroid Build Coastguard Worker help='file containing whitespace separated list of generated source files.') 66*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--resources', dest='resources', action='append', default=[], 67*333d2b36SAndroid Build Coastguard Worker help='file containing whitespace separated list of resource files.') 68*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--classes', dest='classes', action='append', default=[], 69*333d2b36SAndroid Build Coastguard Worker help='file containing the module\'s classes.') 70*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--classpath', dest='classpath', action='append', default=[], 71*333d2b36SAndroid Build Coastguard Worker help='file containing classes from dependencies.') 72*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--extra_checks_jar', dest='extra_checks_jars', action='append', default=[], 73*333d2b36SAndroid Build Coastguard Worker help='file containing extra lint checks.') 74*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--manifest', dest='manifest', 75*333d2b36SAndroid Build Coastguard Worker help='file containing the module\'s manifest.') 76*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--merged_manifest', dest='merged_manifest', 77*333d2b36SAndroid Build Coastguard Worker help='file containing merged manifest for the module and its dependencies.') 78*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--library', dest='library', action='store_true', 79*333d2b36SAndroid Build Coastguard Worker help='mark the module as a library.') 80*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--test', dest='test', action='store_true', 81*333d2b36SAndroid Build Coastguard Worker help='mark the module as a test.') 82*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--cache_dir', dest='cache_dir', 83*333d2b36SAndroid Build Coastguard Worker help='directory to use for cached file.') 84*333d2b36SAndroid Build Coastguard Worker parser.add_argument('--root_dir', dest='root_dir', 85*333d2b36SAndroid Build Coastguard Worker help='directory to use for root dir.') 86*333d2b36SAndroid Build Coastguard Worker group = parser.add_argument_group('check arguments', 'later arguments override earlier ones.') 87*333d2b36SAndroid Build Coastguard Worker group.add_argument('--fatal_check', dest='checks', action=check_action('fatal'), default=[], 88*333d2b36SAndroid Build Coastguard Worker help='treat a lint issue as a fatal error.') 89*333d2b36SAndroid Build Coastguard Worker group.add_argument('--error_check', dest='checks', action=check_action('error'), default=[], 90*333d2b36SAndroid Build Coastguard Worker help='treat a lint issue as an error.') 91*333d2b36SAndroid Build Coastguard Worker group.add_argument('--warning_check', dest='checks', action=check_action('warning'), default=[], 92*333d2b36SAndroid Build Coastguard Worker help='treat a lint issue as a warning.') 93*333d2b36SAndroid Build Coastguard Worker group.add_argument('--disable_check', dest='checks', action=check_action('ignore'), default=[], 94*333d2b36SAndroid Build Coastguard Worker help='disable a lint issue.') 95*333d2b36SAndroid Build Coastguard Worker return parser.parse_args() 96*333d2b36SAndroid Build Coastguard Worker 97*333d2b36SAndroid Build Coastguard Worker 98*333d2b36SAndroid Build Coastguard Workerdef write_project_xml(f, args): 99*333d2b36SAndroid Build Coastguard Worker test_attr = "test='true' " if args.test else "" 100*333d2b36SAndroid Build Coastguard Worker 101*333d2b36SAndroid Build Coastguard Worker f.write("<?xml version='1.0' encoding='utf-8'?>\n") 102*333d2b36SAndroid Build Coastguard Worker f.write("<project>\n") 103*333d2b36SAndroid Build Coastguard Worker if args.root_dir: 104*333d2b36SAndroid Build Coastguard Worker f.write(" <root dir='%s' />\n" % args.root_dir) 105*333d2b36SAndroid Build Coastguard Worker f.write(" <module name='%s' android='true' %sdesugar='full' >\n" % (args.name, "library='true' " if args.library else "")) 106*333d2b36SAndroid Build Coastguard Worker if args.manifest: 107*333d2b36SAndroid Build Coastguard Worker f.write(" <manifest file='%s' %s/>\n" % (args.manifest, test_attr)) 108*333d2b36SAndroid Build Coastguard Worker if args.merged_manifest: 109*333d2b36SAndroid Build Coastguard Worker f.write(" <merged-manifest file='%s' %s/>\n" % (args.merged_manifest, test_attr)) 110*333d2b36SAndroid Build Coastguard Worker for src_file in args.srcs: 111*333d2b36SAndroid Build Coastguard Worker for src in NinjaRspFileReader(src_file): 112*333d2b36SAndroid Build Coastguard Worker f.write(" <src file='%s' %s/>\n" % (src, test_attr)) 113*333d2b36SAndroid Build Coastguard Worker for src_file in args.generated_srcs: 114*333d2b36SAndroid Build Coastguard Worker for src in NinjaRspFileReader(src_file): 115*333d2b36SAndroid Build Coastguard Worker f.write(" <src file='%s' generated='true' %s/>\n" % (src, test_attr)) 116*333d2b36SAndroid Build Coastguard Worker for res_file in args.resources: 117*333d2b36SAndroid Build Coastguard Worker for res in NinjaRspFileReader(res_file): 118*333d2b36SAndroid Build Coastguard Worker f.write(" <resource file='%s' %s/>\n" % (res, test_attr)) 119*333d2b36SAndroid Build Coastguard Worker for classes in args.classes: 120*333d2b36SAndroid Build Coastguard Worker f.write(" <classes jar='%s' />\n" % classes) 121*333d2b36SAndroid Build Coastguard Worker for classpath in args.classpath: 122*333d2b36SAndroid Build Coastguard Worker f.write(" <classpath jar='%s' />\n" % classpath) 123*333d2b36SAndroid Build Coastguard Worker for extra in args.extra_checks_jars: 124*333d2b36SAndroid Build Coastguard Worker f.write(" <lint-checks jar='%s' />\n" % extra) 125*333d2b36SAndroid Build Coastguard Worker f.write(" </module>\n") 126*333d2b36SAndroid Build Coastguard Worker if args.cache_dir: 127*333d2b36SAndroid Build Coastguard Worker f.write(" <cache dir='%s'/>\n" % args.cache_dir) 128*333d2b36SAndroid Build Coastguard Worker f.write("</project>\n") 129*333d2b36SAndroid Build Coastguard Worker 130*333d2b36SAndroid Build Coastguard Worker 131*333d2b36SAndroid Build Coastguard Workerdef write_config_xml(f, args): 132*333d2b36SAndroid Build Coastguard Worker f.write("<?xml version='1.0' encoding='utf-8'?>\n") 133*333d2b36SAndroid Build Coastguard Worker f.write("<lint>\n") 134*333d2b36SAndroid Build Coastguard Worker for check in args.checks: 135*333d2b36SAndroid Build Coastguard Worker f.write(" <issue id='%s' severity='%s' />\n" % (check[1], check[0])) 136*333d2b36SAndroid Build Coastguard Worker f.write("</lint>\n") 137*333d2b36SAndroid Build Coastguard Worker 138*333d2b36SAndroid Build Coastguard Worker 139*333d2b36SAndroid Build Coastguard Workerdef main(): 140*333d2b36SAndroid Build Coastguard Worker """Program entry point.""" 141*333d2b36SAndroid Build Coastguard Worker args = parse_args() 142*333d2b36SAndroid Build Coastguard Worker 143*333d2b36SAndroid Build Coastguard Worker if args.project_out: 144*333d2b36SAndroid Build Coastguard Worker with open(args.project_out, 'w') as f: 145*333d2b36SAndroid Build Coastguard Worker write_project_xml(f, args) 146*333d2b36SAndroid Build Coastguard Worker 147*333d2b36SAndroid Build Coastguard Worker if args.config_out: 148*333d2b36SAndroid Build Coastguard Worker with open(args.config_out, 'w') as f: 149*333d2b36SAndroid Build Coastguard Worker write_config_xml(f, args) 150*333d2b36SAndroid Build Coastguard Worker 151*333d2b36SAndroid Build Coastguard Worker 152*333d2b36SAndroid Build Coastguard Workerif __name__ == '__main__': 153*333d2b36SAndroid Build Coastguard Worker main() 154