#ifndef FASTFILE_FACTORY_H
#define FASTFILE_FACTORY_H

#include "360/fastfile_cod2_360.h"
//#include "360/fastfile_cod3_360.h"
#include "360/fastfile_cod4_360.h"
#include "360/fastfile_cod5_360.h"
#include "360/fastfile_cod6_360.h"
#include "360/fastfile_cod7_360.h"
#include "360/fastfile_cod7_5_360.h"
#include "360/fastfile_cod8_360.h"
#include "360/fastfile_cod9_360.h"
#include "360/fastfile_cod10_360.h"
#include "360/fastfile_cod11_360.h"
#include "360/fastfile_cod12_360.h"

//#include "PS3/fastfile_cod3_ps3.h"
#include "PS3/fastfile_cod4_ps3.h"
#include "PS3/fastfile_cod5_ps3.h"
#include "PS3/fastfile_cod6_ps3.h"
#include "PS3/fastfile_cod7_ps3.h"
#include "PS3/fastfile_cod8_ps3.h"
#include "PS3/fastfile_cod9_ps3.h"
#include "PS3/fastfile_cod10_ps3.h"
#include "PS3/fastfile_cod11_ps3.h"
#include "PS3/fastfile_cod12_ps3.h"

//#include "PC/fastfile_cod3_pc.h"
#include "PC/fastfile_cod4_pc.h"
#include "PC/fastfile_cod5_pc.h"
#include "PC/fastfile_cod6_pc.h"
#include "PC/fastfile_cod7_pc.h"
#include "PC/fastfile_cod8_pc.h"
#include "PC/fastfile_cod9_pc.h"
#include "PC/fastfile_cod10_pc.h"
#include "PC/fastfile_cod11_pc.h"
#include "PC/fastfile_cod12_pc.h"

#include "Wii/fastfile_cod4_wii.h"
#include "Wii/fastfile_cod7_wii.h"
#include "Wii/fastfile_cod8_wii.h"

#include "WiiU/fastfile_cod9_wiiu.h"
#include "WiiU/fastfile_cod10_wiiu.h"

#include <QByteArray>
#include <QString>
#include <memory>
#include <QFile>
#include <QDebug>

class FastFile;

enum FastFile_Platform {
    PLATFORM_NONE = 0x00,
    PLATFORM_PC = 0x01,
    PLATFORM_360 = 0x02,
    PLATFORM_PS3 = 0x03,
    PLATFORM_WII = 0x04,
    PLATFORM_WIIU = 0x05
};

enum FastFile_Game {
    GAME_NONE   = 0x00,
    GAME_COD2   = 0x01,
    GAME_COD4   = 0x02,
    GAME_COD5   = 0x03,
    GAME_COD6   = 0x04,
    GAME_COD7   = 0x05,
    GAME_COD7_5 = 0x06,
    GAME_COD8   = 0x07,
    GAME_COD9   = 0x08,
    GAME_COD10  = 0x09,
    GAME_COD11  = 0x10,
    GAME_COD12  = 0x11
};

