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

#pragma once

#include "CoreTypes.h"
#include "Trace/StoreClient.h"
#include "Containers/UnrealString.h"
#include "ProfilingDebugging/TraceAuxiliary.h" // for FTraceAuxiliary::EConnectionType
#include "TraceServices/Model/AnalysisSession.h"

#define UE_API TRACEINSIGHTS_API

class FAutomationTestBase;

class FInsightsTestUtils
{
public:
	UE_API FInsightsTestUtils(FAutomationTestBase* Test);

	UE_API bool AnalyzeTrace(const TCHAR* Path) const;
	UE_API bool AnalyzeTrace(TSharedPtr<const TraceServices::IAnalysisSession> Session) const;
	UE_API bool FileContainsString(const FString& PathToFile, const FString& ExpectedString, double Timeout) const;
	UE_API bool SetupUTS(double Timeout = 30.0, bool bUseFork = false) const;
	UE_API bool KillUTS(double Timeout = 30.0) const;
	UE_API bool StartTracing(FTraceAuxiliary::EConnectionType ConnectionType, double Timeout = 10.0) const;
	UE_API bool IsUnrealTraceServerReady(const TCHAR* Host = TEXT("localhost"), int32 Port = 0U) const;
	UE_API bool IsTraceHasLiveStatus(const FString& TraceName, const TCHAR* Host = TEXT("localhost"), int32 Port = 0U) const;
	UE_API bool GetMetadata(const FString& TraceFilePath, TMap<FString, FString>& OutMetadata) const;
	UE_API void ResetSession() const;
	UE_API FString GetLiveTrace(const TCHAR* Host = TEXT("localhost"), int32 Port = 0U) const;
	UE_API FString GetUTSPath() const;

private:
	TSharedPtr<const TraceServices::IAnalysisSession> GetSession(const TCHAR* Path) const;
	TSharedPtr<UE::Trace::FStoreClient> CreateStoreClient(const TCHAR* Host, int32 Port) const;
	uint32 GetValidSessionCount(TSharedPtr<UE::Trace::FStoreClient> StoreClient) const;
	const UE::Trace::FStoreClient::FTraceInfo* FindTraceInfoByName(TSharedPtr<UE::Trace::FStoreClient> StoreClient, const FString& TraceName) const;

	FAutomationTestBase* Test;
};

#undef UE_API