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

#include "Commandlets/GatherTextCommandletBase.h"
#include "Misc/Paths.h"
#include "Misc/ConfigCacheIni.h"
#include "Misc/PackageName.h"
#include "ISourceControlOperation.h"
#include "SourceControlOperations.h"
#include "ISourceControlModule.h"
#include "EngineGlobals.h"
#include "AssetData.h"
#include "Editor.h"
#include "IAssetRegistry.h"
#include "ARFilter.h"
#include "PackageHelperFunctions.h"
#include "ObjectTools.h"

DEFINE_LOG_CATEGORY_STATIC(LogGatherTextCommandletBase, Log, All);

FGatherTextDelegates::FGetAdditionalGatherPaths FGatherTextDelegates::GetAdditionalGatherPaths;

//////////////////////////////////////////////////////////////////////////
// UGatherTextCommandletBase

UGatherTextCommandletBase::UGatherTextCommandletBase(const FObjectInitializer& ObjectInitializer)
    : Super(ObjectInitializer)
{
}

void UGatherTextCommandletBase::Initialize(const TSharedRef<FLocTextHelper>& InGatherManifestHelper, const TSharedPtr<FLocalizationSCC>& InSourceControlInfo)
{
    GatherManifestHelper = InGatherManifestHelper;
    SourceControlInfo    = InSourceControlInfo;

    // Cache the split platform info
    SplitPlatforms.Reset();
    if (InGatherManifestHelper->ShouldSplitPlatformData())
    {
        for (const FString& SplitPlatformName: InGatherManifestHelper->GetPlatformsToSplit())
        {
            SplitPlatforms.Add(*SplitPlatformName, FString::Printf(TEXT("/%s/"), *SplitPlatformName));
        }
        SplitPlatforms.KeySort(FNameLexicalLess());
    }
}

void UGatherTextCommandletBase::BeginDestroy()
{
    Super::BeginDestroy();

    GatherManifestHelper.Reset();
    SourceControlInfo.Reset();
}

void UGatherTextCommandletBase::CreateCustomEngine(const FString& Params)
{
    GEngine = GEditor = NULL;  // Force a basic default engine.
}

bool UGatherTextCommandletBase::IsSplitPlatformName(const FName InPlatformName) const
{
    return SplitPlatforms.Contains(InPlatformName);
}

bool UGatherTextCommandletBase::ShouldSplitPlatformForPath(const FString& InPath, FName* OutPlatformName) const
{
    const FName SplitPlatformName = GetSplitPlatformNameFromPath(InPath);
    if (OutPlatformName)
    {
        *OutPlatformName = SplitPlatformName;
    }
    return !SplitPlatformName.IsNone();
}

FName UGatherTextCommandletBase::GetSplitPlatformNameFromPath(const FString& InPath) const
{
    for (const auto& SplitPlatformsPair: SplitPlatforms)
    {
        if (InPath.Contains(SplitPlatformsPair.Value))
        {
            return SplitPlatformsPair.Key;
        }
    }
    return FName();
}

bool UGatherTextCommandletBase::GetBoolFromConfig(const TCHAR* Section, const TCHAR* Key, bool& OutValue, const FString& Filename)
{
    bool bSuccess = GConfig->GetBool(Section, Key, OutValue, Filename);

    if (!bSuccess)
    {
        bSuccess = GConfig->GetBool(TEXT("CommonSettings"), Key, OutValue, Filename);
    }
    return bSuccess;
}

bool UGatherTextCommandletBase::GetStringFromConfig(const TCHAR* Section, const TCHAR* Key, FString& OutValue, const FString& Filename)
{
    bool bSuccess = GConfig->GetString(Section, Key, OutValue, Filename);

    if (!bSuccess)
    {
        bSuccess = GConfig->GetString(TEXT("CommonSettings"), Key, OutValue, Filename);
    }
    return bSuccess;
}

