OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_IMPL_H_ | 5 #ifndef HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_IMPL_H_ |
6 #define HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_IMPL_H_ | 6 #define HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_IMPL_H_ |
7 | 7 |
8 #include "headless/public/headless_browser.h" | 8 #include "headless/public/headless_browser.h" |
9 | 9 |
10 #include <memory> | 10 #include <memory> |
| 11 #include <string> |
11 #include <unordered_map> | 12 #include <unordered_map> |
12 #include <vector> | 13 #include <vector> |
13 | 14 |
| 15 #include "base/memory/weak_ptr.h" |
| 16 #include "headless/lib/browser/headless_devtools_manager_delegate.h" |
14 #include "headless/lib/browser/headless_web_contents_impl.h" | 17 #include "headless/lib/browser/headless_web_contents_impl.h" |
15 | 18 |
16 namespace aura { | 19 namespace aura { |
17 class WindowTreeHost; | 20 class WindowTreeHost; |
18 | 21 |
19 namespace client { | 22 namespace client { |
20 class WindowTreeClient; | 23 class WindowTreeClient; |
21 } | 24 } |
22 } | 25 } |
23 | 26 |
24 namespace headless { | 27 namespace headless { |
25 | 28 |
26 class HeadlessBrowserContext; | 29 class HeadlessBrowserContext; |
27 class HeadlessBrowserMainParts; | 30 class HeadlessBrowserMainParts; |
28 | 31 |
29 class HeadlessBrowserImpl : public HeadlessBrowser { | 32 class HeadlessBrowserImpl : public HeadlessBrowser { |
30 public: | 33 public: |
31 HeadlessBrowserImpl( | 34 HeadlessBrowserImpl( |
32 const base::Callback<void(HeadlessBrowser*)>& on_start_callback, | 35 const base::Callback<void(HeadlessBrowser*)>& on_start_callback, |
33 HeadlessBrowser::Options options); | 36 HeadlessBrowser::Options options); |
34 ~HeadlessBrowserImpl() override; | 37 ~HeadlessBrowserImpl() override; |
35 | 38 |
36 // HeadlessBrowser implementation: | 39 // HeadlessBrowser implementation: |
37 HeadlessWebContents::Builder CreateWebContentsBuilder() override; | |
38 HeadlessBrowserContext::Builder CreateBrowserContextBuilder() override; | 40 HeadlessBrowserContext::Builder CreateBrowserContextBuilder() override; |
39 HeadlessWebContents* CreateWebContents(const GURL& initial_url, | |
40 const gfx::Size& size) override; | |
41 scoped_refptr<base::SingleThreadTaskRunner> BrowserMainThread() | 41 scoped_refptr<base::SingleThreadTaskRunner> BrowserMainThread() |
42 const override; | 42 const override; |
43 scoped_refptr<base::SingleThreadTaskRunner> BrowserFileThread() | 43 scoped_refptr<base::SingleThreadTaskRunner> BrowserFileThread() |
44 const override; | 44 const override; |
45 | 45 |
46 void Shutdown() override; | 46 void Shutdown() override; |
47 | 47 |
48 std::vector<HeadlessWebContents*> GetAllWebContents() override; | 48 std::vector<HeadlessWebContents*> GetAllWebContents() override; |
49 HeadlessWebContents* GetWebContentsForDevtoolsAgentHostId( | 49 HeadlessWebContents* GetWebContentsForDevtoolsAgentHostId( |
50 const std::string& devtools_agent_host_id) override; | 50 const std::string& devtools_agent_host_id) override; |
51 | 51 |
52 void set_browser_main_parts(HeadlessBrowserMainParts* browser_main_parts); | 52 void set_browser_main_parts(HeadlessBrowserMainParts* browser_main_parts); |
53 HeadlessBrowserMainParts* browser_main_parts() const; | 53 HeadlessBrowserMainParts* browser_main_parts() const; |
54 | 54 |
55 void RunOnStartCallback(); | 55 void RunOnStartCallback(); |
56 | 56 |
57 HeadlessBrowser::Options* options() { return &options_; } | 57 HeadlessBrowser::Options* options() { return &options_; } |
58 | 58 |
59 HeadlessWebContents* CreateWebContents(HeadlessWebContents::Builder* builder); | 59 HeadlessWebContents* CreateWebContents(HeadlessWebContents::Builder* builder); |
60 HeadlessWebContentsImpl* RegisterWebContents( | 60 HeadlessWebContentsImpl* RegisterWebContents( |
61 std::unique_ptr<HeadlessWebContentsImpl> web_contents); | 61 std::unique_ptr<HeadlessWebContentsImpl> web_contents); |
62 | 62 |
63 // Close given |web_contents| and delete it. | 63 // Close given |web_contents| and delete it. |
64 void DestroyWebContents(HeadlessWebContentsImpl* web_contents); | 64 void DestroyWebContents(HeadlessWebContentsImpl* web_contents); |
65 | 65 |
66 // Customize the options used by this headless browser instance. Note that | 66 HeadlessDevToolsManagerDelegate* devtools_manager_delegate() const; |
67 // options which take effect before the message loop has been started (e.g., | 67 void set_devtools_manager_delegate( |
68 // custom message pumps) cannot be set via this method. | 68 base::WeakPtr<HeadlessDevToolsManagerDelegate>); |
69 void SetOptionsForTesting(HeadlessBrowser::Options options); | |
70 | 69 |
71 protected: | 70 protected: |
72 base::Callback<void(HeadlessBrowser*)> on_start_callback_; | 71 base::Callback<void(HeadlessBrowser*)> on_start_callback_; |
73 HeadlessBrowser::Options options_; | 72 HeadlessBrowser::Options options_; |
74 HeadlessBrowserMainParts* browser_main_parts_; // Not owned. | 73 HeadlessBrowserMainParts* browser_main_parts_; // Not owned. |
75 std::unique_ptr<aura::WindowTreeHost> window_tree_host_; | 74 std::unique_ptr<aura::WindowTreeHost> window_tree_host_; |
76 std::unique_ptr<aura::client::WindowTreeClient> window_tree_client_; | 75 std::unique_ptr<aura::client::WindowTreeClient> window_tree_client_; |
77 | 76 |
78 std::unordered_map<std::string, std::unique_ptr<HeadlessWebContents>> | 77 std::unordered_map<std::string, std::unique_ptr<HeadlessWebContents>> |
79 web_contents_map_; | 78 web_contents_map_; |
80 | 79 |
| 80 base::WeakPtr<HeadlessDevToolsManagerDelegate> devtools_manager_delegate_; |
| 81 |
81 private: | 82 private: |
82 DISALLOW_COPY_AND_ASSIGN(HeadlessBrowserImpl); | 83 DISALLOW_COPY_AND_ASSIGN(HeadlessBrowserImpl); |
83 }; | 84 }; |
84 | 85 |
85 } // namespace headless | 86 } // namespace headless |
86 | 87 |
87 #endif // HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_IMPL_H_ | 88 #endif // HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_IMPL_H_ |
OLD | NEW |