class FastFileFactory {
public:
    static FastFile* Create(const QString& aFilePath, FastFile_Platform aPlatform = PLATFORM_NONE, FastFile_Game aGame = GAME_NONE) {
        QFile fastFile(aFilePath);
        if (!fastFile.open(QIODevice::ReadOnly)) {
            qDebug() << "Factory failed to open fast file: " << aFilePath;
            return nullptr;
        }

        const QString stem = aFilePath.split('/').last().split('.').first();

        const QByteArray data = fastFile.readAll();
        FastFile* resultFF = nullptr;

        if (aPlatform == PLATFORM_NONE) {
            aPlatform = pGetPlatform(data);
        }
        if (aGame == GAME_NONE) {
            aGame = pGetGame(data);
        }

        if (aPlatform == PLATFORM_360) {
            if (aGame == GAME_COD2) {
                resultFF = new FastFile_COD2_360(aFilePath);
            } else if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_360(aFilePath);
            } else if (aGame == GAME_COD5) {
                resultFF = new FastFile_COD5_360(aFilePath);
            } else if (aGame == GAME_COD6) {
                resultFF = new FastFile_COD6_360(aFilePath);
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_360(aFilePath);
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_360(aFilePath);
            } else if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_360(aFilePath);
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_360(aFilePath);
            } else if (aGame == GAME_COD11) {
                resultFF = new FastFile_COD11_360(aFilePath);
            } else if (aGame == GAME_COD12) {
                resultFF = new FastFile_COD12_360(aFilePath);
            }
        } else if (aPlatform == PLATFORM_PC) {
            if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_PC(aFilePath);
            } else if (aGame == GAME_COD5) {
                resultFF = new FastFile_COD5_PC(aFilePath);
            } else if (aGame == GAME_COD6) {
                resultFF = new FastFile_COD6_PC(aFilePath);
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_PC(aFilePath);
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_PC(aFilePath);
            } else if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_PC(aFilePath);
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_PC(aFilePath);
            } else if (aGame == GAME_COD11) {
                resultFF = new FastFile_COD11_PC(aFilePath);
            } else if (aGame == GAME_COD12) {
                resultFF = new FastFile_COD12_PC(aFilePath);
            }
        } else if (aPlatform == PLATFORM_PS3) {
            if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_PS3(aFilePath);
            } else if (aGame == GAME_COD5) {
                resultFF = new FastFile_COD5_PS3(aFilePath);
            } else if (aGame == GAME_COD6) {
                resultFF = new FastFile_COD6_PS3(aFilePath);
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_PS3(aFilePath);
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_PS3(aFilePath);
            } else if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_PS3(aFilePath);
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_PS3(aFilePath);
            } else if (aGame == GAME_COD11) {
                resultFF = new FastFile_COD11_PS3(aFilePath);
            } else if (aGame == GAME_COD12) {
                resultFF = new FastFile_COD12_PS3(aFilePath);
            }
        } else if (aPlatform == PLATFORM_WII) {
            if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_Wii(aFilePath);
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_Wii(aFilePath);
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_Wii(aFilePath);
            }
        } else if (aPlatform == PLATFORM_WIIU) {
            if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_WiiU(aFilePath);
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_WiiU(aFilePath);
            }
        }
        return resultFF;
    }
    static FastFile* Create(const QByteArray& aData, const QString aStem = "no_name", FastFile_Platform aPlatform = PLATFORM_NONE, FastFile_Game aGame = GAME_NONE) {
        FastFile* resultFF = nullptr;

        if (aPlatform == PLATFORM_NONE) {
            aPlatform = pGetPlatform(aData);
        }
        if (aGame == GAME_NONE) {
            aGame = pGetGame(aData);
        }

        if (aPlatform == PLATFORM_360) {
            if (aGame == GAME_COD2) {
                resultFF = new FastFile_COD2_360();
            } else if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_360();
            } else if (aGame == GAME_COD5) {
                resultFF = new FastFile_COD5_360();
            } else if (aGame == GAME_COD6) {
                resultFF = new FastFile_COD6_360();
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_360();
            } else if (aGame == GAME_COD7_5) {
                resultFF = new FastFile_COD7_5_360();
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_360();
            } else if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_360();
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_360();
            } else if (aGame == GAME_COD11) {
                resultFF = new FastFile_COD11_360();
            } else if (aGame == GAME_COD12) {
                resultFF = new FastFile_COD12_360();
            }
        } else if (aPlatform == PLATFORM_PC) {
            if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_PC();
            } else if (aGame == GAME_COD5) {
                resultFF = new FastFile_COD5_PC();
            } else if (aGame == GAME_COD6) {
                resultFF = new FastFile_COD6_PC();
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_PC();
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_PC();
            } else if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_PC();
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_PC();
            } else if (aGame == GAME_COD11) {
                resultFF = new FastFile_COD11_PC();
            } else if (aGame == GAME_COD12) {
                resultFF = new FastFile_COD12_PC();
            }
        } else if (aPlatform == PLATFORM_PS3) {
            if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_PS3();
            } else if (aGame == GAME_COD5) {
                resultFF = new FastFile_COD5_PS3();
            } else if (aGame == GAME_COD6) {
                resultFF = new FastFile_COD6_PS3();
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_PS3();
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_PS3();
            } else if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_PS3();
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_PS3();
            } else if (aGame == GAME_COD11) {
                resultFF = new FastFile_COD11_PS3();
            } else if (aGame == GAME_COD12) {
                resultFF = new FastFile_COD12_PS3();
            }
        } else if (aPlatform == PLATFORM_WII) {
            if (aGame == GAME_COD4) {
                resultFF = new FastFile_COD4_Wii();
            } else if (aGame == GAME_COD7) {
                resultFF = new FastFile_COD7_Wii();
            } else if (aGame == GAME_COD8) {
                resultFF = new FastFile_COD8_Wii();
            }
        } else if (aPlatform == PLATFORM_WIIU) {
            if (aGame == GAME_COD9) {
                resultFF = new FastFile_COD9_WiiU();
            } else if (aGame == GAME_COD10) {
                resultFF = new FastFile_COD10_WiiU();
            }
        }
        if (resultFF) {
            resultFF->SetStem(aStem);
            resultFF->Load(aData);
        }
        return resultFF;
    }
