| 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/printing/print_view_manager.h" | 5 #include "chrome/browser/printing/print_view_manager.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 #include <utility> | |
| 9 | 8 |
| 10 #include "base/bind.h" | 9 #include "base/bind.h" |
| 11 #include "base/lazy_instance.h" | 10 #include "base/lazy_instance.h" |
| 12 #include "base/memory/ptr_util.h" | |
| 13 #include "chrome/browser/plugins/chrome_plugin_service_filter.h" | 11 #include "chrome/browser/plugins/chrome_plugin_service_filter.h" |
| 14 #include "chrome/browser/printing/print_preview_dialog_controller.h" | 12 #include "chrome/browser/printing/print_preview_dialog_controller.h" |
| 15 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h" | 13 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h" |
| 16 #include "chrome/common/chrome_content_client.h" | 14 #include "chrome/common/chrome_content_client.h" |
| 17 #include "components/printing/common/print_messages.h" | 15 #include "components/printing/common/print_messages.h" |
| 18 #include "content/public/browser/browser_thread.h" | 16 #include "content/public/browser/browser_thread.h" |
| 19 #include "content/public/browser/plugin_service.h" | 17 #include "content/public/browser/plugin_service.h" |
| 20 #include "content/public/browser/render_frame_host.h" | 18 #include "content/public/browser/render_frame_host.h" |
| 21 #include "content/public/browser/render_process_host.h" | 19 #include "content/public/browser/render_process_host.h" |
| 22 #include "content/public/browser/web_contents.h" | 20 #include "content/public/browser/web_contents.h" |
| (...skipping 27 matching lines...) Expand all Loading... |
| 50 render_process_id, render_frame_id, GURL(), pdf_plugin); | 48 render_process_id, render_frame_id, GURL(), pdf_plugin); |
| 51 } | 49 } |
| 52 | 50 |
| 53 } // namespace | 51 } // namespace |
| 54 | 52 |
| 55 namespace printing { | 53 namespace printing { |
| 56 | 54 |
| 57 PrintViewManager::PrintViewManager(content::WebContents* web_contents) | 55 PrintViewManager::PrintViewManager(content::WebContents* web_contents) |
| 58 : PrintViewManagerBase(web_contents), | 56 : PrintViewManagerBase(web_contents), |
| 59 print_preview_state_(NOT_PREVIEWING), | 57 print_preview_state_(NOT_PREVIEWING), |
| 60 print_preview_rfh_(nullptr), | |
| 61 scripted_print_preview_rph_(nullptr) { | 58 scripted_print_preview_rph_(nullptr) { |
| 62 if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents)) { | 59 if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents)) { |
| 63 EnableInternalPDFPluginForContents( | 60 EnableInternalPDFPluginForContents( |
| 64 web_contents->GetRenderProcessHost()->GetID(), | 61 web_contents->GetRenderProcessHost()->GetID(), |
| 65 web_contents->GetMainFrame()->GetRoutingID()); | 62 web_contents->GetMainFrame()->GetRoutingID()); |
| 66 } | 63 } |
| 67 } | 64 } |
| 68 | 65 |
| 69 PrintViewManager::~PrintViewManager() { | 66 PrintViewManager::~PrintViewManager() { |
| 70 DCHECK_EQ(NOT_PREVIEWING, print_preview_state_); | 67 DCHECK_EQ(NOT_PREVIEWING, print_preview_state_); |
| 71 } | 68 } |
| 72 | 69 |
| 73 #if BUILDFLAG(ENABLE_BASIC_PRINTING) | 70 #if BUILDFLAG(ENABLE_BASIC_PRINTING) |
| 74 bool PrintViewManager::PrintForSystemDialogNow( | 71 bool PrintViewManager::PrintForSystemDialogNow( |
| 75 const base::Closure& dialog_shown_callback) { | 72 const base::Closure& dialog_shown_callback) { |
| 76 DCHECK(!dialog_shown_callback.is_null()); | 73 DCHECK(!dialog_shown_callback.is_null()); |
| 77 DCHECK(on_print_dialog_shown_callback_.is_null()); | 74 DCHECK(on_print_dialog_shown_callback_.is_null()); |
| 78 on_print_dialog_shown_callback_ = dialog_shown_callback; | 75 on_print_dialog_shown_callback_ = dialog_shown_callback; |
| 79 | 76 return PrintNowInternal(new PrintMsg_PrintForSystemDialog(routing_id())); |
| 80 SetPrintingRFH(print_preview_rfh_); | |
| 81 int32_t id = print_preview_rfh_->GetRoutingID(); | |
| 82 return PrintNowInternal(print_preview_rfh_, | |
| 83 base::MakeUnique<PrintMsg_PrintForSystemDialog>(id)); | |
| 84 } | 77 } |
| 85 | 78 |
| 86 bool PrintViewManager::BasicPrint(content::RenderFrameHost* rfh) { | 79 bool PrintViewManager::BasicPrint() { |
| 87 PrintPreviewDialogController* dialog_controller = | 80 PrintPreviewDialogController* dialog_controller = |
| 88 PrintPreviewDialogController::GetInstance(); | 81 PrintPreviewDialogController::GetInstance(); |
| 89 if (!dialog_controller) | 82 if (!dialog_controller) |
| 90 return false; | 83 return false; |
| 91 | 84 |
| 92 content::WebContents* print_preview_dialog = | 85 content::WebContents* print_preview_dialog = |
| 93 dialog_controller->GetPrintPreviewForContents(web_contents()); | 86 dialog_controller->GetPrintPreviewForContents(web_contents()); |
| 94 if (!print_preview_dialog) | 87 if (!print_preview_dialog) |
| 95 return PrintNow(rfh); | 88 return PrintNow(); |
| 96 | 89 |
| 97 if (!print_preview_dialog->GetWebUI()) | 90 if (!print_preview_dialog->GetWebUI()) |
| 98 return false; | 91 return false; |
| 99 | 92 |
| 100 PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>( | 93 PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>( |
| 101 print_preview_dialog->GetWebUI()->GetController()); | 94 print_preview_dialog->GetWebUI()->GetController()); |
| 102 print_preview_ui->OnShowSystemDialog(); | 95 print_preview_ui->OnShowSystemDialog(); |
| 103 return true; | 96 return true; |
| 104 } | 97 } |
| 105 #endif // BUILDFLAG(ENABLE_BASIC_PRINTING) | 98 #endif // BUILDFLAG(ENABLE_BASIC_PRINTING) |
| 106 | 99 |
| 107 bool PrintViewManager::PrintPreviewNow(content::RenderFrameHost* rfh, | 100 bool PrintViewManager::PrintPreviewNow(bool selection_only) { |
| 108 bool has_selection) { | |
| 109 // Users can send print commands all they want and it is beyond | 101 // Users can send print commands all they want and it is beyond |
| 110 // PrintViewManager's control. Just ignore the extra commands. | 102 // PrintViewManager's control. Just ignore the extra commands. |
| 111 // See http://crbug.com/136842 for example. | 103 // See http://crbug.com/136842 for example. |
| 112 if (print_preview_state_ != NOT_PREVIEWING) | 104 if (print_preview_state_ != NOT_PREVIEWING) |
| 113 return false; | 105 return false; |
| 114 | 106 |
| 115 auto message = base::MakeUnique<PrintMsg_InitiatePrintPreview>( | 107 if (!PrintNowInternal(new PrintMsg_InitiatePrintPreview(routing_id(), |
| 116 rfh->GetRoutingID(), has_selection); | 108 selection_only))) { |
| 117 if (!PrintNowInternal(rfh, std::move(message))) | |
| 118 return false; | 109 return false; |
| 110 } |
| 119 | 111 |
| 120 DCHECK(!print_preview_rfh_); | |
| 121 print_preview_rfh_ = rfh; | |
| 122 print_preview_state_ = USER_INITIATED_PREVIEW; | 112 print_preview_state_ = USER_INITIATED_PREVIEW; |
| 123 return true; | 113 return true; |
| 124 } | 114 } |
| 125 | 115 |
| 126 void PrintViewManager::PrintPreviewForWebNode() { | 116 void PrintViewManager::PrintPreviewForWebNode() { |
| 127 if (print_preview_state_ != NOT_PREVIEWING) | 117 if (print_preview_state_ != NOT_PREVIEWING) |
| 128 return; | 118 return; |
| 129 print_preview_state_ = USER_INITIATED_PREVIEW; | 119 print_preview_state_ = USER_INITIATED_PREVIEW; |
| 130 } | 120 } |
| 131 | 121 |
| 132 void PrintViewManager::PrintPreviewDone() { | 122 void PrintViewManager::PrintPreviewDone() { |
| 133 DCHECK_CURRENTLY_ON(BrowserThread::UI); | 123 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 134 DCHECK_NE(NOT_PREVIEWING, print_preview_state_); | 124 DCHECK_NE(NOT_PREVIEWING, print_preview_state_); |
| 135 | 125 |
| 136 if (print_preview_state_ == SCRIPTED_PREVIEW) { | 126 if (print_preview_state_ == SCRIPTED_PREVIEW) { |
| 137 auto& map = g_scripted_print_preview_closure_map.Get(); | 127 auto& map = g_scripted_print_preview_closure_map.Get(); |
| 138 auto it = map.find(scripted_print_preview_rph_); | 128 auto it = map.find(scripted_print_preview_rph_); |
| 139 CHECK(it != map.end()); | 129 CHECK(it != map.end()); |
| 140 it->second.Run(); | 130 it->second.Run(); |
| 141 map.erase(it); | 131 map.erase(it); |
| 142 scripted_print_preview_rph_ = nullptr; | 132 scripted_print_preview_rph_ = nullptr; |
| 143 } | 133 } |
| 144 print_preview_state_ = NOT_PREVIEWING; | 134 print_preview_state_ = NOT_PREVIEWING; |
| 145 print_preview_rfh_ = nullptr; | |
| 146 } | 135 } |
| 147 | 136 |
| 148 void PrintViewManager::RenderFrameCreated( | 137 void PrintViewManager::RenderFrameCreated( |
| 149 content::RenderFrameHost* render_frame_host) { | 138 content::RenderFrameHost* render_frame_host) { |
| 150 if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents())) { | 139 if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents())) { |
| 151 EnableInternalPDFPluginForContents(render_frame_host->GetProcess()->GetID(), | 140 EnableInternalPDFPluginForContents(render_frame_host->GetProcess()->GetID(), |
| 152 render_frame_host->GetRoutingID()); | 141 render_frame_host->GetRoutingID()); |
| 153 } | 142 } |
| 154 } | 143 } |
| 155 | 144 |
| 156 void PrintViewManager::RenderFrameDeleted( | 145 void PrintViewManager::RenderProcessGone(base::TerminationStatus status) { |
| 157 content::RenderFrameHost* render_frame_host) { | 146 print_preview_state_ = NOT_PREVIEWING; |
| 158 if (render_frame_host == print_preview_rfh_) | 147 PrintViewManagerBase::RenderProcessGone(status); |
| 159 print_preview_state_ = NOT_PREVIEWING; | |
| 160 PrintViewManagerBase::RenderFrameDeleted(render_frame_host); | |
| 161 } | 148 } |
| 162 | 149 |
| 163 void PrintViewManager::OnDidShowPrintDialog(content::RenderFrameHost* rfh) { | 150 void PrintViewManager::OnDidShowPrintDialog() { |
| 164 if (rfh != print_preview_rfh_) | 151 if (!on_print_dialog_shown_callback_.is_null()) |
| 165 return; | 152 on_print_dialog_shown_callback_.Run(); |
| 166 | |
| 167 if (on_print_dialog_shown_callback_.is_null()) | |
| 168 return; | |
| 169 | |
| 170 on_print_dialog_shown_callback_.Run(); | |
| 171 on_print_dialog_shown_callback_.Reset(); | 153 on_print_dialog_shown_callback_.Reset(); |
| 172 } | 154 } |
| 173 | 155 |
| 174 void PrintViewManager::OnSetupScriptedPrintPreview( | 156 void PrintViewManager::OnSetupScriptedPrintPreview(IPC::Message* reply_msg) { |
| 175 content::RenderFrameHost* rfh, | |
| 176 IPC::Message* reply_msg) { | |
| 177 DCHECK_CURRENTLY_ON(BrowserThread::UI); | 157 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 178 auto& map = g_scripted_print_preview_closure_map.Get(); | 158 auto& map = g_scripted_print_preview_closure_map.Get(); |
| 179 content::RenderProcessHost* rph = rfh->GetProcess(); | 159 content::RenderProcessHost* rph = web_contents()->GetRenderProcessHost(); |
| 180 | 160 |
| 181 if (base::ContainsKey(map, rph)) { | 161 if (base::ContainsKey(map, rph)) { |
| 182 // Renderer already handling window.print(). Abort this attempt to prevent | 162 // Renderer already handling window.print() in another View. |
| 183 // the renderer from having multiple nested loops. If multiple nested loops | 163 Send(reply_msg); |
| 184 // existed, then they have to exit in the right order and that is messy. | |
| 185 rfh->Send(reply_msg); | |
| 186 return; | 164 return; |
| 187 } | 165 } |
| 188 | 166 |
| 189 if (print_preview_state_ != NOT_PREVIEWING) { | 167 if (print_preview_state_ != NOT_PREVIEWING) { |
| 190 // If a print dialog is already open for this tab, ignore the scripted print | 168 // If a user initiated print dialog is already open, ignore the scripted |
| 191 // message. | 169 // print message. |
| 192 rfh->Send(reply_msg); | 170 DCHECK_EQ(USER_INITIATED_PREVIEW, print_preview_state_); |
| 171 Send(reply_msg); |
| 193 return; | 172 return; |
| 194 } | 173 } |
| 195 | 174 |
| 196 PrintPreviewDialogController* dialog_controller = | 175 PrintPreviewDialogController* dialog_controller = |
| 197 PrintPreviewDialogController::GetInstance(); | 176 PrintPreviewDialogController::GetInstance(); |
| 198 if (!dialog_controller) { | 177 if (!dialog_controller) { |
| 199 rfh->Send(reply_msg); | 178 Send(reply_msg); |
| 200 return; | 179 return; |
| 201 } | 180 } |
| 202 | 181 |
| 203 DCHECK(!print_preview_rfh_); | |
| 204 print_preview_rfh_ = rfh; | |
| 205 print_preview_state_ = SCRIPTED_PREVIEW; | 182 print_preview_state_ = SCRIPTED_PREVIEW; |
| 206 map[rph] = base::Bind(&PrintViewManager::OnScriptedPrintPreviewReply, | 183 map[rph] = base::Bind(&PrintViewManager::OnScriptedPrintPreviewReply, |
| 207 base::Unretained(this), reply_msg); | 184 base::Unretained(this), reply_msg); |
| 208 scripted_print_preview_rph_ = rph; | 185 scripted_print_preview_rph_ = rph; |
| 209 } | 186 } |
| 210 | 187 |
| 211 void PrintViewManager::OnShowScriptedPrintPreview(content::RenderFrameHost* rfh, | 188 void PrintViewManager::OnShowScriptedPrintPreview(bool source_is_modifiable) { |
| 212 bool source_is_modifiable) { | |
| 213 DCHECK(print_preview_rfh_); | |
| 214 if (rfh != print_preview_rfh_) | |
| 215 return; | |
| 216 | |
| 217 PrintPreviewDialogController* dialog_controller = | 189 PrintPreviewDialogController* dialog_controller = |
| 218 PrintPreviewDialogController::GetInstance(); | 190 PrintPreviewDialogController::GetInstance(); |
| 219 if (!dialog_controller) { | 191 if (!dialog_controller) { |
| 220 PrintPreviewDone(); | 192 PrintPreviewDone(); |
| 221 return; | 193 return; |
| 222 } | 194 } |
| 223 | 195 |
| 224 dialog_controller->PrintPreview(web_contents()); | 196 dialog_controller->PrintPreview(web_contents()); |
| 225 PrintHostMsg_RequestPrintPreview_Params params; | 197 PrintHostMsg_RequestPrintPreview_Params params; |
| 226 params.is_modifiable = source_is_modifiable; | 198 params.is_modifiable = source_is_modifiable; |
| 227 PrintPreviewUI::SetInitialParams( | 199 PrintPreviewUI::SetInitialParams( |
| 228 dialog_controller->GetPrintPreviewForContents(web_contents()), params); | 200 dialog_controller->GetPrintPreviewForContents(web_contents()), params); |
| 229 } | 201 } |
| 230 | 202 |
| 231 void PrintViewManager::OnScriptedPrintPreviewReply(IPC::Message* reply_msg) { | 203 void PrintViewManager::OnScriptedPrintPreviewReply(IPC::Message* reply_msg) { |
| 232 DCHECK_CURRENTLY_ON(BrowserThread::UI); | 204 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 233 print_preview_rfh_->Send(reply_msg); | 205 Send(reply_msg); |
| 234 } | 206 } |
| 235 | 207 |
| 236 bool PrintViewManager::OnMessageReceived( | 208 bool PrintViewManager::OnMessageReceived(const IPC::Message& message) { |
| 237 const IPC::Message& message, | |
| 238 content::RenderFrameHost* render_frame_host) { | |
| 239 bool handled = true; | 209 bool handled = true; |
| 240 IPC_BEGIN_MESSAGE_MAP_WITH_PARAM(PrintViewManager, message, render_frame_host) | 210 IPC_BEGIN_MESSAGE_MAP(PrintViewManager, message) |
| 241 IPC_MESSAGE_HANDLER(PrintHostMsg_DidShowPrintDialog, OnDidShowPrintDialog) | 211 IPC_MESSAGE_HANDLER(PrintHostMsg_DidShowPrintDialog, OnDidShowPrintDialog) |
| 242 IPC_MESSAGE_HANDLER_WITH_PARAM_DELAY_REPLY( | 212 IPC_MESSAGE_HANDLER_DELAY_REPLY(PrintHostMsg_SetupScriptedPrintPreview, |
| 243 PrintHostMsg_SetupScriptedPrintPreview, OnSetupScriptedPrintPreview) | 213 OnSetupScriptedPrintPreview) |
| 244 IPC_MESSAGE_HANDLER(PrintHostMsg_ShowScriptedPrintPreview, | 214 IPC_MESSAGE_HANDLER(PrintHostMsg_ShowScriptedPrintPreview, |
| 245 OnShowScriptedPrintPreview) | 215 OnShowScriptedPrintPreview) |
| 246 IPC_MESSAGE_UNHANDLED(handled = false) | 216 IPC_MESSAGE_UNHANDLED(handled = false) |
| 247 IPC_END_MESSAGE_MAP() | 217 IPC_END_MESSAGE_MAP() |
| 248 | 218 |
| 249 return handled || | 219 return handled || PrintViewManagerBase::OnMessageReceived(message); |
| 250 PrintViewManagerBase::OnMessageReceived(message, render_frame_host); | |
| 251 } | 220 } |
| 252 | 221 |
| 253 } // namespace printing | 222 } // namespace printing |
| OLD | NEW |