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

#pragma once

#include "CoreMinimal.h"
#include "UObject/ObjectMacros.h"
#include "Factories/Factory.h"
#include "FbxFactory.generated.h"

class IImportSettingsParser;

namespace UnFbx
{
class FFbxImporter;
}

UCLASS(hidecategories = Object)
class UNREALED_API UFbxFactory: public UFactory
{
    GENERATED_UCLASS_BODY()

    UPROPERTY()
    class UFbxImportUI* ImportUI;

    /** Prevent garbage collection of original when overriding ImportUI property */
    UPROPERTY()
    class UFbxImportUI* OriginalImportUI;

    /**  Set import batch **/
    void EnableShowOption() { bShowOption = true; }

    //~ Begin UObject Interface
    virtual bool ConfigureProperties() override;
    virtual void PostInitProperties() override;
    //~ End UObject Interface

    //~ Begin UFactory Interface
    virtual void CleanUp() override;
    virtual bool DoesSupportClass(UClass* Class) override;
    virtual UClass* ResolveSupportedClass() override;
    virtual UObject* FactoryCreateFile(UClass* InClass, UObject* InParent, FName InName, EObjectFlags Flags, const FString& Filename, const TCHAR* Parms, FFeedbackContext* Warn, bool& bOutOperationCanceled) override;
    virtual bool FactoryCanImport(const FString& Filename) override;
    virtual bool CanImportBeCanceled() const override;
    virtual IImportSettingsParser* GetImportSettingsParser() override;

    //~ End UFactory Interface

    /**
     * Detect mesh type to import: Static Mesh or Skeletal Mesh.
     * Only the first mesh will be detected.
     *
     * @param InFilename	FBX file name
     * @return bool	return true if parse the file successfully
     */
    bool DetectImportType(const FString& InFilename);

    void SetDetectImportTypeOnImport(bool bDetectState) { bDetectImportTypeOnImport = bDetectState; }

private:
    /**
     * Delete the newly created objects after an import cancelation.
     */
    void CancelObjectCreation(UnFbx::FFbxImporter* FbxImporter) const;

protected:
    // @todo document
    UObject* RecursiveImportNode(UnFbx::FFbxImporter* FFbxImporter, void* Node, UObject* InParent, FName InName, EObjectFlags Flags, struct FScopedSlowTask& SlowTask, TArray<UObject*>& OutNewAssets);

    // @todo document
    UObject* ImportANode(UnFbx::FFbxImporter* FbxImporter, TArray<void*> VoidNodes, UObject* InParent, FName InName, EObjectFlags Flags, struct FScopedSlowTask& SlowTask, UObject* InMesh = NULL, int LODIndex = 0);

    /**
     * Used to display special task progression text if needs be (like import being cancelled).
     *
     * @param TaskText	The current import task text we want to display.
     * @return			The task text after taking into account the factory's current state.
     */
    FText GetImportTaskText(const FText& TaskText) const;

    bool bShowOption;
    bool bDetectImportTypeOnImport;

    /** true if the import operation was canceled. */
    bool bOperationCanceled;
};