#include "Common.hpp"
#include <QStringList>

#ifndef _WIN32
#include <unistd.h>
#endif

#if __APPLE__
const QString CurPlatformString = QStringLiteral("macos");
#elif _WIN32
const QString CurPlatformString = QStringLiteral("win32");
#elif __linux__
const QString CurPlatformString = QStringLiteral("linux");
#else
#error HECL does not know which OS to fetch for
#endif

QString PlatformToString(Platform plat)
{
    switch (plat)
    {
    case Platform::MacOS:
        return QStringLiteral("macos");
    case Platform::Win32:
        return QStringLiteral("win32");
    case Platform::Linux:
        return QStringLiteral("linux");
    default:
        return QString();
    }
}

Architecture CurArchitecture = Architecture::Invalid;
QString CurArchitectureString;

Platform StringToPlatform(const QString& str)
{
    for (int i = 1; i < int(Platform::MAXPlatform); ++i)
        if (!str.compare(PlatformToString(Platform(i)), Qt::CaseInsensitive))
            return Platform(i);
    return Platform::Invalid;
}

QString ArchitectureToString(Architecture arch)
{
    switch (arch)
    {
    case Architecture::X86:
        return QStringLiteral("x86");
    case Architecture::X86_64:
        return QStringLiteral("x86_64");
    case Architecture::ARM:
        return QStringLiteral("arm");
    case Architecture::AARCH64:
        return QStringLiteral("aarch64");
    default:
        return QString();
    }
}

Architecture StringToArchitecture(const QString& str)
{
    for (int i = 1; i < int(Architecture::MAXArchitecture); ++i)
        if (!str.compare(ArchitectureToString(Architecture(i)), Qt::CaseInsensitive))
            return Architecture(i);
    return Architecture::Invalid;
}

QString VectorISAToString(VectorISA visa)
{
    switch (visa)
    {
    case VectorISA::X87:
        return QStringLiteral("x87");
    case VectorISA::SSE:
        return QStringLiteral("sse");
    case VectorISA::SSE2:
        return QStringLiteral("sse2");
    case VectorISA::SSE3:
        return QStringLiteral("sse3");
    case VectorISA::SSE41:
        return QStringLiteral("sse41");
    case VectorISA::AVX:
        return QStringLiteral("avx");
    case VectorISA::AVX2:
        return QStringLiteral("avx2");
    default:
        return QString();
    }
}

VectorISA StringToVectorISA(const QString& str)
{
    for (int i = 1; i < int(VectorISA::MAXVectorISA); ++i)
        if (!str.compare(VectorISAToString(VectorISA(i)), Qt::CaseInsensitive))
            return VectorISA(i);
    return VectorISA::Invalid;
}

URDEVersion::URDEVersion(const QString& filename)
{
    int idx;
    QString useFilename = filename;
    if ((idx = filename.indexOf('.')) >= 0)
    {
        m_extension = QString(filename).remove(0, idx);
        useFilename.truncate(idx);
    }
    QStringList list = useFilename.split('-');
    if (list.size() >= 2)
        m_version = list[1].toInt();
    if (list.size() >= 3)
        m_platform = StringToPlatform(list[2]);
    if (list.size() >= 4)
        m_architecture = StringToArchitecture(list[3]);
    if (list.size() >= 5)
        m_vectorISA = StringToVectorISA(list[4]);
}

QString URDEVersion::fileString(bool withExtension) const
{
    if (m_version < 0)
        return {};
    if (withExtension && !m_extension.isEmpty())
        return QString("urde-%1-%2-%3-%4%5").arg(QString::number(m_version),
                                                 PlatformToString(m_platform),
                                                 ArchitectureToString(m_architecture),
                                                 VectorISAToString(m_vectorISA),
                                                 m_extension);
    else
        return QString("urde-%1-%2-%3-%4").arg(QString::number(m_version),
                                               PlatformToString(m_platform),
                                               ArchitectureToString(m_architecture),
                                               VectorISAToString(m_vectorISA));
}

void InitializePlatform()
{
#if ZEUS_ARCH_X86_64
    const_cast<Architecture&>(CurArchitecture) = Architecture::X86_64;
#elif ZEUS_ARCH_X86
#if !defined(__APPLE__) && !defined(_WIN32)
    const_cast<Architecture&>(CurArchitecture) =
        (sysconf(_SC_WORD_BIT) == 64 ? Architecture::X86_64 : Architecture::X86);
#elif _WIN32
    bool isWOW = false;
    IsWow64Process(GetCurrentProcess(), &isWOW);
    const_cast<Architecture&>(CurArchitecture) =
        (isWOW ? Architecture::X86_64 : Architecture::X86);
#endif
#endif
    const_cast<QString&>(CurArchitectureString) = ArchitectureToString(CurArchitecture);
}