Allow installer::GetLocalizedString to return mode-specific strings.

chrome/installer/util/prebuild/create_string_rc.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """Generates .h and .rc files for installer strings. Run "python
 create_string_rc.py" for usage details.
 
 This script generates an rc file and header (NAME.{rc,h}) to be included in
 setup.exe. The rc file includes translations for strings pulled from the given
@@ skipping 12 matching lines @@
 #define IDS_MY_STRING_AR 1600
 #define IDS_MY_STRING_BG 1601
 ...
 #define IDS_MY_STRING_BASE IDS_MY_STRING_AR
 
 This allows us to lookup an an ID for a string by adding IDS_MY_STRING_BASE and
 IDS_L10N_OFFSET_* for the language we are interested in.
 """
 
 import argparse
+import exceptions
 import glob
 import io
 import os
 import sys
 from xml import sax
 
 BASEDIR = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(os.path.join(BASEDIR, '../../../../tools/grit'))
 sys.path.append(os.path.join(BASEDIR, '../../../../tools/python'))
 
 from grit.extern import tclib
 
 # The IDs of strings we want to import from the .grd files and include in
 # setup.exe's resources.
 STRING_IDS = [
-  'IDS_PRODUCT_NAME',
-  'IDS_SXS_SHORTCUT_NAME',
-  'IDS_PRODUCT_DESCRIPTION',
   'IDS_ABOUT_VERSION_COMPANY_NAME',
+  'IDS_APP_SHORTCUTS_SUBDIR_NAME',
+  'IDS_APP_SHORTCUTS_SUBDIR_NAME_CANARY',
+  'IDS_INBOUND_MDNS_RULE_DESCRIPTION',
+  'IDS_INBOUND_MDNS_RULE_DESCRIPTION_CANARY',
+  'IDS_INBOUND_MDNS_RULE_NAME',
+  'IDS_INBOUND_MDNS_RULE_NAME_CANARY',
+  'IDS_INSTALL_EXISTING_VERSION_LAUNCHED',
+  'IDS_INSTALL_FAILED',
   'IDS_INSTALL_HIGHER_VERSION',
-  'IDS_INSTALL_FAILED',
-  'IDS_SAME_VERSION_REPAIR_FAILED',
-  'IDS_SETUP_PATCH_FAILED',
+  'IDS_INSTALL_INSUFFICIENT_RIGHTS',
+  'IDS_INSTALL_INVALID_ARCHIVE',
+  'IDS_INSTALL_OS_ERROR',
   'IDS_INSTALL_OS_NOT_SUPPORTED',
-  'IDS_INSTALL_OS_ERROR',
   'IDS_INSTALL_SINGLETON_ACQUISITION_FAILED',
   'IDS_INSTALL_TEMP_DIR_FAILED',
   'IDS_INSTALL_UNCOMPRESSION_FAILED',
-  'IDS_INSTALL_INVALID_ARCHIVE',
-  'IDS_INSTALL_INSUFFICIENT_RIGHTS',
+  'IDS_PRODUCT_DESCRIPTION',
+  'IDS_PRODUCT_NAME',
+  'IDS_SAME_VERSION_REPAIR_FAILED',
+  'IDS_SETUP_PATCH_FAILED',
+  'IDS_SHORTCUT_NEW_WINDOW',
   'IDS_SHORTCUT_TOOLTIP',
-  'IDS_SHORTCUT_NEW_WINDOW',
-  'IDS_APP_SHORTCUTS_SUBDIR_NAME',
-  'IDS_APP_SHORTCUTS_SUBDIR_NAME_CANARY',
-  'IDS_INBOUND_MDNS_RULE_NAME',
-  'IDS_INBOUND_MDNS_RULE_NAME_CANARY',
-  'IDS_INBOUND_MDNS_RULE_DESCRIPTION',
-  'IDS_INBOUND_MDNS_RULE_DESCRIPTION_CANARY',
-  'IDS_INSTALL_EXISTING_VERSION_LAUNCHED',
+  'IDS_SXS_SHORTCUT_NAME',
 ]
 
