mirror of
https://bitbucket.org/chromiumembedded/cef
synced 2024-12-13 10:06:28 +01:00
325 lines
8.9 KiB
C++
325 lines
8.9 KiB
C++
// Copyright (c) 2014 The Chromium Embedded Framework Authors.
|
|
// Portions 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 "libcef/browser/printing/print_dialog_linux.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "libcef/browser/extensions/browser_extensions_util.h"
|
|
#include "libcef/browser/print_settings_impl.h"
|
|
#include "libcef/browser/thread_util.h"
|
|
#include "libcef/common/content_client.h"
|
|
|
|
#include "base/bind.h"
|
|
#include "base/files/file_util.h"
|
|
#include "base/lazy_instance.h"
|
|
#include "base/logging.h"
|
|
#include "base/strings/utf_string_conversions.h"
|
|
#include "base/values.h"
|
|
#include "printing/metafile.h"
|
|
#include "printing/print_job_constants.h"
|
|
#include "printing/print_settings.h"
|
|
|
|
using content::BrowserThread;
|
|
using printing::PageRanges;
|
|
using printing::PrintSettings;
|
|
|
|
class CefPrintDialogCallbackImpl : public CefPrintDialogCallback {
|
|
public:
|
|
explicit CefPrintDialogCallbackImpl(CefRefPtr<CefPrintDialogLinux> dialog)
|
|
: dialog_(dialog) {}
|
|
|
|
void Continue(CefRefPtr<CefPrintSettings> settings) override {
|
|
if (CEF_CURRENTLY_ON_UIT()) {
|
|
if (dialog_.get()) {
|
|
dialog_->OnPrintContinue(settings);
|
|
dialog_ = NULL;
|
|
}
|
|
} else {
|
|
CEF_POST_TASK(CEF_UIT, base::Bind(&CefPrintDialogCallbackImpl::Continue,
|
|
this, settings));
|
|
}
|
|
}
|
|
|
|
void Cancel() override {
|
|
if (CEF_CURRENTLY_ON_UIT()) {
|
|
if (dialog_.get()) {
|
|
dialog_->OnPrintCancel();
|
|
dialog_ = NULL;
|
|
}
|
|
} else {
|
|
CEF_POST_TASK(CEF_UIT,
|
|
base::Bind(&CefPrintDialogCallbackImpl::Cancel, this));
|
|
}
|
|
}
|
|
|
|
void Disconnect() { dialog_ = NULL; }
|
|
|
|
private:
|
|
CefRefPtr<CefPrintDialogLinux> dialog_;
|
|
|
|
IMPLEMENT_REFCOUNTING(CefPrintDialogCallbackImpl);
|
|
DISALLOW_COPY_AND_ASSIGN(CefPrintDialogCallbackImpl);
|
|
};
|
|
|
|
class CefPrintJobCallbackImpl : public CefPrintJobCallback {
|
|
public:
|
|
explicit CefPrintJobCallbackImpl(CefRefPtr<CefPrintDialogLinux> dialog)
|
|
: dialog_(dialog) {}
|
|
|
|
void Continue() override {
|
|
if (CEF_CURRENTLY_ON_UIT()) {
|
|
if (dialog_.get()) {
|
|
dialog_->OnJobCompleted();
|
|
dialog_ = NULL;
|
|
}
|
|
} else {
|
|
CEF_POST_TASK(CEF_UIT,
|
|
base::Bind(&CefPrintJobCallbackImpl::Continue, this));
|
|
}
|
|
}
|
|
|
|
void Disconnect() { dialog_ = NULL; }
|
|
|
|
private:
|
|
CefRefPtr<CefPrintDialogLinux> dialog_;
|
|
|
|
IMPLEMENT_REFCOUNTING(CefPrintJobCallbackImpl);
|
|
DISALLOW_COPY_AND_ASSIGN(CefPrintJobCallbackImpl);
|
|
};
|
|
|
|
// static
|
|
printing::PrintDialogGtkInterface* CefPrintDialogLinux::CreatePrintDialog(
|
|
PrintingContextLinux* context) {
|
|
CEF_REQUIRE_UIT();
|
|
return new CefPrintDialogLinux(context);
|
|
}
|
|
|
|
// static
|
|
gfx::Size CefPrintDialogLinux::GetPdfPaperSize(
|
|
printing::PrintingContextLinux* context) {
|
|
CEF_REQUIRE_UIT();
|
|
|
|
gfx::Size size;
|
|
|
|
CefRefPtr<CefApp> app = CefContentClient::Get()->application();
|
|
if (app.get()) {
|
|
CefRefPtr<CefBrowserProcessHandler> browser_handler =
|
|
app->GetBrowserProcessHandler();
|
|
if (browser_handler.get()) {
|
|
CefRefPtr<CefPrintHandler> handler = browser_handler->GetPrintHandler();
|
|
if (handler.get()) {
|
|
const printing::PrintSettings& settings = context->settings();
|
|
CefSize cef_size =
|
|
handler->GetPdfPaperSize(settings.device_units_per_inch());
|
|
size.SetSize(cef_size.width, cef_size.height);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (size.IsEmpty()) {
|
|
LOG(ERROR) << "Empty size value returned in GetPdfPaperSize; "
|
|
"PDF printing will fail.";
|
|
}
|
|
return size;
|
|
}
|
|
|
|
// static
|
|
void CefPrintDialogLinux::OnPrintStart(int render_process_id,
|
|
int render_routing_id) {
|
|
if (!CEF_CURRENTLY_ON(CEF_UIT)) {
|
|
CEF_POST_TASK(CEF_UIT, base::Bind(&CefPrintDialogLinux::OnPrintStart,
|
|
render_process_id, render_routing_id));
|
|
return;
|
|
}
|
|
|
|
CefRefPtr<CefApp> app = CefContentClient::Get()->application();
|
|
if (!app.get())
|
|
return;
|
|
|
|
CefRefPtr<CefBrowserProcessHandler> browser_handler =
|
|
app->GetBrowserProcessHandler();
|
|
if (!browser_handler.get())
|
|
return;
|
|
|
|
CefRefPtr<CefPrintHandler> handler = browser_handler->GetPrintHandler();
|
|
if (!handler.get())
|
|
return;
|
|
|
|
CefRefPtr<CefBrowserHostImpl> browser =
|
|
extensions::GetOwnerBrowserForFrameRoute(render_process_id,
|
|
render_routing_id, NULL);
|
|
if (browser.get())
|
|
handler->OnPrintStart(browser.get());
|
|
}
|
|
|
|
CefPrintDialogLinux::CefPrintDialogLinux(PrintingContextLinux* context)
|
|
: context_(context) {
|
|
DCHECK(context_);
|
|
browser_ = extensions::GetOwnerBrowserForFrameRoute(
|
|
context_->render_process_id(), context_->render_frame_id(), NULL);
|
|
DCHECK(browser_);
|
|
}
|
|
|
|
CefPrintDialogLinux::~CefPrintDialogLinux() {
|
|
// It's not safe to dereference |context_| during the destruction of this
|
|
// object because the PrintJobWorker which owns |context_| may already have
|
|
// been deleted.
|
|
CEF_REQUIRE_UIT();
|
|
ReleaseHandler();
|
|
}
|
|
|
|
void CefPrintDialogLinux::UseDefaultSettings() {
|
|
UpdateSettings(std::make_unique<PrintSettings>(), true);
|
|
}
|
|
|
|
void CefPrintDialogLinux::UpdateSettings(
|
|
std::unique_ptr<PrintSettings> settings) {
|
|
UpdateSettings(std::move(settings), false);
|
|
}
|
|
|
|
void CefPrintDialogLinux::ShowDialog(
|
|
gfx::NativeView parent_view,
|
|
bool has_selection,
|
|
PrintingContextLinux::PrintSettingsCallback callback) {
|
|
CEF_REQUIRE_UIT();
|
|
|
|
SetHandler();
|
|
if (!handler_.get()) {
|
|
std::move(callback).Run(PrintingContextLinux::CANCEL);
|
|
return;
|
|
}
|
|
|
|
callback_ = std::move(callback);
|
|
|
|
CefRefPtr<CefPrintDialogCallbackImpl> callback_impl(
|
|
new CefPrintDialogCallbackImpl(this));
|
|
|
|
if (!handler_->OnPrintDialog(browser_.get(), has_selection,
|
|
callback_impl.get())) {
|
|
callback_impl->Disconnect();
|
|
OnPrintCancel();
|
|
}
|
|
}
|
|
|
|
void CefPrintDialogLinux::PrintDocument(
|
|
const printing::MetafilePlayer& metafile,
|
|
const base::string16& document_name) {
|
|
// This runs on the print worker thread, does not block the UI thread.
|
|
DCHECK(!CEF_CURRENTLY_ON_UIT());
|
|
|
|
// The document printing tasks can outlive the PrintingContext that created
|
|
// this dialog.
|
|
AddRef();
|
|
|
|
bool success = base::CreateTemporaryFile(&path_to_pdf_);
|
|
|
|
if (success) {
|
|
base::File file;
|
|
file.Initialize(path_to_pdf_,
|
|
base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE);
|
|
success = metafile.SaveTo(&file);
|
|
file.Close();
|
|
if (!success)
|
|
base::DeleteFile(path_to_pdf_, false);
|
|
}
|
|
|
|
if (!success) {
|
|
LOG(ERROR) << "Saving metafile failed";
|
|
// Matches AddRef() above.
|
|
Release();
|
|
return;
|
|
}
|
|
|
|
// No errors, continue printing.
|
|
CEF_POST_TASK(CEF_UIT, base::Bind(&CefPrintDialogLinux::SendDocumentToPrinter,
|
|
this, document_name));
|
|
}
|
|
|
|
void CefPrintDialogLinux::AddRefToDialog() {
|
|
AddRef();
|
|
}
|
|
|
|
void CefPrintDialogLinux::ReleaseDialog() {
|
|
Release();
|
|
}
|
|
|
|
void CefPrintDialogLinux::SetHandler() {
|
|
if (handler_.get())
|
|
return;
|
|
|
|
CefRefPtr<CefApp> app = CefContentClient::Get()->application();
|
|
if (app.get()) {
|
|
CefRefPtr<CefBrowserProcessHandler> browser_handler =
|
|
app->GetBrowserProcessHandler();
|
|
if (browser_handler.get())
|
|
handler_ = browser_handler->GetPrintHandler();
|
|
}
|
|
}
|
|
|
|
void CefPrintDialogLinux::ReleaseHandler() {
|
|
if (handler_.get()) {
|
|
handler_->OnPrintReset(browser_.get());
|
|
handler_ = NULL;
|
|
}
|
|
}
|
|
|
|
bool CefPrintDialogLinux::UpdateSettings(
|
|
std::unique_ptr<PrintSettings> settings,
|
|
bool get_defaults) {
|
|
CEF_REQUIRE_UIT();
|
|
|
|
SetHandler();
|
|
if (!handler_.get())
|
|
return false;
|
|
|
|
CefRefPtr<CefPrintSettingsImpl> settings_impl(
|
|
new CefPrintSettingsImpl(std::move(settings), false));
|
|
handler_->OnPrintSettings(browser_.get(), settings_impl.get(), get_defaults);
|
|
|
|
context_->InitWithSettings(settings_impl->TakeOwnership());
|
|
return true;
|
|
}
|
|
|
|
void CefPrintDialogLinux::SendDocumentToPrinter(
|
|
const base::string16& document_name) {
|
|
CEF_REQUIRE_UIT();
|
|
|
|
if (!handler_.get()) {
|
|
OnJobCompleted();
|
|
return;
|
|
}
|
|
|
|
CefRefPtr<CefPrintJobCallbackImpl> callback_impl(
|
|
new CefPrintJobCallbackImpl(this));
|
|
|
|
if (!handler_->OnPrintJob(browser_.get(), document_name, path_to_pdf_.value(),
|
|
callback_impl.get())) {
|
|
callback_impl->Disconnect();
|
|
OnJobCompleted();
|
|
}
|
|
}
|
|
|
|
void CefPrintDialogLinux::OnPrintContinue(
|
|
CefRefPtr<CefPrintSettings> settings) {
|
|
CefPrintSettingsImpl* impl =
|
|
static_cast<CefPrintSettingsImpl*>(settings.get());
|
|
context_->InitWithSettings(impl->TakeOwnership());
|
|
std::move(callback_).Run(PrintingContextLinux::OK);
|
|
}
|
|
|
|
void CefPrintDialogLinux::OnPrintCancel() {
|
|
std::move(callback_).Run(PrintingContextLinux::CANCEL);
|
|
}
|
|
|
|
void CefPrintDialogLinux::OnJobCompleted() {
|
|
CEF_POST_BACKGROUND_TASK(
|
|
base::Bind(base::IgnoreResult(&base::DeleteFile), path_to_pdf_, false));
|
|
|
|
// Printing finished. Matches AddRef() in PrintDocument();
|
|
Release();
|
|
}
|