Injected CSS localization fix (see bug no.)

chrome/browser/extensions/user_script_master.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/browser/extensions/user_script_master.h"
 
+#include <map>
 #include <string>
 #include <vector>
 
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/pickle.h"
 #include "base/stl_util.h"
 #include "base/string_util.h"
 #include "base/threading/thread.h"
 #include "base/version.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/extensions/extension.h"
+#include "chrome/common/extensions/extension_file_util.h"
 #include "chrome/common/extensions/extension_messages.h"
+#include "chrome/common/extensions/extension_message_bundle.h"
 #include "chrome/common/extensions/extension_resource.h"
 #include "content/browser/renderer_host/render_process_host.h"
 #include "content/common/notification_service.h"
 
 // Helper function to parse greasesmonkey headers
 static bool GetDeclarationValue(const base::StringPiece& line,
                                 const base::StringPiece& prefix,
                                 std::string* value) {
   base::StringPiece::size_type index = line.find(prefix);
   if (index == base::StringPiece::npos)
     return false;
 
   std::string temp(line.data() + index + prefix.length(),
                    line.length() - index - prefix.length());
 
   if (temp.empty() || !IsWhitespace(temp[0]))
     return false;
 
   TrimWhitespaceASCII(temp, TRIM_ALL, value);
   return true;
 }
 
 UserScriptMaster::ScriptReloader::ScriptReloader(UserScriptMaster* master)
     : master_(master) {
   CHECK(BrowserThread::GetCurrentThreadIdentifier(&master_thread_id_));
+
+  // Gather extensions information needed for localization.
+  if (master && master->profile_ && master->profile_->GetExtensionInfoMap()) {
+    const ExtensionInfoMap* info_map = master->profile_->GetExtensionInfoMap();
+    info_map->extensions().GetExtensionsPathAndDefaultLocale(extensions_info_);
+  }
 }
 
 // static
 bool UserScriptMaster::ScriptReloader::ParseMetadataHeader(
       const base::StringPiece& script_text, UserScript* script) {
   // http://wiki.greasespot.net/Metadata_block
   base::StringPiece line;
   size_t line_start = 0;
   size_t line_end = line_start;
   bool in_metadata = false;
@@ skipping 80 matching lines @@
 }
 
 void UserScriptMaster::ScriptReloader::StartLoad(
     const UserScriptList& user_scripts) {
   // Add a reference to ourselves to keep ourselves alive while we're running.
   // Balanced by NotifyMaster().
   AddRef();
   BrowserThread::PostTask(
       BrowserThread::FILE, FROM_HERE,
       NewRunnableMethod(
           this, &UserScriptMaster::ScriptReloader::RunLoad, user_scripts));

[Nebojša Ćirić, 2011/08/10 20:38:57]

Do you wanna pass your map here to file thread?

[adriansc, 2011/08/10 23:20:53]

I could, but the methods executing on BrowserThread::FILE belong to
ScriptReloader, so while loading takes place, the object cannot die (enforced by
AddRef() above). Since I put the map inside the ScriptReloader object, it will
be freely accessible by the FILE thread.

 }
 
 void UserScriptMaster::ScriptReloader::NotifyMaster(
     base::SharedMemory* memory) {
   // The master went away, so these new scripts aren't useful anymore.
   if (!master_)
     delete memory;
   else
     master_->NewScriptsAvailable(memory);
 
   // Drop our self-reference.
   // Balances StartLoad().
   Release();
 }
 
-static bool LoadScriptContent(UserScript::File* script_file) {
+static bool LoadScriptContent(UserScript::File* script_file,
+                              const SubstitutionMap& localization_messages) {
   std::string content;
   const FilePath& path = ExtensionResource::GetFilePath(
       script_file->extension_root(), script_file->relative_path());
   if (path.empty() || !file_util::ReadFileToString(path, &content)) {
     LOG(WARNING) << "Failed to load user script file: " << path.value();
     return false;
   }
 
+  // Localize the content.
+  if (!localization_messages.empty()) {
+    std::string error;
+    LOG(INFO) << "$$$_BEFORE_LOCALIZATION " << content;

[Nebojša Ćirić, 2011/08/10 20:38:57]

Remove LOG(INFO).

[adriansc, 2011/08/10 23:20:53]

Done.

+    ExtensionMessageBundle::ReplaceMessagesWithExternalDictionary(
+        localization_messages, &content, &error);
+    LOG(INFO) << "@@@_AFTER__LOCALIZATION " << content;
+  }
+
   // Remove BOM from the content.
   std::string::size_type index = content.find(kUtf8ByteOrderMark);
   if (index == 0) {
     script_file->set_content(content.substr(strlen(kUtf8ByteOrderMark)));
   } else {
     script_file->set_content(content);
   }
 
   return true;
 }
 
-// static
 void UserScriptMaster::ScriptReloader::LoadUserScripts(
     UserScriptList* user_scripts) {
   for (size_t i = 0; i < user_scripts->size(); ++i) {
     UserScript& script = user_scripts->at(i);
+    SubstitutionMap localization_messages;
+    if (extensions_info_.find(script.extension_id()) !=
+        extensions_info_.end()) {
+      localization_messages =
+          extension_file_util::LoadExtensionMessageBundleSubstitutionMap(
+              extensions_info_[script.extension_id()].first,
+              script.extension_id(),
+              extensions_info_[script.extension_id()].second);

[Nebojša Ćirić, 2011/08/10 20:38:57]

Refactor this into a method call.

[adriansc, 2011/08/10 23:20:53]

Done.

+    }
     for (size_t k = 0; k < script.js_scripts().size(); ++k) {
       UserScript::File& script_file = script.js_scripts()[k];
       if (script_file.GetContent().empty())
-        LoadScriptContent(&script_file);
+        LoadScriptContent(&script_file, localization_messages);
     }
     for (size_t k = 0; k < script.css_scripts().size(); ++k) {
       UserScript::File& script_file = script.css_scripts()[k];
       if (script_file.GetContent().empty())
-        LoadScriptContent(&script_file);
+        LoadScriptContent(&script_file, localization_messages);
     }
   }
 }
 
 // Pickle user scripts and return pointer to the shared memory.
 static base::SharedMemory* Serialize(const UserScriptList& scripts) {
   Pickle pickle;
   pickle.WriteSize(scripts.size());
   for (size_t i = 0; i < scripts.size(); i++) {
     const UserScript& script = scripts[i];
@@ skipping 167 matching lines @@
   if (!handle)
     return;
 
   base::SharedMemoryHandle handle_for_process;
   if (!shared_memory->ShareToProcess(handle, &handle_for_process))
     return;  // This can legitimately fail if the renderer asserts at startup.
 
   if (base::SharedMemory::IsHandleValid(handle_for_process))
     process->Send(new ExtensionMsg_UpdateUserScripts(handle_for_process));
 }