UnrealEngineUWP/Engine/Source/Editor/DetailCustomizations/Private/AssetImportDataCustomization.cpp

// Copyright 1998-2019 Epic Games, Inc. All Rights Reserved.

#include "AssetImportDataCustomization.h"
#include "Widgets/DeclarativeSyntaxSupport.h"
#include "Engine/GameViewportClient.h"
#include "Widgets/SBoxPanel.h"
#include "Widgets/Images/SImage.h"
#include "Widgets/Text/STextBlock.h"
#include "Widgets/Input/SEditableText.h"
#include "Widgets/Input/SButton.h"
#include "EditorReimportHandler.h"
#include "EditorFramework/AssetImportData.h"
#include "PropertyHandle.h"
#include "IDetailChildrenBuilder.h"
#include "DetailWidgetRow.h"
#include "DetailLayoutBuilder.h"

#include "ScopedTransaction.h"

#define LOCTEXT_NAMESPACE "AssetImportDataCustomization"

TSharedRef<IPropertyTypeCustomization> FAssetImportDataCustomization::MakeInstance()
{
	return MakeShareable( new FAssetImportDataCustomization );
}

void FAssetImportDataCustomization::CustomizeHeader( TSharedRef<IPropertyHandle> InPropertyHandle, FDetailWidgetRow& HeaderRow, IPropertyTypeCustomizationUtils& CustomizationUtils )
{
}

void FAssetImportDataCustomization::CustomizeChildren( TSharedRef<IPropertyHandle> InPropertyHandle, IDetailChildrenBuilder& ChildBuilder, IPropertyTypeCustomizationUtils& CustomizationUtils )
{
	PropertyHandle = InPropertyHandle;

	FAssetImportInfo* Info = GetEditStruct();
	if (!Info)
	{
		return;
	}

	auto Font = IDetailLayoutBuilder::GetDetailFont();

	const FText SourceFileText = LOCTEXT("SourceFile", "Source File");
	int32 NumSourceFiles = Info->SourceFiles.Num() ? Info->SourceFiles.Num() : 1;

	for (int32 Index = 0; Index < NumSourceFiles; ++Index)
	{
		FText SourceFileLabel = SourceFileText;
		if (Info->SourceFiles.IsValidIndex(Index) && Info->SourceFiles[Index].DisplayLabelName.Len() > 0)
		{
			SourceFileLabel = FText::FromString(SourceFileText.ToString() + TEXT(" (") + Info->SourceFiles[Index].DisplayLabelName + TEXT(")"));
		}

		ChildBuilder.AddCustomRow(SourceFileLabel)
		.NameContent()
		[
			SNew(STextBlock)
			.Text(SourceFileLabel)
			.Font(Font)
		]
		.ValueContent()
		.HAlign(HAlign_Fill)
		.MaxDesiredWidth(TOptional<float>())
		[
			SNew(SHorizontalBox)
			+ SHorizontalBox::Slot()
			.VAlign(VAlign_Center)
			[
				SNew(SEditableText)
				.IsReadOnly(true)
				.Text(this, &FAssetImportDataCustomization::GetFilenameText, Index)
				.ToolTipText(this, &FAssetImportDataCustomization::GetFilenameText, Index)
				.Font(Font)
			]

			+SHorizontalBox::Slot()
			.VAlign(VAlign_Center)
			.AutoWidth()
			[
				SNew(SButton)
				.OnClicked(this, &FAssetImportDataCustomization::OnChangePathClicked, Index)
				.ToolTipText(LOCTEXT("ChangePath_Tooltip", "Browse for a new source file path"))
				[
					SNew(STextBlock)
					.Text(LOCTEXT("...", "..."))
					.Font(Font)
				]
			]

			+SHorizontalBox::Slot()
			.AutoWidth()
			[
				SNew(SButton)
				.VAlign(VAlign_Center)
				.HAlign(HAlign_Center)
				.ButtonStyle(FEditorStyle::Get(), "HoverHintOnly")
				.OnClicked(this, &FAssetImportDataCustomization::OnClearPathClicked, Index)
				.ToolTipText(LOCTEXT("ClearPath_Tooltip", "Clear this source file information from the asset"))
				[
					SNew(SImage)
					.Image(FEditorStyle::GetBrush("Cross"))
				]
			]
		];

		ChildBuilder.AddCustomRow(SourceFileText)
			.ValueContent()
			.HAlign(HAlign_Fill)
			.MaxDesiredWidth(TOptional<float>())
			[
				SNew(SHorizontalBox)
				+ SHorizontalBox::Slot()
				.VAlign(VAlign_Center)
				.FillWidth(1.0f)
				[
					SNew(SEditableText)
					.IsReadOnly(true)
					.Text(this, &FAssetImportDataCustomization::GetTimestampText, Index)
					.Font(Font)
				]
				+ SHorizontalBox::Slot()
				.VAlign(VAlign_Center)
				.HAlign(HAlign_Center)
				.AutoWidth()
				[
					SNew(SButton)
					.VAlign(VAlign_Center)
					.HAlign(HAlign_Center)
					.ButtonStyle(FEditorStyle::Get(), "HoverHintOnly")
					.IsEnabled(this, &FAssetImportDataCustomization::IsPropagateFromAbovePathEnable, Index)
					.OnClicked(this, &FAssetImportDataCustomization::OnPropagateFromAbovePathClicked, Index)
					.ToolTipText(LOCTEXT("PropagateFromAbovePath_Tooltip", "Use the above source path to set this path."))
					[
						SNew(SImage)
						.Image(FEditorStyle::GetBrush("ArrowDown"))
					]
				]
				+ SHorizontalBox::Slot()
				.VAlign(VAlign_Center)
				.HAlign(HAlign_Center)
				.AutoWidth()
				[
					SNew(SButton)
					.VAlign(VAlign_Center)
					.HAlign(HAlign_Center)
					.ButtonStyle(FEditorStyle::Get(), "HoverHintOnly")
					.IsEnabled(this, &FAssetImportDataCustomization::IsPropagateFromBelowPathEnable, Index)
					.OnClicked(this, &FAssetImportDataCustomization::OnPropagateFromBelowPathClicked, Index)
					.ToolTipText(LOCTEXT("PropagateFromBelowPath_Tooltip", "Use the below source path to set this path."))
					[
						SNew(SImage)
						.Image(FEditorStyle::GetBrush("ArrowUp"))
					]
				]
			];
	}
}

