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

chrome/installer/util/l10n_string_util.cc

 // 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.
 //
 // This file defines utility functions for fetching localized resources.
 
 #include "chrome/installer/util/l10n_string_util.h"
 
 #include <atlbase.h>
 #include <stdint.h>
 
 #include <algorithm>
 #include <limits>
 #include <memory>
 
 #include "base/logging.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
+#include "chrome/install_static/install_details.h"
+#include "chrome/install_static/install_modes.h"
+#include "chrome/installer/util/installer_util_strings.h"
 #include "chrome/installer/util/language_selector.h"
 
 namespace {
 
 const installer::LanguageSelector& GetLanguageSelector() {
   static const installer::LanguageSelector instance;
 
   return instance;
 }
 
 installer::TranslationDelegate* g_translation_delegate = NULL;
 
+// Returns the mode-specific message id for |base_message_id| given the current
+// install mode. Returns |base_message_id| itself if it does not have per-mode
+// variants. See MODE_SPECIFIC_STRINGS in create_string_rc.py for details..
+int GetBaseMessageIdForMode(int base_message_id) {
+// Generate the constants holding the mode-specific resource ID arrays.
+#define HANDLE_MODE_STRING(id, ...)                                   \
+  static constexpr int k##id##Strings[] = {__VA_ARGS__};              \
+  static_assert(                                                      \
+      arraysize(k##id##Strings) == install_static::NUM_INSTALL_MODES, \
+      "resource " #id                                                 \
+      " has the wrong number of mode-specific "                       \
+      "strings.");
+  DO_MODE_STRINGS
+#undef HANDLE_MODE_STRING
+
+  const int* mode_strings = nullptr;
+  switch (base_message_id) {
+// Generate the cases mapping each mode-specific resource ID to its array.
+#define HANDLE_MODE_STRING(id, ...)    \
+  case id:                             \
+    mode_strings = &k##id##Strings[0]; \
+    break;
+    DO_MODE_STRINGS
+#undef HANDLE_MODE_STRING
+    default:
+      // This ID has no per-mode variants.
+      return base_message_id;
+  }
+
+  // Return the variant of |base_message_id| for the current mode.
+  return mode_strings[install_static::InstallDetails::Get()
+                          .install_mode_index()];
+}
+
 }  // namespace
 
 namespace installer {
 
 TranslationDelegate::~TranslationDelegate() {
 }
 
 void SetTranslationDelegate(TranslationDelegate* delegate) {
   g_translation_delegate = delegate;
 }
 
 std::wstring GetLocalizedString(int base_message_id) {
+  // Map |base_message_id| to the base id for the current install mode.
+  base_message_id = GetBaseMessageIdForMode(base_message_id);
+
   if (g_translation_delegate)
     return g_translation_delegate->GetLocalizedString(base_message_id);
 
   std::wstring localized_string;
 
   int message_id = base_message_id + GetLanguageSelector().offset();
   const ATLSTRINGRESOURCEIMAGE* image = AtlGetStringResourceImage(
       _AtlBaseModule.GetModuleInstance(), message_id);
   if (image) {
     localized_string = std::wstring(image->achString, image->nLength);
@@ skipping 46 matching lines @@
   if (SUCCEEDED(hr))
     return std::wstring(url_canon.get());
   return url_path;
 }
 
 std::wstring GetCurrentTranslation() {
   return GetLanguageSelector().selected_translation();
 }
 
 }  // namespace installer