Android: convert kEnumName to ENUM_NAME in java_cpp_enum.py.

build/android/gyp/java_cpp_enum.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 collections
 from datetime import date
 import re
 import optparse
@@ skipping 36 matching lines @@
     self.comments[key] = value
 
   @property
   def class_name(self):
     return self.class_name_override or self.original_enum_name
 
   def Finalize(self):
     self._Validate()
     self._AssignEntryIndices()
     self._StripPrefix()
+    self._NormalizeNames()
 
   def _Validate(self):
     assert self.class_name
     assert self.enum_package
     assert self.entries
     if self.fixed_type and self.fixed_type not in ENUM_FIXED_TYPE_WHITELIST:
       raise Exception('Fixed type %s for enum %s not whitelisted.' %
           (self.fixed_type, self.class_name))
 
   def _AssignEntryIndices(self):
@@ skipping 18 matching lines @@
     prefix_to_strip = self.prefix_to_strip
     if not prefix_to_strip:
       prefix_to_strip = self.original_enum_name
       prefix_to_strip = re.sub('(?!^)([A-Z]+)', r'_\1', prefix_to_strip).upper()
       prefix_to_strip += '_'
       if not all([w.startswith(prefix_to_strip) for w in self.entries.keys()]):
         prefix_to_strip = ''
 
     def StripEntries(entries):
       ret = collections.OrderedDict()
-      for (k, v) in entries.iteritems():
+      for k, v in entries.iteritems():
         stripped_key = k.replace(prefix_to_strip, '', 1)
         if isinstance(v, basestring):
-          stripped_value = v.replace(prefix_to_strip, '', 1)
+          stripped_value = v.replace(prefix_to_strip, '')
         else:
           stripped_value = v
         ret[stripped_key] = stripped_value
 
       return ret
 
     self.entries = StripEntries(self.entries)
     self.comments = StripEntries(self.comments)
 
+  def _NormalizeNames(self):
+    self.entries = _TransformKeys(self.entries, _KCamelToShouty)
+    self.comments = _TransformKeys(self.comments, _KCamelToShouty)
+
+
+def _TransformKeys(d, func):
+  """Normalize keys in |d| and update references to old keys in |d| values."""
+  normal_keys = {k: func(k) for k in d}
+  ret = collections.OrderedDict()
+  for k, v in d.iteritems():
+    if isinstance(v, basestring):

[agrieve, 2017/04/13 13:52:34]

nit: can you just mention in a comment here when things are a string vs. not.

[estevenson, 2017/04/13 14:11:14]

Done.

+      ret[normal_keys[k]] = reduce(

[agrieve, 2017/04/13 13:52:34]

nit: this is pretty hard to mentally grok (at least for me). Probably would be
easier to follow if written out as a loop.

[estevenson, 2017/04/13 14:11:13]

Done.

+          lambda x, y: x.replace(y, normal_keys[y]), normal_keys.keys(), v)
+    else:
+      ret[normal_keys[k]] = v
+  return ret
+
+
+def _KCamelToShouty(s):
+  """Convert |s| from kCamelCase or CamelCase to SHOUTY_CASE.
+
+  kFooBar -> FOO_BAR
+  FooBar -> FOO_BAR
+  FooBAR9 -> FOO_BAR9
+  FooBARBaz -> FOO_BAR_BAZ
+  """
+  if not re.match(r'^k?([A-Z][^A-Z]+|[A-Z0-9]+)+$', s):
+    return s
+  # Strip the leading k.
+  s = re.sub(r'^k', '', s)
+  # Add _ between title words and anything else.
+  s = re.sub(r'([^_])([A-Z][^A-Z_0-9]+)', r'\1_\2', s)
+  # Add _ between lower -> upper transitions.
+  s = re.sub(r'([^A-Z_0-9])([A-Z])', r'\1_\2', s)
+  return s.upper()
+
+
 class DirectiveSet(object):
   class_name_override_key = 'CLASS_NAME_OVERRIDE'
   enum_package_key = 'ENUM_PACKAGE'
   prefix_to_strip_key = 'PREFIX_TO_STRIP'
 
   known_keys = [class_name_override_key, enum_package_key, prefix_to_strip_key]
 
   def __init__(self):
     self._directives = {}
 
@@ skipping 279 matching lines @@
   with zipfile.ZipFile(options.srcjar, 'w', zipfile.ZIP_STORED) as srcjar:
     for output_path, data in DoGenerate(input_paths):
       build_utils.AddToZipHermetic(srcjar, output_path, data=data)
 
   if options.depfile:
     build_utils.WriteDepfile(options.depfile, options.srcjar)
 
 
 if __name__ == '__main__':
   DoMain(sys.argv[1:])