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

#include "AssetViewerSettings.h"
#include "UObject/UnrealType.h"
#include "Editor/EditorPerProjectUserSettings.h"
#include "Editor.h"

FName UDefaultEditorProfiles::DefaultProfileName = FName("Epic Headquarters");
FName UDefaultEditorProfiles::EditingProfileName = FName("Grey Wireframe");
FName UDefaultEditorProfiles::GreyAmbientProfileName = FName("Grey Ambient");

const FPreviewSceneProfile* UDefaultEditorProfiles::GetProfile(const FString& ProfileName)
{
	for (const FPreviewSceneProfile& Profile : Profiles)
	{
		if (Profile.ProfileName == ProfileName)
		{
			return &Profile;
		}
	}

	return nullptr;
}

UAssetViewerSettings::UAssetViewerSettings()
{
}

UAssetViewerSettings::~UAssetViewerSettings()
{
	if (GEditor)
	{
		GEditor->UnregisterForUndo(this);
	}
}

UAssetViewerSettings* UAssetViewerSettings::Get()
{
	// This is a singleton, use default object
	UAssetViewerSettings* DefaultSettings = GetMutableDefault<UAssetViewerSettings>();

	// Load environment map textures (once)
	static bool bInitialized = false;
	if (!bInitialized)
	{
		DefaultSettings->SetFlags(RF_Transactional);

		for (FPreviewSceneProfile& Profile : DefaultSettings->Profiles)
		{
			Profile.LoadEnvironmentMap();
		}
		bInitialized = true;

		if (GEditor)
		{
			GEditor->RegisterForUndo(DefaultSettings);
		}
	}

	return DefaultSettings;
}

FPreviewSceneProfile& UAssetViewerSettings::GetCurrentUserProjectProfile()
{
	return Get()->Profiles[GetMutableDefault<UEditorPerProjectUserSettings>()->AssetViewerProfileIndex];
}

void UAssetViewerSettings::Save(bool bWarnIfFail)
{
	ULocalProfiles* LocalProfilesObject = GetMutableDefault<ULocalProfiles>();
	USharedProfiles* SharedProfilesObject = GetMutableDefault<USharedProfiles>();

	TArray<FPreviewSceneProfile>& LocalProfiles = GetMutableDefault<ULocalProfiles>()->Profiles;
	TArray<FPreviewSceneProfile>& SharedProfiles = GetMutableDefault<USharedProfiles>()->Profiles;

	LocalProfiles.Empty();
	SharedProfiles.Empty();

	// Divide profiles up in corresponding arrays
	for (FPreviewSceneProfile& Profile : Profiles)
	{
		if (Profile.bSharedProfile)
		{
			SharedProfiles.Add(Profile);
		}
		else
		{
			LocalProfiles.Add(Profile);
		}
	}

	LocalProfilesObject->SaveConfig();

	SharedProfilesObject->SaveConfig();
	SharedProfilesObject->TryUpdateDefaultConfigFile(FString(), bWarnIfFail);
}

void UAssetViewerSettings::PostEditChangeProperty(struct FPropertyChangedEvent& PropertyChangedEvent)
{
	FName PropertyName = (PropertyChangedEvent.Property != nullptr) ? PropertyChangedEvent.Property->GetFName() : NAME_None;	
	UObject* Outer = (PropertyChangedEvent.Property != nullptr) ? PropertyChangedEvent.Property->GetOwner<UObject>() : nullptr;
	if (Outer != nullptr && ( Outer->GetName() == "PostProcessSettings" || Outer->GetName() == "WeightedBlendable" || Outer->GetName() == "Vector" || Outer->GetName() == "Vector4" || Outer->GetName() == "LinearColor"))
	{
		PropertyName = GET_MEMBER_NAME_CHECKED(FPreviewSceneProfile, PostProcessingSettings);
	}

	// Store path to the set enviroment map texture
	if (PropertyName == GET_MEMBER_NAME_CHECKED(FPreviewSceneProfile, EnvironmentCubeMap))
	{		
		int32 ProfileIndex = GetMutableDefault<UEditorPerProjectUserSettings>()->AssetViewerProfileIndex;
		Profiles[ProfileIndex].EnvironmentCubeMapPath = Profiles[ProfileIndex].EnvironmentCubeMap.ToString();
	}

	if (NumProfiles != Profiles.Num())
	{
		OnAssetViewerProfileAddRemovedEvent.Broadcast();
		NumProfiles = Profiles.Num();
	}
	
	OnAssetViewerSettingsChangedEvent.Broadcast(PropertyName);
}

void UAssetViewerSettings::PostInitProperties()
{
	Super::PostInitProperties();

	Profiles.Empty();
	
	TArray<FPreviewSceneProfile>& SharedProfiles = GetMutableDefault<USharedProfiles>()->Profiles;
	for (FPreviewSceneProfile& Profile : SharedProfiles)
	{
		Profiles.Add(Profile);
	}

	TArray<FPreviewSceneProfile>& LocalProfiles = GetMutableDefault<ULocalProfiles>()->Profiles;
	for (FPreviewSceneProfile& Profile : LocalProfiles)
	{
		Profiles.Add(Profile);
	}
	
	TArray<FPreviewSceneProfile>& DefaultEditorProfiles = GetMutableDefault<UDefaultEditorProfiles>()->Profiles;
	for (FPreviewSceneProfile& ProfileToAdd : DefaultEditorProfiles)
	{
		// add the default profile if it's not already stored
		// default editor profiles should be marked as "shared" profiles to allow user overrides
		// therefore, they are only added once and maintained by the user thereafter
		if (!Profiles.ContainsByPredicate([&ProfileToAdd](const FPreviewSceneProfile& Profile)
			{
				return Profile.ProfileName == ProfileToAdd.ProfileName;
			}))
		{
			Profiles.Add(ProfileToAdd);
		}
	}

	// the UDefaultEditorProfiles should always add at least one default profile
	if (!ensure(!Profiles.IsEmpty()))
	{
		// Make sure there always is one profile as default
		Profiles.AddDefaulted(1);
		Profiles[0].ProfileName = TEXT("Profile_0");
	}

	NumProfiles = Profiles.Num();

	UEditorPerProjectUserSettings* ProjectSettings = GetMutableDefault<UEditorPerProjectUserSettings>();

	// Find last saved profile index
	int32 SelectedProfileIndex = INDEX_NONE;
	for (int32 ProfileIndex = 0; ProfileIndex < Profiles.Num(); ++ProfileIndex)
	{
		if (Profiles[ProfileIndex].ProfileName == ProjectSettings->AssetViewerProfileName)
		{
			SelectedProfileIndex = ProfileIndex;
			break;
		}
	}

	// Only need to set profile index if we found a correct value since it defaults to 0
	if (SelectedProfileIndex != INDEX_NONE)
	{
		ProjectSettings->AssetViewerProfileIndex = SelectedProfileIndex;
	}
}

void UAssetViewerSettings::PostUndo(bool bSuccess)
{
	if (bSuccess)
	{
		OnAssetViewerSettingsPostUndoEvent.Broadcast();
	}
}

void UAssetViewerSettings::PostRedo(bool bSuccess)
{
	PostUndo(bSuccess);
}