Android: Add exclude_xxxhdpi GN parameter to android_apk

build/android/gyp/package_resources.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.
 
 # pylint: disable=C0301
 """Package resources into an apk.
 
 See https://android.googlesource.com/platform/tools/base/+/master/legacy/ant-tasks/src/main/java/com/android/ant/AaptExecTask.java
@@ skipping 117 matching lines @@
       action='store_true',
       help='Enables density splits')
   parser.add_option('--language-splits',
                     default='[]',
                     help='GN list of languages to create splits for')
   parser.add_option('--locale-whitelist',
                     default='[]',
                     help='GN list of languages to include. All other language '
                          'configs will be stripped out. List may include '
                          'a combination of Android locales or Chrome locales.')
-
   parser.add_option('--apk-path',
                     help='Path to output (partial) apk.')
+  parser.add_option('--exclude-xxxhdpi', action='store_true',
+                    help='Do not include xxxhdpi drawables.')
+  parser.add_option('--xxxhdpi-whitelist',
+                    default='[]',
+                    help='GN list of globs that say which xxxhdpi images to '
+                         'include even when --exclude-xxxhdpi is set.')
 
   options, positional_args = parser.parse_args(args)
 
   if positional_args:
     parser.error('No positional arguments should be given.')
 
   # Check that required options have been provided.
   required_options = ('android_sdk_jar', 'aapt_path', 'configuration_name',
                       'android_manifest', 'version_code', 'version_name',
                       'apk_path')
 
   build_utils.CheckOptions(options, parser, required=required_options)
 
   options.resource_zips = build_utils.ParseGnList(options.resource_zips)
   options.language_splits = build_utils.ParseGnList(options.language_splits)
   options.locale_whitelist = build_utils.ParseGnList(options.locale_whitelist)
+  options.xxxhdpi_whitelist = build_utils.ParseGnList(options.xxxhdpi_whitelist)
   return options
 
 
 def _ToAaptLocales(locale_whitelist):
   """Converts the list of Chrome locales to aapt config locales."""
   ret = set()
   for locale in locale_whitelist:
     locale = _CHROME_TO_ANDROID_LOCALE_MAP.get(locale, locale)
     if locale is None or ('-' in locale and '-r' not in locale):
       raise Exception('_CHROME_TO_ANDROID_LOCALE_MAP needs updating.'
@@ skipping 34 matching lines @@
 
 def PackageArgsForExtractedZip(d):
   """Returns the aapt args for an extracted resources zip.
 
   A resources zip either contains the resources for a single target or for
   multiple targets. If it is multiple targets merged into one, the actual
   resource directories will be contained in the subdirectories 0, 1, 2, ...
   """
   subdirs = [os.path.join(d, s) for s in os.listdir(d)]
   subdirs = [s for s in subdirs if os.path.isdir(s)]
-  is_multi = '0' in [os.path.basename(s) for s in subdirs]
+  is_multi = any(os.path.basename(s).isdigit() for s in subdirs)
   if is_multi:
     res_dirs = sorted(subdirs, key=lambda p : int(os.path.basename(p)))
   else:
     res_dirs = [d]
   package_command = []
   for d in res_dirs:
     MoveImagesToNonMdpiFolders(d)
     package_command += ['-S', d]
   return package_command
 
@@ skipping 73 matching lines @@
   if 'Debug' in options.configuration_name:
     package_command += ['--debug-mode']
 
   if options.locale_whitelist:
     aapt_locales = _ToAaptLocales(options.locale_whitelist)
     package_command += ['-c', ','.join(aapt_locales)]
 
   return package_command
 
 
+def _ResourceNameFromPath(path):
+  return os.path.splitext(os.path.basename(path))[0]
+
+
+def _CreateExtractPredicate(dep_zips, exclude_xxxhdpi, xxxhdpi_whitelist):
+  if not exclude_xxxhdpi:
+    # Do not extract dotfiles (e.g. ".gitkeep"). aapt ignores them anyways.
+    return lambda path: os.path.basename(path)[0] != '.'
+
+  # Returns False only for xxxhdpi non-mipmap, non-whitelisted drawables.
+  naive_predicate = lambda path: (
+      not re.search(r'[/-]xxxhdpi[/-]', path) or
+      re.search(r'[/-]mipmap[/-]', path) or
+      build_utils.MatchesGlob(path, xxxhdpi_whitelist))
+
+  # Build a set of all non-xxxhdpi drawables to ensure that we never exclude any
+  # xxxhdpi drawable that does not exist in other densities.
+  non_xxxhdpi_drawables = set()
+  for resource_zip_path in dep_zips:
+    with zipfile.ZipFile(resource_zip_path) as zip_file:
+      for path in zip_file.namelist():
+        if re.search(r'[/-]drawable[/-]', path) and naive_predicate(path):
+          non_xxxhdpi_drawables.add(_ResourceNameFromPath(path))
+
+  return lambda path: (naive_predicate(path) or
+                       _ResourceNameFromPath(path) not in non_xxxhdpi_drawables)
+
+
 def _OnStaleMd5(package_command, options):
   with build_utils.TempDir() as temp_dir:
     if options.resource_zips:
       dep_zips = options.resource_zips
+      extract_predicate = _CreateExtractPredicate(
+          dep_zips, options.exclude_xxxhdpi, options.xxxhdpi_whitelist)
       for z in dep_zips:
         subdir = os.path.join(temp_dir, os.path.basename(z))
         if os.path.exists(subdir):
           raise Exception('Resource zip name conflict: ' + os.path.basename(z))
-        build_utils.ExtractAll(z, path=subdir)
-        package_command += PackageArgsForExtractedZip(subdir)
+        if build_utils.ExtractAll(z, path=subdir, predicate=extract_predicate):
+          package_command += PackageArgsForExtractedZip(subdir)
 
     build_utils.CheckOutput(
         package_command, print_stdout=False, print_stderr=False)
 
     if options.create_density_splits or options.language_splits:
       CheckForMissedConfigs(options.apk_path, options.create_density_splits,
                             options.language_splits)
 
     if options.create_density_splits:
       RenameDensitySplits(options.apk_path)
@@ skipping 33 matching lines @@
   build_utils.CallAndWriteDepfileIfStale(
       lambda: _OnStaleMd5(package_command, options),
       options,
       input_paths=input_paths,
       input_strings=input_strings,
       output_paths=output_paths)
 
 
 if __name__ == '__main__':
   main(sys.argv[1:])