File indexing completed on 2024-12-29 04:50:01

 /*
     SPDX-FileCopyrightText: 2023 Volker Krause <vkrause@kde.org>
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "extractorengine.h"
 #include "logging.h"
 
 #include "text/pricefinder_p.h"
 
 #include <KItinerary/ExtractorDocumentNodeFactory>
 
 #include <QJSValue>
 #include <QJSValueIterator>
 #include <QScopeGuard>
 
 using namespace KItinerary;
 
 static constexpr inline auto RecursionDepthLimit = 10;
 
 JsApi::ExtractorEngine::ExtractorEngine(QObject *parent)
     : QObject(parent)
 {
 }
 
 JsApi::ExtractorEngine::~ExtractorEngine() = default;
 
 void JsApi::ExtractorEngine::setEngine(KItinerary::ExtractorEngine *engine)
 {
     m_engine = engine;
 }
 
 void JsApi::ExtractorEngine::setCurrentNode(const ExtractorDocumentNode& node)
 {
     m_currentNode = node;
 }
 
 void JsApi::ExtractorEngine::clear()
 {
     m_currentNode = {};
 }
 
 ExtractorDocumentNode JsApi::ExtractorEngine::extract(const QByteArray &data)
 {
     if (m_recursionDepth > RecursionDepthLimit) {
         qCWarning(Log) << "Recursion depth limit reached, aborting";
         return {};
     }
 
     const auto preHints = m_engine->hints();
     const auto prevNode = m_currentNode;
 
     m_engine->setHints(preHints | KItinerary::ExtractorEngine::ExtractFullPageRasterImages);
     auto node = m_engine->documentNodeFactory()->createNode(data);
     m_currentNode.appendChild(node);
 
     ++m_recursionDepth;
     m_engine->processNode(node);
     --m_recursionDepth;
 
     m_engine->setHints(preHints);
     m_currentNode = prevNode;
 
     return node;
 }
 
 ExtractorDocumentNode JsApi::ExtractorEngine::extract(const QVariant &content, const QString &mimeType)
 {
     if (m_recursionDepth > RecursionDepthLimit) {
         qCWarning(Log) << "Recursion depth limit reached, aborting";
         return {};
     }
 
     const auto preHints = m_engine->hints();
     const auto prevNode = m_currentNode;
 
     m_engine->setHints(preHints | KItinerary::ExtractorEngine::ExtractFullPageRasterImages);
     auto node = m_engine->documentNodeFactory()->createNode(content, mimeType);
     m_currentNode.appendChild(node);
 
     ++m_recursionDepth;
     m_engine->processNode(node);
     --m_recursionDepth;
 
     m_engine->setHints(preHints);
     m_currentNode = prevNode;
 
     return node;
 }
 
 static void applyPrice(const PriceFinder::Result &price, QJSValue obj)
 {
     if (!obj.isObject()) {
         return;
     }
     obj.setProperty(QStringLiteral("totalPrice"), price.value);
     obj.setProperty(QStringLiteral("priceCurrency"), price.currency);
 }
 
 void JsApi::ExtractorEngine::extractPrice(const QString &text, QJSValue result) const
 {
     PriceFinder finder;
     const auto price = finder.findHighest(text);
     if (!price.hasResult()) {
         return;
     }
 
     if (result.isArray()) {
         QJSValueIterator it(result);
         while (it.hasNext()) {
             it.next();
             applyPrice(price, it.value());
         }
     } else {
         applyPrice(price, result);
     }
 }
 
 #include "moc_extractorengine.cpp"