Add commands to manage tabs and contexts to Browser Domain

headless/lib/browser/headless_browser_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_browser_impl.h"
 
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/memory/ptr_util.h"
@@ skipping 87 matching lines @@
 }
 
 void HeadlessBrowserImpl::Shutdown() {
   DCHECK(BrowserMainThread()->BelongsToCurrentThread());
   BrowserMainThread()->PostTask(FROM_HERE,
                                 base::MessageLoop::QuitWhenIdleClosure());
 }
 
 std::vector<HeadlessWebContents*> HeadlessBrowserImpl::GetAllWebContents() {
   std::vector<HeadlessWebContents*> result;
-  result.reserve(web_contents_.size());
+  result.reserve(web_contents_map_.size());
 
-  for (const auto& web_contents_pair : web_contents_) {
-    result.push_back(web_contents_pair.first);
+  for (const auto& web_contents_pair : web_contents_map_) {
+    result.push_back(web_contents_pair.second.get());
   }
 
   return result;
 }
 
 HeadlessBrowserMainParts* HeadlessBrowserImpl::browser_main_parts() const {
   DCHECK(BrowserMainThread()->BelongsToCurrentThread());
   return browser_main_parts_;
 }
 
@@ skipping 15 matching lines @@
       new HeadlessWindowTreeClient(window_tree_host_->window()));
 
   on_start_callback_.Run(this);
   on_start_callback_ = base::Callback<void(HeadlessBrowser*)>();
 }
 
 HeadlessWebContentsImpl* HeadlessBrowserImpl::RegisterWebContents(
     std::unique_ptr<HeadlessWebContentsImpl> web_contents) {
   DCHECK(web_contents);
   HeadlessWebContentsImpl* unowned_web_contents = web_contents.get();
-  web_contents_[unowned_web_contents] = std::move(web_contents);
+  web_contents_map_[unowned_web_contents->GetDevtoolsAgentHostId()] =
+      std::move(web_contents);
   return unowned_web_contents;
 }
 
 void HeadlessBrowserImpl::DestroyWebContents(
     HeadlessWebContentsImpl* web_contents) {
-  auto it = web_contents_.find(web_contents);
-  DCHECK(it != web_contents_.end());
-  web_contents_.erase(it);
+  auto it = web_contents_map_.find(web_contents->GetDevtoolsAgentHostId());
+  DCHECK(it != web_contents_map_.end());
+  web_contents_map_.erase(it);
+}
+
+HeadlessWebContents* HeadlessBrowserImpl::GetWebContentsForDevtoolsAgentHostId(
+    const std::string& devtools_agent_host_id) {
+  auto it = web_contents_map_.find(devtools_agent_host_id);
+  if (it == web_contents_map_.end())
+    return nullptr;
+  return it->second.get();
 }
 
 void HeadlessBrowserImpl::SetOptionsForTesting(
     HeadlessBrowser::Options options) {
   options_ = std::move(options);
   browser_main_parts()->default_browser_context()->SetOptionsForTesting(
       &options_);
 }
 
 void RunChildProcessIfNeeded(int argc, const char** argv) {
@@ skipping 18 matching lines @@
 
   // Child processes should not end up here.
   base::CommandLine command_line(options.argc, options.argv);
   DCHECK(!command_line.HasSwitch(switches::kProcessType));
 #endif
   return RunContentMain(std::move(options),
                         std::move(on_browser_start_callback));
 }
 
 }  // namespace headless