headless: Initial headless embedder API implementation

headless/lib/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/headless_browser_impl.h"
+
+#include "base/thread_task_runner_handle.h"
+#include "content/public/app/content_main.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/web_contents.h"
+#include "headless/lib/browser/headless_browser_context.h"
+#include "headless/lib/browser/headless_browser_main_parts.h"
+#include "headless/lib/headless_content_main_delegate.h"
+#include "headless/lib/headless_web_contents_impl.h"
+#include "ui/gfx/geometry/size.h"
+
+namespace headless {
+
+HeadlessBrowserImpl::HeadlessBrowserImpl(
+    const StartCallback& on_start_callback,
+    const HeadlessBrowser::Options& options)
+    : on_start_callback_(on_start_callback),
+      options_(options),
+      browser_main_parts_(nullptr) {}
+
+HeadlessBrowserImpl::~HeadlessBrowserImpl() {}
+
+scoped_ptr<HeadlessWebContents> HeadlessBrowserImpl::CreateWebContents(
+    const gfx::Size& size) {
+  DCHECK(BrowserMainThread()->BelongsToCurrentThread());
+  return make_scoped_ptr(new HeadlessWebContentsImpl(browser_context(), size));
+}
+
+scoped_refptr<base::SingleThreadTaskRunner>
+HeadlessBrowserImpl::BrowserMainThread() {
+  return content::BrowserThread::GetMessageLoopProxyForThread(
+      content::BrowserThread::UI);
+}
+
+void HeadlessBrowserImpl::Stop() {
+  DCHECK(BrowserMainThread()->BelongsToCurrentThread());
+  BrowserMainThread()->PostTask(FROM_HERE,
+                                base::MessageLoop::QuitWhenIdleClosure());
+}
+
+HeadlessBrowserContext* HeadlessBrowserImpl::browser_context() {
+  DCHECK(BrowserMainThread()->BelongsToCurrentThread());
+  DCHECK(browser_main_parts());
+  return browser_main_parts()->browser_context();
+}
+
+HeadlessBrowserMainParts* HeadlessBrowserImpl::browser_main_parts() {
+  DCHECK(BrowserMainThread()->BelongsToCurrentThread());
+  return browser_main_parts_;
+}
+
+void HeadlessBrowserImpl::set_browser_main_parts(
+    HeadlessBrowserMainParts* browser_main_parts) {
+  DCHECK(!browser_main_parts_);
+  browser_main_parts_ = browser_main_parts;
+}
+
+void HeadlessBrowserImpl::RunOnStartCallback() {
+  if (!on_start_callback_.is_null()) {
+    on_start_callback_.Run(this);
+    on_start_callback_ = StartCallback();
+  }
+}
+
+int HeadlessBrowser::Run(const Options& options,
+                         const StartCallback& on_browser_start_callback) {
+  scoped_ptr<HeadlessBrowserImpl> browser(
+      new HeadlessBrowserImpl(on_browser_start_callback, options));
+
+  // TODO(skyostil): Implement custom message pumps.
+  DCHECK(!options.message_pump);
+
+  headless::HeadlessContentMainDelegate delegate(std::move(browser));
+  content::ContentMainParams params(&delegate);
+  params.argc = options.argc;
+  params.argv = options.argv;
+  return content::ContentMain(params);
+}
+
+}  // namespace headless