chrome/test/chromedriver/embed_networks_in_cpp.py

   1 #!/usr/bin/env python
   2 # Copyright 2015 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Embeds standalone JavaScript snippets in C++ code.
   7
   8 The script requires the devtools/front_end/toolbox/OverridesUI.js file from
   9 WebKit that lists the preset network conditions to be passed in as the only
  10 argument. The list of network conditions will be written to a C-style string to
  11 be parsed with JSONReader.
  12 """
  13
  14 import optparse
  15 import re
  16 import subprocess
  17 import sys
  18
  19 import cpp_source
  20
  21 UNLIMITED_THROUGHPUT = ('WebInspector.OverridesSupport'
  22                         '.NetworkThroughputUnlimitedValue')
  23
  24
  25 def quotizeKeys(s, keys):
  26   """Returns the string s with each instance of each key wrapped in quotes.
  27
  28   Args:
  29     s: a string containing keys that need to be wrapped in quotes.
  30     keys: an iterable of keys to be wrapped in quotes in the string.
  31   """
  32   for key in keys:
  33     s = re.sub('%s: ' % key, '"%s": ' % key, s)
  34   return s
  35
  36
  37 def evaluateMultiplications(s):
  38   """Returns the string s with each bare multiplication evaluated.
  39
  40   Since the source is JavaScript, which includes bare arithmetic, and the
  41   output must be JSON format, we must evaluate all expressions.
  42
  43   Args:
  44     s: a string containing bare multiplications that need to be evaluated.
  45   """
  46   def evaluateBinaryMultiplication(match):
  47     return str(float(match.group(1)) * float(match.group(2)))
  48
  49   return re.sub('([0-9\.]+) \* ([0-9\.]+)', evaluateBinaryMultiplication, s)
  50
  51
  52 def main():
  53   parser = optparse.OptionParser()
  54   parser.add_option(
  55       '', '--directory', type='string', default='.',
  56       help='Path to directory where the cc/h files should be created')
  57   options, args = parser.parse_args()
  58
  59   networks = '['
  60   file_name = args[0]
  61   inside_list = False
  62   with open(file_name, 'r') as f:
  63     for line in f:
  64       if not inside_list:
  65         if 'WebInspector.OverridesUI._networkConditionsPresets = [' in line:
  66           inside_list = True
  67       else:
  68         if line.strip() == '];':
  69           inside_list = False
  70           continue
  71         line = line.replace(UNLIMITED_THROUGHPUT, "-1")
  72         networks += line.strip()
  73
  74   output_dir = 'chrome/test/chromedriver/chrome'
  75   networks += ']'
  76   networks = quotizeKeys(networks, ['id', 'title', 'throughput', 'latency'])
  77   networks = evaluateMultiplications(networks)
  78   cpp_source.WriteSource('network_list',
  79                          output_dir,
  80                          options.directory, {'kNetworks': networks})
  81
  82   clang_format = ['clang-format', '-i']
  83   subprocess.Popen(clang_format + ['%s/network_list.cc' % output_dir])
  84   subprocess.Popen(clang_format + ['%s/network_list.h' % output_dir])
  85
  86
  87 if __name__ == '__main__':
  88   sys.exit(main())