#ifndef __URDE_CRESFACTORY_HPP__
#define __URDE_CRESFACTORY_HPP__

#include <unordered_map>
#include "IFactory.hpp"
#include "CResLoader.hpp"
#include "IVParamObj.hpp"
#include "MP1/MP1OriginalIDs.hpp"
#include "CToken.hpp"

namespace urde
{
class IDvdRequest;
class CSimplePool;

class CResFactory : public IFactory
{
    CResLoader x4_loader;
    CFactoryMgr x5c_factoryMgr;
    TLockedToken<MP1OriginalIDs> m_origIds;

public:
    struct SLoadingData
    {
        SObjectTag x0_tag;
        std::shared_ptr<IDvdRequest> x8_dvdReq;
        std::unique_ptr<IObj>* xc_targetPtr = nullptr;
        std::unique_ptr<u8[]> x10_loadBuffer;
        u32 x14_resSize = 0;
        CVParamTransfer x18_cvXfer;
        bool m_compressed = false;
        CObjectReference* m_selfRef = nullptr;

        SLoadingData() = default;
        SLoadingData(const SObjectTag& tag, std::unique_ptr<IObj>* ptr, const CVParamTransfer& xfer,
                     bool compressed, CObjectReference* selfRef)
        : x0_tag(tag), xc_targetPtr(ptr), x18_cvXfer(xfer), m_compressed(compressed), m_selfRef(selfRef) {}
    };
private:
    std::list<SLoadingData> m_loadList;
    std::unordered_map<SObjectTag, std::list<SLoadingData>::iterator> m_loadMap;
    void AddToLoadList(SLoadingData&& data);
    CFactoryFnReturn BuildSync(const SObjectTag&, const CVParamTransfer&, CObjectReference* selfRef);
    bool PumpResource(SLoadingData& data);
public:
    CResLoader& GetLoader() {return x4_loader;}
    std::unique_ptr<IObj> Build(const SObjectTag&, const CVParamTransfer&, CObjectReference* selfRef);
    void BuildAsync(const SObjectTag&, const CVParamTransfer&, std::unique_ptr<IObj>*, CObjectReference* selfRef);
    void AsyncIdle();
    void CancelBuild(const SObjectTag&);

    bool CanBuild(const SObjectTag& tag)
    {
        return x4_loader.ResourceExists(tag);
    }

    u32 ResourceSize(const urde::SObjectTag& tag)
    {
        return x4_loader.ResourceSize(tag);
    }

    std::unique_ptr<u8[]> LoadResourceSync(const urde::SObjectTag& tag)
    {
        return x4_loader.LoadResourceSync(tag);
    }

    std::unique_ptr<u8[]> LoadResourcePartSync(const urde::SObjectTag& tag, u32 size, u32 off)
    {
        return x4_loader.LoadResourcePartSync(tag, size, off);
    }

    void GetTagListForFile(const char* pakName, std::vector<SObjectTag>& out) const
    {
        return x4_loader.GetTagListForFile(pakName, out);
    }

    std::shared_ptr<IDvdRequest> LoadResourceAsync(const urde::SObjectTag& tag, void* target)
    {
        return x4_loader.LoadResourceAsync(tag, target);
    }

    std::shared_ptr<IDvdRequest> LoadResourcePartAsync(const urde::SObjectTag& tag, u32 size, u32 off, void* target)
    {
        return x4_loader.LoadResourcePartAsync(tag, size, off, target);
    }

    const SObjectTag* GetResourceIdByName(const char* name) const
    {
        return x4_loader.GetResourceIdByName(name);
    }

    FourCC GetResourceTypeById(CAssetId id) const
    {
        return x4_loader.GetResourceTypeById(id);
    }

    std::vector<std::pair<std::string, SObjectTag>> GetResourceIdToNameList() const
    {
        return x4_loader.GetResourceIdToNameList();
    }

    void EnumerateResources(const std::function<bool(const SObjectTag&)>& lambda) const
    {
        return x4_loader.EnumerateResources(lambda);
    }

    void EnumerateNamedResources(const std::function<bool(const std::string&, const SObjectTag&)>& lambda) const
    {
        return x4_loader.EnumerateNamedResources(lambda);
    }

    void LoadOriginalIDs(CSimplePool& sp);
    CAssetId TranslateOriginalToNew(CAssetId id) const;
    CAssetId TranslateNewToOriginal(CAssetId id) const;

    CResLoader* GetResLoader() { return &x4_loader; }
    CFactoryMgr* GetFactoryMgr() { return &x5c_factoryMgr; }
};

}

#endif // __URDE_CRESFACTORY_HPP__