#include "ld.h"
#include "helper.h"
#include "libgen.h"
#include "target.h"
#include "tier0/platform.h"
#include "tier1/interface.h"
#include "tier1/utlstring.h"

class CClangLinker : public ILinker
{
public:
	virtual CUtlString Link( LinkProject_t *pProject ) override;
};

EXPOSE_INTERFACE(CClangLinker, ILinker, CLANG_LINKER_INTERFACE_NAME);

CUtlString CClangLinker::Link( LinkProject_t *pProject )
{
	CUtlString szFileName;
	unsigned int hash = pProject->GenerateProjectHash();
	switch(pProject->linkType)
	{
	case ELINK_EXECUTABLE:
		if (pProject->m_target.kernel == TARGET_KERNEL_WINDOWS)
			szFileName = CUtlString("%s.exe", pProject->m_szName.GetString());	
		else if (pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
			szFileName = CUtlString("lib%s.so", pProject->m_szName.GetString());
		else
			szFileName = CUtlString("%s", pProject->m_szName.GetString());
		break;
	case ELINK_STATIC_LIBRARY:
		szFileName = CUtlString("lib%s.a", pProject->m_szName.GetString());
		break;
	case ELINK_DYNAMIC_LIBRARY:
		if (pProject->m_target.kernel == TARGET_KERNEL_DARWIN)
			szFileName = CUtlString("lib%s.dylib", pProject->m_szName.GetString());
		if (pProject->m_target.kernel == TARGET_KERNEL_LINUX)
			szFileName = CUtlString("lib%s.so", pProject->m_szName.GetString());
		break;
	}

	CUtlString szTarget = pProject->m_target.GetTriplet();
	CUtlString szOutputFile = CUtlString("%s/%s/ld/%u_%s/%s",FPC_TEMPORAL_DIRNAME, szTarget.GetString(), hash, pProject->m_szName.GetString(), szFileName.GetString());
	CUtlString szOutputDir = szOutputFile;
	szOutputDir = dirname(szOutputDir);
	filesystem2->MakeDirectory(szOutputDir);
	if (pProject->linkType == ELINK_STATIC_LIBRARY)
	{
		V_printf("  AR     %s\n", pProject->m_szName.GetString());
		bool shouldRecompile = false;
		CUtlVector<CUtlString> args;
		for (auto object: pProject->objects)
		{
			if (filesystem2->ShouldRecompile(object.m_szObjectFile,szOutputFile))
			{
				shouldRecompile = true;
				break;
			}
		}
		if (!shouldRecompile)
			goto compiled;
		args = {
			"rcs",
			szOutputFile
		};
		for (auto object: pProject->objects)
			args.AppendTail(object.m_szObjectFile);
		runner->Run("ar", args);
		runner->Wait();	
	} else {
		V_printf("  LINK     %s\n", pProject->m_szName.GetString());
		bool shouldRecompile = false;
		CUtlVector<CUtlString> args;
		for (auto object: pProject->objects)
		{
			if (filesystem2->ShouldRecompile(object.m_szObjectFile,szOutputFile))
			{
				shouldRecompile = true;
				break;
			}
		}
		if (!shouldRecompile)
			goto compiled;
		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);
		}
		args = {
			"-o",
			szOutputFile,
			"-target",
			szCompiledTarget,
		};
		if (pProject->bNoStdLib)
		{
			args.AppendTail("-nostdlib");
		}
		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");
			args.AppendTail("-Wl,-export_dynamic");
			args.AppendTail("-undefined");
			args.AppendTail("dynamic_lookup");
		}
		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");
			args.AppendTail("-Wl,-rpath,@executable_path");
			args.AppendTail("-Wl,-all_load");
			args.AppendTail("-w");
		} 
		else if (pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
		{
			args.AppendTail(CUtlString("--sysroot=%s/sysroot", pProject->m_target.szSysroot));
			args.AppendTail("-static-libstdc++");
		}
		else if (pProject->m_target.szSysroot)
		{
			args.AppendTail(CUtlString("--sysroot=%s", pProject->m_target.szSysroot));
		}
		if (pProject->m_target.kernel == TARGET_KERNEL_WINDOWS)
		{
			args.AppendTail("-fuse-ld=ld");
		}
		if (pProject->m_target.kernel == TARGET_KERNEL_LINUX || pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
		{
			args.AppendTail("-rdynamic");
		}
		if (pProject->linkType == ELINK_DYNAMIC_LIBRARY || pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
		{
			args.AppendTail("-shared");
		}
		if (pProject->m_target.kernel == TARGET_KERNEL_WINDOWS || pProject->m_target.kernel == TARGET_KERNEL_LINUX || pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
			args.AppendTail("-Wl,--whole-archive");
		for (auto object: pProject->objects)
			args.AppendTail(object.m_szObjectFile);
		if (pProject->m_target.kernel == TARGET_KERNEL_WINDOWS || pProject->m_target.kernel == TARGET_KERNEL_LINUX || pProject->m_target.kernel == TARGET_KERNEL_ANDROID)
			args.AppendTail("-Wl,--no-whole-archive");
		for (auto lib: pProject->libraries)
		{
			args.AppendTail("-l");
			args.AppendTail(lib);
		}
		for (auto &directory: pProject->frameworkDirectories)
		{
			args.AppendTail("-F");
			args.AppendTail(directory);
		}
		for (auto &framework: pProject->frameworks)
		{
			args.AppendTail("-framework");
			args.AppendTail(framework);
		}
		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);
		}
		runner->Wait();	
	}
compiled:
	return szOutputFile;
};