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

#pragma once

#include "UbaFile.h"
#include "UbaStringBuffer.h"

#define CHECK_TRUE(x) \
	if (!(x)) \
		return logger.Error(TC("Failed %s (%s:%u)"), TC(#x), TC("") __FILE__, __LINE__);


namespace uba
{
	class Logger;
	class LoggerWithWriter;
	class ProcessHandle;
	class SessionServer;
	struct ProcessStartInfo;

	bool CreateTestFile(Logger& logger, StringView testRootDir, StringView fileName, StringView content, u32 attributes = DefaultAttributes());
	bool CreateTestFile(StringBufferBase& outFile, Logger& logger, StringView testRootDir, StringView fileName, StringView content, u32 attributes = DefaultAttributes());
	bool DeleteTestFile(Logger& logger, StringView testRootDir, StringView fileName);
	bool FileExists(Logger& logger, StringView dir, StringView fileName);

	using RunProcessFunction = Function<ProcessHandle(const ProcessStartInfo&)>;
	using TestSessionFunction = Function<bool(LoggerWithWriter& logger, SessionServer& session, const tchar* workingDir, const RunProcessFunction& runProcess, void* extraData)>;
	bool RunLocal(LoggerWithWriter& logger, const StringBufferBase& testRootDir, const TestSessionFunction& testFunc, void* extraData = nullptr, bool enableDetour = true);
	bool RunRemote(LoggerWithWriter& logger, const StringBufferBase& testRootDir, const TestSessionFunction& testFunc, bool deleteAll = true, bool serverShouldListen = true);
	void GetTestAppPath(LoggerWithWriter& logger, StringBufferBase& out);
	const tchar* GetSystemApplication();
	const tchar* GetSystemArguments();
}