#include "helper.h"
#include "c.h"
#include "ld.h"
#include "runner.h"
#include "target.h"
#include "tier1/utlstring.h"
#include "tier1/commandline.h"

void build_shader( const char *szName )
{
	IFileSystem2::MakeDirectory(CUtlString("build/funnygame/metal/gfx"));
	IFileSystem2::MakeDirectory(CUtlString("build/funnygame/vulkan/gfx"));

	CUtlVector<CUtlString> slang_args = {
		CUtlString("funnyassets/gfx_shaders/%s.slang", szName),
		"-g",
		"-target",
		"spirv",
		"-D",
		"__SPIRV__=1",
		"-o",
		CUtlString("build/funnygame/vulkan/gfx/%s.shader", szName),
	};

	IRunner::Run("slangc", slang_args);
	slang_args[3] = "metal";
	slang_args[5] = "__METAL__=1";
	slang_args[7] = CUtlString("build/funnygame/metal/gfx/%s.shader", szName);
	IRunner::Run("slangc", slang_args);
	IRunner::Wait();	
};

int assets_build()
{
	IFileSystem2::CopyDirectory("build", "tools");

	IFileSystem2::CopyDirectory(CUtlString("build/funnygame/assets"), "funnyassets/maps");
	IFileSystem2::CopyDirectory(CUtlString("build/funnygame/assets"), "funnyassets/gfx");
	IFileSystem2::CopyDirectory(CUtlString("build/funnygame/assets"), "funnyassets/textures");
	IFileSystem2::CopyDirectory(CUtlString("build/funnygame/assets"), "funnyassets/materials");
	build_shader("mesh_frag");
	build_shader("mesh_vert");
	CUtlVector<CUtlString> python_args = {
		"build/tools/makepak64.py",
		CUtlString("build/funnygame/assets"),
		bStaticBuild ? CUtlString("%s/bin/%s.pak", szOutputDir.GetString(), "rtt") : CUtlString("%s/funnygame/%s.pak", szOutputDir.GetString(), "rtt"),
	};
	IRunner::Run("python3", python_args);
	if (Target_t::DefaultTarget().kernel == TARGET_KERNEL_IOS ||Target_t::DefaultTarget().kernel == TARGET_KERNEL_DARWIN)
	{
		python_args = {
			"build/tools/makepak64.py",
			CUtlString("build/funnygame/metal"),
			bStaticBuild ? CUtlString("%s/bin/%s.pak", szOutputDir.GetString(), "metal") : CUtlString("%s/funnygame/%s.pak", szOutputDir.GetString(), "metal"),
		};
		IRunner::Run("python3", python_args);
	} else {
		python_args = {
			"build/tools/makepak64.py",
			CUtlString("build/funnygame/vulkan"),
			bStaticBuild ? CUtlString("%s/bin/%s.pak", szOutputDir.GetString(), "vulkan") : CUtlString("%s/funnygame/%s.pak", szOutputDir.GetString(), "vulkan"),
		};
		IRunner::Run("python3", python_args);
	}
	IRunner::Wait();	

	if (bStaticBuild)
		IFileSystem2::CopyFile(CUtlString("%s/bin", szOutputDir.GetString()), "funnyassets/default.cfg");
	else
		IFileSystem2::CopyFile(szOutputDir.GetString(), "funnyassets/default.cfg");
	return 0;
};
DECLARE_BUILD_STAGE(assets, assets_build);