| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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_preview_message_handler.h" | 5 #include "chrome/browser/printing/print_preview_message_handler.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 10 #include "base/memory/ref_counted_memory.h" | 10 #include "base/memory/ref_counted_memory.h" |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 86 const PrintHostMsg_DidGetPreviewPageCount_Params& params) { | 86 const PrintHostMsg_DidGetPreviewPageCount_Params& params) { |
| 87 if (params.page_count <= 0) | 87 if (params.page_count <= 0) |
| 88 return; | 88 return; |
| 89 | 89 |
| 90 TabContents* print_preview_tab = GetPrintPreviewTab(); | 90 TabContents* print_preview_tab = GetPrintPreviewTab(); |
| 91 if (!print_preview_tab || !print_preview_tab->web_ui()) | 91 if (!print_preview_tab || !print_preview_tab->web_ui()) |
| 92 return; | 92 return; |
| 93 | 93 |
| 94 PrintPreviewUI* print_preview_ui = | 94 PrintPreviewUI* print_preview_ui = |
| 95 static_cast<PrintPreviewUI*>(print_preview_tab->web_ui()); | 95 static_cast<PrintPreviewUI*>(print_preview_tab->web_ui()); |
| 96 |
| 97 if (!params.is_modifiable || params.clear_preview_data) |
| 98 print_preview_ui->ClearAllPreviewData(); |
| 99 |
| 96 print_preview_ui->OnDidGetPreviewPageCount(params); | 100 print_preview_ui->OnDidGetPreviewPageCount(params); |
| 97 } | 101 } |
| 98 | 102 |
| 99 void PrintPreviewMessageHandler::OnDidPreviewPage( | 103 void PrintPreviewMessageHandler::OnDidPreviewPage( |
| 100 const PrintHostMsg_DidPreviewPage_Params& params) { | 104 const PrintHostMsg_DidPreviewPage_Params& params) { |
| 101 RenderViewHost* rvh = tab_contents()->render_view_host(); | 105 RenderViewHost* rvh = tab_contents()->render_view_host(); |
| 102 TabContents* print_preview_tab = GetPrintPreviewTab(); | 106 TabContents* print_preview_tab = GetPrintPreviewTab(); |
| 103 if (!print_preview_tab || !print_preview_tab->web_ui()) { | 107 if (!print_preview_tab || !print_preview_tab->web_ui()) { |
| 104 // Can't find print preview tab means we should abort. | 108 // Can't find print preview tab means we should abort. |
| 105 rvh->Send(new PrintMsg_AbortPreview(rvh->routing_id())); | 109 rvh->Send(new PrintMsg_AbortPreview(rvh->routing_id())); |
| 106 return; | 110 return; |
| 107 } | 111 } |
| 108 | 112 |
| 109 PrintPreviewUI* print_preview_ui = | 113 PrintPreviewUI* print_preview_ui = |
| 110 static_cast<PrintPreviewUI*>(print_preview_tab->web_ui()); | 114 static_cast<PrintPreviewUI*>(print_preview_tab->web_ui()); |
| 111 bool has_pending = print_preview_ui->HasPendingRequests(); | 115 bool has_pending = print_preview_ui->HasPendingRequests(); |
| 112 if (has_pending) { | 116 if (has_pending) { |
| 113 // Cancel. Next print preview request will cancel the current one. Just do | 117 // Cancel. Next print preview request will cancel the current one. Just do |
| 114 // the required maintenance work here. | 118 // the required maintenance work here. |
| 115 StopWorker(print_preview_ui->document_cookie()); | 119 StopWorker(print_preview_ui->document_cookie()); |
| 116 print_preview_ui->OnPrintPreviewCancelled(); | 120 print_preview_ui->OnPrintPreviewCancelled(); |
| 117 return; | 121 return; |
| 118 } | 122 } |
| 119 | 123 |
| 120 int requested_preview_page_index = INVALID_PAGE_INDEX; | |
| 121 int page_number = params.page_number; | 124 int page_number = params.page_number; |
| 122 | |
| 123 if (page_number == FIRST_PAGE_INDEX) | |
| 124 print_preview_ui->ClearAllPreviewData(); | |
| 125 | |
| 126 if (page_number >= FIRST_PAGE_INDEX && params.data_size) { | 125 if (page_number >= FIRST_PAGE_INDEX && params.data_size) { |
| 127 RefCountedBytes* data_bytes = | 126 RefCountedBytes* data_bytes = |
| 128 GetDataFromHandle(params.metafile_data_handle, params.data_size); | 127 GetDataFromHandle(params.metafile_data_handle, params.data_size); |
| 129 DCHECK(data_bytes); | 128 DCHECK(data_bytes); |
| 130 | 129 |
| 131 print_preview_ui->SetPrintPreviewDataForIndex(page_number, data_bytes); | 130 print_preview_ui->SetPrintPreviewDataForIndex(page_number, data_bytes); |
| 132 print_preview_ui->OnDidPreviewPage(page_number, params.preview_request_id); | 131 print_preview_ui->OnDidPreviewPage(page_number, params.preview_request_id); |
| 133 // TODO(kmadhusu): Query |PrintPreviewUI| and update | |
| 134 // |requested_preview_page_index| accordingly. | |
| 135 } | 132 } |
| 136 | 133 |
| 137 rvh->Send(new PrintMsg_ContinuePreview(rvh->routing_id(), | 134 rvh->Send(new PrintMsg_ContinuePreview(rvh->routing_id())); |
| 138 requested_preview_page_index)); | |
| 139 } | 135 } |
| 140 | 136 |
| 141 void PrintPreviewMessageHandler::OnPagesReadyForPreview( | 137 void PrintPreviewMessageHandler::OnMetafileReadyForPrinting( |
| 142 const PrintHostMsg_DidPreviewDocument_Params& params) { | 138 const PrintHostMsg_DidPreviewDocument_Params& params) { |
| 143 StopWorker(params.document_cookie); | 139 StopWorker(params.document_cookie); |
| 144 | 140 |
| 145 // Get the print preview tab. | 141 // Get the print preview tab. |
| 146 TabContents* print_preview_tab = GetPrintPreviewTab(); | 142 TabContents* print_preview_tab = GetPrintPreviewTab(); |
| 147 // User might have closed it already. | 143 // User might have closed it already. |
| 148 if (!print_preview_tab) | 144 if (!print_preview_tab) |
| 149 return; | 145 return; |
| 150 | 146 |
| 151 PrintPreviewUI* print_preview_ui = | 147 PrintPreviewUI* print_preview_ui = |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 211 bool PrintPreviewMessageHandler::OnMessageReceived( | 207 bool PrintPreviewMessageHandler::OnMessageReceived( |
| 212 const IPC::Message& message) { | 208 const IPC::Message& message) { |
| 213 bool handled = true; | 209 bool handled = true; |
| 214 IPC_BEGIN_MESSAGE_MAP(PrintPreviewMessageHandler, message) | 210 IPC_BEGIN_MESSAGE_MAP(PrintPreviewMessageHandler, message) |
| 215 IPC_MESSAGE_HANDLER(PrintHostMsg_RequestPrintPreview, | 211 IPC_MESSAGE_HANDLER(PrintHostMsg_RequestPrintPreview, |
| 216 OnRequestPrintPreview) | 212 OnRequestPrintPreview) |
| 217 IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount, | 213 IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount, |
| 218 OnDidGetPreviewPageCount) | 214 OnDidGetPreviewPageCount) |
| 219 IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage, | 215 IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage, |
| 220 OnDidPreviewPage) | 216 OnDidPreviewPage) |
| 221 IPC_MESSAGE_HANDLER(PrintHostMsg_PagesReadyForPreview, | 217 IPC_MESSAGE_HANDLER(PrintHostMsg_MetafileReadyForPrinting, |
| 222 OnPagesReadyForPreview) | 218 OnMetafileReadyForPrinting) |
| 223 IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewFailed, | 219 IPC_MESSAGE_HANDLER(PrintHostMsg_PrintPreviewFailed, |
| 224 OnPrintPreviewFailed) | 220 OnPrintPreviewFailed) |
| 225 IPC_MESSAGE_UNHANDLED(handled = false) | 221 IPC_MESSAGE_UNHANDLED(handled = false) |
| 226 IPC_END_MESSAGE_MAP() | 222 IPC_END_MESSAGE_MAP() |
| 227 return handled; | 223 return handled; |
| 228 } | 224 } |
| 229 | 225 |
| 230 void PrintPreviewMessageHandler::DidStartLoading() { | 226 void PrintPreviewMessageHandler::DidStartLoading() { |
| 231 if (tab_contents()->delegate() && | 227 if (tab_contents()->delegate() && |
| 232 printing::PrintPreviewTabController::IsPrintPreviewTab(tab_contents())) { | 228 printing::PrintPreviewTabController::IsPrintPreviewTab(tab_contents())) { |
| 233 tab_contents()->SetContentRestrictions(CONTENT_RESTRICTION_PRINT); | 229 tab_contents()->SetContentRestrictions(CONTENT_RESTRICTION_PRINT); |
| 234 } | 230 } |
| 235 } | 231 } |
| 236 | 232 |
| 237 } // namespace printing | 233 } // namespace printing |
| OLD | NEW |