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

#include "CacheDataService.h"
#include "HAL/FileManagerGeneric.h"
#include "JsonObjectConverter.h"
#include "Misc/Paths.h"

#include "Logging/SubmitToolLog.h"
#include "Parameters/SubmitToolParameters.h"

FCacheDataService::FCacheDataService(const FGeneralParameters& InGeneralParameters) : Parameters(InGeneralParameters)
{
	LoadFromFile(FConfiguration::SubstituteAndNormalizeFilename(Parameters.CacheFile));
	CleanOldData();
}

FCacheDataService::~FCacheDataService()
{
}

bool FCacheDataService::GetChangelistCacheData(const FString& InCLId, FChangelistCacheData& OutCLData) const
{
	OutCLData = CacheData->CLCacheData.FindOrAdd(InCLId);
	return true;
}

const FDateTime FCacheDataService::GetLastValidationDate(const FString& InCLId, const FName& InValidatorId, const FString& InValidatorConfig, const FString& InFilePath) const
{
	FChangelistCacheData& Data = CacheData->CLCacheData.FindOrAdd(InCLId);
	Data.LastAccessed = FDateTime::UtcNow();
	return Data.GetLastValidationDate(InFilePath, InValidatorId, InValidatorConfig);
}

void FCacheDataService::UpdateLastValidationForFiles(const FString& InCLId, const FName& InValidatorId, const FString& InValidatorConfig, const TArray<FSourceControlStateRef>& InFiles, const FDateTime& InNewTimestamp)
{
	FChangelistCacheData& Data = CacheData->CLCacheData.FindOrAdd(InCLId);
	Data.LastAccessed = FDateTime::UtcNow();

	for(const FSourceControlStateRef& File : InFiles)
	{
		FValidationRecord& ValidationRecord = Data.LastFileValidations.FindOrAdd(File->GetFilename()).LastValidationDateTimes.FindOrAdd(InValidatorId);
		ValidationRecord.ValidatorConfig = InValidatorConfig;
		ValidationRecord.LastValidationSuccess = InNewTimestamp;
	}
}

FString FCacheDataService::GetIntegrationFieldValue(const FString& InCLId, const FString& InIntegrationFieldId) const
{
	FChangelistCacheData& Data = CacheData->CLCacheData.FindOrAdd(InCLId);
	Data.LastAccessed = FDateTime::UtcNow();	
	return Data.IntegrationFields.Contains(InIntegrationFieldId) ? Data.IntegrationFields[InIntegrationFieldId] : FString();
}

void FCacheDataService::SetIntegrationFieldValue(const FString& InCLId, const FString& InIntegrationFieldId, const FString& InValue)
{
	FChangelistCacheData& Data = CacheData->CLCacheData.FindOrAdd(InCLId);
	Data.LastAccessed = FDateTime::UtcNow();

	Data.IntegrationFields.FindOrAdd(InIntegrationFieldId) = InValue;
}

void FCacheDataService::SaveCacheToDisk()
{
	CleanOldData();
	SaveToFile(FConfiguration::SubstituteAndNormalizeFilename(Parameters.CacheFile));
}


void FCacheDataService::LoadFromFile(const FString& InFilePath)
{
	CacheData = MakeUnique<FCacheFile>();
	if(FPaths::FileExists(InFilePath))
	{
		FString InText;
		FArchive* File = IFileManager::Get().CreateFileReader(*InFilePath, EFileRead::FILEREAD_None);
		*File << InText;
		File->Close();
		delete File;
		File = nullptr;

		FString OutputText;
		FText Errors;

		if(!FJsonObjectConverter::JsonObjectStringToUStruct(InText, CacheData.Get(), 0, 0, false, &Errors))
		{
			UE_LOG(LogSubmitTool, Log, TEXT("Error loading cache data file %s"), *Errors.ToString());
		}
		else
		{
			UE_LOG(LogSubmitToolDebug, Verbose, TEXT("Loaded Cache Data from %s:\n%s"), *InFilePath, *InText);
		}
	}
	else
	{
		UE_LOG(LogSubmitToolDebug, Log, TEXT("No cache data available %s."), *InFilePath);
	}
}

void FCacheDataService::SaveToFile(const FString& InFilePath) const
{
	FString OutputText;
	FJsonObjectConverter::UStructToJsonObjectString(*CacheData, OutputText);

	FArchive* File = IFileManager::Get().CreateFileWriter(*InFilePath, EFileWrite::FILEWRITE_EvenIfReadOnly);
	*File << OutputText;
	File->Close();
	delete File;
	File = nullptr;
	UE_LOG(LogSubmitToolDebug, Verbose, TEXT("Saved Cache Data to %s:\n%s"), *InFilePath, *OutputText);
}

void FCacheDataService::CleanOldData()
{
	TArray<FString> OldData;
	for(const TPair<FString, FChangelistCacheData>& Pair : CacheData->CLCacheData)
	{
		if((FDateTime::UtcNow() - Pair.Value.LastAccessed).GetTotalHours() > Parameters.InvalidateCacheHours)
		{
			OldData.Add(Pair.Key);
		}
	}

	for(const FString& Key : OldData)
	{
		CacheData->CLCacheData.Remove(Key);
	}
}