FText FAssetImportDataCustomization::GetFilenameText(int32 Index) const
{
	FAssetImportInfo* Info = GetEditStruct();
	if (Info)
	{
		if (Info->SourceFiles.IsValidIndex(Index))
		{
			return FText::FromString(Info->SourceFiles[Index].RelativeFilename);
		}
	}
	return LOCTEXT("NoFilenameFound", "No Source Path Set");
}

FText FAssetImportDataCustomization::GetTimestampText(int32 Index) const
{
	FAssetImportInfo* Info = GetEditStruct();
	if (Info)
	{
		if (Info->SourceFiles.IsValidIndex(Index))
		{
			return FText::FromString(Info->SourceFiles[Index].Timestamp.ToString());
		}
	}
	return FText();
}


FAssetImportInfo* FAssetImportDataCustomization::GetEditStruct() const
{
	static TArray<FAssetImportInfo*> AssetImportInfo;
	AssetImportInfo.Reset();

	if(PropertyHandle->IsValidHandle())
	{
		PropertyHandle->AccessRawData(reinterpret_cast<TArray<void*>&>(AssetImportInfo));
	}

	if (AssetImportInfo.Num() == 1)
	{
		return AssetImportInfo[0];
	}
	return nullptr;
}

UAssetImportData* FAssetImportDataCustomization::GetOuterClass() const
{
	static TArray<UObject*> OuterObjects;
	OuterObjects.Reset();

	PropertyHandle->GetOuterObjects(OuterObjects);

	return OuterObjects.Num() ? Cast<UAssetImportData>(OuterObjects[0]) : nullptr;
}

class FImportDataSourceFileTransactionScope
{
public:
	FImportDataSourceFileTransactionScope(FText TransactionName, UAssetImportData* InImportData)
	{
		check(InImportData);
		ImportData = InImportData;
		FScopedTransaction Transaction(TransactionName);

		bIsTransactionnal = (ImportData->GetFlags() & RF_Transactional) > 0;
		if (!bIsTransactionnal)
		{
			ImportData->SetFlags(RF_Transactional);
		}

		ImportData->Modify();
	}
	
