OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 #include "chrome/browser/extensions/extension_page_capture_api.h" | 5 #include "chrome/browser/extensions/extension_page_capture_api.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/file_util.h" | 8 #include "base/file_util.h" |
9 #include "chrome/browser/browser_process.h" | 9 #include "chrome/browser/browser_process.h" |
10 #include "chrome/browser/extensions/extension_tab_util.h" | 10 #include "chrome/browser/extensions/extension_tab_util.h" |
11 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" | 11 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" |
12 #include "chrome/common/extensions/extension_messages.h" | 12 #include "chrome/common/extensions/extension_messages.h" |
13 #include "content/browser/renderer_host/render_view_host.h" | 13 #include "content/browser/renderer_host/render_view_host.h" |
14 #include "content/browser/download/mhtml_generation_manager.h" | |
15 #include "content/public/browser/child_process_security_policy.h" | 14 #include "content/public/browser/child_process_security_policy.h" |
16 #include "content/public/browser/notification_details.h" | 15 #include "content/public/browser/notification_details.h" |
17 #include "content/public/browser/notification_source.h" | 16 #include "content/public/browser/notification_source.h" |
18 #include "content/public/browser/notification_types.h" | 17 #include "content/public/browser/notification_types.h" |
19 #include "content/public/browser/render_process_host.h" | 18 #include "content/public/browser/render_process_host.h" |
20 #include "content/public/browser/web_contents.h" | 19 #include "content/public/browser/web_contents.h" |
21 | 20 |
22 using content::BrowserThread; | 21 using content::BrowserThread; |
23 using content::ChildProcessSecurityPolicy; | 22 using content::ChildProcessSecurityPolicy; |
24 using content::WebContents; | 23 using content::WebContents; |
(...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
120 | 119 |
121 if (test_delegate_) | 120 if (test_delegate_) |
122 test_delegate_->OnTemporaryFileCreated(mhtml_path_); | 121 test_delegate_->OnTemporaryFileCreated(mhtml_path_); |
123 | 122 |
124 WebContents* web_contents = GetWebContents(); | 123 WebContents* web_contents = GetWebContents(); |
125 if (!web_contents) { | 124 if (!web_contents) { |
126 ReturnFailure(kTabClosedError); | 125 ReturnFailure(kTabClosedError); |
127 return; | 126 return; |
128 } | 127 } |
129 | 128 |
130 MHTMLGenerationManager::GenerateMHTMLCallback callback = | 129 web_contents->GenerateMHTML( |
131 base::Bind(&PageCaptureSaveAsMHTMLFunction::MHTMLGenerated, this); | 130 mhtml_path_, |
132 | 131 base::Bind(&PageCaptureSaveAsMHTMLFunction::MHTMLGenerated, this)); |
133 g_browser_process->mhtml_generation_manager()->GenerateMHTML( | |
134 web_contents, mhtml_path_, callback); | |
135 } | 132 } |
136 | 133 |
137 void PageCaptureSaveAsMHTMLFunction::MHTMLGenerated(const FilePath& file_path, | 134 void PageCaptureSaveAsMHTMLFunction::MHTMLGenerated(const FilePath& file_path, |
138 int64 mhtml_file_size) { | 135 int64 mhtml_file_size) { |
139 DCHECK(mhtml_path_ == file_path); | 136 DCHECK(mhtml_path_ == file_path); |
140 if (mhtml_file_size <= 0) { | 137 if (mhtml_file_size <= 0) { |
141 ReturnFailure(kMHTMLGenerationFailedError); | 138 ReturnFailure(kMHTMLGenerationFailedError); |
142 return; | 139 return; |
143 } | 140 } |
144 | 141 |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
188 WebContents* PageCaptureSaveAsMHTMLFunction::GetWebContents() { | 185 WebContents* PageCaptureSaveAsMHTMLFunction::GetWebContents() { |
189 Browser* browser = NULL; | 186 Browser* browser = NULL; |
190 TabContentsWrapper* tab_contents_wrapper = NULL; | 187 TabContentsWrapper* tab_contents_wrapper = NULL; |
191 | 188 |
192 if (!ExtensionTabUtil::GetTabById(tab_id_, profile(), include_incognito(), | 189 if (!ExtensionTabUtil::GetTabById(tab_id_, profile(), include_incognito(), |
193 &browser, NULL, &tab_contents_wrapper, NULL)) { | 190 &browser, NULL, &tab_contents_wrapper, NULL)) { |
194 return NULL; | 191 return NULL; |
195 } | 192 } |
196 return tab_contents_wrapper->web_contents(); | 193 return tab_contents_wrapper->web_contents(); |
197 } | 194 } |
OLD | NEW |