Add commands to manage tabs and contexts to Browser Domain

headless/lib/browser/headless_web_contents_impl.cc

 // Copyright 2015 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 "headless/lib/browser/headless_web_contents_impl.h"
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/weak_ptr.h"
 #include "base/trace_event/trace_event.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/common/bindings_policy.h"
 #include "content/public/renderer/render_frame.h"
 #include "headless/lib/browser/headless_browser_context_impl.h"
 #include "headless/lib/browser/headless_browser_impl.h"
 #include "headless/lib/browser/headless_browser_main_parts.h"
 #include "headless/lib/browser/headless_devtools_client_impl.h"
 #include "ui/aura/window.h"
 
 namespace headless {
 
+// static
+HeadlessWebContentsImpl* HeadlessWebContentsImpl::From(
+    HeadlessWebContents* web_contents) {
+  // This downcast is safe because there is only one implementation of
+  // HeadlessWebContents.
+  return static_cast<HeadlessWebContentsImpl*>(web_contents);
+}
+
 class WebContentsObserverAdapter : public content::WebContentsObserver {
  public:
   WebContentsObserverAdapter(content::WebContents* web_contents,
                              HeadlessWebContents::Observer* observer)
       : content::WebContentsObserver(web_contents), observer_(observer) {}
 
   ~WebContentsObserverAdapter() override {}
 
   void RenderViewReady() override {
     DCHECK(web_contents()->GetMainFrame()->IsRenderFrameLive());
@@ skipping 75 matching lines @@
     : web_contents_delegate_(new HeadlessWebContentsImpl::Delegate(browser)),
       web_contents_(web_contents),
       browser_(browser) {
   web_contents_->SetDelegate(web_contents_delegate_.get());
 }
 
 HeadlessWebContentsImpl::~HeadlessWebContentsImpl() {
   web_contents_->Close();
 }
 
+content::DevToolsAgentHost* HeadlessWebContentsImpl::EnsureDevToolsAgentHost() {
+  if (!agent_host_)
+    agent_host_ = content::DevToolsAgentHost::GetOrCreateFor(web_contents());
+  return agent_host_.get();
+}
+
 bool HeadlessWebContentsImpl::OpenURL(const GURL& url) {
   if (!url.is_valid())
     return false;
   content::NavigationController::LoadURLParams params(url);
   params.transition_type = ui::PageTransitionFromInt(
       ui::PAGE_TRANSITION_TYPED | ui::PAGE_TRANSITION_FROM_ADDRESS_BAR);
   web_contents_->GetController().LoadURLWithParams(params);
   web_contents_->Focus();
   return true;
 }
 
 void HeadlessWebContentsImpl::Close() {
   browser_->DestroyWebContents(this);
 }
 
+std::string HeadlessWebContentsImpl::GetAgentHostId() {
+  return EnsureDevToolsAgentHost()->GetId();
+}
+
 void HeadlessWebContentsImpl::AddObserver(Observer* observer) {
   DCHECK(observer_map_.find(observer) == observer_map_.end());
   observer_map_[observer] = base::WrapUnique(
       new WebContentsObserverAdapter(web_contents_.get(), observer));
 }
 
 void HeadlessWebContentsImpl::RemoveObserver(Observer* observer) {
   ObserverMap::iterator it = observer_map_.find(observer);
   DCHECK(it != observer_map_.end());
   observer_map_.erase(it);
 }
 
 HeadlessDevToolsTarget* HeadlessWebContentsImpl::GetDevToolsTarget() {
   return web_contents()->GetMainFrame()->IsRenderFrameLive() ? this : nullptr;
 }
 
 void HeadlessWebContentsImpl::AttachClient(HeadlessDevToolsClient* client) {
-  if (!agent_host_)
-    agent_host_ = content::DevToolsAgentHost::GetOrCreateFor(web_contents());
-  HeadlessDevToolsClientImpl::From(client)->AttachToHost(agent_host_.get());
+  HeadlessDevToolsClientImpl::From(client)->AttachToHost(
+      EnsureDevToolsAgentHost());
 }
 
 void HeadlessWebContentsImpl::DetachClient(HeadlessDevToolsClient* client) {
   DCHECK(agent_host_);
   HeadlessDevToolsClientImpl::From(client)->DetachFromHost(agent_host_.get());
 }
 
 content::WebContents* HeadlessWebContentsImpl::web_contents() const {
   return web_contents_.get();
 }
@@ skipping 23 matching lines @@
     HeadlessBrowserContext* browser_context) {
   browser_context_ = browser_context;
   return *this;
 }
 
 HeadlessWebContents* HeadlessWebContents::Builder::Build() {
   return browser_->CreateWebContents(this);
 }
 
 }  // namespace headless