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 |