#include "fastfile_cod5_ps3.h"
#include "zonefile_cod5_ps3.h"

#include "utils.h"
#include "compression.h"
#include "statusbarmanager.h"

#include <QFile>
#include <QDebug>

FastFile_COD5_PS3::FastFile_COD5_PS3()
    : FastFile() {
    SetCompany(COMPANY_INFINITY_WARD);
    SetType(FILETYPE_FAST_FILE);
    SetSignage(SIGNAGE_UNSIGNED);
    SetMagic(0);
    SetVersion(0);
    SetGame("COD5");
    SetPlatform("PS3");
}

FastFile_COD5_PS3::FastFile_COD5_PS3(const QByteArray& aData)
    : FastFile_COD5_PS3() {

    if (!aData.isEmpty()) {
        Load(aData);
    }
}

FastFile_COD5_PS3::FastFile_COD5_PS3(const QString aFilePath)
    : FastFile_COD5_PS3() {

    if (!aFilePath.isEmpty()) {
        Load(aFilePath);
    }
}

FastFile_COD5_PS3::~FastFile_COD5_PS3() {

}

QByteArray FastFile_COD5_PS3::GetBinaryData() const {
    return QByteArray();
}

bool FastFile_COD5_PS3::Load(const QString aFilePath) {
    StatusBarManager::instance().updateStatus("Loading COD5 Fast File w/path", 1000);

    if (aFilePath.isEmpty()) {
        return false;
    }

    // Check fastfile can be read
    QFile *file = new QFile(aFilePath);
    if (!file->open(QIODevice::ReadOnly)) {
        qDebug() << QString("Error: Failed to open FastFile: %1!").arg(aFilePath);
        return false;
    }

    // Decompress fastfile and close
    const QString fastFileStem = aFilePath.section("/", -1, -1);
    SetStem(fastFileStem);
    if (!Load(file->readAll())) {
        qDebug() << "Error: Failed to load fastfile: " << fastFileStem;
        return false;
    }

    file->close();

    // Open zone file after decompressing ff and writing
    return true;
}

bool FastFile_COD5_PS3::Load(const QByteArray aData) {
    StatusBarManager::instance().updateStatus("Loading COD5 Fast File w/data", 1000);
    QByteArray decompressedData;

    // Create a XDataStream on the input data.
    XDataStream fastFileStream(aData);
    fastFileStream.setByteOrder(XDataStream::LittleEndian);

    // Parse header values.
    SetCompany(pParseFFCompany(&fastFileStream));
    SetType(pParseFFFileType(&fastFileStream));
    SetSignage(pParseFFSignage(&fastFileStream));
    SetMagic(pParseFFMagic(&fastFileStream));
    SetVersion(pParseFFVersion(&fastFileStream));

    // Loop until EOF or invalid chunk
    fastFileStream.setByteOrder(XDataStream::BigEndian);
    while (!fastFileStream.atEnd()) {
        // Read 2-byte BIG-ENDIAN chunk size
        quint16 chunkSize;
        fastFileStream >> chunkSize;

        if (chunkSize == 0 || fastFileStream.device()->pos() + chunkSize > aData.size()) {
            qWarning() << "Invalid or incomplete chunk detected, stopping.";
            break;
        }

        QByteArray compressedChunk(chunkSize, Qt::Uninitialized);
        fastFileStream.readRawData(compressedChunk.data(), chunkSize);

        QByteArray decompressedChunk = Compression::DecompressDeflate(compressedChunk);
        decompressedData.append(decompressedChunk);
    }
    Utils::ExportData(GetBaseStem() + ".zone", decompressedData);

    ZoneFile_COD5_PS3* zoneFile = new ZoneFile_COD5_PS3();
    zoneFile->SetStem(GetBaseStem() + ".zone");
    if (!zoneFile->Load(decompressedData)) {
        qWarning() << "Failed to load ZoneFile!";
        return false;
    }
    SetZoneFile(zoneFile);

    return true;
}