Add src_files to src_jar GN template

components/cronet/tools/jar_src.py

 #!/usr/bin/env python
 #
 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import optparse
 import os
 import sys
 import tempfile
 
 REPOSITORY_ROOT = os.path.abspath(os.path.join(
     os.path.dirname(__file__), '..', '..', '..'))
 
 sys.path.append(os.path.join(REPOSITORY_ROOT, 'build/android/gyp/util'))
 import build_utils
 
+JAVA_PACKAGE_PREFIX = 'org/chromium/'
 
-def JarSources(src_dir, jar_path):
+def JarSources(src_dir, src_files, jar_path):
   # The paths of the files in the jar will be the same as they are passed in to
   # the command. Because of this, the command should be run in
   # options.src_dir so the .java file paths in the jar are correct.
   jar_cwd = src_dir
   jar_path = os.path.abspath(jar_path)
   if os.path.exists(jar_path):
-    jar_cmd = ['jar', 'uf', jar_path, '.']
+    jar_cmd = ['jar', 'uf', jar_path]
   else:
-    jar_cmd = ['jar', 'cf', jar_path, '.']
+    jar_cmd = ['jar', 'cf', jar_path]
 
+  jar_cmd.extend(src_files)
   build_utils.CheckOutput(jar_cmd, cwd=jar_cwd)
 
 # Uncompress source jars so that they can be combined with other sources
 def UnzipSourceJar(jar, unzipped_jar_path):
   if os.path.exists(jar):
     jar_cmd = ['jar', 'xf', os.path.abspath(jar)]
     build_utils.CheckOutput(jar_cmd, cwd=unzipped_jar_path)
   else:
     raise Exception('Jar file does not exist %s' % jar)
 
 
 def main():
   parser = optparse.OptionParser()
   build_utils.AddDepfileOption(parser)
-  parser.add_option('--src-dir', action="append",
-      help='Directory containing .java files.')
+  parser.add_option('--src-search-dirs', action="append",
+      help='A list of directories that should be searched '
+           'for the source files.')
+  parser.add_option('--src-files', action="append",
+      help='A list of source files to jar.')
   parser.add_option('--src-jars', action="append",
       help='A list of source jars to include in addition to source files.')
   parser.add_option('--jar-path', help='Jar output path.')
   parser.add_option('--stamp', help='Path to touch on success.')
 
   options, _ = parser.parse_args()
 
   # A temporary directory to put the output of jar files.
   unzipped_jar_path = None
-  generated_src_dirs = []
   if options.src_jars:
     unzipped_jar_path = tempfile.mkdtemp(dir=os.path.dirname(options.jar_path))
-    generated_src_dirs.append(unzipped_jar_path)
     jar_list = []
     for gn_list in options.src_jars:
       jar_list.extend(build_utils.ParseGnList(gn_list))
 
     for jar in jar_list:
       UnzipSourceJar(jar, unzipped_jar_path)
 
-  src_dirs = []
-  for src_dir in options.src_dir:
-    src_dirs.extend(build_utils.ParseGnList(src_dir))
+  src_search_dirs = []
+  for gn_src_search_dirs in options.src_search_dirs:
+    src_search_dirs.extend(build_utils.ParseGnList(gn_src_search_dirs))
 
-  for src_dir in src_dirs + generated_src_dirs:
-    JarSources(src_dir, options.jar_path)
+  src_files = []
+  for gn_src_files in options.src_files:
+    src_files.extend(build_utils.ParseGnList(gn_src_files))
+
+  # 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:
+      src_files[i] = s[prefix_position:]
+
+  # 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 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 src_search_dirs:
+      if os.path.isfile(os.path.join(src_search_dir, src_file)):
+        number_of_file_instances += 1
+        dir_to_files_map[src_search_dir].append(src_file)
+    if (number_of_file_instances > 1):
+      raise Exception(
+          'There is more than one instance of file %s in %s'
+          % (src_file, src_search_dirs))
+    if (number_of_file_instances < 1):
+      raise Exception(
+          'Unable to find file %s in %s' % (src_file, src_search_dirs))
+
+  # Jar the sources from every source search directory.
+  for src_search_dir in src_search_dirs:
+    if len(dir_to_files_map[src_search_dir]) > 0:
+      JarSources(src_search_dir, dir_to_files_map[src_search_dir],
+                 options.jar_path)
+    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 unzipped_jar_path:
+    JarSources(unzipped_jar_path, ['.'], options.jar_path)
 
   if options.depfile:
     deps = []
-    for src_dir in src_dirs:
+    for src_dir in src_search_dirs:
       for root, _, filenames in os.walk(src_dir):
         deps.extend(os.path.join(root, f) for f in filenames)
     # Srcjar deps already captured in GN rules (no need to list them here).
     build_utils.WriteDepfile(options.depfile, options.jar_path, deps)
 
   # Clean up temporary output directory.
   if unzipped_jar_path:
     build_utils.DeleteDirectory(unzipped_jar_path)
 
   if options.stamp:
     build_utils.Touch(options.stamp)
 
 if __name__ == '__main__':
   sys.exit(main())