//  
// Copyright (c) 2023 articy Software GmbH & Co. KG. All rights reserved.  
//

#include "Slate/UserInterfaceHelperFunctions.h"
#include "ArticyObject.h"
#include "Interfaces/ArticyObjectWithPreviewImage.h"
#include "ArticyAsset.h"
#include "Interfaces/ArticyObjectWithSpeaker.h"
#include "Interfaces/ArticyObjectWithDisplayName.h"
#include "Interfaces/ArticyObjectWithText.h"
#include "Kismet/KismetStringLibrary.h"
#include "ArticyImportData.h"
#include "ArticyEditorStyle.h"
#include "ArticyFlowClasses.h"
#include "ArticyEntity.h"
#include "Interfaces/ArticyObjectWithColor.h"
#include "Kismet/KismetSystemLibrary.h"
#include "Styling/ISlateStyle.h"
#include "Styling/SlateBrush.h"

/**
 * Retrieves the type image associated with an Articy object based on its type and specified size.
 *
 * @param ArticyObject The Articy object for which to retrieve the type image.
 * @param Size The desired size of the image (Small, Medium, or Large).
 * @return The Slate brush representing the type image for the specified Articy object.
 */
const FSlateBrush* UserInterfaceHelperFunctions::GetArticyTypeImage(const UArticyObject* ArticyObject, UserInterfaceHelperFunctions::EImageSize Size)
{
	if (!ArticyObject)
	{
		return FArticyEditorStyle::Get().GetBrush(FName(TEXT("ArticyImporter.ArticyImporter.64")));
	}

	FString SizeString = "";
	if (Size == EImageSize::Large)
	{
		SizeString.Append("64");
	}
	else if (Size == EImageSize::Medium)
	{
		SizeString.Append("32");
	}
	else
	{
		SizeString.Append("16");
	}

	if (ArticyObject->IsA<UArticyDialogueFragment>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.DialogueFragment.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyDialogue>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Dialogue.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyEntity>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Entity.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyFlowFragment>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.FlowFragment.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyHub>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Hub.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyJump>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Jump.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyInstruction>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Instruction.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyCondition>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Condition.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyAsset>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Asset.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyUserFolder>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.UserFolder.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyZone>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Zone.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyLocation>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Location.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyLocationImage>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.LocationImage.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyLocationText>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.LocationText.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyLink>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Link.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyComment>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Comment.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyTextObject>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.TextObject.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyPath>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Path.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticyDocument>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Document.").Append(SizeString)));
	}
	else if (ArticyObject->IsA<UArticySpot>())
	{
		return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.Type.Spot.").Append(SizeString)));
	}

	return FArticyEditorStyle::Get().GetBrush(FName(*FString("ArticyImporter.ArticyImporter.64")));
}

/**
 * Retrieves the display image (preview image) associated with an Articy object.
 *
 * @param ArticyObject The Articy object for which to retrieve the display image.
 * @return The texture representing the display image for the specified Articy object.
 */
UTexture2D* UserInterfaceHelperFunctions::GetDisplayImage(const UArticyObject* ArticyObject)
{
	UTexture2D* Image = nullptr;

	if (!ArticyObject)
	{
		return Image;
	}

	// if there is a preview image, use it as display image
	const IArticyObjectWithPreviewImage* ObjectWithPreviewImage = Cast<IArticyObjectWithPreviewImage>(ArticyObject);
	if (ObjectWithPreviewImage)
	{
		const FArticyId AssetID = ObjectWithPreviewImage->GetPreviewImage()->Asset;
		const UArticyObject* AssetObject = UArticyObject::FindAsset(AssetID);

		if (AssetObject)
		{
			Image = (Cast<UArticyAsset>(AssetObject))->LoadAsTexture2D();
			return Image;
		}
	}

	return Image;
}

