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 view_instance_id,
+    const std::set<int>& script_ids) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+  for (auto& render_id_info : web_view_info_map_) {
+    WebViewInfo& info = render_id_info.second;
+    if (info.embedder_process_id == embedder_process_id &&
+        info.instance_id == view_instance_id) {
+      for (int id : script_ids) {
+        info.content_script_ids.insert(id);
+      }
+      return;
+    }
+  }
+}
+
+void WebViewRendererState::RemoveContentScriptIDs(
+    int embedder_process_id,
+    int view_instance_id,
+    const std::set<int>& script_ids) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+  for (auto& render_id_info : web_view_info_map_) {
+    WebViewInfo& info = render_id_info.second;
+    if (info.embedder_process_id == embedder_process_id &&
+        info.instance_id == view_instance_id) {
+      for (int id : script_ids)
+        info.content_script_ids.erase(id);
+      return;
+    }
+  }
+}
+
 }  // namespace extensions