[headless] Remove default browser context.

headless/lib/browser/headless_devtools_manager_delegate.cc

 // Copyright 2016 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_devtools_manager_delegate.h"
 
 #include <string>
+#include <utility>
 
 #include "base/guid.h"
 #include "content/public/browser/web_contents.h"
 #include "headless/lib/browser/headless_browser_context_impl.h"
 #include "headless/lib/browser/headless_browser_impl.h"
 #include "headless/lib/browser/headless_web_contents_impl.h"
 #include "headless/public/domains/browser.h"
 
 namespace headless {
 
 HeadlessDevToolsManagerDelegate::HeadlessDevToolsManagerDelegate(
     HeadlessBrowserImpl* browser)
-    : browser_(browser) {
+    : browser_(browser), weak_ptr_factory_(this) {
   command_map_["Browser.createTarget"] =
       &HeadlessDevToolsManagerDelegate::CreateTarget;
   command_map_["Browser.closeTarget"] =
       &HeadlessDevToolsManagerDelegate::CloseTarget;
   command_map_["Browser.createBrowserContext"] =
       &HeadlessDevToolsManagerDelegate::CreateBrowserContext;
   command_map_["Browser.disposeBrowserContext"] =
       &HeadlessDevToolsManagerDelegate::DisposeBrowserContext;
 }
 
@@ skipping 17 matching lines @@
   std::unique_ptr<base::Value> cmd_result(((this)->*command_fn_ptr)(params));
   if (!cmd_result)
     return nullptr;
 
   std::unique_ptr<base::DictionaryValue> result(new base::DictionaryValue());
   result->SetInteger("id", id);
   result->Set("result", std::move(cmd_result));
   return result.release();
 }
 
+void HeadlessDevToolsManagerDelegate::Shutdown() {
+  default_browser_context_.reset();
+  browser_context_map_.clear();
+}
+
+base::WeakPtr<HeadlessDevToolsManagerDelegate>
+HeadlessDevToolsManagerDelegate::GetWeakPtr() {
+  return weak_ptr_factory_.GetWeakPtr();
+}
+
 std::unique_ptr<base::Value> HeadlessDevToolsManagerDelegate::CreateTarget(
     const base::DictionaryValue* params) {
   std::string url;
   std::string browser_context_id;
   int width = browser_->options()->window_size.width();
   int height = browser_->options()->window_size.height();
   params->GetString("url", &url);
   params->GetString("browserContextId", &browser_context_id);
   params->GetInteger("width", &width);
   params->GetInteger("height", &height);
-  HeadlessWebContentsImpl* web_contents_impl;
+
+  HeadlessBrowserContext* context;
   auto find_it = browser_context_map_.find(browser_context_id);
   if (find_it != browser_context_map_.end()) {
-    web_contents_impl = HeadlessWebContentsImpl::From(
-        browser_->CreateWebContentsBuilder()
-            .SetInitialURL(GURL(url))
-            .SetWindowSize(gfx::Size(width, height))
-            .SetBrowserContext(find_it->second.get())
-            .Build());
+    context = find_it->second.get();
   } else {
-    web_contents_impl = HeadlessWebContentsImpl::From(
-        browser_->CreateWebContentsBuilder()
-            .SetInitialURL(GURL(url))
-            .SetWindowSize(gfx::Size(width, height))
-            .Build());
+    if (!default_browser_context_) {
+      default_browser_context_ =
+          browser_->CreateBrowserContextBuilder().Build();
+    }
+    context = default_browser_context_.get();
   }
+
+  HeadlessWebContentsImpl* web_contents_impl =
+      HeadlessWebContentsImpl::From(context->CreateWebContentsBuilder()
+                                        .SetInitialURL(GURL(url))
+                                        .SetWindowSize(gfx::Size(width, height))
+                                        .Build());
+
   return browser::CreateTargetResult::Builder()
       .SetTargetId(web_contents_impl->GetDevtoolsAgentHostId())
       .Build()
       ->Serialize();
 }
 
 std::unique_ptr<base::Value> HeadlessDevToolsManagerDelegate::CloseTarget(
     const base::DictionaryValue* params) {
   std::string target_id;
   if (!params->GetString("targetId", &target_id)) {
@@ skipping 51 matching lines @@
     if (success)
       browser_context_map_.erase(find_it);
   }
   return browser::DisposeBrowserContextResult::Builder()
       .SetSuccess(success)
       .Build()
       ->Serialize();
 }
 
 }  // namespace headless