// Copyright Epic Games, Inc. All Rights Reserved.

#define CATCH_CONFIG_RUNNER

#include "TestRunner.h"
#include "TestHarness.h"
#include "Modules/ModuleManager.h"
#include "HAL/PlatformTLS.h"

#include "TestCommon/CoreUtilities.h"

// Test run interceptor
struct TestRunListener : public Catch::TestEventListenerBase {
	using TestEventListenerBase::TestEventListenerBase; // inherit constructor
public:
	void testCaseStarting(Catch::TestCaseInfo  const& TestInfo) override {
		if (bGDebug)
		{
			std::cout << TestInfo.lineInfo.file << ":" << TestInfo.lineInfo.line << " with tags " << TestInfo.tagsAsString() << std::endl;
		}
	}
};

CATCH_REGISTER_LISTENER(TestRunListener);

void LoadBaseTestModule(FString BaseModuleName)
{
	if (!BaseModuleName.IsEmpty())
	{
		FModuleManager::Get().LoadModule(*BaseModuleName);
	}
}

void UnloadBaseTestModule(FString BaseModuleName)
{
	if (!BaseModuleName.IsEmpty())
	{
		FModuleManager::Get().UnloadModule(*BaseModuleName);
	}
}

int RunTests(int argc, const char* argv[])
{
	// remember thread id of the main thread
	GGameThreadId = FPlatformTLS::GetCurrentThreadId();
	GIsGameThreadIdInitialized = true;

#ifdef SLEEP_ON_INIT
	// Sleep to allow sync with Gauntlet
	std::this_thread::sleep_for(std::chrono::milliseconds(5000));
#endif

	//Read command-line from file (if any). Some platforms do this earlier.
#ifndef PLATFORM_SKIP_ADDITIONAL_ARGS
	int ArgsOverrideNum = 0;
	const char** ArgsOverride = ReadAndAppendAdditionalArgs(GetProcessExecutablePath(), &ArgsOverrideNum, argv, argc);
	if (ArgsOverride != nullptr && ArgsOverrideNum > 1)
	{
		argc = ArgsOverrideNum;
		argv = ArgsOverride;
	}
#endif

	int CatchArgc;
	TUniquePtr<const char* []> CatchArgv = MakeUnique<const char* []>(argc);

	// Everything past a "--" argument, if present, is not sent to the catch test runner.
	for (CatchArgc = 0; CatchArgc < argc; ++CatchArgc)
	{
		if (std::strcmp(argv[CatchArgc], "--") == 0)
		{
			break;
		}
		CatchArgv.Get()[CatchArgc] = argv[CatchArgc];
	}

	// By default don't wait for input
	bool bWaitForInputToTerminate = false;

	
	FString LoadModuleName = TEXT("");

	for (int i = CatchArgc; i < argc; ++i)
	{
		if (std::strcmp(argv[i], "--wait") == 0)
		{
			bWaitForInputToTerminate = true;
		}
		else if (std::strcmp(argv[i], "--no-wait") == 0)
		{
			bWaitForInputToTerminate = false;
		}
		if (std::strcmp(argv[i], "--no-log") == 0)
		{
			bGAllowLogging = false;
		}
		if (std::strcmp(argv[i], "--no-mt") == 0)
		{
			bGMultithreaded = false;
		}
		if (std::strcmp(argv[i], "--debug") == 0)
		{
			bGDebug = true;
		}
		// If we have --base-global-module parse proceeding argument as its option-value
		if (std::strcmp(argv[i], "--base-global-module") == 0 &&
			i + 1 < argc)
		{
			LoadModuleName = argv[i + 1];
			++i;
		}
	}

	LoadBaseTestModule(LoadModuleName);

	int SessionResult = 0;
	{
		TGuardValue<bool> CatchRunning(bCatchIsRunning, true);
		SessionResult = Catch::Session().run(CatchArgc, CatchArgv.Get());
		CatchArgv.Reset();
	}

	UnloadBaseTestModule(LoadModuleName);

	// Required, will crash on exit otherwise...
	CleanupPlatform();

#if PLATFORM_DESKTOP
	if (bWaitForInputToTerminate)
	{
		std::cout << "Press enter to exit..." << std::endl;
		std::cin.ignore();
	}
#endif

	return SessionResult;
};