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

/**
 * UI to pick options when importing a data table
 */

#pragma once

#include "CoreMinimal.h"
#include "Factories/CSVImportFactory.h"
#include "Layout/Visibility.h"
#include "Widgets/DeclarativeSyntaxSupport.h"
#include "Input/Reply.h"
#include "Widgets/SWidget.h"
#include "Widgets/SCompoundWidget.h"
#include "Widgets/Input/SComboBox.h"

enum class ECSVImportOptionDlgResponse : uint8
{
    Import,
    ImportAll,
    Cancel
};

class UNREALED_API SCSVImportOptions: public SCompoundWidget
{
private:
    /** Typedef for curve enum pointers */
    typedef TSharedPtr<ERichCurveInterpMode> CurveInterpModePtr;

public:
    SLATE_BEGIN_ARGS(SCSVImportOptions)
        : _WidgetWindow(), _FullPath(), _TempImportDataTable()
    {}

    SLATE_ARGUMENT(TSharedPtr<SWindow>, WidgetWindow)
    SLATE_ARGUMENT(FText, FullPath)
    SLATE_ARGUMENT(UDataTable*, TempImportDataTable)
    SLATE_END_ARGS()

    SCSVImportOptions()
        : UserDlgResponse(ECSVImportOptionDlgResponse::Cancel), SelectedImportType(ECSVImportType::ECSV_DataTable), SelectedStruct(nullptr), TempImportDataTable(nullptr)
    {}

    void Construct(const FArguments& InArgs);

    /** If we should import */
    bool ShouldImport();

    /** If the current settings should be applied to all items being imported */
    bool ShouldImportAll();

    /** Get the row struct we selected */
    UScriptStruct* GetSelectedRowStruct();

    /** Get the import type we selected */
    ECSVImportType GetSelectedImportType();

    /** Get the interpolation mode we selected */
    ERichCurveInterpMode GetSelectedCurveIterpMode();

    /** Whether to show table row options */
    EVisibility GetTableRowOptionVis() const;

    /** Whether to show curve type options */
    EVisibility GetCurveTypeVis() const;

    /** Whether to show details panel */
    EVisibility GetDetailsPanelVis() const;

    FString GetImportTypeText(TSharedPtr<ECSVImportType> Type) const;

    /** Called to create a widget for each struct */
    TSharedRef<SWidget> MakeImportTypeItemWidget(TSharedPtr<ECSVImportType> Type);

    /** Called when import type changes */
    void OnImportTypeSelected(TSharedPtr<ECSVImportType> Selection, ESelectInfo::Type SelectionType);

    /** Called when datatable row is selected */
    void OnStructSelected(UScriptStruct* NewStruct);

    FString GetCurveTypeText(CurveInterpModePtr InterpMode) const;

    /** Called to create a widget for each curve interpolation enum */
    TSharedRef<SWidget> MakeCurveTypeWidget(CurveInterpModePtr InterpMode);

    /** Called when 'Apply' button is pressed */
    FReply OnImport();

    /** Do we have all of the data we need to import this asset? */
    bool CanImport() const;

    /** Called when 'Cancel' button is pressed */
    FReply OnCancel();

    FText GetSelectedItemText() const;

    FText GetSelectedCurveTypeText() const;

private:
    FReply HandleImport();
    FReply OnImportAll();

    /** Whether we should go ahead with import */
    ECSVImportOptionDlgResponse UserDlgResponse;

    /** Window that owns us */
    TWeakPtr<SWindow> WidgetWindow;

    // Import type

    /** List of import types to pick from, drives combo box */
    TArray<TSharedPtr<ECSVImportType>> ImportTypes;

    /** The combo box */
    TSharedPtr<SComboBox<TSharedPtr<ECSVImportType>>> ImportTypeCombo;

    /** Indicates what kind of asset we want to make from the CSV file */
    ECSVImportType SelectedImportType;

    // Row type

    /** The row struct combo box */
    TSharedPtr<SWidget> RowStructCombo;

    /** The selected row struct */
    UScriptStruct* SelectedStruct;

    /** Temp DataTable to hold import options */
    TWeakObjectPtr<UDataTable> TempImportDataTable;

    /** The curve interpolation combo box */
    TSharedPtr<SComboBox<CurveInterpModePtr>> CurveInterpCombo;

    /** A property view to edit advanced options */
    TSharedPtr<class IDetailsView> PropertyView;

    /** All available curve interpolation modes */
    TArray<CurveInterpModePtr> CurveInterpModes;

    /** The selected curve interpolation type */
    ERichCurveInterpMode SelectedCurveInterpMode;
};