Extend tweak_info_plist template to support multiple plist file as input.

build/config/mac/plist_util.py

 # Copyright 2016 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 argparse
 import plistlib
 import os
 import re
 import subprocess
 import sys
 import tempfile
 import shlex
 
 
 # Xcode substitutes variables like ${PRODUCT_NAME} when compiling Info.plist.
 # It also supports supports modifiers like :identifier or :rfc1034identifier.
 # SUBST_RE matches a variable substitution pattern with an optional modifier,
 # while IDENT_RE matches all characters that are not valid in an "identifier"
 # value (used when applying the modifier).
 SUBST_RE = re.compile(r'\$\{(?P<id>[^}]*?)(?P<modifier>:[^}]*)?\}')
 IDENT_RE = re.compile(r'[_/\s]')
 
 
-class ArgumentParser(argparse.ArgumentParser):
-  """Subclass of argparse.ArgumentParser to work with GN response files.
-
-  GN response file writes all the arguments on a single line and assumes
-  that the python script uses shlext.split() to extract them. Since the
-  default ArgumentParser expects a single argument per line, we need to
-  provide a subclass to have the correct support for @{{response_file_name}}.
-  """
-
-  def convert_arg_line_to_args(self, arg_line):
-    return shlex.split(arg_line)
-
-
 def InterpolateList(values, substitutions):
   """Interpolates variable references into |value| using |substitutions|.
 
   Inputs:
     values: a list of values
     substitutions: a mapping of variable names to values
 
   Returns:
     A new list of values with all variables references ${VARIABLE} replaced
     by their value in |substitutions| or None if any of the variable has no
@@ skipping 128 matching lines @@
     if key in plist2:
       value = plist2[key]
     else:
       value = plist1[key]
     if isinstance(value, dict):
       value = MergePList(plist1.get(key, None), plist2.get(key, None))
     result[key] = value
   return result
 
 
-def main():
-  parser = ArgumentParser(
-      description='A script to generate iOS application Info.plist.',
-      fromfile_prefix_chars='@')
-  parser.add_argument('-o', '--output', required=True,
-                      help='Path to output plist file.')
-  parser.add_argument('-s', '--subst', action='append', default=[],
-                      help='Substitution rule in the format "key=value".')
-  parser.add_argument('-f', '--format', required=True,
-                      help='Plist format (e.g. binary1, xml1) to output.')
-  parser.add_argument('path', nargs="+", help='Path to input plist files.')
+class Action(object):
+  """Class implementing one action supported by the script."""
+
+  @classmethod
+  def Register(cls, subparsers):
+    parser = subparsers.add_parser(cls.name, help=cls.help)
+    parser.set_defaults(func=cls._Execute)
+    cls._Register(parser)
+
+
+class MergeAction(Action):
+  """Class to merge multiple plist files."""
+
+  name = 'merge'
+  help = 'merge multiple plist files'
+
+  @staticmethod
+  def _Register(parser):
+    parser.add_argument(
+        '-o', '--output', required=True,
+        help='path to the output plist file')
+    parser.add_argument(
+        '-f', '--format', required=True, choices=('xml1', 'binary1', 'json'),
+        help='format of the plist file to generate')
+    parser.add_argument(
+          'path', nargs="+",
+          help='path to plist files to merge')
+
+  @staticmethod
+  def _Execute(args):
+    data = {}
+    for filename in args.path:
+      data = MergePList(data, LoadPList(filename))
+    SavePList(args.output, args.format, data)
+
+
+class SubstituteAction(Action):
+  """Class implementing the variable substitution in a plist file."""
+
+  name = 'substitute'
+  help = 'perform pattern substitution in a plist file'
+
+  @staticmethod
+  def _Register(parser):
+    parser.add_argument(
+        '-o', '--output', required=True,
+        help='path to the output plist file')
+    parser.add_argument(
+        '-t', '--template', required=True,
+        help='path to the template file')
+    parser.add_argument(
+        '-s', '--substitution', action='append', default=[],
+        help='substitution rule in the format key=value')
+    parser.add_argument(
+        '-f', '--format', required=True, choices=('xml1', 'binary1', 'json'),
+        help='format of the plist file to generate')
+
+  @staticmethod
+  def _Execute(args):
+    substitutions = {}
+    for substitution in args.substitution:
+      key, value = substitution.split('=', 1)
+      substitutions[key] = value
+    data = Interpolate(LoadPList(args.template), substitutions)
+    SavePList(args.output, args.format, data)
+
+
+def Main():
+  parser = argparse.ArgumentParser(description='manipulate plist files')
+  subparsers = parser.add_subparsers()
+
+  for action in [MergeAction, SubstituteAction]:
+    action.Register(subparsers)
+
   args = parser.parse_args()
-  substitutions = {}
-  for subst in args.subst:
-    key, value = subst.split('=', 1)
-    substitutions[key] = value
-  data = {}
-  for filename in args.path:
-    data = MergePList(data, LoadPList(filename))
-  data = Interpolate(data, substitutions)
-  SavePList(args.output, args.format, data)
-  return 0
+  args.func(args)
+
 
 if __name__ == '__main__':
-  sys.exit(main())
+  sys.exit(Main())