OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 COMPONENTS_DOM_DISTILLER_CONTENT_DISTILLER_PAGE_WEB_CONTENTS_H_ | 5 #ifndef COMPONENTS_DOM_DISTILLER_CONTENT_DISTILLER_PAGE_WEB_CONTENTS_H_ |
6 #define COMPONENTS_DOM_DISTILLER_CONTENT_DISTILLER_PAGE_WEB_CONTENTS_H_ | 6 #define COMPONENTS_DOM_DISTILLER_CONTENT_DISTILLER_PAGE_WEB_CONTENTS_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
11 #include "components/dom_distiller/core/distiller_page.h" | 11 #include "components/dom_distiller/core/distiller_page.h" |
12 #include "content/public/browser/web_contents.h" | 12 #include "content/public/browser/web_contents.h" |
13 #include "content/public/browser/web_contents_observer.h" | 13 #include "content/public/browser/web_contents_observer.h" |
14 #include "url/gurl.h" | 14 #include "url/gurl.h" |
15 | 15 |
16 namespace content { | 16 namespace content { |
17 class RenderViewHost; | 17 class RenderViewHost; |
18 } | 18 } |
19 | 19 |
20 using content::RenderViewHost; | 20 using content::RenderViewHost; |
21 | 21 |
22 namespace dom_distiller { | 22 namespace dom_distiller { |
23 | 23 |
24 class SourcePageHandleWebContents : public SourcePageHandle { | 24 class DistillerContext; |
25 public: | |
26 explicit SourcePageHandleWebContents( | |
27 scoped_ptr<content::WebContents> web_contents); | |
28 virtual ~SourcePageHandleWebContents(); | |
29 | |
30 scoped_ptr<content::WebContents> GetWebContents(); | |
31 | |
32 private: | |
33 // The WebContents this class owns. | |
34 scoped_ptr<content::WebContents> web_contents_; | |
35 }; | |
36 | 25 |
37 class DistillerPageWebContentsFactory : public DistillerPageFactory { | 26 class DistillerPageWebContentsFactory : public DistillerPageFactory { |
38 public: | 27 public: |
39 explicit DistillerPageWebContentsFactory( | 28 explicit DistillerPageWebContentsFactory( |
40 content::BrowserContext* browser_context) | 29 content::BrowserContext* browser_context) |
41 : DistillerPageFactory(), browser_context_(browser_context) {} | 30 : DistillerPageFactory(), browser_context_(browser_context) {} |
42 virtual ~DistillerPageWebContentsFactory() {} | 31 virtual ~DistillerPageWebContentsFactory() {} |
43 | 32 |
44 virtual scoped_ptr<DistillerPage> CreateDistillerPage() const OVERRIDE; | 33 virtual scoped_ptr<DistillerPage> CreateDistillerPage() const OVERRIDE; |
45 virtual scoped_ptr<DistillerPage> CreateDistillerPageWithHandle( | |
46 scoped_ptr<SourcePageHandle> handle) const OVERRIDE; | |
47 | 34 |
48 private: | 35 private: |
49 content::BrowserContext* browser_context_; | 36 content::BrowserContext* browser_context_; |
50 }; | 37 }; |
51 | 38 |
52 class DistillerPageWebContents : public DistillerPage, | 39 class DistillerPageWebContents : public DistillerPage, |
53 public content::WebContentsObserver { | 40 public content::WebContentsObserver { |
54 public: | 41 public: |
55 DistillerPageWebContents( | 42 DistillerPageWebContents(content::BrowserContext* browser_context); |
56 content::BrowserContext* browser_context, | |
57 scoped_ptr<SourcePageHandleWebContents> optional_web_contents_handle); | |
58 virtual ~DistillerPageWebContents(); | 43 virtual ~DistillerPageWebContents(); |
59 | 44 |
60 // content::WebContentsObserver implementation. | 45 // content::WebContentsObserver implementation. |
61 virtual void DocumentLoadedInFrame(int64 frame_id, | 46 virtual void DocumentLoadedInFrame(int64 frame_id, |
62 RenderViewHost* render_view_host) OVERRIDE; | 47 RenderViewHost* render_view_host) OVERRIDE; |
63 | 48 |
64 virtual void DidFailLoad(int64 frame_id, | 49 virtual void DidFailLoad(int64 frame_id, |
65 const GURL& validated_url, | 50 const GURL& validated_url, |
66 bool is_main_frame, | 51 bool is_main_frame, |
67 int error_code, | 52 int error_code, |
68 const base::string16& error_description, | 53 const base::string16& error_description, |
69 RenderViewHost* render_view_host) OVERRIDE; | 54 RenderViewHost* render_view_host) OVERRIDE; |
70 | 55 |
71 protected: | 56 protected: |
72 virtual void DistillPageImpl(const GURL& url, | 57 virtual void DistillPageImpl(const GURL& url, |
73 const std::string& script) OVERRIDE; | 58 const std::string& script) OVERRIDE; |
74 | 59 |
75 private: | 60 private: |
76 friend class TestDistillerPageWebContents; | |
77 | |
78 enum State { | 61 enum State { |
79 // The page distiller is idle. | 62 // The page distiller is idle. |
80 IDLE, | 63 IDLE, |
81 // A page is currently loading. | 64 // A page is currently loading. |
82 LOADING_PAGE, | 65 LOADING_PAGE, |
83 // There was an error processing the page. | 66 // There was an error processing the page. |
84 PAGELOAD_FAILED, | 67 PAGELOAD_FAILED, |
85 // JavaScript is executing within the context of the page. When the | 68 // JavaScript is executing within the context of the page. When the |
86 // JavaScript completes, the state will be returned to |IDLE|. | 69 // JavaScript completes, the state will be returned to |IDLE|. |
87 EXECUTING_JAVASCRIPT | 70 EXECUTING_JAVASCRIPT |
88 }; | 71 }; |
89 | 72 |
90 // Creates a new WebContents, adds |this| as an observer, and loads the | |
91 // |url|. | |
92 virtual void CreateNewWebContents(const GURL& url); | |
93 | |
94 // Injects and executes JavaScript in the context of a loaded page. This | 73 // Injects and executes JavaScript in the context of a loaded page. This |
95 // must only be called after the page has successfully loaded. | 74 // must only be called after the page has successfully loaded. |
96 void ExecuteJavaScript(); | 75 void ExecuteJavaScript(); |
97 | 76 |
98 // Called when the distillation is done or if the page load failed. | 77 // Called when the distillation is done or if the page load failed. |
99 void OnWebContentsDistillationDone(const GURL& page_url, | 78 void OnWebContentsDistillationDone(const GURL& page_url, |
100 const base::Value* value); | 79 const base::Value* value); |
101 | 80 |
102 // The current state of the |DistillerPage|, initially |IDLE|. | 81 // The current state of the |DistillerPage|, initially |IDLE|. |
103 State state_; | 82 State state_; |
104 | 83 |
105 // The JavaScript to inject to extract content. | 84 // The JavaScript to inject to extract content. |
106 std::string script_; | 85 std::string script_; |
107 | 86 |
108 scoped_ptr<content::WebContents> web_contents_; | 87 scoped_ptr<content::WebContents> web_contents_; |
109 content::BrowserContext* browser_context_; | 88 content::BrowserContext* browser_context_; |
110 DISALLOW_COPY_AND_ASSIGN(DistillerPageWebContents); | 89 DISALLOW_COPY_AND_ASSIGN(DistillerPageWebContents); |
111 }; | 90 }; |
112 | 91 |
113 } // namespace dom_distiller | 92 } // namespace dom_distiller |
114 | 93 |
115 #endif // COMPONENTS_DOM_DISTILLER_CONTENT_DISTILLER_PAGE_WEB_CONTENTS_H_ | 94 #endif // COMPONENTS_DOM_DISTILLER_CONTENT_DISTILLER_PAGE_WEB_CONTENTS_H_ |
OLD | NEW |