#pragma once

#include "CoreMinimal.h"
#include "Toolkits/AssetEditorToolkit.h"
#include "Widgets/Docking/SDockTab.h"

class SDockTab;
class SGraphEditor;
class SPropertyTreeView;
class SAssetBrowser;
class SMatrixInputWidget;

/**
 * FLESH Main Editor
 * Provides the main editing functionality for the dismemberment system
 */
class FLESHEDITOR_API FFLESHEditor : public FAssetEditorToolkit
{
public:
    FFLESHEditor();
    virtual ~FFLESHEditor();

    // Initialize the editor
    void InitFLESHEditor(const EToolkitMode::Type Mode, const TSharedPtr<IToolkitHost>& InitToolkitHost);

    // FAssetEditorToolkit interface
    virtual void RegisterTabSpawners(const TSharedRef<FTabManager>& TabManager) override;
    virtual void UnregisterTabSpawners(const TSharedRef<FTabManager>& TabManager) override;
    virtual FName GetToolkitFName() const override;
    virtual FText GetBaseToolkitName() const override;
    virtual FString GetWorldCentricTabPrefix() const override;
    virtual FLinearColor GetWorldCentricTabColorScale() const override;
    // End of FAssetEditorToolkit interface

    // Open the editor
    static void OpenEditor();

private:
    // Tab spawners
    TSharedRef<SDockTab> SpawnTab_Viewport(const FSpawnTabArgs& Args);
    TSharedRef<SDockTab> SpawnTab_Details(const FSpawnTabArgs& Args);
    TSharedRef<SDockTab> SpawnTab_AssetBrowser(const FSpawnTabArgs& Args);
    TSharedRef<SDockTab> SpawnTab_MatrixEditor(const FSpawnTabArgs& Args);
    TSharedRef<SDockTab> SpawnTab_GraphEditor(const FSpawnTabArgs& Args);
    TSharedRef<SDockTab> SpawnTab_Toolbar(const FSpawnTabArgs& Args);

    // Create viewport widget
    TSharedRef<SWidget> CreateViewportWidget();

    // Create details panel
    TSharedRef<SWidget> CreateDetailsWidget();

    // Create asset browser
    TSharedRef<SWidget> CreateAssetBrowserWidget();

    // Create matrix editor
    TSharedRef<SWidget> CreateMatrixEditorWidget();

    // Create graph editor
    TSharedRef<SWidget> CreateGraphEditorWidget();

    // Create toolbar
    TSharedRef<SWidget> CreateToolbarWidget();

    // Create command list
    void CreateCommandList();

    // Command handlers
    void OnOpenDismembermentGraphEditor();
    void OnOpenAnatomicalLayerEditor();
    void OnOpenBooleanCutTool();
    void OnOpenBloodSystemEditor();
    void OnImportCharacterModel();
    void OnImportOrganModel();
    void OnImportSkeletonModel();
    void OnImportPhysicsAsset();
    void OnTestMatrix();

    // Viewport widget
    TSharedPtr<class SViewport> ViewportWidget;

    // Details panel
    TSharedPtr<class IDetailsView> DetailsWidget;

    // Asset browser
    TSharedPtr<SAssetBrowser> AssetBrowserWidget;

    // Matrix editor
    TSharedPtr<SMatrixInputWidget> MatrixEditorWidget;

    // Graph editor
    TSharedPtr<SGraphEditor> GraphEditorWidget;

    // Toolbar
    TSharedPtr<class SBorder> ToolbarWidget;

    // Command list
    TSharedPtr<FUICommandList> CommandList;

    // Tab IDs
    static const FName ViewportTabId;
    static const FName DetailsTabId;
    static const FName AssetBrowserTabId;
    static const FName MatrixEditorTabId;
    static const FName GraphEditorTabId;
    static const FName ToolbarTabId;
};