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

extensions/browser/guest_view/web_view/web_view_guest.h

 // 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.
 
 #ifndef EXTENSIONS_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_GUEST_H_
 #define EXTENSIONS_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_GUEST_H_
 
 #include <vector>
 
 #include "base/observer_list.h"
 #include "content/public/browser/javascript_dialog_manager.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "extensions/browser/guest_view/guest_view.h"
 #include "extensions/browser/guest_view/web_view/javascript_dialog_helper.h"
 #include "extensions/browser/guest_view/web_view/web_view_find_helper.h"
 #include "extensions/browser/guest_view/web_view/web_view_guest_delegate.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/script_executor.h"
 
 namespace blink {
 struct WebFindOptions;
 }  // nanespace blink
 
 namespace extensions {
 
+class DeclarativeUserScriptMaster;
 class WebViewInternalFindFunction;
 
 // A WebViewGuest provides the browser-side implementation of the <webview> API
 // and manages the dispatch of <webview> extension events. WebViewGuest is
 // created on attachment. That is, when a guest WebContents is associated with
 // a particular embedder WebContents. This happens on either initial navigation
 // or through the use of the New Window API, when a new window is attached to
 // a particular <webview>.
 class WebViewGuest : public GuestView<WebViewGuest>,
                      public content::NotificationObserver {
@@ skipping 190 matching lines @@
   void Terminate();
 
   // Clears data in the storage partition of this guest.
   //
   // Partition data that are newer than |removal_since| will be removed.
   // |removal_mask| corresponds to bitmask in StoragePartition::RemoveDataMask.
   bool ClearData(const base::Time remove_since,
                  uint32 removal_mask,
                  const base::Closure& callback);
 
+  // Adds content script to inject on the guest.
+  void AddContentScripts(const std::string& script_name,
+                         const UserScript& script);
+
+  // Removes content script by |script_name|.
+  void RemoveContentScripts(const std::string& script_name);
+
+  // Removes all content scripts added by the guest.
+  void RemoveAllContentScripts();
+
+  // Returns whether |host_id_| is nullptr.
+  bool IsHostIDEmpty();
+
   ScriptExecutor* script_executor() { return script_executor_.get(); }
+  const HostID& host_id() { return *host_id_; }
+  void set_host_id(scoped_ptr<const HostID> host_id) {
+    host_id_ = host_id.Pass();
+  }
 
  private:
   friend class WebViewPermissionHelper;
 
+  using UserScriptMap = std::map<std::string, UserScript>;
+
   explicit WebViewGuest(content::WebContents* owner_web_contents);
 
   ~WebViewGuest() override;
 
   void AttachWebViewHelpers(content::WebContents* contents);
 
   void OnWebViewNewWindowResponse(int new_window_instance_id,
                                   bool allow,
                                   const std::string& user_input);
 
@@ skipping 112 matching lines @@
 
   using PendingWindowMap = std::map<WebViewGuest*, NewWindowInfo>;
   PendingWindowMap pending_new_windows_;
 
   // Stores the current zoom factor.
   double current_zoom_factor_;
 
   // Determines if this guest accepts pinch-zoom gestures.
   bool allow_scaling_;
 
+  // The ID of the injection host of the guest.
+  scoped_ptr<const HostID> host_id_;
+
+  DeclarativeUserScriptMaster* master_;
+
+  UserScriptMap user_script_map_;
+
   // This is used to ensure pending tasks will not fire after this object is
   // destroyed.
   base::WeakPtrFactory<WebViewGuest> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(WebViewGuest);
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_GUEST_VIEW_WEB_VIEW_WEB_VIEW_GUEST_H_