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

#pragma once

#include "CoreMinimal.h"
#include "Engine/UserDefinedEnum.h"
#include "Kismet2/ListenerManager.h"

class UNREALED_API FEnumEditorUtils
{
    static void PrepareForChange(UUserDefinedEnum* Enum);
    static void BroadcastChanges(const UUserDefinedEnum* Enum, const TArray<TPair<FName, int64>>& OldNames, bool bResolveData = true);

    /** copy full enumeratos names from given enum to OutEnumNames, the last '_MAX' enumerator is skipped */
    static void CopyEnumeratorsWithoutMax(const UEnum* Enum, TArray<TPair<FName, int64>>& OutEnumNames);

public:
    enum EEnumEditorChangeInfo
    {
        Changed,
    };

    class FEnumEditorManager: public FListenerManager<UUserDefinedEnum, EEnumEditorChangeInfo>
    {
        FEnumEditorManager() {}

    public:
        UNREALED_API static FEnumEditorManager& Get();

        class UNREALED_API ListenerType: public InnerListenerType<FEnumEditorManager>
        {
        };
    };

    typedef FEnumEditorManager::ListenerType INotifyOnEnumChanged;

    //////////////////////////////////////////////////////////////////////////
    // User defined enumerations

    /** Creates new user defined enum in given blueprint. */
    static UEnum* CreateUserDefinedEnum(UObject* InParent, FName EnumName, EObjectFlags Flags);

    /** return if an enum can be named/renamed with given name*/
    static bool IsNameAvailebleForUserDefinedEnum(FName Name);

    /** Updates enumerators names after name or path of the Enum was changed */
    static void UpdateAfterPathChanged(UEnum* Enum);

    /** adds new enumerator (with default unique name) for user defined enum */
    static void AddNewEnumeratorForUserDefinedEnum(class UUserDefinedEnum* Enum);

    /** Removes enumerator from enum*/
    static void RemoveEnumeratorFromUserDefinedEnum(class UUserDefinedEnum* Enum, int32 EnumeratorIndex);

    /** Reorder enumerators in enum. Swap an enumerator with given name, with previous or next (based on bDirectionUp parameter) */
    static void MoveEnumeratorInUserDefinedEnum(class UUserDefinedEnum* Enum, int32 EnumeratorIndex, bool bDirectionUp);

    /** Check if the enumerator-as-bitflags meta data is set */
    static bool IsEnumeratorBitflagsType(class UUserDefinedEnum* Enum);

    /** Set the state of the enumerator-as-bitflags meta data */
    static void SetEnumeratorBitflagsTypeState(class UUserDefinedEnum* Enum, bool bBitflagsType);

    /** check if NewName is a short name and is acceptable as name in given enum */
    static bool IsProperNameForUserDefinedEnumerator(const UEnum* Enum, FString NewName);

    /** Handles necessary notifications when the Enum has had a transaction undone or redone on it. */
    static void PostEditUndo(UUserDefinedEnum* Enum);

    /*
     *	Try to update an out-of-date enum index after an enum's change
     *
     *	@param Enum - new version of enum
     *	@param Ar - special archive
     *	@param EnumeratorIndex - old enumerator index
     *
     *	@return new enum
     */
    static int64 ResolveEnumerator(const UEnum* Enum, FArchive& Ar, int64 EnumeratorValue);

    // DISPLAY NAME
    static bool SetEnumeratorDisplayName(UUserDefinedEnum* Enum, int32 EnumeratorIndex, FText NewDisplayName);
    static bool IsEnumeratorDisplayNameValid(const UUserDefinedEnum* Enum, int32 EnumeratorIndex, FText NewDisplayName);
    static void EnsureAllDisplayNamesExist(class UUserDefinedEnum* Enum);
    static void UpgradeDisplayNamesFromMetaData(class UUserDefinedEnum* Enum);
};