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

#pragma once

#include "CoreMinimal.h"
#include "Textures/SlateIcon.h"
#include "Styling/SlateIconFinder.h"

class AActor;
struct FAssetData;
class ISlateStyle;
class UBlueprint;
struct FSlateBrush;

class FClassIconFinder
{
public:
    /** Find the best fitting small icon to use for the supplied actor array */
    UNREALED_API static const FSlateBrush* FindIconForActors(const TArray<TWeakObjectPtr<AActor>>& InActors, UClass*& CommonBaseClass);

    /** Find the small icon to use for the supplied actor */
    UNREALED_API static const FSlateBrush* FindIconForActor(const TWeakObjectPtr<const AActor>& InActor);

    /** Find the small icon to use for the supplied actor */
    UNREALED_API static FSlateIcon FindSlateIconForActor(const TWeakObjectPtr<const AActor>& InActor);

    /** Utility function to convert a Blueprint into the most suitable class possible for use by the icon finder */
    UNREALED_API static const UClass* GetIconClassForBlueprint(const UBlueprint* InBlueprint);

    /** Utility function to convert an asset into the most suitable class possible for use by the icon finder */
    UNREALED_API static const UClass* GetIconClassForAssetData(const FAssetData& InAssetData, bool* bOutIsClassType = nullptr);

    /** Find the large thumbnail name to use for the supplied class */
    UNREALED_API static const FSlateBrush* FindThumbnailForClass(const UClass* InClass, const FName& InDefaultName = FName())
    {
        return FSlateIconFinder::FindCustomIconBrushForClass(InClass, TEXT("ClassThumbnail"), InDefaultName);
    }

private:
    UNREALED_API static const FSlateBrush* LookupBrush(FName IconName);
};