+# Certain strings are conditional on a brand's install mode (see
+# chrome/install_static/install_modes.h for details). This allows
+# installer::GetLocalizedString to return a resource specific to the current
+# install mode at runtime (e.g., "Google Chrome SxS" as IDS_SHORTCUT_NAME for
+# the localized shortcut name for Google Chrome's canary channel).
+# l10n_util::GetStringUTF16 (used within the rest of Chrome) is unaffected, and
+# will always return the requested string.
+#
+# This mapping provides brand- and mode-specific string ids for a given input id
+# as described here:
+# {
+#   resource_id_1: {  # A resource ID for use with GetLocalizedString.
+#     brand_1: [  # 'google_chrome', for example.
+#       string_id_1,  # Strings listed in order of the brand's modes, as
+#       string_id_2,  # specified in install_static::InstallConstantIndex.
+#       ...
+#       string_id_N,
+#     ],
+#     brand_2: [  # 'chromium', for example.
+#       ...
+#     ],
+#   },
+#   resource_id_2:  ...
+# }
+# 'resource_id_1' names an existing string ID. All calls to
+# installer::GetLocalizedString with this string ID will map to the
+# mode-specific string.
+#
+# Note: Update the test expectations in GetBaseMessageIdForMode.GoogleStringIds
+# when adding to/modifying this structure.
+MODE_SPECIFIC_STRINGS = {
+  'IDS_APP_SHORTCUTS_SUBDIR_NAME': {
+    'google_chrome': [
+      'IDS_APP_SHORTCUTS_SUBDIR_NAME',
+      'IDS_APP_SHORTCUTS_SUBDIR_NAME_CANARY',
+    ],
+    'chromium': [
+      'IDS_APP_SHORTCUTS_SUBDIR_NAME',
+    ],
+  },
+  'IDS_INBOUND_MDNS_RULE_DESCRIPTION': {
+    'google_chrome': [
+      'IDS_INBOUND_MDNS_RULE_DESCRIPTION',
+      'IDS_INBOUND_MDNS_RULE_DESCRIPTION_CANARY',
+    ],
+    'chromium': [
+      'IDS_INBOUND_MDNS_RULE_DESCRIPTION',
+    ],
+  },
+  'IDS_INBOUND_MDNS_RULE_NAME': {
+    'google_chrome': [
+      'IDS_INBOUND_MDNS_RULE_NAME',
+      'IDS_INBOUND_MDNS_RULE_NAME_CANARY',
+    ],
+    'chromium': [
+      'IDS_INBOUND_MDNS_RULE_NAME',
+    ],
+  },
+  # In contrast to the strings above, this one (IDS_PRODUCT_NAME) is used
+  # throughout Chrome in mode-independent contexts. Within the installer (the
+  # place where this mapping matters), it is only used for mode-specific strings
+  # such as the name of Chrome's shortcut.
+  'IDS_PRODUCT_NAME': {
+    'google_chrome': [
+      'IDS_PRODUCT_NAME',
+      'IDS_SXS_SHORTCUT_NAME',
+    ],
+    'chromium': [
+      'IDS_PRODUCT_NAME',
+    ],
+  },
+}
+# Note: Update the test expectations in GetBaseMessageIdForMode.GoogleStringIds
+# when adding to/modifying the above structure.
+
 # The ID of the first resource string.
 FIRST_RESOURCE_ID = 1600
 
 
 class GrdHandler(sax.handler.ContentHandler):
   """Extracts selected strings from a .grd file.
 
   Attributes:
     messages: A dict mapping string identifiers to their corresponding messages.
   """
@@ skipping 124 matching lines @@
       translated_string = ''.join(self.__text_scraps).strip()
       for string_id in self.__string_ids:
         self.translations[string_id] = translated_string
       self.__string_ids = None
       self.__text_scraps = []
       self.__characters_callback = None
 
 
 class StringRcMaker(object):
   """Makes .h and .rc files containing strings and translations."""
-  def __init__(self, name, inputs, outdir):
+  def __init__(self, name, inputs, outdir, brand):
     """Constructs a maker.
 
     Args:
       name: The base name of the generated files (e.g.,
         'installer_util_strings').
       inputs: A list of (grd_file, xtb_dir) pairs containing the source data.
       outdir: The directory into which the files will be generated.
     """
     self.name = name
     self.inputs = inputs
     self.outdir = outdir
+    self.brand = brand
 
   def MakeFiles(self):
     translated_strings = self.__ReadSourceAndTranslatedStrings()
     self.__WriteRCFile(translated_strings)
     self.__WriteHeaderFile(translated_strings)
 
   class __TranslationData(object):
     """A container of information about a single translation."""
     def __init__(self, resource_id_str, language, translation):
       self.resource_id_str = resource_id_str
