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

#pragma once

#include "CoreMinimal.h"

/** Counts all persistent text references from within a package, using the specified comparison mode */
class COREUOBJECT_API FTextReferenceCollector
{
public:
    enum class EComparisonMode : uint8
    {
        /** Detect a reference if it matches the given ID (ignoring the source text) */
        MatchId,
        /** Detect a reference if it matches the given ID and source string */
        MatchSource,
        /** Detect a reference if it matches the given ID but has a different source string */
        MismatchSource,
    };

    typedef TFunction<void(UObject*, FArchive&)> FTextReferenceCollectorCallback;
    typedef TMap<const UClass*, FTextReferenceCollectorCallback> FTextReferenceCollectorCallbackMap;

    FTextReferenceCollector(const UPackage* const InPackage, const EComparisonMode InComparisonMode, const FString& InTextNamespace, const FString& InTextKey, const FString& InTextSource, int32& OutCount);

    static FTextReferenceCollectorCallbackMap& GetTypeSpecificTextReferenceCollectorCallbacks();
};

/** Struct to automatically register a callback when it's constructed */
struct FAutoRegisterTextReferenceCollectorCallback
{
    FORCEINLINE FAutoRegisterTextReferenceCollectorCallback(const UClass* InClass, const FTextReferenceCollector::FTextReferenceCollectorCallback& InCallback)
    {
        FTextReferenceCollector::GetTypeSpecificTextReferenceCollectorCallbacks().Add(InClass, InCallback);
    }
};