// SPDX-FileCopyrightText: 2023 Joshua Goins <josh@redstrate.com>
// SPDX-License-Identifier: GPL-3.0-or-later

#include "exdpart.h"

#include <QDebug>
#include <QFile>
#include <QTableWidget>
#include <QVBoxLayout>
#include <QJsonDocument>
#include <QJsonObject>
#include <QJsonArray>
#include <physis.hpp>

EXDPart::EXDPart(GameData* data) : data(data) {
    pageTabWidget = new QTabWidget();

    auto layout = new QVBoxLayout();
    layout->addWidget(pageTabWidget);
    setLayout(layout);
}

void EXDPart::loadSheet(const QString& name) {
    qDebug() << "Loading" << name;

    pageTabWidget->clear();

    QFile definitionFile(QStringLiteral("Achievement.json"));
    definitionFile.open(QIODevice::ReadOnly);

    auto document = QJsonDocument::fromJson(definitionFile.readAll());
    auto definitionList = document.object()[QLatin1String("definitions")].toArray();

    for (auto definition : definitionList) {
        if (definition.toObject().contains(QLatin1String("converter"))
            && definition.toObject()[QLatin1String("converter")].toObject()[QLatin1String("type")].toString() == QStringLiteral("link")) {
            auto linkName = definition.toObject()[QLatin1String("converter")].toObject()[QLatin1String("target")].toString();

            auto linkExh = physis_gamedata_read_excel_sheet_header(data, linkName.toStdString().c_str());
            auto linkExd = physis_gamedata_read_excel_sheet(data, linkName.toStdString().c_str(), linkExh, getSuitableLanguage(linkExh), 0);

            if (linkExd.p_ptr != nullptr) {
                cachedExcelSheets[linkName] = CachedExcel{linkExh, linkExd};
            }
        }
    }

    auto exh = physis_gamedata_read_excel_sheet_header(data, name.toStdString().c_str());

    for(int i = 0; i < exh->page_count; i++) {
        auto tableWidget = new QTableWidget();

        tableWidget->setColumnCount(exh->column_count);

        auto exd = physis_gamedata_read_excel_sheet(data, name.toStdString().c_str(), exh, getSuitableLanguage(exh), i);

        tableWidget->setRowCount(exd.row_count);

        for(int z = 0; z < exd.column_count; z++) {
            auto columnData = exd.row_data[0].column_data[z];

            QString columnType;
            switch (columnData.tag) {
                case physis_ColumnData::Tag::String:
                columnType = QStringLiteral("String");
                break;
                case physis_ColumnData::Tag::Bool:
                columnType = QStringLiteral("Bool");
                break;
                case physis_ColumnData::Tag::Int8:
                columnType = QStringLiteral("Int8");
                break;
                case physis_ColumnData::Tag::UInt8:
                columnType = QStringLiteral("UInt8");
                break;
                case physis_ColumnData::Tag::Int16:
                columnType = QStringLiteral("Int16");
                break;
                case physis_ColumnData::Tag::UInt16:
                columnType = QStringLiteral("UInt16");
                break;
                case physis_ColumnData::Tag::Int32:
                columnType = QStringLiteral("Int32");
                break;
                case physis_ColumnData::Tag::UInt32:
                columnType = QStringLiteral("UInt32");
                break;
                case physis_ColumnData::Tag::Float32:
                columnType = QStringLiteral("Float32");
                break;
                case physis_ColumnData::Tag::Int64:
                columnType = QStringLiteral("Int64");
                break;
                case physis_ColumnData::Tag::UInt64:
                columnType = QStringLiteral("UInt64");
                break;
            }

            columnType = definitionList[z].toObject()[QLatin1String("name")].toString();

            auto headerItem = new QTableWidgetItem();
            headerItem->setText(columnType);

            tableWidget->setHorizontalHeaderItem(z, headerItem);
        }

        for (int j = 0; j < exd.row_count; j++) {
            for(int z = 0; z < exd.column_count; z++) {
                auto columnData = exd.row_data[j].column_data[z];

                QString columnString;
                int columnRow;
                switch (columnData.tag) {
                    case physis_ColumnData::Tag::String:
                        columnString = QString::fromStdString(columnData.string._0);
                        break;
                    case physis_ColumnData::Tag::Bool:
                        columnString = columnData.bool_._0 ? QStringLiteral("True") : QStringLiteral("False");
                        break;
                    case physis_ColumnData::Tag::Int8:
                        columnString = QString::number(columnData.int8._0);
                        columnRow = columnData.int8._0;
                        break;
                    case physis_ColumnData::Tag::UInt8:
                        columnString = QString::number(columnData.u_int8._0);
                        columnRow = columnData.u_int8._0;
                        break;
                    case physis_ColumnData::Tag::Int16:
                        columnString = QString::number(columnData.int16._0);
                        columnRow = columnData.int16._0;
                        break;
                    case physis_ColumnData::Tag::UInt16:
                        columnString = QString::number(columnData.u_int16._0);
                        columnRow = columnData.u_int16._0;
                        break;
                    case physis_ColumnData::Tag::Int32:
                        columnString = QString::number(columnData.int32._0);
                        columnRow = columnData.int32._0;
                        break;
                    case physis_ColumnData::Tag::UInt32:
                        columnString = QString::number(columnData.u_int32._0);
                        columnRow = columnData.u_int32._0;
                        break;
                    case physis_ColumnData::Tag::Float32:
                        columnString = QString::number(columnData.float32._0);
                        break;
                    case physis_ColumnData::Tag::Int64:
                        columnString = QString::number(columnData.int64._0);
                        columnRow = columnData.int64._0;
                        break;
                    case physis_ColumnData::Tag::UInt64:
                        columnString = QString::number(columnData.u_int64._0);
                        columnRow = columnData.u_int64._0;
                        break;
                }

                auto definition = definitionList[z].toObject();
                if (definition.contains(QLatin1String("converter"))
                    && definition[QLatin1String("converter")].toObject()[QLatin1String("type")].toString() == QLatin1String("link")) {
                        auto linkName = definition[QLatin1String("converter")].toObject()[QLatin1String("target")].toString();

                        if (cachedExcelSheets.contains(linkName)) {
                            auto cachedExcel = cachedExcelSheets[linkName];
                            if (columnRow < cachedExcel.exd.row_count) {
                                columnString = QString::fromStdString(cachedExcel.exd.row_data[columnRow].column_data->string._0);
                            }
                        }
                }

                auto newItem = new QTableWidgetItem(columnString);

                tableWidget->setItem(j, z, newItem);
            }
        }

        tableWidget->resizeColumnsToContents();

        pageTabWidget->addTab(tableWidget, QStringLiteral("Page %1").arg(i));
    }
}

Language EXDPart::getSuitableLanguage(physis_EXH* pExh) {
    for (int i = 0; i < pExh->language_count; i++) {
        if (pExh->languages[i] == Language::English) {
            return Language::English;
        }
    }

    return Language::None;
}