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

#pragma once

#include "Containers/Array.h"
#include "Containers/ArrayView.h"
#include "Containers/UnrealString.h"
#include "ContentBrowserItem.h"
#include "CoreMinimal.h"
#include "DragAndDrop/AssetDragDropOp.h"
#include "HAL/Platform.h"
#include "Input/DragAndDrop.h"
#include "Internationalization/Text.h"
#include "Templates/SharedPointer.h"

class UActorFactory;
struct FAssetData;

/**
 * Additional params for FContentBrowserDataDragDropOp to override the thumbnail used if dragging only folders
 */
struct CONTENTBROWSERDATA_API FThumbnailOverrideParams
{
	FThumbnailOverrideParams() {}

	/** Folder brush name to use */
	FName FolderBrushName = NAME_None;

	/** Folder shadow brush name to use */
	FName FolderShadowBrushName = NAME_None;

	/** Override color for the final brush */
	FLinearColor FolderColorOverride = FLinearColor::Gray;
};

class CONTENTBROWSERDATA_API FContentBrowserDataDragDropOp : public FAssetDragDropOp
{
public:
	DRAG_DROP_OPERATOR_TYPE(FContentBrowserDataDragDropOp, FAssetDragDropOp)

	static TSharedRef<FContentBrowserDataDragDropOp> New(TArrayView<const FContentBrowserItem> InDraggedItems, FThumbnailOverrideParams InThumbnailOverrideParams = FThumbnailOverrideParams());

	static TSharedRef<FContentBrowserDataDragDropOp> Legacy_New(TArrayView<const FAssetData> InAssetData, TArrayView<const FString> InAssetPaths = TArrayView<const FString>(), UActorFactory* InActorFactory = nullptr);

	const TArray<FContentBrowserItem>& GetDraggedItems() const
	{
		return DraggedItems;
	}

	const TArray<FContentBrowserItem>& GetDraggedFiles() const
	{
		return DraggedFiles;
	}

	const TArray<FContentBrowserItem>& GetDraggedFolders() const
	{
		return DraggedFolders;
	}

private:
	void Init(TArrayView<const FContentBrowserItem> InDraggedItems);
	void LegacyInit(TArrayView<const FAssetData> InAssetData, TArrayView<const FString> InAssetPaths, UActorFactory* ActorFactory);
	virtual void InitThumbnail() override;

	virtual bool HasFiles() const override;
	virtual bool HasFolders() const override;

	virtual int32 GetTotalCount() const override;
	virtual FText GetFirstItemText() const override;

	/** Return whether or not it should override the thumbnail widget */
	bool ShouldOverrideThumbnailWidget() const;

	/** Return the folder widget or nullptr if the BrushName is not valid */
	TSharedPtr<SWidget> GetFolderWidget() const;

	/** Return the folder widget for DragAndDrop or nullptr if there is no basic folder widget  */
	TSharedPtr<SWidget> GetFolderWidgetDragAndDrop() const;

private:
	FName FolderBrushName;
	FName FolderShadowBrushName;
	FLinearColor FolderColorOverride = FLinearColor::Gray;
	TArray<FContentBrowserItem> DraggedItems;
	TArray<FContentBrowserItem> DraggedFiles;
	TArray<FContentBrowserItem> DraggedFolders;
};