File indexing completed on 2024-04-28 05:08:25

 /***************************************************************************
     Copyright (C) 2020 Robby Stephenson <robby@periapsis.org>
  ***************************************************************************/
 
 /***************************************************************************
  *                                                                         *
  *   This program is free software; you can redistribute it and/or         *
  *   modify it under the terms of the GNU General Public License as        *
  *   published by the Free Software Foundation; either version 2 of        *
  *   the License or (at your option) version 3 or any later version        *
  *   accepted by the membership of KDE e.V. (or its successor approved     *
  *   by the membership of KDE e.V.), which shall act as a proxy            *
  *   defined in Section 14 of version 3 of the license.                    *
  *                                                                         *
  *   This program is distributed in the hope that it will be useful,       *
  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  *   GNU General Public License for more details.                          *
  *                                                                         *
  *   You should have received a copy of the GNU General Public License     *
  *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
  *                                                                         *
  ***************************************************************************/
 
 #include "printhandler.h"
 #include "document.h"
 #include "controller.h"
 #include "translators/htmlexporter.h"
 #include "utils/cursorsaver.h"
 #include "config/tellico_config.h"
 #include "../tellico_debug.h"
 
 #ifdef USE_KHTML
 #include <KHTMLPart>
 #include <KHTMLView>
 #include <KAboutData>
 #else
 #include <QWebEngineView>
 #include <QWebEnginePage>
 #include <QWebEngineSettings>
 #include <QPrinter>
 #include <QPrinterInfo>
 #include <QPrintDialog>
 #include <QPrintPreviewDialog>
 #include <QPrintPreviewWidget>
 #include <QEventLoop>
 #endif
 
 using Tellico::PrintHandler;
 
 #ifndef USE_KHTML
 class WebPagePrintable : public QWebEnginePage {
 Q_OBJECT
 
 public:
   WebPagePrintable(QObject* parent) : QWebEnginePage(parent) {
     QWebEngineSettings* settings = this->settings();
     settings->setAttribute(QWebEngineSettings::JavascriptEnabled, false);
     settings->setAttribute(QWebEngineSettings::PluginsEnabled, false);
     settings->setAttribute(QWebEngineSettings::LocalContentCanAccessRemoteUrls, true);
     settings->setAttribute(QWebEngineSettings::LocalContentCanAccessFileUrls, true);
   }
 
 public Q_SLOTS:
   void printDocument(QPrinter* printer) {
     Tellico::GUI::CursorSaver cs(Qt::WaitCursor);
     QEventLoop loop;
     print(printer, [&](bool) { loop.quit(); });
     loop.exec();
   }
 };
 #endif
 
 PrintHandler::PrintHandler(QObject* parent_) : QObject(parent_)
     , m_inPrintPreview(false) {
 }
 
 PrintHandler::~PrintHandler() {
 }
 
 void PrintHandler::setEntries(const Tellico::Data::EntryList& entries_) {
   m_entries = entries_;
   m_html.clear();
 }
 
 void PrintHandler::setColumns(const QStringList& columns_) {
   m_columns = columns_;
   m_html.clear();
 }
 
 void PrintHandler::print() {
   GUI::CursorSaver cs(Qt::WaitCursor);
 
   if(m_html.isEmpty()) {
     m_html = generateHtml();
     if(m_html.isEmpty()) {
       myDebug() << "PrintHandler - empty html output";
       return;
     }
   }
 
 #ifdef USE_KHTML
   KHTMLPart w;
 
   // KHTMLPart printing was broken in KDE until KHTML 5.16
   // see https://git.reviewboard.kde.org/r/125681/
   const QString version =  w.componentData().version();
   const uint major = version.section(QLatin1Char('.'), 0, 0).toUInt();
   const uint minor = version.section(QLatin1Char('.'), 1, 1).toUInt();
   if(major == 5 && minor < 16) {
     myWarning() << "Printing is broken for KDE Frameworks < 5.16. Please upgrade";
     return;
   }
 
   w.setJScriptEnabled(false);
   w.setJavaEnabled(false);
   w.setMetaRefreshEnabled(false);
   w.setPluginsEnabled(false);
   w.begin(Data::Document::self()->URL());
   w.write(m_html);
   w.end();
   w.view()->print();
 #else
   QScopedPointer<QWebEngineView> view(new QWebEngineView);
   WebPagePrintable* page = new WebPagePrintable(view.data());
   view->setPage(page);
   view->setHtml(m_html, Data::Document::self()->URL());
 
   // don't have busy cursor when showing the print dialog
   cs.restore();
 
   auto info = QPrinterInfo::defaultPrinter();
   QScopedPointer<QPrinter> printer;
   if(info.isNull()) {
     printer.reset(new QPrinter);
   } else {
     printer.reset(new QPrinter(info));
   }
   printer->setResolution(300);
   QPointer<QPrintDialog> dialog = new QPrintDialog(printer.data(), view.data());
   if(dialog->exec() != QDialog::Accepted) {
     return;
   }
   page->printDocument(printer.data());
 #endif
 }
 
 void PrintHandler::printPreview() {
 // print preview only works with WebEngine
 #ifndef USE_KHTML
   if(m_inPrintPreview) {
     return;
   }
   GUI::CursorSaver cs(Qt::WaitCursor);
 
   if(m_html.isEmpty()) {
     m_html = generateHtml();
     if(m_html.isEmpty()) {
       myDebug() << "PrintHandler - empty html output in preview";
       return;
     }
   }
 
   QScopedPointer<QWebEngineView> view(new QWebEngineView);
   WebPagePrintable* page = new WebPagePrintable(view.data());
   view->setPage(page);
   view->setHtml(m_html, Data::Document::self()->URL());
 
   // don't have busy cursor when showing the print dialog
   cs.restore();
 
   m_inPrintPreview = true;
   auto info = QPrinterInfo::defaultPrinter();
   QScopedPointer<QPrinter> printer;
   if(info.isNull()) {
     printer.reset(new QPrinter);
   } else {
     printer.reset(new QPrinter(info));
   }
   printer->setResolution(300);
   QPrintPreviewDialog preview(printer.data(), view.data());
   connect(&preview, &QPrintPreviewDialog::paintRequested,
           page, &WebPagePrintable::printDocument);
   {
     // this is a workaround for ensuring the initial dailog open shows the preview already
     // with Qt 5.15.2, it didn't seem to get previewed initially
     QList<QPrintPreviewWidget*> list = preview.findChildren<QPrintPreviewWidget*>();
     QPrintPreviewWidget* w = list.first();
     if(w) w->updatePreview();
   }
   preview.exec();
   m_inPrintPreview = false;
 #endif
 }
 
 QString PrintHandler::generateHtml() const {
   Export::HTMLExporter exporter(Data::Document::self()->collection());
   // only print visible entries
   exporter.setEntries(m_entries);
   exporter.setXSLTFile(QStringLiteral("tellico-printing.xsl"));
   exporter.setPrintHeaders(Config::printFieldHeaders());
   exporter.setPrintGrouped(Config::printGrouped());
   exporter.setGroupBy(Controller::self()->expandedGroupBy());
   if(!Config::printGrouped()) { // the sort titles are only used if the entries are not grouped
     exporter.setSortTitles(Controller::self()->sortTitles());
   }
   exporter.setColumns(m_columns);
   exporter.setMaxImageSize(Config::maxImageWidth(), Config::maxImageHeight());
   if(Config::printFormatted()) {
     exporter.setOptions(Export::ExportUTF8 | Export::ExportFormatted);
   } else {
     exporter.setOptions(Export::ExportUTF8);
   }
 
   return exporter.text();
 }
 
 #ifndef USE_KHTML
 #include "printhandler.moc"
 #endif