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"
@@ skipping 588 matching lines @@
       removal_mask,
       content::StoragePartition::QUOTA_MANAGED_STORAGE_MASK_ALL,
       GURL(),
       content::StoragePartition::OriginMatcherFunction(),
       remove_since,
       base::Time::Now(),
       callback);
   return true;
 }
 
+void WebViewGuest::AddContentScripts(const std::string& script_name,

[Fady Samuel, 2015/03/11 11:43:17]

This name is wrong. It adds a single script. Really I think we should simply
move all of this code to web_view_content_scripts_helper

+                                     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);
+}
+
+void WebViewGuest::RemoveContentScripts(const std::string& script_name) {

[Fady Samuel, 2015/03/11 11:43:17]

This name is wrong. It removes a single script. Really I think we should simply
move all of this code to web_view_content_scripts_helper

+  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