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

/**
 * Factory for importing table from text (CSV)
 */

#pragma once

#include "CoreMinimal.h"
#include "UObject/ObjectMacros.h"
#include "Curves/RichCurve.h"
#include "Factories/Factory.h"
#include "Factories/ImportSettings.h"
#include "EditorReimportHandler.h"
#include "CSVImportFactory.generated.h"

DECLARE_LOG_CATEGORY_EXTERN(LogCSVImportFactory, Log, All);

/** Enum to indicate what to import CSV as */
UENUM(BlueprintType)
enum class ECSVImportType : uint8
{
    /** Import as UDataTable */
    ECSV_DataTable,
    /** Import as UCurveTable */
    ECSV_CurveTable,
    /** Import as a UCurveFloat */
    ECSV_CurveFloat,
    /** Import as a UCurveVector */
    ECSV_CurveVector,
    /** Import as a UCurveLinearColor */
    ECSV_CurveLinearColor,
};

USTRUCT(BlueprintType)
struct FCSVImportSettings
{
    GENERATED_BODY()

    FCSVImportSettings();

    UPROPERTY(BlueprintReadWrite, Category = "Misc")
    UScriptStruct* ImportRowStruct;

    UPROPERTY(BlueprintReadWrite, Category = "Misc")
    ECSVImportType ImportType;

    UPROPERTY(BlueprintReadWrite, Category = "Misc")
    TEnumAsByte<ERichCurveInterpMode> ImportCurveInterpMode;

    FString DataToImport;
};

UCLASS(hidecategories = Object)
class UNREALED_API UCSVImportFactory: public UFactory
    , public IImportSettingsParser
{
    GENERATED_UCLASS_BODY()

    //~ Begin UFactory Interface
    virtual FText GetDisplayName() const override;
    virtual UObject* FactoryCreateText(UClass* InClass, UObject* InParent, FName InName, EObjectFlags Flags, UObject* Context, const TCHAR* Type, const TCHAR*& Buffer, const TCHAR* BufferEnd, FFeedbackContext* Warn, bool& bOutOperationCanceled) override;
    virtual bool DoesSupportClass(UClass* Class) override;
    virtual bool FactoryCanImport(const FString& Filename) override;
    virtual IImportSettingsParser* GetImportSettingsParser() override;
    virtual void CleanUp() override;

    /* Reimport an object that was created based on a CSV */
    EReimportResult::Type ReimportCSV(UObject* Obj);

    /**
     * IImportSettings interface
     */
    virtual void ParseFromJson(TSharedRef<class FJsonObject> ImportSettingsJson) override;

protected:
    virtual TArray<FString> DoImportDataTable(const FCSVImportSettings& ImportSettings, class UDataTable* TargetDataTable);
    virtual TArray<FString> DoImportCurveTable(const FCSVImportSettings& ImportSettings, class UCurveTable* TargetCurveTable);
    virtual TArray<FString> DoImportCurve(const FCSVImportSettings& InImportSettings, class UCurveBase* TargetCurve);

private:
    /* Reimport object from the given path*/
    EReimportResult::Type Reimport(UObject* Obj, const FString& Path);

    bool bImportAll = false;

public:
    UPROPERTY(BlueprintReadWrite, Category = "Automation")
    FCSVImportSettings AutomatedImportSettings;

    /** Temporary data table to use to display import options */
    UPROPERTY()
    UDataTable* DataTableImportOptions;
};