#include "BSPLinker.h"

#include "ComWorldLinker.h"
#include "ClipMapLinker.h"
#include "GameWorldMpLinker.h"
#include "GfxWorldLinker.h"
#include "MapEntsLinker.h"
#include "SkinnedVertsLinker.h"

namespace BSP
{
    FootstepTableDef* BSPLinker::addEmptyFootstepTableAsset(std::string assetName)
    {
        if (assetName.length() == 0)
            return nullptr;

        FootstepTableDef* footstepTable = m_memory.Alloc<FootstepTableDef>();
        footstepTable->name = m_memory.Dup(assetName.c_str());
        memset(footstepTable->sndAliasTable, 0, sizeof(footstepTable->sndAliasTable));

        m_context.AddAsset<AssetFootstepTable>(assetName, footstepTable);

        return footstepTable;
    }

    bool BSPLinker::addDefaultRequiredAssets(BSPData* bsp)
    {
        if (m_context.LoadDependency<AssetScript>("maps/mp/" + bsp->name + ".gsc") == nullptr)
            return false;
        if (m_context.LoadDependency<AssetScript>("maps/mp/" + bsp->name + "_amb.gsc") == nullptr)
            return false;
        if (m_context.LoadDependency<AssetScript>("maps/mp/" + bsp->name + "_fx.gsc") == nullptr)
            return false;

        if (m_context.LoadDependency<AssetScript>("clientscripts/mp/" + bsp->name + ".csc") == nullptr)
            return false;
        if (m_context.LoadDependency<AssetScript>("clientscripts/mp/" + bsp->name + "_amb.csc") == nullptr)
            return false;
        if (m_context.LoadDependency<AssetScript>("clientscripts/mp/" + bsp->name + "_fx.csc") == nullptr)
            return false;

        addEmptyFootstepTableAsset("default_1st_person");
        addEmptyFootstepTableAsset("default_3rd_person");
        addEmptyFootstepTableAsset("default_1st_person_quiet");
        addEmptyFootstepTableAsset("default_3rd_person_quiet");
        addEmptyFootstepTableAsset("default_3rd_person_loud");
        addEmptyFootstepTableAsset("default_ai");

        if (m_context.LoadDependency<AssetRawFile>("animtrees/fxanim_props.atr") == nullptr)
            return false;

        return true;
    }

	BSPLinker::BSPLinker(MemoryManager& memory, ISearchPath& searchPath, AssetCreationContext& context) 
		: m_memory(memory),
		  m_search_path(searchPath),
		  m_context(context)
	{
	}

	AssetCreationResult BSPLinker::linkBSP(BSPData* bsp)
	{
        if (!addDefaultRequiredAssets(bsp))
            return AssetCreationResult::Failure();

        ComWorldLinker comWorldLinker(m_memory, m_search_path, m_context);
        ClipMapLinker clipMapLinker(m_memory, m_search_path, m_context);
        GameWorldMpLinker gameWorldMpLinker(m_memory, m_search_path, m_context);
        GfxWorldLinker gfxWorldLinker(m_memory, m_search_path, m_context);
        MapEntsLinker mapEntsLinker(m_memory, m_search_path, m_context);
        SkinnedVertsLinker skinnedVertsLinker(m_memory, m_search_path, m_context);

        if (comWorldLinker.linkComWorld(bsp).HasFailed())
            return AssetCreationResult::Failure();

        if (mapEntsLinker.linkMapEnts(bsp).HasFailed())
            return AssetCreationResult::Failure();

        gameWorldMpLinker.linkGameWorldMp(bsp);
        if (gameWorldMpLinker.linkGameWorldMp(bsp).HasFailed())
            return AssetCreationResult::Failure();

        if (skinnedVertsLinker.linkSkinnedVerts(bsp).HasFailed())
            return AssetCreationResult::Failure();

        auto result = gfxWorldLinker.linkGfxWorld(bsp); // requires mapents asset
        if (result.HasFailed())
            return AssetCreationResult::Failure();

        if (clipMapLinker.linkClipMap(bsp).HasFailed()) // requires gfxworld and mapents asset
            return AssetCreationResult::Failure();

        return result;
	}
}