/**
 * Retrieves the preview image associated with an Articy object and assigns it to the specified Slate brush.
 *
 * @param ArticyObject The Articy object for which to retrieve the preview image.
 * @param OutSlateBrush The Slate brush to which the preview image will be assigned.
 * @return True if a preview image was successfully retrieved and assigned; otherwise, false.
 */
bool UserInterfaceHelperFunctions::RetrievePreviewImage(const UArticyObject* ArticyObject, FSlateBrush& OutSlateBrush)
{
	UTexture2D* PreviewImage = GetDisplayImage(ArticyObject);

	if (PreviewImage)
	{
		OutSlateBrush.SetResourceObject(PreviewImage);
		return true;
	}

	return false;
}

/**
 * Retrieves the preview image associated with the speaker of an Articy object and assigns it to the specified Slate brush.
 *
 * @param ArticyObject The Articy object for which to retrieve the speaker's preview image.
 * @param OutSlateBrush The Slate brush to which the speaker's preview image will be assigned.
 * @return True if a speaker's preview image was successfully retrieved and assigned; otherwise, false.
 */
bool UserInterfaceHelperFunctions::RetrieveSpeakerPreviewImage(const UArticyObject* ArticyObject, FSlateBrush& OutSlateBrush)
{
	const IArticyObjectWithSpeaker* ArticyObjectWithSpeaker = Cast<IArticyObjectWithSpeaker>(ArticyObject);

	if (ArticyObjectWithSpeaker)
	{
		UArticyObject* Speaker = UArticyObject::FindAsset(ArticyObjectWithSpeaker->GetSpeakerId());
		return RetrievePreviewImage(Speaker, OutSlateBrush);
	}

	return false;
}

/**
 * Retrieves the target ID associated with an Articy object.
 *
 * @param ArticyObject The Articy object for which to retrieve the target ID.
 * @return A pointer to the target ID associated with the specified Articy object.
 */
const FArticyId* UserInterfaceHelperFunctions::GetTargetID(const UArticyObject* ArticyObject)
{
	if (!ArticyObject)
	{
		return nullptr;
	}

	static const auto& PropName = FName("Target");
	FProperty* Prop = ArticyObject->GetProperty(PropName);

	if (!Prop)
	{
		return nullptr;
	}

	const FArticyId* TargetID = Prop->ContainerPtrToValuePtr<FArticyId>(ArticyObject);

	if (TargetID)
	{
		return TargetID;
	}

	return nullptr;
}

/**
 * Retrieves the display name associated with an Articy object.
 *
 * @param ArticyObject The Articy object for which to retrieve the display name.
 * @return The string representing the display name for the specified Articy object.
 */
const FString UserInterfaceHelperFunctions::GetDisplayName(const UArticyObject* ArticyObject)
{
	FString DisplayName = "None";

	if (!ArticyObject)
	{
		return DisplayName;
	}

	const FArticyId* TargetID = UserInterfaceHelperFunctions::GetTargetID(ArticyObject);
	if (TargetID)
	{
		const UArticyObject* TargetObject = UArticyObject::FindAsset(*TargetID);

		if (TargetObject)
		{
			return GetDisplayName(TargetObject);
		}
	}

	// use the display name as display name, if available
	const IArticyObjectWithDisplayName* ArticyObjectWithDisplayName = Cast<IArticyObjectWithDisplayName>(ArticyObject);
	if (ArticyObjectWithDisplayName)
	{
		DisplayName = ArticyObjectWithDisplayName->GetDisplayName().ToString();

		if (!DisplayName.IsEmpty())
		{
			return DisplayName;
		}
	}

	// use a text preview as display name, if available
	const IArticyObjectWithText* ArticyObjectWithText = Cast<IArticyObjectWithText>(ArticyObject);
	if (ArticyObjectWithText)
	{
		const FText Text = ArticyObjectWithText->GetText();

		// cut the string to the first X characters and attach "..." as a preview
		// primarily used for dialogue fragments, as they have neither preview image nor display name
		if (!Text.IsEmpty())
		{
			DisplayName = Text.ToString();
			DisplayName = UKismetStringLibrary::GetSubstring(DisplayName, 0, 14);

			if (Text.ToString().Len() > 14)
			{
				DisplayName.Append("...");
			}

			return DisplayName;
		}
	}

	return DisplayName;
}

