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
3 // found in the LICENSE file.
5 #include "chrome/browser/printing/background_printing_manager.h"
7 #include "base/stl_util.h"
8 #include "chrome/browser/chrome_notification_types.h"
9 #include "chrome/browser/printing/print_job.h"
10 #include "chrome/browser/printing/print_preview_dialog_controller.h"
11 #include "content/public/browser/browser_thread.h"
12 #include "content/public/browser/notification_details.h"
13 #include "content/public/browser/notification_source.h"
14 #include "content/public/browser/render_view_host.h"
15 #include "content/public/browser/web_contents.h"
16 #include "content/public/browser/web_contents_delegate.h"
17 #include "content/public/browser/web_contents_observer.h"
19 using content::BrowserThread
;
20 using content::WebContents
;
24 class BackgroundPrintingManager::Observer
25 : public content::WebContentsObserver
{
27 Observer(BackgroundPrintingManager
* manager
, WebContents
* web_contents
);
30 virtual void RenderProcessGone(base::TerminationStatus status
) OVERRIDE
;
31 virtual void WebContentsDestroyed() OVERRIDE
;
33 BackgroundPrintingManager
* manager_
;
36 BackgroundPrintingManager::Observer::Observer(
37 BackgroundPrintingManager
* manager
, WebContents
* web_contents
)
38 : content::WebContentsObserver(web_contents
),
42 void BackgroundPrintingManager::Observer::RenderProcessGone(
43 base::TerminationStatus status
) {
44 manager_
->DeletePreviewContents(web_contents());
46 void BackgroundPrintingManager::Observer::WebContentsDestroyed() {
47 manager_
->DeletePreviewContents(web_contents());
50 BackgroundPrintingManager::BackgroundPrintingManager() {
51 DCHECK_CURRENTLY_ON(BrowserThread::UI
);
54 BackgroundPrintingManager::~BackgroundPrintingManager() {
55 DCHECK(CalledOnValidThread());
56 // The might be some WebContentses still in |printing_contents_map_| at this
57 // point (e.g. when the last remaining tab closes and there is still a print
58 // preview WebContents trying to print). In such a case it will fail to print,
59 // but we should at least clean up the observers.
60 // TODO(thestig): Handle this case better.
61 STLDeleteValues(&printing_contents_map_
);
64 void BackgroundPrintingManager::OwnPrintPreviewDialog(
65 WebContents
* preview_dialog
) {
66 DCHECK(CalledOnValidThread());
67 DCHECK(PrintPreviewDialogController::IsPrintPreviewDialog(preview_dialog
));
68 CHECK(!HasPrintPreviewDialog(preview_dialog
));
70 printing_contents_map_
[preview_dialog
] = new Observer(this, preview_dialog
);
72 // Watch for print jobs finishing. Everything else is watched for by the
73 // Observer. TODO(avi, cait): finish the job of removing this last
75 registrar_
.Add(this, chrome::NOTIFICATION_PRINT_JOB_RELEASED
,
76 content::Source
<WebContents
>(preview_dialog
));
78 // Activate the initiator.
79 PrintPreviewDialogController
* dialog_controller
=
80 PrintPreviewDialogController::GetInstance();
81 if (!dialog_controller
)
83 WebContents
* initiator
= dialog_controller
->GetInitiator(preview_dialog
);
86 initiator
->GetDelegate()->ActivateContents(initiator
);
89 void BackgroundPrintingManager::Observe(
91 const content::NotificationSource
& source
,
92 const content::NotificationDetails
& details
) {
93 DCHECK_EQ(chrome::NOTIFICATION_PRINT_JOB_RELEASED
, type
);
94 DeletePreviewContents(content::Source
<WebContents
>(source
).ptr());
97 void BackgroundPrintingManager::DeletePreviewContents(
98 WebContents
* preview_contents
) {
99 WebContentsObserverMap::iterator i
=
100 printing_contents_map_
.find(preview_contents
);
101 if (i
== printing_contents_map_
.end()) {
102 // Everyone is racing to be the first to delete the |preview_contents|. If
103 // this case is hit, someone else won the race, so there is no need to
104 // continue. <http://crbug.com/100806>
108 // Stop all observation ...
109 registrar_
.Remove(this, chrome::NOTIFICATION_PRINT_JOB_RELEASED
,
110 content::Source
<WebContents
>(preview_contents
));
111 Observer
* observer
= i
->second
;
112 printing_contents_map_
.erase(i
);
115 // ... and mortally wound the contents. (Deletion immediately is not a good
116 // idea in case this was called from RenderViewGone.)
117 base::MessageLoop::current()->DeleteSoon(FROM_HERE
, preview_contents
);
120 std::set
<content::WebContents
*> BackgroundPrintingManager::CurrentContentSet() {
121 std::set
<content::WebContents
*> result
;
122 for (WebContentsObserverMap::iterator i
= printing_contents_map_
.begin();
123 i
!= printing_contents_map_
.end(); ++i
) {
124 result
.insert(i
->first
);
129 bool BackgroundPrintingManager::HasPrintPreviewDialog(
130 WebContents
* preview_dialog
) {
131 return ContainsKey(printing_contents_map_
, preview_dialog
);
134 } // namespace printing