void ResolveLocalizationPath(FString& InOutPath)
{
    static const bool bIsEngineTarget        = FPaths::ProjectDir().IsEmpty();
    static const FString AbsoluteEnginePath  = FPaths::ConvertRelativePathToFull(FPaths::EngineDir()) / FString();
    static const FString AbsoluteProjectPath = FPaths::ConvertRelativePathToFull(FPaths::ProjectDir()) / FString();

    InOutPath.ReplaceInline(TEXT("%LOCENGINEROOT%"), *AbsoluteEnginePath, ESearchCase::CaseSensitive);
    InOutPath.ReplaceInline(TEXT("%LOCPROJECTROOT%"), *AbsoluteProjectPath, ESearchCase::CaseSensitive);

    if (FPaths::IsRelative(InOutPath))
    {
        InOutPath.InsertAt(0, bIsEngineTarget ? AbsoluteEnginePath : AbsoluteProjectPath);
    }

    FPaths::CollapseRelativeDirectories(InOutPath);
}

bool UGatherTextCommandletBase::GetPathFromConfig(const TCHAR* Section, const TCHAR* Key, FString& OutValue, const FString& Filename)
{
    const bool bSuccess = GetStringFromConfig(Section, Key, OutValue, Filename);
    if (bSuccess)
    {
        ResolveLocalizationPath(OutValue);
    }
    return bSuccess;
}

int32 UGatherTextCommandletBase::GetStringArrayFromConfig(const TCHAR* Section, const TCHAR* Key, TArray<FString>& OutArr, const FString& Filename)
{
    int32 count = GConfig->GetArray(Section, Key, OutArr, Filename);

    if (count == 0)
    {
        count = GConfig->GetArray(TEXT("CommonSettings"), Key, OutArr, Filename);
    }
    return count;
}

int32 UGatherTextCommandletBase::GetPathArrayFromConfig(const TCHAR* Section, const TCHAR* Key, TArray<FString>& OutArr, const FString& Filename)
{
    int32 count = GetStringArrayFromConfig(Section, Key, OutArr, Filename);

    for (FString& Path: OutArr)
    {
        ResolveLocalizationPath(Path);
    }

    return count;
}

FFuzzyPathMatcher::FFuzzyPathMatcher(const TArray<FString>& InIncludePathFilters, const TArray<FString>& InExcludePathFilters)
{
    FuzzyPaths.Reserve(InIncludePathFilters.Num() + InExcludePathFilters.Num());

    for (const FString& IncludePath: InIncludePathFilters)
    {
        FuzzyPaths.Add(FFuzzyPath(FPaths::ConvertRelativePathToFull(IncludePath), EPathType::Include));
    }

    for (const FString& ExcludePath: InExcludePathFilters)
    {
        FuzzyPaths.Add(FFuzzyPath(FPaths::ConvertRelativePathToFull(ExcludePath), EPathType::Exclude));
    }

    // Sort the paths so that deeper paths with fewer wildcards appear first in the list
    FuzzyPaths.Sort([](const FFuzzyPath& PathOne, const FFuzzyPath& PathTwo) -> bool
                    {
                        auto GetFuzzRating = [](const FFuzzyPath& InFuzzyPath) -> int32
                        {
                            int32 PathDepth = 0;
                            int32 PathFuzz  = 0;
                            for (const TCHAR Char: InFuzzyPath.PathFilter)
                            {
                                if (Char == TEXT('/') || Char == TEXT('\\'))
                                {
                                    ++PathDepth;
                                }
                                else if (Char == TEXT('*') || Char == TEXT('?'))
                                {
                                    ++PathFuzz;
                                }
                            }

                            return (100 - PathDepth) + (PathFuzz * 1000);
                        };

                        const int32 PathOneFuzzRating = GetFuzzRating(PathOne);
                        const int32 PathTwoFuzzRating = GetFuzzRating(PathTwo);
                        return PathOneFuzzRating < PathTwoFuzzRating;
                    });
}

FFuzzyPathMatcher::EPathMatch FFuzzyPathMatcher::TestPath(const FString& InPathToTest) const
{
    for (const FFuzzyPath& FuzzyPath: FuzzyPaths)
    {
        if (InPathToTest.MatchesWildcard(FuzzyPath.PathFilter))
        {
            return (FuzzyPath.PathType == EPathType::Include) ? EPathMatch::Included : EPathMatch::Excluded;
        }
    }

    return EPathMatch::NoMatch;
}