webengineexportpdfpagejob.cpp

/*
   SPDX-FileCopyrightText: 2020-2024 Laurent Montel <montel@kde.org>
 
   SPDX-License-Identifier: LGPL-2.0-or-later
*/
 
#include "webengineexportpdfpagejob.h"
#include "webengineviewer_debug.h"
#include <KLocalizedString>
#include <QPageSetupDialog>
#include <QPrinter>
#include <QWebEngineView>
 
using namespace WebEngineViewer;
WebEngineExportPdfPageJob::WebEngineExportPdfPageJob(QObject *parent)
    : QObject(parent)
{
}
 
WebEngineExportPdfPageJob::~WebEngineExportPdfPageJob() = default;
 
void WebEngineExportPdfPageJob::start()
{
    if (!canStart()) {
        qCWarning(WEBENGINEVIEWER_LOG) << "webengineview not defined or path is not defined.! It's a bug";
        Q_EMIT exportPdfFailed();
        deleteLater();
        return;
    }
    auto printer = new QPrinter();
    printer->setOutputFormat(QPrinter::PdfFormat);
    printer->setOutputFileName(mPdfPath);
 
    auto dialog = new QPageSetupDialog(printer, mWebEngineView);
    connect(mWebEngineView->page(), &QWebEnginePage::pdfPrintingFinished, this, &WebEngineExportPdfPageJob::slotExportPdfFinished);
    if (dialog->exec() == QDialog::Accepted) {
        if (dialog->printer()->outputFormat() == QPrinter::PdfFormat) {
            mWebEngineView->page()->printToPdf(dialog->printer()->outputFileName(), dialog->printer()->pageLayout());
            delete dialog;
            delete printer;
        }
    }
}
 
void WebEngineExportPdfPageJob::slotExportPdfFinished(const QString &filePath, bool success)
{
    if (success) {
        Q_EMIT exportToPdfSuccess();
    } else {
        qCWarning(WEBENGINEVIEWER_LOG) << "Failed to export to pdf to " << filePath;
        Q_EMIT exportPdfFailed();
    }
    deleteLater();
}
 
QWebEngineView *WebEngineExportPdfPageJob::engineView() const
{
    return mWebEngineView;
}
 
void WebEngineExportPdfPageJob::setEngineView(QWebEngineView *engineView)
{
    mWebEngineView = engineView;
}
 
QString WebEngineExportPdfPageJob::pdfPath() const
{
    return mPdfPath;
}
 
void WebEngineExportPdfPageJob::setPdfPath(const QString &pdfPath)
{
    mPdfPath = pdfPath;
}
 
bool WebEngineExportPdfPageJob::canStart() const
{
    return mWebEngineView && !mPdfPath.isEmpty();
}
 
#include "moc_webengineexportpdfpagejob.cpp"