| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/printing/background_printing_manager.h" |
| |
| #include "base/location.h" |
| #include "base/single_thread_task_runner.h" |
| #include "base/stl_util.h" |
| #include "base/threading/thread_task_runner_handle.h" |
| #include "chrome/browser/chrome_notification_types.h" |
| #include "chrome/browser/printing/print_job.h" |
| #include "chrome/browser/printing/print_preview_dialog_controller.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/notification_details.h" |
| #include "content/public/browser/notification_source.h" |
| #include "content/public/browser/web_contents.h" |
| #include "content/public/browser/web_contents_delegate.h" |
| #include "content/public/browser/web_contents_observer.h" |
| |
| using content::BrowserContext; |
| using content::BrowserThread; |
| using content::WebContents; |
| |
| namespace printing { |
| |
| class BackgroundPrintingManager::Observer |
| : public content::WebContentsObserver { |
| public: |
| Observer(BackgroundPrintingManager* manager, WebContents* web_contents); |
| |
| private: |
| void RenderProcessGone(base::TerminationStatus status) override; |
| |
| BackgroundPrintingManager* manager_; |
| }; |
| |
| BackgroundPrintingManager::Observer::Observer( |
| BackgroundPrintingManager* manager, WebContents* web_contents) |
| : content::WebContentsObserver(web_contents), |
| manager_(manager) { |
| } |
| |
| void BackgroundPrintingManager::Observer::RenderProcessGone( |
| base::TerminationStatus status) { |
| manager_->DeletePreviewContents(web_contents()); |
| } |
| |
| BackgroundPrintingManager::BackgroundPrintingManager() { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| } |
| |
| BackgroundPrintingManager::~BackgroundPrintingManager() { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| // The might be some WebContentses still in |printing_contents_map_| at this |
| // point (e.g. when the last remaining tab closes and there is still a print |
| // preview WebContents trying to print). In such a case it will fail to print, |
| // but we should at least clean up the observers. |
| // TODO(thestig): Handle this case better. |
| } |
| |
| void BackgroundPrintingManager::OwnPrintPreviewDialog( |
| std::unique_ptr<WebContents> preview_dialog) { |
| DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
| DCHECK(PrintPreviewDialogController::IsPrintPreviewURL( |
| preview_dialog->GetURL())); |
| CHECK(!HasPrintPreviewDialog(preview_dialog.get())); |
| |
| WebContents* raw_preview_dialog = preview_dialog.get(); |
| PrintingContents printing_contents; |
| printing_contents.observer = |
| std::make_unique<Observer>(this, raw_preview_dialog); |
| printing_contents.contents = std::move(preview_dialog); |
| printing_contents_map_[raw_preview_dialog] = std::move(printing_contents); |
| |
| // Watch for print jobs finishing. Everything else is watched for by the |
| // Observer. TODO(avi, cait): finish the job of removing this last |
| // notification. |
| registrar_.Add(this, chrome::NOTIFICATION_PRINT_JOB_RELEASED, |
| content::Source<WebContents>(raw_preview_dialog)); |
| |
| // Activate the initiator. |
| PrintPreviewDialogController* dialog_controller = |
| PrintPreviewDialogController::GetInstance(); |
| if (!dialog_controller) |
| return; |
| WebContents* initiator = dialog_controller->GetInitiator(raw_preview_dialog); |
| if (!initiator) |
| return; |
| initiator->GetDelegate()->ActivateContents(initiator); |
| } |
| |
| void BackgroundPrintingManager::Observe( |
| int type, |
| const content::NotificationSource& source, |
| const content::NotificationDetails& details) { |
| DCHECK_EQ(chrome::NOTIFICATION_PRINT_JOB_RELEASED, type); |
| DeletePreviewContents(content::Source<WebContents>(source).ptr()); |
| } |
| |
| void BackgroundPrintingManager::DeletePreviewContentsForBrowserContext( |
| BrowserContext* browser_context) { |
| std::vector<WebContents*> preview_contents_to_delete; |
| for (const auto& iter : printing_contents_map_) { |
| WebContents* preview_contents = iter.first; |
| if (preview_contents->GetBrowserContext() == browser_context) { |
| preview_contents_to_delete.push_back(preview_contents); |
| } |
| } |
| |
| for (size_t i = 0; i < preview_contents_to_delete.size(); i++) { |
| DeletePreviewContents(preview_contents_to_delete[i]); |
| } |
| } |
| |
| void BackgroundPrintingManager::OnPrintRequestCancelled( |
| WebContents* preview_contents) { |
| DeletePreviewContents(preview_contents); |
| } |
| |
| void BackgroundPrintingManager::DeletePreviewContents( |
| WebContents* preview_contents) { |
| auto i = printing_contents_map_.find(preview_contents); |
| if (i == printing_contents_map_.end()) { |
| // Everyone is racing to be the first to delete the |preview_contents|. If |
| // this case is hit, someone else won the race, so there is no need to |
| // continue. <http://crbug.com/100806> |
| return; |
| } |
| |
| // Stop all observation ... |
| registrar_.Remove(this, chrome::NOTIFICATION_PRINT_JOB_RELEASED, |
| content::Source<WebContents>(preview_contents)); |
| std::unique_ptr<WebContents> contents_to_delete = |
| std::move(i->second.contents); |
| printing_contents_map_.erase(i); |
| |
| // ... and mortally wound the contents. Deletion immediately is not a good |
| // idea in case this was triggered by |preview_contents| far up the |
| // callstack. (Trace where the NOTIFICATION_PRINT_JOB_RELEASED comes from.) |
| base::ThreadTaskRunnerHandle::Get()->DeleteSoon( |
| FROM_HERE, std::move(contents_to_delete)); |
| } |
| |
| std::set<content::WebContents*> BackgroundPrintingManager::CurrentContentSet() { |
| std::set<content::WebContents*> result; |
| for (const auto& entry : printing_contents_map_) |
| result.insert(entry.first); |
| |
| return result; |
| } |
| |
| bool BackgroundPrintingManager::HasPrintPreviewDialog( |
| WebContents* preview_dialog) { |
| return base::ContainsKey(printing_contents_map_, preview_dialog); |
| } |
| |
| BackgroundPrintingManager::PrintingContents::PrintingContents() = default; |
| BackgroundPrintingManager::PrintingContents::~PrintingContents() = default; |
| BackgroundPrintingManager::PrintingContents::PrintingContents( |
| PrintingContents&&) = default; |
| BackgroundPrintingManager::PrintingContents& |
| BackgroundPrintingManager::PrintingContents::operator=(PrintingContents&&) = |
| default; |
| |
| } // namespace printing |