File indexing completed on 2024-05-05 11:56:06

 /*
     SPDX-License-Identifier: GPL-2.0-or-later
     SPDX-FileCopyrightText: 2019 Nikita Sirgienko <warquark@gmail.com>
 */
 
 #include "mimeresult.h"
 
 #include <QDebug>
 #include <QJsonArray>
 #include <QJsonDocument>
 #include <QFile>
 #include <KLocalizedString>
 
 #include "jupyterutils.h"
 
 using namespace Cantor;
 
 class Cantor::MimeResultPrivate
 {
 public:
     MimeResultPrivate() = default;
 
     QString plain;
     QJsonObject mimeBundle;
 };
 
 MimeResult::MimeResult(const QJsonObject& mimeBundle) : d(new MimeResultPrivate)
 {
     bool isOriginalPlain = mimeBundle.contains(QLatin1String("text/plain"));
     if (isOriginalPlain)
         d->plain = JupyterUtils::fromJupyterMultiline(mimeBundle.value(QLatin1String("text/plain")));
     else
         d->plain = i18n("This is unsupported Jupyter content of types ('%1')", mimeBundle.keys().join(QLatin1String(", ")));
     d->mimeBundle = mimeBundle;
 }
 
 MimeResult::~MimeResult()
 {
     delete d;
 }
 
 QString MimeResult::toHtml()
 {
     return QLatin1String("<pre>") + d->plain.toHtmlEscaped() + QLatin1String("</pre>");
 }
 
 int MimeResult::type()
 {
     return MimeResult::Type;
 }
 
 QString MimeResult::mimeType()
 {
     return QLatin1String("application/json");
 }
 
 QVariant MimeResult::data()
 {
     return d->mimeBundle;
 }
 
 QString MimeResult::plain()
 {
     return d->plain;
 }
 
 QDomElement MimeResult::toXml(QDomDocument& doc)
 {
     qDebug()<<"saving mime result with types" << d->mimeBundle.keys();
     QDomElement e=doc.createElement(QStringLiteral("Result"));
     e.setAttribute(QStringLiteral("type"), QStringLiteral("mime"));
 
     for (const QString& key : d->mimeBundle.keys())
     {
         QJsonDocument jsonDoc;
         QJsonObject obj;
         obj.insert(QLatin1String("content"), d->mimeBundle[key]);
         jsonDoc.setObject(obj);
 
         QDomElement content = doc.createElement(QStringLiteral("Content"));
         content.setAttribute(QStringLiteral("key"), key);
         content.appendChild(doc.createTextNode(QString::fromUtf8(jsonDoc.toJson())));
         e.appendChild(content);
     }
 
     return e;
 }
 
 QJsonValue Cantor::MimeResult::toJupyterJson()
 {
     QJsonObject root;
     if (executionIndex() != -1)
     {
         root.insert(QLatin1String("output_type"), QLatin1String("execute_result"));
         root.insert(QLatin1String("execution_count"), executionIndex());
     }
     else
         root.insert(QLatin1String("output_type"), QLatin1String("display_data"));
 
     root.insert(QLatin1String("data"), d->mimeBundle);
     root.insert(QLatin1String("metadata"), jupyterMetadata());
 
     return root;
 }
 
 void Cantor::MimeResult::save(const QString& filename)
 {
     QFile file(filename);
 
     if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
         return;
 
     QTextStream stream(&file);
 
     QJsonDocument jsonDoc;
     jsonDoc.setObject(d->mimeBundle);
 
     stream << jsonDoc.toJson();
 
     file.close();
 }