@@ skipping 100 matching lines @@
                        escaped_text))
       outfile.write(FOOTER_TEXT)
 
   def __WriteHeaderFile(self, translated_strings):
     """Writes a .h file with resource ids."""
     # TODO(grt): Stream the lines to the file rather than building this giant
     # list of lines first.
     lines = []
     do_languages_lines = ['\n#define DO_LANGUAGES']
     installer_string_mapping_lines = ['\n#define DO_INSTALLER_STRING_MAPPING']
+    do_mode_strings_lines = ['\n#define DO_MODE_STRINGS']
 
     # Write the values for how the languages ids are offset.
     seen_languages = set()
     offset_id = 0
     for translation_data in translated_strings:
       lang = translation_data.language
       if lang not in seen_languages:
         seen_languages.add(lang)
         lines.append('#define IDS_L10N_OFFSET_%s %s' % (lang, offset_id))
         do_languages_lines.append('  HANDLE_LANGUAGE(%s, IDS_L10N_OFFSET_%s)'
                                   % (lang.replace('_', '-').lower(), lang))
         offset_id += 1
       else:
         break
 
     # Write the resource ids themselves.
     resource_id = FIRST_RESOURCE_ID
     for translation_data in translated_strings:
       lines.append('#define %s %s' % (translation_data.resource_id_str + '_' +
                                       translation_data.language,
                                       resource_id))
       resource_id += 1
 
+    # Handle mode-specific strings.
+    for string_id, brands in MODE_SPECIFIC_STRINGS.iteritems():
+      # Populate the DO_MODE_STRINGS macro.
+      brand_strings = brands.get(self.brand)
+      if not brand_strings:
+        raise exceptions.RuntimeError(
+          'No strings declared for brand \'%s\' in MODE_SPECIFIC_STRINGS for '
+          'message %s' % (self.brand, string_id))
+      do_mode_strings_lines.append(
+        '  HANDLE_MODE_STRING(%s_BASE, %s)'
+        % (string_id, ', '.join([ ('%s_BASE' % s) for s in brand_strings])))
+
     # Write out base ID values.
     for string_id in STRING_IDS:
       lines.append('#define %s_BASE %s_%s' % (string_id,
                                               string_id,
                                               translated_strings[0].language))
       installer_string_mapping_lines.append('  HANDLE_STRING(%s_BASE, %s)'
                                             % (string_id, string_id))
 
     with open(os.path.join(self.outdir, self.name + '.h'), 'wb') as outfile:
       outfile.write('\n'.join(lines))
       outfile.write('\n#ifndef RC_INVOKED')
       outfile.write(' \\\n'.join(do_languages_lines))
       outfile.write(' \\\n'.join(installer_string_mapping_lines))
+      outfile.write(' \\\n'.join(do_mode_strings_lines))
       # .rc files must end in a new line
       outfile.write('\n#endif  // ndef RC_INVOKED\n')
 
 
 def ParseCommandLine():
   def GrdPathAndXtbDirPair(string):
     """Returns (grd_path, xtb_dir) given a colon-separated string of the same.
     """
     parts = string.split(':')
     if len(parts) is not 2:
       raise argparse.ArgumentTypeError('%r is not grd_path:xtb_dir')
     return (parts[0], parts[1])
 
   parser = argparse.ArgumentParser(
     description='Generate .h and .rc files for installer strings.')
+  brands = [b for b in MODE_SPECIFIC_STRINGS.itervalues().next().iterkeys()]
+  parser.add_argument('-b',
+                      choices=brands,
+                      required=True,
+                      help='identifier of the browser brand (e.g., chromium).',
+                      dest='brand')
   parser.add_argument('-i', action='append',
                       type=GrdPathAndXtbDirPair,
                       required=True,
                       help='path to .grd file:relative path to .xtb dir',
                       metavar='GRDFILE:XTBDIR',
                       dest='inputs')
   parser.add_argument('-o',
                       required=True,
                       help='output directory for generated .rc and .h files',
                       dest='outdir')
   parser.add_argument('-n',
                       required=True,
                       help='base name of generated .rc and .h files',
                       dest='name')
   return parser.parse_args()
 
 
 def main():
   args = ParseCommandLine()
-  StringRcMaker(args.name, args.inputs, args.outdir).MakeFiles()
+  StringRcMaker(args.name, args.inputs, args.outdir, args.brand).MakeFiles()
   return 0
 
 
 if '__main__' == __name__:
   sys.exit(main())