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

extensions/browser/guest_view/web_view/web_view_renderer_state.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 "content/public/browser/browser_thread.h"
 #include "extensions/browser/guest_view/web_view/web_view_renderer_state.h"
 
 using content::BrowserThread;
 
 namespace extensions {
 
+WebViewRendererState::WebViewInfo::WebViewInfo() {
+}
+
+WebViewRendererState::WebViewInfo::~WebViewInfo() {
+}
+
 // static
 WebViewRendererState* WebViewRendererState::GetInstance() {
   return Singleton<WebViewRendererState>::get();
 }
 
 WebViewRendererState::WebViewRendererState() {
 }
 
 WebViewRendererState::~WebViewRendererState() {
 }
@@ skipping 74 matching lines @@
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   auto iter = web_view_partition_id_map_.find(guest_process_id);
   if (iter != web_view_partition_id_map_.end()){
     *partition_id = iter->second.partition_id;
     return true;
   }
   return false;
 }
 
+void WebViewRendererState::AddContentScriptIDs(int embedder_process_id,
+                                               int guest_view_instance_id,
+                                               std::set<int> script_ids) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+  for (auto& x : web_view_info_map_) {

[Fady Samuel, 2015/03/31 23:40:12]

const auto& render_id_info : web_view_info_map_

[Xi Han, 2015/04/01 22:14:41]

It cannot be const, since we insert ids to the info.content_script_ids.

+    WebViewInfo& info = x.second;

[Fady Samuel, 2015/03/31 23:40:12]

const WebViewInfo&

[Xi Han, 2015/04/01 22:14:41]

Same as above.

+    if (info.embedder_process_id == embedder_process_id &&
+        info.instance_id == guest_view_instance_id) {
+      for (int id : script_ids) {
+        info.content_script_ids.insert(id);
+      }
+      return;
+    }
+  }
+}
+
+void WebViewRendererState::RemoveContentScriptIDs(int embedder_process_id,
+                                                  int guest_view_instance_id,
+                                                  std::set<int> script_ids) {

[Fady Samuel, 2015/03/31 23:40:12]

const std::set<int>&

[Xi Han, 2015/04/01 22:14:41]

Done.

+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+  for (auto& x : web_view_info_map_) {
+    WebViewInfo& info = x.second;
+    if (info.embedder_process_id == embedder_process_id &&
+        info.instance_id == guest_view_instance_id) {
+      for (int id : script_ids)
+        info.content_script_ids.erase(id);
+      return;
+    }
+  }
+}
+
 }  // namespace extensions