#include "c.h"
#include "helper.h"
#include "obj.h"
#include "target.h"
#include "tier0/lib.h"
#include "tier0/platform.h"
#include "tier1/interface.h"
#include "tier1/utlstring.h"
#include "tier1/utlvector.h"
#include "libgen.h"
#include "ctype.h"

struct ClangFile_t
{
	CUtlString m_szName;
	CUtlVector<CUtlString> m_szArguments;
};

class CClangCompiler : public ICCompiler
{
public:
	virtual LinkProject_t Compile( CProject_t *pProject ) override;
	virtual void GenerateLinterData( void ) override;
};

EXPOSE_INTERFACE(CClangCompiler, ICCompiler, CLANG_C_COMPILER_INTERFACE_NAME);

CUtlVector<ClangFile_t> g_clangFiles;

LinkProject_t CClangCompiler::Compile( CProject_t *pProject )
{
	LinkProject_t proj = {};
	proj.m_szName = pProject->m_szName;
	proj.m_target = pProject->m_target;
	proj.m_androidmanifest = pProject->m_androidmanifest;
	unsigned int hash = pProject->GenerateProjectHash();
	for (auto &file: pProject->files)
	{
		CUtlString szTarget = pProject->m_target.GetTriplet();
		CUtlString szOutputFile = CUtlString("%s/%s/cc/%u_%s/%s/%s.o",FPC_TEMPORAL_DIRNAME, szTarget.GetString(), hash, pProject->m_szName.GetString(), filesystem2->BuildDirectory(), file.GetString());
		CUtlString szOutputDir = szOutputFile;
		szOutputDir = dirname(szOutputDir);
		filesystem2->MakeDirectory(szOutputDir);
	}
	for (auto &file: pProject->files)
	{
		CUtlVector<CUtlString> args;
		V_printf("  CC       %s\n", file.GetString());
		CUtlString szTarget = pProject->m_target.GetTriplet();
		CUtlString szCompiledTarget = szTarget; 
		if (pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
		{
			szCompiledTarget = CUtlString("%s%u", szTarget.GetString(), pProject->m_androidmanifest.m_nTargetVersion);
		}
		CUtlString szOutputFile = CUtlString("%s/%s/cc/%u_%s/%s/%s.o",FPC_TEMPORAL_DIRNAME, szTarget.GetString(), hash, pProject->m_szName.GetString(), filesystem2->BuildDirectory(), file.GetString());

		args = {
			"-target",
			szCompiledTarget,
			"-g",
			"-c",
			"-o",
			szOutputFile,
			file,
		};
		if (!strcmp(Plat_GetExtension(file),"cpp"))
			args.AppendTail("-std=c++17"); 
		else if (!strcmp(Plat_GetExtension(file),"mm"))
			;
		else
			args.AppendTail("-std=c99"); 

		if (pProject->m_target.kernel == TARGET_KERNEL_DARWIN)
		{
			args.AppendTail("-isysroot");
			args.AppendTail("/Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/MacOSX.sdk");
		} else if (pProject->m_target.kernel == TARGET_KERNEL_IOS)
		{
			args.AppendTail("-isysroot");
			args.AppendTail("/Applications/Xcode.app/Contents/Developer/Platforms/iPhoneOS.platform/Developer/SDKs/iPhoneOS.sdk");
			args.AppendTail("-miphoneos-version-min=18.0 ");
			args.AppendTail("-fembed-bitcode");
		}

		if (pProject->m_target.szSysroot)
		{
			args.AppendTail(CUtlString("--sysroot=%s", pProject->m_target.szSysroot));
		}

		if (pProject->bFPIC)
			args.AppendTail("-fPIC");
		if (pProject->bFPIE)
			args.AppendTail("-fPIE");
		for (auto &macro: pProject->macros)
		{
			args.AppendTail("-D");
			args.AppendTail(CUtlString("%s=%s", (char*)macro.szName, (char*)macro.szValue));
		}
		for (auto &include: pProject->includeDirectories)
		{
			args.AppendTail("-I");
			args.AppendTail(include);
		}
		for (auto &include: pProject->includeFiles)
		{
			args.AppendTail("-include");
			args.AppendTail(include);
		}
		if (!filesystem2->ShouldRecompile(file, szOutputFile))
			goto skipcompile;

		if (pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
		{
			if (!pProject->m_target.szSysroot)
				Plat_FatalErrorFunc("-sysroot must be specified for android\n");
			runner->Run(CUtlString("%s/bin/clang",pProject->m_target.szSysroot), args);

		}
		else 
		{
			runner->Run("clang", args);
		}
skipcompile:
		proj.objects.AppendTail((Object_t){szOutputFile});

		ClangFile_t cfile = {};
		cfile.m_szName = file;
		cfile.m_szArguments = args;
		if (pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
		{
			if (!pProject->m_target.szSysroot)
				Plat_FatalErrorFunc("-sysroot must be specified for android\n");
			cfile.m_szArguments.AppendHead(CUtlString("%s/bin/clang",pProject->m_target.szSysroot));
		}
		else 
			cfile.m_szArguments.AppendHead("clang");

		g_clangFiles.AppendTail(cfile);
	}
	return proj;
}

void CClangCompiler::GenerateLinterData()
{
	FILE* f = V_fopen("compile_commands.json", "wb");
	V_fprintf(f, "[\n");
	uint32_t i = 0;
	for (auto &file: g_clangFiles)
	{
		V_fprintf(f, "\t{\n");
		V_fprintf(f, "\t\t\"arguments\": [\n");
		for (auto &arg: file.m_szArguments)
			V_fprintf(f, "\t\t\t\"%s\",\n",arg.GetString());
	
		V_fseek(f, -2, SEEK_CUR);
		V_fprintf(f, "\n\t\t],\n");
		V_fprintf(f, "\t\t\"file\": \"%s\",\n", file.m_szName.GetString());
		V_fprintf(f, "\t\t\"directory\": \"%s\"\n", filesystem2->BuildDirectory());
		V_fprintf(f, "\t},\n");
	};
	V_fseek(f, -2, SEEK_CUR);
	V_fprintf(f, "\n]\n");
	V_fclose(f);
};