/**
 * Retrieves the color associated with an Articy object.
 *
 * @param ArticyObject The Articy object for which to retrieve the color.
 * @return The linear color associated with the specified Articy object.
 */
const FLinearColor UserInterfaceHelperFunctions::GetColor(const UArticyObject* ArticyObject)
{
	FLinearColor Color(0.577, 0.76, 0.799);

	if (!ArticyObject)
	{
		return Color;
	}

	const FArticyId* TargetID = UserInterfaceHelperFunctions::GetTargetID(ArticyObject);

	if (TargetID)
	{
		const UArticyObject* TargetObject = UArticyObject::FindAsset(*TargetID);

		if (TargetObject)
		{
			return GetColor(TargetObject);
		}
	}

	const IArticyObjectWithColor* ObjectWithColor = Cast<IArticyObjectWithColor>(ArticyObject);
	if (ObjectWithColor)
	{
		Color = ObjectWithColor->GetColor();
		return Color;
	}

	return Color;
}

/**
 * Displays an Articy object in the Articy software.
 *
 * @param ArticyObject The Articy object to be displayed.
 * @return True if the Articy object was successfully displayed; otherwise, false.
 */
const bool UserInterfaceHelperFunctions::ShowObjectInArticy(const UArticyObject* ArticyObject)
{
	if (ArticyObject == nullptr)
	{
		return false;
	}

	bool bNewTab = false;
	if (ArticyObject->IsA(UArticyNode::StaticClass()))
	{
		bNewTab = false;
	}
	else
	{
		bNewTab = true;
	}

	return ShowObjectInArticy(ArticyObject->GetId(), bNewTab);
}

/**
 * Displays an Articy object with the specified ID in the Articy software.
 *
 * @param ArticyId The ID of the Articy object to be displayed.
 * @param bNewTab Specifies whether to open the object in a new tab (true) or the current tab (false).
 * @return True if the Articy object was successfully displayed; otherwise, false.
 */
const bool UserInterfaceHelperFunctions::ShowObjectInArticy(const FArticyId ArticyId, bool bNewTab)
{
	FAssetRegistryModule& AssetRegistryModule = FModuleManager::LoadModuleChecked<FAssetRegistryModule>("AssetRegistry");

	TArray<FAssetData> OutAssetData;

#if ENGINE_MAJOR_VERSION >= 5 && ENGINE_MINOR_VERSION >0
	AssetRegistryModule.Get().GetAssetsByClass(UArticyPackage::StaticClass()->GetClassPathName(), OutAssetData, false);
#else
	AssetRegistryModule.Get().GetAssetsByClass(UArticyImportData::StaticClass()->GetFName(), OutAssetData, false);
#endif

	FString TabURL = bNewTab ? FString("new") : FString("current");
	if (OutAssetData.Num() == 1)
	{
		UArticyImportData* ImportData = Cast<UArticyImportData>(OutAssetData[0].GetAsset());
		FString ProjectGuid = ImportData->GetProject().Guid;

		// restore the original Id in decimal
		const int32 High = ArticyId.High;
		const int32 Low = ArticyId.Low;
		const uint64 HighOriginal = (uint64)High << 32;
		const uint64 LowOriginal = Low;
		const uint64 OriginalId = HighOriginal + LowOriginal;

		FString ArticyObjectId = FString::Printf(TEXT("%llu"), OriginalId);
		FString ArticyURL = FString::Printf(TEXT("articy://localhost/view/%s/%s?window=main&pane=first&tab=%s"), *ProjectGuid, *ArticyObjectId, *TabURL);
		UKismetSystemLibrary::LaunchURL(ArticyURL);
	}

	return true;
}