103 lines
		
	
	
	
		
			3.3 KiB
			
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			103 lines
		
	
	
	
		
			3.3 KiB
			
		
	
	
	
		
			C++
		
	
	
	
	
	
// Copyright (c) 2014 GitHub, Inc.
 | 
						|
// Use of this source code is governed by the MIT license that can be
 | 
						|
// found in the LICENSE file.
 | 
						|
 | 
						|
#ifndef ATOM_BROWSER_WEB_VIEW_MANAGER_H_
 | 
						|
#define ATOM_BROWSER_WEB_VIEW_MANAGER_H_
 | 
						|
 | 
						|
#include <map>
 | 
						|
 | 
						|
#include "base/files/file_path.h"
 | 
						|
#include "base/synchronization/lock.h"
 | 
						|
#include "content/public/browser/browser_plugin_guest_manager.h"
 | 
						|
 | 
						|
namespace content {
 | 
						|
class BrowserContext;
 | 
						|
class RenderProcessHost;
 | 
						|
}
 | 
						|
 | 
						|
namespace atom {
 | 
						|
 | 
						|
class WebViewManager : public content::BrowserPluginGuestManager {
 | 
						|
 public:
 | 
						|
  struct WebViewInfo {
 | 
						|
    int guest_instance_id;
 | 
						|
    content::WebContents* embedder;
 | 
						|
    bool node_integration;
 | 
						|
    bool plugins;
 | 
						|
    bool disable_web_security;
 | 
						|
    base::FilePath preload_script;
 | 
						|
  };
 | 
						|
 | 
						|
  // Finds the WebViewManager attached with |process| and returns the
 | 
						|
  // WebViewInfo of it.
 | 
						|
  static bool GetInfoForProcess(content::RenderProcessHost* process,
 | 
						|
                                WebViewInfo* info);
 | 
						|
 | 
						|
  // Updates the guest process ID.
 | 
						|
  static void UpdateGuestProcessID(content::RenderProcessHost* old_process,
 | 
						|
                                   content::RenderProcessHost* new_process);
 | 
						|
 | 
						|
  explicit WebViewManager(content::BrowserContext* context);
 | 
						|
  virtual ~WebViewManager();
 | 
						|
 | 
						|
  void AddGuest(int guest_instance_id,
 | 
						|
                int element_instance_id,
 | 
						|
                content::WebContents* embedder,
 | 
						|
                content::WebContents* web_contents,
 | 
						|
                const WebViewInfo& info);
 | 
						|
  void RemoveGuest(int guest_instance_id);
 | 
						|
 | 
						|
  // Looks up the information for the embedder <webview> for a given render
 | 
						|
  // view, if one exists. Called on the IO thread.
 | 
						|
  bool GetInfo(int guest_process_id, WebViewInfo* webview_info);
 | 
						|
 | 
						|
 protected:
 | 
						|
  // content::BrowserPluginGuestManager:
 | 
						|
  content::WebContents* GetGuestByInstanceID(int owner_process_id,
 | 
						|
                                             int element_instance_id) override;
 | 
						|
  bool ForEachGuest(content::WebContents* embedder,
 | 
						|
                    const GuestCallback& callback) override;
 | 
						|
 | 
						|
 private:
 | 
						|
  struct WebContentsWithEmbedder {
 | 
						|
    content::WebContents* web_contents;
 | 
						|
    content::WebContents* embedder;
 | 
						|
  };
 | 
						|
  // guest_instance_id => (web_contents, embedder)
 | 
						|
  std::map<int, WebContentsWithEmbedder> web_contents_embdder_map_;
 | 
						|
 | 
						|
  struct ElementInstanceKey {
 | 
						|
    int embedder_process_id;
 | 
						|
    int element_instance_id;
 | 
						|
 | 
						|
    ElementInstanceKey(int embedder_process_id, int element_instance_id)
 | 
						|
        : embedder_process_id(embedder_process_id),
 | 
						|
          element_instance_id(element_instance_id) {}
 | 
						|
 | 
						|
    bool operator<(const ElementInstanceKey& other) const {
 | 
						|
      if (embedder_process_id != other.embedder_process_id)
 | 
						|
        return embedder_process_id < other.embedder_process_id;
 | 
						|
      return element_instance_id < other.element_instance_id;
 | 
						|
    }
 | 
						|
 | 
						|
    bool operator==(const ElementInstanceKey& other) const {
 | 
						|
      return (embedder_process_id == other.embedder_process_id) &&
 | 
						|
          (element_instance_id == other.element_instance_id);
 | 
						|
    }
 | 
						|
  };
 | 
						|
  // (embedder_process_id, element_instance_id) => guest_instance_id
 | 
						|
  std::map<ElementInstanceKey, int> element_instance_id_to_guest_map_;
 | 
						|
 | 
						|
  typedef std::map<int, WebViewInfo> WebViewInfoMap;
 | 
						|
  // guest_process_id => (guest_instance_id, embedder, ...)
 | 
						|
  WebViewInfoMap webview_info_map_;
 | 
						|
 | 
						|
  base::Lock lock_;
 | 
						|
 | 
						|
  DISALLOW_COPY_AND_ASSIGN(WebViewManager);
 | 
						|
};
 | 
						|
 | 
						|
}  // namespace atom
 | 
						|
 | 
						|
#endif  // ATOM_BROWSER_WEB_VIEW_MANAGER_H_
 |