Implement <webview>.addContentScript/removeContentScript API [1]

extensions/browser/guest_view/web_view/web_view_guest.cc

 // 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.
 
 #include "extensions/browser/guest_view/web_view/web_view_guest.h"
 
 #include "base/message_loop/message_loop.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/public/browser/browser_context.h"
@@ skipping 16 matching lines @@
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/common/media_stream_request.h"
 #include "content/public/common/page_zoom.h"
 #include "content/public/common/result_codes.h"
 #include "content/public/common/stop_find_action.h"
 #include "content/public/common/url_constants.h"
 #include "extensions/browser/api/declarative/rules_registry_service.h"
 #include "extensions/browser/api/extensions_api_client.h"
 #include "extensions/browser/api/web_request/web_request_api.h"
 #include "extensions/browser/api/web_view/web_view_internal_api.h"
+#include "extensions/browser/declarative_user_script_manager.h"
+#include "extensions/browser/declarative_user_script_master.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/browser/guest_view/guest_view_manager.h"
 #include "extensions/browser/guest_view/web_view/web_view_constants.h"
 #include "extensions/browser/guest_view/web_view/web_view_permission_helper.h"
 #include "extensions/browser/guest_view/web_view/web_view_permission_types.h"
 #include "extensions/browser/guest_view/web_view/web_view_renderer_state.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension_messages.h"
 #include "extensions/common/guest_view/guest_view_constants.h"
 #include "extensions/strings/grit/extensions_strings.h"
 #include "ipc/ipc_message_macros.h"
 #include "net/base/escape.h"
 #include "net/base/net_errors.h"
 #include "ui/base/models/simple_menu_model.h"
 #include "url/url_constants.h"
 
 using base::UserMetricsAction;
 using content::RenderFrameHost;
 using content::ResourceType;
 using content::WebContents;
 
 namespace extensions {
 
 namespace {
 
+

[Fady Samuel, 2015/03/04 02:02:10]

Remove this extra line.

[Xi Han, 2015/03/06 17:29:07]

Done.

 std::string WindowOpenDispositionToString(
   WindowOpenDisposition window_open_disposition) {
   switch (window_open_disposition) {
     case IGNORE_ACTION:
       return "ignore";
     case SAVE_TO_DISK:
       return "save_to_disk";
     case CURRENT_TAB:
       return "current_tab";
     case NEW_BACKGROUND_TAB:
@@ skipping 563 matching lines @@
       removal_mask,
       content::StoragePartition::QUOTA_MANAGED_STORAGE_MASK_ALL,
       GURL(),
       content::StoragePartition::OriginMatcherFunction(),
       remove_since,
       base::Time::Now(),
       callback);
   return true;
 }
 
+bool WebViewGuest::IsHostIDEmpty() {

[Fady Samuel, 2015/03/04 02:02:10]

What does this mean?

[Xi Han, 2015/03/06 17:29:07]

This function returns whether the host_id_ is nullptr. I remove it for now and
update the get function of host_id_.

+  return !host_id_;
+}
+
+void WebViewGuest::AddContentScripts(const std::string& script_name,
+                                     const UserScript& script) {
+  if (!master_) {
+    CHECK(host_id_);
+    master_ = ExtensionSystem::Get(browser_context())
+                  ->declarative_user_script_manager()
+                  ->GetDeclarativeUserScriptMasterByID(*host_id_);
+  }
+  // If a script with the same script name exists, we delete the old one and
+  // then add the new one.
+  UserScriptMap::iterator iter = user_script_map_.find(script_name);
+  if (iter != user_script_map_.end()) {
+    master_->RemoveScript(iter->second);
+    user_script_map_.erase(iter);
+  }
+
+  user_script_map_.insert(
+      std::pair<std::string, UserScript>(script_name, script));
+  master_->AddScript(script);

[Fady Samuel, 2015/03/04 02:02:10]

Does this plumbing work now?

[Xi Han, 2015/03/06 17:29:07]

For webUI, since webUI needs a different way to load user script, it doesn't
work for sure. The support for webUI will be in a different patch. 

For extensions, it has worked:)

+}
+
+void WebViewGuest::RemoveContentScripts(const std::string& script_name) {
+  auto iter = user_script_map_.find(script_name);
+  if (iter == user_script_map_.end() || !master_)
+    return;
+  master_->RemoveScript(iter->second);
+  user_script_map_.erase(iter);
+}
+
+void WebViewGuest::RemoveAllContentScripts() {
+  if (!master_)
+    return;
+  auto iter = user_script_map_.begin();
+  while(iter != user_script_map_.end()) {
+    master_->RemoveScript(iter->second);
+    iter = user_script_map_.erase(iter);
+  }
+}
+
 WebViewGuest::WebViewGuest(content::WebContents* owner_web_contents)
     : GuestView<WebViewGuest>(owner_web_contents),
       rules_registry_id_(RulesRegistryService::kInvalidRulesRegistryID),
       find_helper_(this),
       is_overriding_user_agent_(false),
       guest_opaque_(true),
       javascript_dialog_helper_(this),
       current_zoom_factor_(1.0),
       allow_scaling_(false),
+      host_id_(nullptr),
+      master_(nullptr),
       weak_ptr_factory_(this) {
   web_view_guest_delegate_.reset(
       ExtensionsAPIClient::Get()->CreateWebViewGuestDelegate(this));
 }
 
 WebViewGuest::~WebViewGuest() {
 }
 
 void WebViewGuest::DidCommitProvisionalLoadForFrame(
     content::RenderFrameHost* render_frame_host,
@@ skipping 605 matching lines @@
       WebViewGuest::From(owner_web_contents()->GetRenderProcessHost()->GetID(),
                          new_window_instance_id);
   if (!guest)
     return;
 
   if (!allow)
     guest->Destroy();
 }
 
 }  // namespace extensions