private:
    static QStringList pGetDataSections(const QByteArray& aData) {
        QStringList sections;

        sections << aData.mid(0, 2).toHex().toUpper();
        sections << aData.mid(2, 2).toHex().toUpper();
        sections << aData.mid(4, 2).toHex().toUpper();
        sections << aData.mid(6, 2).toHex().toUpper();
        sections << aData.mid(8, 2).toHex().toUpper();
        sections << aData.mid(10, 2).toHex().toUpper();
        sections << aData.mid(12, 2).toHex().toUpper();
        sections << aData.mid(14, 2).toHex().toUpper();

        return sections;
    }

    static FastFile_Platform pGetPlatform(const QByteArray& aData) {
        const QStringList sections = pGetDataSections(aData);
        if (sections[0] == "0000" || sections[0] == "4E58") {
            return PLATFORM_360;
        } else if (sections[4] == "0000") {
            if (sections[5] == "0001" && sections[6] == "78DA") {
                return PLATFORM_360;
            } else if (sections[5] == "0001" && sections[6] == "4957") {
                return PLATFORM_360;
            } else if (sections[5] == "0183" && sections[6] == "7801") {
                return PLATFORM_360;
            } else if (sections[5] == "01D9" && sections[6] == "0000") {
                return PLATFORM_360;
            } else if (sections[6] == "0101" && sections[7] == "CA3E") {
                return PLATFORM_360;
            } else if (sections[6] == "0000" && sections[7] == "0001") {
                return PLATFORM_360;
            } else if (sections[6] == "0101" && sections[7] == "CC76") {
                return PLATFORM_360;
            } else if (sections[6] == "0101" && sections[7] == "0101") {
                return PLATFORM_360;
            } else if (sections[2] == "7831") {
                return PLATFORM_360;
            } else if (sections[0] == "5331" && sections[2] == "7531") {
                return PLATFORM_360;
            } else if (sections[2] == "3030" && sections[3] == "3030") {
                return PLATFORM_360;
            } else if (sections[5] == "01A2" && sections[6] == "7801") {
                return PLATFORM_WII;
            } else if (sections[5] == "01DD" && sections[6] == "7801") {
                return PLATFORM_WII;
            } else if (sections[5] == "006B") {
                return PLATFORM_WII;
            }
        } else if (sections[5] == "0000") {
            return PLATFORM_PC;
        }else if (sections[5] == "0094" || sections[6] == "0100") {
            return PLATFORM_WIIU;
        }
        return PLATFORM_PS3;
    }

    static FastFile_Game pGetGame(const QByteArray& aData) {
        const QStringList sections = pGetDataSections(aData);
        if (sections[0] == "0000") {
            return GAME_COD2;
        } else if (sections[4] == "0500" || sections[5] == "01A2" || sections[5] == "0001") {
            return GAME_COD4;
        } else if (sections[4] == "8301" || sections[5] == "0183") {
            return GAME_COD5;
        } else if (sections[4] == "1401" || sections[5] == "010D") {
            return GAME_COD6;
        } else if (sections[4] == "D901" || sections[5] == "01DD" || sections[5] == "01D9") {
            return GAME_COD7;
        } else if (sections[0] == "4E58") {
            return GAME_COD7_5;
        } else if (sections[4] == "0100" || sections[5] == "006B" || sections[5] == "0070") {
            return GAME_COD8;
        } else if (sections[4] == "9300" || sections[5] == "0092"
                   || sections[5] == "0094") {
            return GAME_COD9;
        } else if (sections[4] == "3502" || sections[5] == "022E") {
            return GAME_COD10;
        } else if (sections[0] == "5331") {
            return GAME_COD11;
        } else if (sections[2] == "3030") {
            return GAME_COD12;
        }
        return GAME_NONE;
    }
};

#endif // FASTFILE_FACTORY_H