From c941024ecf4d603a7086d8b982b6a98e6a42fced Mon Sep 17 00:00:00 2001
From: baz <bazzadobs@live.co.uk>
Date: Tue, 21 May 2024 22:07:26 +0100
Subject: [PATCH] Add Save functions to NakatomiGameInstance

---
 Source/Nakatomi/NakatomiGameInstance.cpp | 89 ++++++++++++++++++++++++
 Source/Nakatomi/NakatomiGameInstance.h   | 20 ++++++
 2 files changed, 109 insertions(+)

diff --git a/Source/Nakatomi/NakatomiGameInstance.cpp b/Source/Nakatomi/NakatomiGameInstance.cpp
index b91c284..9951662 100644
--- a/Source/Nakatomi/NakatomiGameInstance.cpp
+++ b/Source/Nakatomi/NakatomiGameInstance.cpp
@@ -3,6 +3,9 @@
 
 #include "NakatomiGameInstance.h"
 
+#include "PlayerCharacter.h"
+#include "Kismet/GameplayStatics.h"
+
 UNakatomiAIAttackTokenManager* UNakatomiGameInstance::GetAIAttackTokenManager()
 {
 	return IsValid(AIAttackTokenManager) ? AIAttackTokenManager : (AIAttackTokenManager = NewObject<UNakatomiAIAttackTokenManager>(this, TEXT("AI Attack Token Manager")));
@@ -19,3 +22,89 @@ void UNakatomiGameInstance::SetCurrentLevelManager(UNakatomiLevelManager* NewLev
 {
 	currentLevelManager = NewLevelManager;
 }
+
+UNakatomiSaveGame* UNakatomiGameInstance::LoadGameFromSlot(FString SaveSlotName)
+{
+	if (UNakatomiSaveGame* Save = Cast<UNakatomiSaveGame>(UGameplayStatics::LoadGameFromSlot(SaveSlotName, 0)))
+	{
+		SaveGameObject = Save;
+		return SaveGameObject;
+	}
+
+	return nullptr;
+}
+
+bool UNakatomiGameInstance::SaveGame(bool ResetDefaults)
+{
+	if (SaveGameObject == nullptr)
+	{
+		UE_LOG(LogTemp, Warning, TEXT("SaveGame called without a loaded save game object."));
+		return false;
+	}
+
+	if (ResetDefaults)
+	{
+		SaveGameObject->ResetPlayerValuesToDefault();
+	}
+	else
+	{
+		APlayerCharacter* Player = Cast<APlayerCharacter>(UGameplayStatics::GetPlayerCharacter(GetWorld(), 0));
+		SaveGameObject->PlayerHealth = Player->GetCurrentHealthCount();
+		SaveGameObject->WeaponInventory = Player->WeaponInventory;
+		SaveGameObject->CurrentInventorySlot = Player->GetCurrentInventorySlot();
+	}
+
+	SaveGameObject->LevelName = GetWorld()->GetMapName();
+	SaveGameObject->LevelName.RemoveFromStart(GetWorld()->StreamingLevelsPrefix);
+
+	return UGameplayStatics::SaveGameToSlot(SaveGameObject, SaveGameObject->PlayerName, 0);
+}
+
+UNakatomiSaveGame* UNakatomiGameInstance::GetSaveGameObject()
+{
+	return SaveGameObject;
+}
+
+// TODO: This should either fail or append number if a save already exists with the same name
+UNakatomiSaveGame* UNakatomiGameInstance::CreateNewSaveGame(FString PlayerName)
+{
+	if (UNakatomiSaveGame* Save = Cast<UNakatomiSaveGame>(
+		UGameplayStatics::CreateSaveGameObject(UNakatomiSaveGame::StaticClass())))
+	{
+		SaveGameObject->PlayerName = PlayerName;
+
+		if (UGameplayStatics::SaveGameToSlot(Save, PlayerName, 0))
+		{
+			SaveGameObject = Save;
+			return SaveGameObject;
+		}
+	}
+	
+	return nullptr;
+}
+
+TArray<FNakatomiSaveFileInfo> UNakatomiGameInstance::GetAllSaveFilesFromDisk()
+{
+	TArray<FNakatomiSaveFileInfo> Files;
+
+	IFileManager& FileManager = IFileManager::Get();
+	FString SavedGamesFolder = FString{FPaths::ProjectSavedDir()} + "SaveGames/";
+
+	TArray<FString> FileNames;
+	FileManager.FindFiles(FileNames, *SavedGamesFolder, TEXT("*.sav"));
+
+	for (FString FileName : FileNames)
+	{
+		FileName.RemoveFromEnd(".sav");
+
+		if (UNakatomiSaveGame* saveGameObject = Cast<UNakatomiSaveGame>(UGameplayStatics::LoadGameFromSlot(FileName, 0)))
+		{
+			Files.Add({FileName,
+							 saveGameObject->PlayerName,
+							 saveGameObject->LevelName,
+				FileManager.GetTimeStamp(*(SavedGamesFolder + FileName + ".sav")).ToString()});
+		}
+	}
+	
+	return Files;
+}
diff --git a/Source/Nakatomi/NakatomiGameInstance.h b/Source/Nakatomi/NakatomiGameInstance.h
index ddde7d2..6bd4528 100644
--- a/Source/Nakatomi/NakatomiGameInstance.h
+++ b/Source/Nakatomi/NakatomiGameInstance.h
@@ -6,6 +6,8 @@
 #include "Engine/GameInstance.h"
 #include "NakatomiAIAttackTokenManager.h"
 #include "NakatomiLevelManager.h"
+#include "NakatomiSaveFileInfo.h"
+#include "NakatomiSaveGame.h"
 #include "NakatomiGameInstance.generated.h"
 
 /**
@@ -25,6 +27,9 @@ private:
 	UPROPERTY()
 	UNakatomiAIAttackTokenManager* AIAttackTokenManager;
 
+	UPROPERTY()
+	UNakatomiSaveGame* SaveGameObject = nullptr;
+
 public:
 
 	UFUNCTION(BlueprintCallable)
@@ -35,4 +40,19 @@ public:
 
 	UFUNCTION(BlueprintCallable)
 	void SetCurrentLevelManager(UNakatomiLevelManager* NewLevelManager);
+
+	UFUNCTION(BlueprintCallable)
+	UNakatomiSaveGame* LoadGameFromSlot(FString SaveSlotName);
+
+	UFUNCTION(BlueprintCallable)
+	bool SaveGame(bool ResetDefaults);
+
+	UFUNCTION(BlueprintCallable)
+	UNakatomiSaveGame* GetSaveGameObject();
+
+	UFUNCTION(BlueprintCallable)
+	UNakatomiSaveGame* CreateNewSaveGame(FString PlayerName);
+
+	UFUNCTION(BlueprintCallable)
+	TArray<FNakatomiSaveFileInfo> GetAllSaveFilesFromDisk();
 };