File indexing completed on 2024-05-12 05:10:12

 /***************************************************************************
     Copyright (C) 2019 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 "librarythingimporter.h"
 #include "../collections/bookcollection.h"
 #include "../core/filehandler.h"
 #include "../utils/mapvalue.h"
 #include "../utils/isbnvalidator.h"
 #include "../tellico_debug.h"
 
 #include <KLocalizedString>
 #include <KUrlRequester>
 
 #include <QVBoxLayout>
 #include <QFormLayout>
 #include <QGroupBox>
 #include <QLabel>
 #include <QJsonDocument>
 #include <QJsonObject>
 #include <QJsonArray>
 #include <QJsonParseError>
 
 using Tellico::Import::LibraryThingImporter;
 
 LibraryThingImporter::LibraryThingImporter() : Import::Importer()
   , m_widget(nullptr)
   , m_URLRequester(nullptr) {
 }
 
 LibraryThingImporter::LibraryThingImporter(const QUrl& url) : Import::Importer(url)
   , m_widget(nullptr)
   , m_URLRequester(nullptr) {
 }
 
 bool LibraryThingImporter::canImport(int type) const {
   return type == Data::Collection::Book;
 }
 
 Tellico::Data::CollPtr LibraryThingImporter::collection() {
   if(m_coll) {
     return m_coll;
   }
 
   QUrl jsonUrl;
   if(m_widget && m_URLRequester) {
     jsonUrl = m_URLRequester->url();
   } else if(!url().isEmpty()) {
     jsonUrl = url();
   } else {
     myWarning() << "no widget!";
     return Data::CollPtr();
   }
 
   if(jsonUrl.isEmpty() || !jsonUrl.isValid()) {
     myDebug() << "Bad jsonUrl:" << jsonUrl;
     return Data::CollPtr();
   }
 
   const QByteArray data = Tellico::FileHandler::readDataFile(jsonUrl, false /* quiet */);
   QJsonParseError parseError;
   QJsonDocument doc = QJsonDocument::fromJson(data, &parseError);
   if(doc.isNull()) {
     myDebug() << "Bad json data:" << parseError.errorString();
     return Data::CollPtr();
   }
 
   QRegularExpression digits(QStringLiteral("\\d+"));
   QRegularExpression pubRx(QStringLiteral("^(.+?)[\\(,]"));
 
   m_coll = new Data::BookCollection(true);
   bool defaultTitle = true;
   Data::EntryList entries;
   QVariantMap map = doc.object().toVariantMap();
   QMapIterator<QString, QVariant> i(map);
   while(i.hasNext()) {
     i.next();
     QVariantMap valueMap = i.value().toMap();
     Data::EntryPtr entry(new Data::Entry(m_coll));
     entry->setField(QStringLiteral("title"), mapValue(valueMap, "title"));
     entry->setField(QStringLiteral("pub_year"), mapValue(valueMap, "date"));
     entry->setField(QStringLiteral("keyword"), mapValue(valueMap, "tags"));
     entry->setField(QStringLiteral("genre"), mapValue(valueMap, "genre"));
     entry->setField(QStringLiteral("series"), mapValue(valueMap, "series"));
     entry->setField(QStringLiteral("language"), mapValue(valueMap, "language"));
 
     const QString collName = mapValue(valueMap, "collections");
     // default to using a collection title based on the first entry
     if(defaultTitle && !collName.isEmpty()) {
       defaultTitle = false;
       m_coll->setTitle(collName);
     }
 
     const QString cdate = mapValue(valueMap, "entrydate");
     if(!cdate.isEmpty()) {
       entry->setField(QStringLiteral("cdate"), cdate);
     }
 
     QString pub = mapValue(valueMap, "publication");
     // try to clean up the publisher
     QRegularExpressionMatch pubMatch = pubRx.match(pub);
     if(pubMatch.hasMatch()) {
       pub = pubMatch.captured(1).simplified();
     }
     entry->setField(QStringLiteral("publisher"), pub);
 
     QJsonArray authorArray = valueMap.value(QStringLiteral("authors")).toJsonArray();
     QStringList authors;
     for(int i = 0; i < authorArray.size(); ++i) {
       QVariantMap m = authorArray.at(i).toObject().toVariantMap();
       // TODO: read config option for author formatting?
       // use first-lastname for now
       const QString role = mapValue(m, "role");
       if(role.isEmpty() || role == QLatin1String("Author")) {
         authors += mapValue(m, "fl");
       }
     }
     entry->setField(QStringLiteral("author"), authors.join(FieldFormat::delimiterString()));
 
     QJsonArray formatArray = valueMap.value(QStringLiteral("format")).toJsonArray();
     if(!formatArray.isEmpty()) {
       // use the first one
       const QVariantMap m = formatArray.at(0).toObject().toVariantMap();
       const QString format = mapValue(m, "text");
       if(format == QLatin1String("Paperback")) {
         entry->setField(QStringLiteral("binding"), i18n("Paperback"));
       } else if(format == QLatin1String("Hardcover")) {
         entry->setField(QStringLiteral("binding"), i18n("Hardback"));
       } else {
         // just in case there's a value there
         entry->setField(QStringLiteral("binding"), format);
       }
     }
 
     QString isbn = mapValue(valueMap, "originalisbn");
     ISBNValidator::staticFixup(isbn);
     entry->setField(QStringLiteral("isbn"), isbn);
 
     // grab first set of digits
     QRegularExpressionMatch match = digits.match(mapValue(valueMap, "pages"));
     if(match.hasMatch()) {
       entry->setField(QStringLiteral("pages"), match.captured(0));
     }
     entries += entry;
   }
   m_coll->addEntries(entries);
   return m_coll;
 }
 
 QWidget* LibraryThingImporter::widget(QWidget* parent_) {
   if(m_widget) {
     return m_widget;
   }
   m_widget = new QWidget(parent_);
   QVBoxLayout* l = new QVBoxLayout(m_widget);
 
   QGroupBox* gbox = new QGroupBox(i18n("LibraryThing Options"), m_widget);
   QFormLayout* lay = new QFormLayout(gbox);
 
   auto label = new QLabel(i18n("Export your LibraryThing collection in "
                                "<a href=\"https://www.librarything.com/export.php?export_type=json\">JSON format</a>."), gbox);
   label->setOpenExternalLinks(true);
   lay->addRow(label);
 
   m_URLRequester = new KUrlRequester(gbox);
   // these are in the old KDE4 filter format, not the Qt5 format
   QString filter = QLatin1String("*.json|") + i18n("JSON Files")
                  + QLatin1Char('\n')
                  + QLatin1String("*|") + i18n("All Files");
   m_URLRequester->setFilter(filter);
 
   lay->addRow(i18n("LibraryThing file:"), m_URLRequester);
 
   l->addWidget(gbox);
   l->addStretch(1);
 
   return m_widget;
 }