#!/usr/bin/env python
#
# Copyright 2014 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 os
import sys
import zipfile
REPOSITORY_ROOT = os.path.abspath(
os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, os.pardir))
# pylint: disable=wrong-import-position
# pylint: disable=import-error
sys.path.insert(0, os.path.join(REPOSITORY_ROOT, 'build/android/gyp'))
from util import build_utils
import action_helpers # build_utils adds //build to sys.path.
import zip_helpers
# pylint: enable=import-error
# pylint: enable=wrong-import-position
JAVA_PACKAGE_PREFIX = 'org/chromium/'
JNI_ZERO_PACKAGE_PREFIX = 'org/jni_zero/'
def main():
parser = argparse.ArgumentParser()
action_helpers.add_depfile_arg(parser)
parser.add_argument(
'--excluded-classes',
help='A list of .class file patterns to exclude from the jar.')
parser.add_argument(
'--src-search-dirs',
action='append',
help='A list of directories that should be searched'
' for the source files.')
parser.add_argument(
'--src-files', action='append', help='A list of source files to jar.')
parser.add_argument(
'--src-jars',
action='append',
help='A list of source jars to include in addition to source files.')
parser.add_argument(
'--src-list-files',
action='append',
help='A list of files that contain a list of sources,'
' e.g. a list of \'.sources\' files generated by GN.')
parser.add_argument('--jar-path', help='Jar output path.', required=True)
options = parser.parse_args()
options.src_jars = action_helpers.parse_gn_list(options.src_jars)
options.src_search_dirs = action_helpers.parse_gn_list(
options.src_search_dirs)
options.src_list_files = action_helpers.parse_gn_list(options.src_list_files)
options.src_files = action_helpers.parse_gn_list(options.src_files)
options.excluded_classes = action_helpers.parse_gn_list(
options.excluded_classes)
src_files = options.src_files
# Add files from --source_list_files
for src_list_file in options.src_list_files:
with open(src_list_file, 'r') as f:
src_files.extend(f.read().splitlines())
# Preprocess source files by removing any prefix that comes before
# the Java package name.
for i, s in enumerate(src_files):
prefix_position = s.find(JAVA_PACKAGE_PREFIX)
if prefix_position == -1:
prefix_position = s.find(JNI_ZERO_PACKAGE_PREFIX)
if prefix_position != -1:
src_files[i] = s[prefix_position:]
excluded_classes = [
f.replace('.class', '.java') for f in options.excluded_classes
]
predicate = None
if excluded_classes:
predicate = lambda f: not build_utils.MatchesGlob(f, excluded_classes)
# Create a dictionary that maps every source directory
# to source files that it contains.
dir_to_files_map = {}
# Initialize the map.
for src_search_dir in options.src_search_dirs:
dir_to_files_map[src_search_dir] = []
# Fill the map.
for src_file in src_files:
number_of_file_instances = 0
for src_search_dir in options.src_search_dirs:
target_path = os.path.join(src_search_dir, src_file)
if os.path.isfile(target_path):
number_of_file_instances += 1
if not predicate or predicate(src_file):
dir_to_files_map[src_search_dir].append(target_path)
if (number_of_file_instances > 1):
raise Exception('There is more than one instance of file %s in %s' %
(src_file, options.src_search_dirs))
if (number_of_file_instances < 1):
raise Exception('Unable to find file %s in %s' %
(src_file, options.src_search_dirs))
# Jar the sources from every source search directory.
with action_helpers.atomic_output(options.jar_path) as o, \
zipfile.ZipFile(o, 'w', zipfile.ZIP_DEFLATED) as z:
for src_search_dir in options.src_search_dirs:
subpaths = dir_to_files_map[src_search_dir]
if subpaths:
zip_helpers.add_files_to_zip(subpaths, z, base_dir=src_search_dir)
else:
raise Exception(
'Directory %s does not contain any files and can be'
' removed from the list of directories to search' % src_search_dir)
# Jar additional src jars
if options.src_jars:
zip_helpers.merge_zips(z, options.src_jars, compress=True)
if options.depfile:
deps = []
for sources in dir_to_files_map.values():
deps.extend(sources)
# Srcjar deps already captured in GN rules (no need to list them here).
action_helpers.write_depfile(options.depfile, options.jar_path, deps)
if __name__ == '__main__':
sys.exit(main())