8bbdc224ac
* chore: bump chromium in DEPS to 7fb9778894d73378bff51087ce869ea5aa6e5d5d * chore: bump chromium in DEPS to 83da426e53d423f0530fc23433b6d2c4d0548442 * update patches * remove chromeos-only TtsControllerDelegate Refs: https://chromium-review.googlesource.com/c/chromium/src/+/2255314 * SharedUserScriptMaster -> SharedUserScriptManager Refs: https://chromium-review.googlesource.com/c/chromium/src/+/2258357 * avoid deprecated DISALLOW_COPY_AND_ASSIGN https://groups.google.com/a/chromium.org/forum/#!msg/cxx/qwH2hxaEjac/TUKq6eqfCwAJ * chore: bump chromium in DEPS to b2eaf9ff4e6b03267bf279583ea20ceb2b25e9d0 * update patches * rename GetHighContrastColorScheme -> GetPlatformHighContrastColorScheme Refs: https://chromium-review.googlesource.com/c/chromium/src/+/2250224 * remove vulkan info collection Refs: https://chromium-review.googlesource.com/c/chromium/src/+/2252818 * add max_xcode_version build var Refs: https://chromium-review.googlesource.com/c/chromium/src/+/2264867 * add missing headers * chore: bump chromium in DEPS to cded18ca1138f7e8efc904f077ddcca34f0135cf * update patches * add empty floc blocklist to BrowserProcessImpl Refs: https://chromium-review.googlesource.com/c/chromium/src/+/2240873 * chore: bump chromium in DEPS to f06602226cd80bf677b2ce013a94a2fb7f6ac58d * chore: bump chromium in DEPS to 747aa4bfc74fc6cf7f08ee72624cd69ae41ae28d * chore: bump chromium in DEPS to 31c0105e50fcc4e94de33e5c8602c755ace4a32b * chore: update patches * Reland "[base] Stop including check.h, notreached.h, etc. in logging.h" https://chromium-review.googlesource.com/c/chromium/src/+/2264297 * X11 and Ozone: make sure gfx::AcceleratedWidget to be uint32_t https://chromium-review.googlesource.com/c/chromium/src/+/2260554 * Move zygote from //services/service_manager back to //content https://chromium-review.googlesource.com/c/chromium/src/+/2252466 * chore: update v8 patches * [XProto] Remove usage of Shape extension https://chromium-review.googlesource.com/c/chromium/src/+/2262113 * fixup! add empty floc blocklist to BrowserProcessImpl * Require macOS 10.15.1 sdk https://chromium-review.googlesource.com/c/chromium/src/+/2238504 * Use newer Xcode version 11.5.0 * update src cache * chore: bump chromium in DEPS to 60a9883e35db3f6f91916f0878e88e1849c17b11 * chore: update patches * Reland "Reland "New toolchain for Windows 10 19041 SDK"" https://chromium-review.googlesource.com/c/chromium/src/+/2255527 * update patches * Convert raw NonClientFrameViews to unique_ptrs https://chromium-review.googlesource.com/c/chromium/src/+/2240417 * [printing] Move PrintHostMsg_DidPreviewDocument_Params to print.mojom https://chromium-review.googlesource.com/c/chromium/src/+/2257035 * chore: bump chromium in DEPS to 12c233c2a85bfa28fb279f390121ba681e52a71b * chore: update patches * Removing oppressive language for the directory chrome/browser/apps https://chromium-review.googlesource.com/c/chromium/src/+/2269822 * Inclusion: rename SpellcheckLanguageBlacklistPolicyHandler https://chromium-review.googlesource.com/c/chromium/src/+/2267646 * Clean up duplicate WebContents "is fullscreen" functions https://chromium-review.googlesource.com/c/chromium/src/+/2275148 * Adds icon loading service with sandbox for Windows. https://chromium-review.googlesource.com/c/chromium/src/+/1987273 * No more Vulkan info collection for UMA on Windows https://chromium-review.googlesource.com/c/chromium/src/+/2252818 * fix lint * chore: update buildflag conditions * chore: bump chromium in DEPS to a837d4c4230ace4f10b2768728f4044b7995dfa5 * update hunspell files * chore: update patches * Make content::FileSelectListener a RefCounted https://chromium-review.googlesource.com/c/chromium/src/+/2275338 * fix build failures on MAS * update patches * fixup! Reland "[base] Stop including check.h, notreached.h, etc. in logging.h" * fix build on windows * Check for GDI exhaustion if window creation fails https://chromium-review.googlesource.com/c/chromium/src/+/2244124 * chore: bump chromium in DEPS to 2c9b2a73be4ef9ec22d8b6da8e174cb80753f125 * chore: update patches * Network Service: Move DeleteCookiePredicate into public folder https://chromium-review.googlesource.com/c/chromium/src/+/2264186 * chore: bump chromium in DEPS to fa2606299bcc02c362528d26b5dcf8c8a0db0735 * chore: bump chromium in DEPS to d9c235d1227204dbae3708daae851573a3566b94 * chore: bump chromium in DEPS to 2f82c284243c035f49a747fd1ead6c44b4b31093 * chore: update patches * Move creating the LayerTreeSettings into blink. https://chromium-review.googlesource.com/c/chromium/src/+/2267720 * chore: bump chromium in DEPS to 914112f1d9af9e4974059dc403da62699a55550f * update patches * chore: bump chromium in DEPS to e0bc1ffae6393fc543a2da94c88167df75859b36 * refactor: match upstream print preview handling (#24452) * update patches * chore: bump chromium in DEPS to 0881423156abe084164b51ab58ce93a8bd380524 * update patches * update patches * give a type to pendingPromise * chore: bump chromium in DEPS to 11a8c1534b16d130075d378a64187360ba4e7820 * update patches * 2272609: Move //services/service_manager/sandbox to //sandbox/policy. https://chromium-review.googlesource.com/c/chromium/src/+/2272609 * update patches * fixup! 2272609: Move //services/service_manager/sandbox to //sandbox/policy. * fixup! 2272609: Move //services/service_manager/sandbox to //sandbox/policy. * 2264511: Cookies: Update SetCanonicalCookie to return CookieAccessResult https://chromium-review.googlesource.com/c/chromium/src/+/2264511 * chore: fix setAlwaysOnTop test The window must be visible for state to be updated properly. * Revert "Migrate modules/desktop_capture and modules/video_capture to webrtc::Mutex." https://webrtc-review.googlesource.com/c/src/+/179080 * update patches Co-authored-by: Andy Locascio <andy@slack-corp.com> Co-authored-by: deepak1556 <hop2deep@gmail.com> Co-authored-by: Samuel Attard <samuel.r.attard@gmail.com> Co-authored-by: John Kleinschmidt <jkleinsc@github.com> Co-authored-by: Electron Bot <anonymous@electronjs.org> Co-authored-by: Shelley Vohr <shelley.vohr@gmail.com> Co-authored-by: Jeremy Rose <nornagon@nornagon.net> Co-authored-by: Samuel Attard <marshallofsound@electronjs.org>
284 lines
9.8 KiB
C++
284 lines
9.8 KiB
C++
// Copyright (c) 2018 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "shell/browser/printing/print_preview_message_handler.h"
|
|
|
|
#include <memory>
|
|
#include <utility>
|
|
|
|
#include "base/bind.h"
|
|
#include "base/memory/read_only_shared_memory_region.h"
|
|
#include "base/memory/ref_counted.h"
|
|
#include "base/memory/ref_counted_memory.h"
|
|
#include "base/task/post_task.h"
|
|
#include "chrome/browser/browser_process.h"
|
|
#include "chrome/browser/printing/print_job_manager.h"
|
|
#include "chrome/browser/printing/printer_query.h"
|
|
#include "components/printing/browser/print_composite_client.h"
|
|
#include "components/printing/browser/print_manager_utils.h"
|
|
#include "components/printing/common/print_messages.h"
|
|
#include "components/services/print_compositor/public/cpp/print_service_mojo_types.h"
|
|
#include "content/public/browser/browser_task_traits.h"
|
|
#include "content/public/browser/browser_thread.h"
|
|
#include "content/public/browser/render_frame_host.h"
|
|
#include "content/public/browser/web_contents.h"
|
|
#include "mojo/public/cpp/bindings/callback_helpers.h"
|
|
#include "shell/common/gin_helper/locker.h"
|
|
#include "third_party/blink/public/common/associated_interfaces/associated_interface_provider.h"
|
|
|
|
#include "shell/common/node_includes.h"
|
|
|
|
using content::BrowserThread;
|
|
|
|
namespace electron {
|
|
|
|
namespace {
|
|
|
|
void StopWorker(int document_cookie) {
|
|
if (document_cookie <= 0)
|
|
return;
|
|
scoped_refptr<printing::PrintQueriesQueue> queue =
|
|
g_browser_process->print_job_manager()->queue();
|
|
std::unique_ptr<printing::PrinterQuery> printer_query =
|
|
queue->PopPrinterQuery(document_cookie);
|
|
if (printer_query.get()) {
|
|
base::PostTask(FROM_HERE, {BrowserThread::IO},
|
|
base::BindOnce(&printing::PrinterQuery::StopWorker,
|
|
std::move(printer_query)));
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
PrintPreviewMessageHandler::PrintPreviewMessageHandler(
|
|
content::WebContents* web_contents)
|
|
: content::WebContentsObserver(web_contents), weak_ptr_factory_(this) {
|
|
DCHECK(web_contents);
|
|
}
|
|
|
|
PrintPreviewMessageHandler::~PrintPreviewMessageHandler() = default;
|
|
|
|
bool PrintPreviewMessageHandler::OnMessageReceived(
|
|
const IPC::Message& message,
|
|
content::RenderFrameHost* render_frame_host) {
|
|
bool handled = true;
|
|
IPC_BEGIN_MESSAGE_MAP_WITH_PARAM(PrintPreviewMessageHandler, message,
|
|
render_frame_host)
|
|
IPC_MESSAGE_HANDLER(PrintHostMsg_MetafileReadyForPrinting,
|
|
OnMetafileReadyForPrinting)
|
|
IPC_MESSAGE_HANDLER(PrintHostMsg_DidPreviewPage, OnDidPreviewPage)
|
|
IPC_MESSAGE_HANDLER(PrintHostMsg_DidPrepareDocumentForPreview,
|
|
OnDidPrepareForDocumentToPdf)
|
|
IPC_MESSAGE_UNHANDLED(handled = false)
|
|
IPC_END_MESSAGE_MAP()
|
|
return handled;
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::OnMetafileReadyForPrinting(
|
|
content::RenderFrameHost* render_frame_host,
|
|
const printing::mojom::DidPreviewDocumentParams& params,
|
|
const PrintHostMsg_PreviewIds& ids) {
|
|
// Always try to stop the worker.
|
|
StopWorker(params.document_cookie);
|
|
|
|
if (params.expected_pages_count == 0) {
|
|
RejectPromise(ids.request_id);
|
|
return;
|
|
}
|
|
|
|
const base::ReadOnlySharedMemoryRegion& metafile =
|
|
params.content->metafile_data_region;
|
|
|
|
if (printing::IsOopifEnabled()) {
|
|
auto* client =
|
|
printing::PrintCompositeClient::FromWebContents(web_contents());
|
|
DCHECK(client);
|
|
|
|
auto callback = base::BindOnce(
|
|
&PrintPreviewMessageHandler::OnCompositeDocumentToPdfDone,
|
|
weak_ptr_factory_.GetWeakPtr(), ids);
|
|
|
|
client->DoCompleteDocumentToPdf(
|
|
params.document_cookie, params.expected_pages_count,
|
|
mojo::WrapCallbackWithDefaultInvokeIfNotRun(
|
|
std::move(callback),
|
|
printing::mojom::PrintCompositor::Status::kCompositingFailure,
|
|
base::ReadOnlySharedMemoryRegion()));
|
|
} else {
|
|
ResolvePromise(
|
|
ids.request_id,
|
|
base::RefCountedSharedMemoryMapping::CreateFromWholeRegion(metafile));
|
|
}
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::OnPrepareForDocumentToPdfDone(
|
|
const PrintHostMsg_PreviewIds& ids,
|
|
printing::mojom::PrintCompositor::Status status) {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
if (status != printing::mojom::PrintCompositor::Status::kSuccess) {
|
|
LOG(ERROR) << "Preparing document for pdf failed with error " << status;
|
|
}
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::OnDidPrepareForDocumentToPdf(
|
|
content::RenderFrameHost* render_frame_host,
|
|
int document_cookie,
|
|
const PrintHostMsg_PreviewIds& ids) {
|
|
if (printing::IsOopifEnabled()) {
|
|
auto* client =
|
|
printing::PrintCompositeClient::FromWebContents(web_contents());
|
|
DCHECK(client);
|
|
|
|
if (client->GetIsDocumentConcurrentlyComposited(document_cookie))
|
|
return;
|
|
|
|
client->DoPrepareForDocumentToPdf(
|
|
document_cookie, render_frame_host,
|
|
mojo::WrapCallbackWithDefaultInvokeIfNotRun(
|
|
base::BindOnce(
|
|
&PrintPreviewMessageHandler::OnPrepareForDocumentToPdfDone,
|
|
weak_ptr_factory_.GetWeakPtr(), ids),
|
|
printing::mojom::PrintCompositor::Status::kCompositingFailure));
|
|
}
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::OnCompositeDocumentToPdfDone(
|
|
const PrintHostMsg_PreviewIds& ids,
|
|
printing::mojom::PrintCompositor::Status status,
|
|
base::ReadOnlySharedMemoryRegion region) {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
|
|
if (status != printing::mojom::PrintCompositor::Status::kSuccess) {
|
|
LOG(ERROR) << "Compositing pdf failed with error " << status;
|
|
RejectPromise(ids.request_id);
|
|
return;
|
|
}
|
|
|
|
ResolvePromise(
|
|
ids.request_id,
|
|
base::RefCountedSharedMemoryMapping::CreateFromWholeRegion(region));
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::OnCompositePdfPageDone(
|
|
int page_number,
|
|
int document_cookie,
|
|
const PrintHostMsg_PreviewIds& ids,
|
|
printing::mojom::PrintCompositor::Status status,
|
|
base::ReadOnlySharedMemoryRegion region) {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
if (status != printing::mojom::PrintCompositor::Status::kSuccess) {
|
|
LOG(ERROR) << "Compositing pdf failed on page: " << page_number
|
|
<< " with error: " << status;
|
|
}
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::OnDidPreviewPage(
|
|
content::RenderFrameHost* render_frame_host,
|
|
const printing::mojom::DidPreviewPageParams& params,
|
|
const PrintHostMsg_PreviewIds& ids) {
|
|
int page_number = params.page_number;
|
|
const printing::mojom::DidPrintContentParams& content = *params.content;
|
|
|
|
if (page_number < printing::FIRST_PAGE_INDEX ||
|
|
!content.metafile_data_region.IsValid()) {
|
|
RejectPromise(ids.request_id);
|
|
return;
|
|
}
|
|
|
|
if (printing::IsOopifEnabled()) {
|
|
auto* client =
|
|
printing::PrintCompositeClient::FromWebContents(web_contents());
|
|
DCHECK(client);
|
|
|
|
// Use utility process to convert skia metafile to pdf.
|
|
client->DoCompositePageToPdf(
|
|
params.document_cookie, render_frame_host, content,
|
|
mojo::WrapCallbackWithDefaultInvokeIfNotRun(
|
|
base::BindOnce(&PrintPreviewMessageHandler::OnCompositePdfPageDone,
|
|
weak_ptr_factory_.GetWeakPtr(), page_number,
|
|
params.document_cookie, ids),
|
|
printing::mojom::PrintCompositor::Status::kCompositingFailure,
|
|
base::ReadOnlySharedMemoryRegion()));
|
|
}
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::PrintPreviewFailed(int32_t document_cookie,
|
|
int32_t request_id) {
|
|
StopWorker(document_cookie);
|
|
|
|
RejectPromise(request_id);
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::PrintPreviewCancelled(int32_t document_cookie,
|
|
int32_t request_id) {
|
|
StopWorker(document_cookie);
|
|
|
|
RejectPromise(request_id);
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::PrintToPDF(
|
|
base::DictionaryValue options,
|
|
gin_helper::Promise<v8::Local<v8::Value>> promise) {
|
|
int request_id;
|
|
options.GetInteger(printing::kPreviewRequestID, &request_id);
|
|
promise_map_.emplace(request_id, std::move(promise));
|
|
|
|
auto* focused_frame = web_contents()->GetFocusedFrame();
|
|
auto* rfh = focused_frame && focused_frame->HasSelection()
|
|
? focused_frame
|
|
: web_contents()->GetMainFrame();
|
|
|
|
if (!print_render_frame_.is_bound()) {
|
|
rfh->GetRemoteAssociatedInterfaces()->GetInterface(&print_render_frame_);
|
|
}
|
|
if (!receiver_.is_bound()) {
|
|
print_render_frame_->SetPrintPreviewUI(
|
|
receiver_.BindNewEndpointAndPassRemote());
|
|
}
|
|
print_render_frame_->PrintPreview(options.Clone());
|
|
}
|
|
|
|
gin_helper::Promise<v8::Local<v8::Value>>
|
|
PrintPreviewMessageHandler::GetPromise(int request_id) {
|
|
auto it = promise_map_.find(request_id);
|
|
DCHECK(it != promise_map_.end());
|
|
|
|
gin_helper::Promise<v8::Local<v8::Value>> promise = std::move(it->second);
|
|
promise_map_.erase(it);
|
|
|
|
return promise;
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::ResolvePromise(
|
|
int request_id,
|
|
scoped_refptr<base::RefCountedMemory> data_bytes) {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
|
|
gin_helper::Promise<v8::Local<v8::Value>> promise = GetPromise(request_id);
|
|
|
|
v8::Isolate* isolate = promise.isolate();
|
|
gin_helper::Locker locker(isolate);
|
|
v8::HandleScope handle_scope(isolate);
|
|
v8::Context::Scope context_scope(
|
|
v8::Local<v8::Context>::New(isolate, promise.GetContext()));
|
|
|
|
v8::Local<v8::Value> buffer =
|
|
node::Buffer::Copy(isolate,
|
|
reinterpret_cast<const char*>(data_bytes->front()),
|
|
data_bytes->size())
|
|
.ToLocalChecked();
|
|
|
|
promise.Resolve(buffer);
|
|
}
|
|
|
|
void PrintPreviewMessageHandler::RejectPromise(int request_id) {
|
|
DCHECK_CURRENTLY_ON(BrowserThread::UI);
|
|
|
|
gin_helper::Promise<v8::Local<v8::Value>> promise = GetPromise(request_id);
|
|
promise.RejectWithErrorMessage("Failed to generate PDF");
|
|
}
|
|
|
|
WEB_CONTENTS_USER_DATA_KEY_IMPL(PrintPreviewMessageHandler)
|
|
|
|
} // namespace electron
|