ArticyEditorCustomizationManager.h

//  
// Copyright (c) 2023 articy Software GmbH & Co. KG. All rights reserved.  
//
 
#pragma once
 
#include "CoreMinimal.h"
#include "ArticyRef.h"
#include "Framework/MultiBox/MultiBoxExtender.h"


struct FArticyIdPropertyWidgetCustomizationInfo
{
    TSharedPtr<FExtender> ExtraButtonExtender;
};

class FArticyIdPropertyWidgetCustomizationBuilder
{
public:
    explicit FArticyIdPropertyWidgetCustomizationBuilder(const UArticyObject* InObject) : ArticyObject(InObject)
    {}

    void AddCustomization(const FArticyIdPropertyWidgetCustomizationInfo& Customization) { Customizations.Add(Customization); }

    const TArray<FArticyIdPropertyWidgetCustomizationInfo>& GetCustomizations() const { return Customizations; }

    const UArticyObject* GetArticyObject() const { return ArticyObject; }
 
private:
    const UArticyObject* ArticyObject;
    TArray<FArticyIdPropertyWidgetCustomizationInfo> Customizations;
};

class ARTICYEDITOR_API IArticyIdPropertyWidgetCustomization
{
public:
    virtual ~IArticyIdPropertyWidgetCustomization() {}

    virtual void RegisterArticyIdPropertyWidgetCustomization(FArticyIdPropertyWidgetCustomizationBuilder& Builder) = 0;

    virtual void UnregisterArticyIdPropertyWidgetCustomization() = 0;
};

class ARTICYEDITOR_API IArticyIdPropertyWidgetCustomizationFactory
{
public:
    virtual ~IArticyIdPropertyWidgetCustomizationFactory() {}

    virtual TSharedPtr<IArticyIdPropertyWidgetCustomization> CreateCustomization() = 0;

    virtual bool SupportsType(const UArticyObject* InObject) = 0;
};

DECLARE_DELEGATE_RetVal(TSharedRef<IArticyIdPropertyWidgetCustomizationFactory>, FOnCreateArticyIdPropertyWidgetCustomizationFactory);

class ARTICYEDITOR_API FArticyEditorCustomizationManager
{
public:
    FArticyEditorCustomizationManager();

    IArticyIdPropertyWidgetCustomizationFactory* RegisterArticyIdPropertyWidgetCustomizationFactory(FOnCreateArticyIdPropertyWidgetCustomizationFactory GetCustomizationDelegate);

    void UnregisterArticyIdPropertyWidgetCustomizationFactory(const IArticyIdPropertyWidgetCustomizationFactory*);

    void CreateArticyIdPropertyWidgetCustomizations(const UArticyObject* ArticyObject, TArray<TSharedPtr<IArticyIdPropertyWidgetCustomization>>& OutCustomizations);
 
private:
    void Shutdown();

    FArticyEditorCustomizationManager(const FArticyEditorCustomizationManager&) = delete;
    FArticyEditorCustomizationManager& operator=(const FArticyEditorCustomizationManager&) = delete;
 
    TArray<TSharedPtr<IArticyIdPropertyWidgetCustomizationFactory>> ArticyIdPropertyWidgetCustomizationFactories;
 
    friend class FArticyEditorModule;
};