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

extensions/browser/guest_view/web_view/web_view_renderer_state.cc

Index: extensions/browser/guest_view/web_view/web_view_renderer_state.cc
diff --git a/extensions/browser/guest_view/web_view/web_view_renderer_state.cc b/extensions/browser/guest_view/web_view/web_view_renderer_state.cc
index 9851e617562b2c817cd294a591e543aa689aad37..46cf9c8a859336d4f46d7430b963b8d084366111 100644
--- a/extensions/browser/guest_view/web_view/web_view_renderer_state.cc
+++ b/extensions/browser/guest_view/web_view/web_view_renderer_state.cc
@@ -9,6 +9,12 @@ using content::BrowserThread;
 
 namespace extensions {
 
+WebViewRendererState::WebViewInfo::WebViewInfo() {
+}
+
+WebViewRendererState::WebViewInfo::~WebViewInfo() {
+}
+
 // static
 WebViewRendererState* WebViewRendererState::GetInstance() {
   return Singleton<WebViewRendererState>::get();
@@ -103,4 +109,39 @@ bool WebViewRendererState::GetPartitionID(int guest_process_id,
   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) {

[Devlin, 2015/04/13 19:21:48]

no brackets

[Xi Han, 2015/04/14 19:05:52]

Done.

+        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