	~FImportDataSourceFileTransactionScope()
	{
		if (!bIsTransactionnal)
		{
			ImportData->ClearFlags(RF_Transactional);
		}
		ImportData->MarkPackageDirty();
	}
private:
	bool bIsTransactionnal;
	UAssetImportData* ImportData;
};

FReply FAssetImportDataCustomization::OnChangePathClicked(int32 Index) const
{
	UAssetImportData* ImportData = GetOuterClass();
	UObject* Obj = ImportData ? ImportData->GetOuter() : nullptr;

	FAssetImportInfo* Info = GetEditStruct();

	if (!Obj)
	{
		return FReply::Handled();
	}

	TArray<FString> OpenFilenames;
	FReimportManager::Instance()->GetNewReimportPath(Obj, OpenFilenames);
	if (OpenFilenames.Num() == 1)
	{
		FImportDataSourceFileTransactionScope TransactionScope(LOCTEXT("SourceReimportChangePath", "Change source file path"), ImportData);
		if (!Info || !Info->SourceFiles.IsValidIndex(Index))
		{
			ImportData->UpdateFilenameOnly(FPaths::ConvertRelativePathToFull(OpenFilenames[0]));
		}
		else
		{
			ImportData->UpdateFilenameOnly(FPaths::ConvertRelativePathToFull(OpenFilenames[0]), Index);
		}
	}
	return FReply::Handled();
}

FReply FAssetImportDataCustomization::OnClearPathClicked(int32 Index) const
{
	UAssetImportData* ImportData = GetOuterClass();
	if (ImportData && ImportData->SourceData.SourceFiles.IsValidIndex(Index))
	{
		FImportDataSourceFileTransactionScope TransactionScope(LOCTEXT("SourceReimportClearPath", "Clear Source file path"), ImportData);
		ImportData->SourceData.SourceFiles[Index] = FAssetImportInfo::FSourceFile(FString());
	}

	return FReply::Handled();
}

bool FAssetImportDataCustomization::IsPropagateFromAbovePathEnable(int32 Index) const
{
	UAssetImportData* ImportData = GetOuterClass();
	return (ImportData && ImportData->SourceData.SourceFiles.IsValidIndex(Index) && ImportData->SourceData.SourceFiles.IsValidIndex(Index - 1));
}

bool FAssetImportDataCustomization::IsPropagateFromBelowPathEnable(int32 Index) const
{
	UAssetImportData* ImportData = GetOuterClass();
	return (ImportData && ImportData->SourceData.SourceFiles.IsValidIndex(Index) && ImportData->SourceData.SourceFiles.IsValidIndex(Index + 1));
}

void FAssetImportDataCustomization::PropagatePath(int32 SrcIndex, int32 DstIndex) const
{
	UAssetImportData* ImportData = GetOuterClass();
	if (ImportData && ImportData->SourceData.SourceFiles.IsValidIndex(DstIndex) && ImportData->SourceData.SourceFiles.IsValidIndex(SrcIndex))
	{
		FImportDataSourceFileTransactionScope TransactionScope(LOCTEXT("SourceReimportPropagateFromAbove", "Propagate source file path"), ImportData);
		FString OriginalLabel = ImportData->SourceData.SourceFiles[DstIndex].DisplayLabelName;
		ImportData->SourceData.SourceFiles[DstIndex] = ImportData->SourceData.SourceFiles[SrcIndex];
		ImportData->SourceData.SourceFiles[DstIndex].DisplayLabelName = OriginalLabel;
	}
}

FReply FAssetImportDataCustomization::OnPropagateFromAbovePathClicked(int32 Index) const
{
	PropagatePath(Index - 1, Index);
	return FReply::Handled();
}


FReply FAssetImportDataCustomization::OnPropagateFromBelowPathClicked(int32 Index) const
{
	PropagatePath(Index + 1, Index);
	return FReply::Handled();
}

#undef LOCTEXT_NAMESPACE