oracleimporter.h

Go to the documentation of this file.

#ifndef ORACLEIMPORTER_H
#define ORACLEIMPORTER_H
 
#include "libcockatrice/card/printing/printing_info.h"
#include "libcockatrice/card/set/card_set.h"
#include "raw_json_scanner.h"
 
#include <QAtomicInt>
#include <QByteArray>
#include <QDate>
#include <QHash>
#include <QJsonArray>
#include <QJsonObject>
#include <QList>
#include <QMap>
#include <QObject>
#include <QString>
#include <QStringList>
#include <libcockatrice/card/card_info.h>
#include <qnamespace.h>
#include <qtmetamacros.h>
#include <utility>
 
class CardRelation;
class QRegularExpression;
 
// many users prefer not to see these sets with non english arts
// they will given priority PriorityLowest
const QStringList nonEnglishSets = {"4BB", "FBB", "PS11", "PSAL", "REN", "RIN"};
const QMap<QString, CardSet::Priority> setTypePriorities{
    {"core", CardSet::PriorityPrimary},
    {"expansion", CardSet::PriorityPrimary},
 
    {"commander", CardSet::PrioritySecondary},
    {"starter", CardSet::PrioritySecondary},
    {"draft_innovation", CardSet::PrioritySecondary},
    {"duel_deck", CardSet::PrioritySecondary},
 
    {"archenemy", CardSet::PriorityReprint},
    {"arsenal", CardSet::PriorityReprint},
    {"box", CardSet::PriorityReprint},
    {"eternal", CardSet::PriorityReprint},
    {"from_the_vault", CardSet::PriorityReprint},
    {"masterpiece", CardSet::PriorityReprint},
    {"masters", CardSet::PriorityReprint},
    {"memorabilia", CardSet::PriorityReprint},
    {"planechase", CardSet::PriorityReprint},
    {"premium_deck", CardSet::PriorityReprint},
    {"promo", CardSet::PriorityReprint},
    {"spellbook", CardSet::PriorityReprint},
    {"token", CardSet::PriorityReprint},
    {"treasure_chest", CardSet::PriorityReprint},
 
    {"alchemy", CardSet::PriorityOther},
    {"funny", CardSet::PriorityOther},
    {"minigame", CardSet::PriorityOther},
    {"vanguard", CardSet::PriorityOther},
};
 
class SetToDownload
{
private:
    QString shortName, longName;
    QDate releaseDate;
    QString setType;
    CardSet::Priority priority;
    // Byte range of this set's object within the importer's raw JSON text. Parsing
    // one set at a time keeps peak memory low instead of holding the whole document.
    RawJson::SetDataRange rawRange;
 
public:
    const QString &getShortName() const
    {
        return shortName;
    }
    const QString &getLongName() const
    {
        return longName;
    }
    const QString &getSetType() const
    {
        return setType;
    }
    const QDate &getReleaseDate() const
    {
        return releaseDate;
    }
    CardSet::Priority getPriority() const
    {
        return priority;
    }
    const RawJson::SetDataRange &getRawRange() const
    {
        return rawRange;
    }
    SetToDownload(QString _shortName,
                  QString _longName,
                  CardSet::Priority _priority,
                  QString _setType = QString(),
                  const QDate &_releaseDate = QDate())
        : shortName(std::move(_shortName)), longName(std::move(_longName)), releaseDate(_releaseDate),
          setType(std::move(_setType)), priority(_priority)
    {
    }
    void setRawRange(const RawJson::SetDataRange &_rawRange)
    {
        rawRange = _rawRange;
    }
    bool operator<(const SetToDownload &set) const
    {
        return longName.compare(set.longName, Qt::CaseInsensitive) < 0;
    }
};
 
class SplitCardPart
{
public:
    SplitCardPart(const QString &_name,
                  const QString &_text,
                  const QHash<QString, QString> &_properties,
                  const PrintingInfo &_printingInfo,
                  const QString &_localizedText = QString());
    inline const QString &getName() const
    {
        return name;
    }
    inline const QString &getText() const
    {
        return text;
    }

    inline const QString &getLocalizedText() const
    {
        return localizedText;
    }
    inline const QHash<QString, QString> &getProperties() const
    {
        return properties;
    }
    inline const PrintingInfo &getPrintingInfo() const
    {
        return printingInfo;
    }
 
private:
    QString name;
    QString text;
    QString localizedText;
    QHash<QString, QString> properties;
    PrintingInfo printingInfo;
};
 
struct LocalizedCardEntry
{
    QString name;
    QString text;
    CardSet::Priority priority = CardSet::PriorityLowest;
};
 
class OracleImporter : public QObject
{
    Q_OBJECT
private:
    static const QRegularExpression formatRegex;

    CardNameMap cards;

    SetNameMap sets;
 
    QList<SetToDownload> allSets;

    QByteArray rawSetsData;

    bool progressReporting = true;

    QAtomicInt importCancelled;

    QString cardLang = "en";

    bool localizationEnabled = false;

    QMap<QString, LocalizedCardEntry> localizedEntries;

    QMap<QString, LocalizedCardEntry> splitLocalizedTexts;
 
    CardInfoPtr addCard(QString name,
                        const QString &text,
                        bool isToken,
                        QHash<QString, QString> properties,
                        const QList<CardRelation *> &relatedCards,
                        const PrintingInfo &printingInfo);

    void collectForeignData(const QString &cardKey,
                            const CardSetPtr &currentSet,
                            const QJsonObject &card,
                            bool collectText = true);
signals:
    void setIndexChanged(int cardsImported, int setIndex, const QString &setName);
    void dataReadProgress(int bytesRead, int totalBytes);
 
public:
    explicit OracleImporter(QObject *parent = nullptr);
    void setProgressReporting(bool enabled)
    {
        progressReporting = enabled;
    }

    void setCardLang(const QString &lang);
    const QString &getCardLang() const
    {
        return cardLang;
    }

    bool readSetsFromByteArray(QByteArray data);
    int startImport();
    void cancelImport()
    {
        importCancelled.storeRelease(1);
    }

    void applyLocalizedData();
    bool saveToFile(const QString &fileName, const QString &sourceUrl, const QString &sourceVersion);
    int importCardsFromSet(const CardSetPtr &currentSet, const QJsonArray &cardsList);
    const FormatRulesNameMap &createDefaultMagicFormats();
    const CardNameMap &getCardList() const
    {
        return cards;
    }
    QList<SetToDownload> &getSets()
    {
        return allSets;
    }
    const QByteArray &getRawSetsData() const
    {
        return rawSetsData;
    }
    void releaseSetData();
    void clear();
};
 
#endif