Support script response files in GN.

components/domain_reliability/bake_in_configs.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.
 
 
 """Takes the JSON files in components/domain_reliability/baked_in_configs and
 encodes their contents as an array of C strings that gets compiled in to Chrome
 and loaded at runtime."""
 
 
 import ast
 import json
+import optparse
 import os
+import shlex
 import sys
 
 
 # A whitelist of domains that the script will accept when baking configs in to
 # Chrome, to ensure incorrect ones are not added accidentally. Subdomains of
 # whitelist entries are also allowed (e.g. maps.google.com, ssl.gstatic.com).
 DOMAIN_WHITELIST = (
   '2mdn.net',
   'admob.biz',
   'admob.co.in',
@@ skipping 443 matching lines @@
 """
 
 
 def read_json_files_from_gypi(gypi_file):
   with open(gypi_file, 'r') as f:
     gypi_text = f.read()
   json_files = ast.literal_eval(gypi_text)['variables']['baked_in_configs']
   return json_files
 
 
+def read_json_files_from_file(list_file):
+  with open(list_file, 'r') as f:
+    list_text = f.read()
+  return shlex.split(list_text)
+
+
 def domain_is_whitelisted(domain):
   return any(domain == e or domain.endswith('.' + e)  for e in DOMAIN_WHITELIST)
 
 
 def quote_and_wrap_text(text, width=79, prefix='  "', suffix='"'):
   max_length = width - len(prefix) - len(suffix)
   output = prefix
   line_length = 0
   for c in text:
     if c == "\"":
       c = "\\\""
     elif c == "\n":
       c = "\\n"
     elif c == "\\":
       c = "\\\\"
     if line_length + len(c) > max_length:
       output += suffix + "\n" + prefix
       line_length = 0
     output += c
     line_length += len(c)
   output += suffix
   return output
 
 
 def main():
-  if len(sys.argv) != 4:
-    print >> sys.stderr, (('Usage: %s <JSON pathname base directory> ' +
-                           '<input .gypi file> <output .cpp file>') %
-                          sys.argv[0])
-    print >> sys.stderr, sys.modules[__name__].__doc__
+  parser = optparse.OptionParser(usage="bake_in_configs.py [options]")
+  parser.add_option("", "--output", metavar="FILE",
+                    help="[Required] Name of the .cc file to write.")
+
+  # For response file reading.
+  parser.add_option("", "--file-list", metavar="FILE",
+                    help="File containing whitespace separated names of "
+                         "the baked in configs files.")
+
+  # For .gypi file reading.
+  parser.add_option("", "--gypi-file", metavar="FILE",
+                    help=".gypi file containing baked_in_configs variable.")
+  parser.add_option("", "--gypi-relative-to", metavar="PATH",
+                    help="Directory the baked_in_configs in the --gypi-file"
+                         "are relative to.""")
+
+  opts, args = parser.parse_args()
+
+  if not opts.output:
+    print >> sys.stderr, "--output argument required"
     return 1
-  json_path = sys.argv[1]
-  gypi_file = sys.argv[2]
-  cpp_file = sys.argv[3]
 
-  json_files = read_json_files_from_gypi(gypi_file)
-  json_files = [ os.path.join(json_path, f) for f in json_files ]
-  json_files = [ os.path.normpath(f) for f in json_files ]
+  if opts.gypi_file:
+    # .gypi-style input.
+    if not opts.gypi_relative_to:
+      print >> sys.stderr, "--gypi-relative-to is required with --gypi-file"
+      return 1
+    json_files = read_json_files_from_gypi(opts.gypi_file)
+    json_files = [ os.path.join(opts.gypi_relative_to, f) for f in json_files ]
+    json_files = [ os.path.normpath(f) for f in json_files ]
+  elif opts.file_list:
+    # Regular file list input.
+    json_files = read_json_files_from_file(opts.file_list)
+  else:
+    print >> sys.stderr, "Either --file-list or --gypi-file is required."
+    return 1
 
   cpp_code = CC_HEADER
   found_invalid_config = False
 
   for json_file in json_files:
     with open(json_file, 'r') as f:
       json_text = f.read()
     try:
       config = json.loads(json_text)
     except ValueError, e:
@@ skipping 16 matching lines @@
 
     cpp_code += "  // " + json_file + ":\n"
     cpp_code += quote_and_wrap_text(dumped_json_text) + ",\n"
     cpp_code += "\n"
 
   cpp_code += CC_FOOTER
 
   if found_invalid_config:
     return 1
 
-  with open(cpp_file, 'wb') as f:
+  with open(opts.output, 'wb') as f:
     f.write(cpp_code)
 
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())