ArticyIdCustomization.h

//  
// Copyright (c) 2023 articy Software GmbH & Co. KG. All rights reserved.  
//
 
#pragma once
 
#include "CoreMinimal.h"
#include <IPropertyTypeCustomization.h>
#include "Slate/SArticyIdProperty.h"
#include "ClassViewerFilter.h"
#include "ClassViewerModule.h"
#include "ArticyObject.h"

class FArticyRefClassFilter final : public IClassViewerFilter
{
public:
    explicit FArticyRefClassFilter(UClass* GivenClass = UArticyObject::StaticClass(), bool bInRequiresExactClass = false);
 
    virtual bool IsClassAllowed(const FClassViewerInitializationOptions& InInitOptions, const UClass* InClass, TSharedRef< FClassViewerFilterFuncs > InFilterFuncs) override
    {
        if (bRequiresExactClass)
        {
            return InClass == GivenClass;
        }
 
        return InClass->IsChildOf(GivenClass);
    }
 
    virtual bool IsUnloadedClassAllowed(const FClassViewerInitializationOptions& InInitOptions, const TSharedRef< const IUnloadedBlueprintData > InUnloadedClassData, TSharedRef< FClassViewerFilterFuncs > InFilterFuncs) override
    {
        return false;
    }
 
    UClass* GivenClass = nullptr; 
    bool bRequiresExactClass = false; 
};

class FArticyIdCustomization final : public IPropertyTypeCustomization
{
public:
    static TSharedRef<IPropertyTypeCustomization> MakeInstance();

    virtual void CustomizeHeader(TSharedRef<IPropertyHandle> PropertyHandle, FDetailWidgetRow& HeaderRow, IPropertyTypeCustomizationUtils& CustomizationUtils) override;
 
    virtual void CustomizeChildren(TSharedRef<IPropertyHandle> PropertyHandle, IDetailChildrenBuilder& ChildBuilder, IPropertyTypeCustomizationUtils& CustomizationUtils) override;

    static FArticyId* RetrieveArticyId(IPropertyHandle* ArticyIdHandle);

    static FArticyId GetIdFromValueString(FString SourceString);
 
private:
    TSharedPtr<IPropertyHandle> ArticyIdPropertyHandle; 
    TSharedPtr<SArticyIdProperty> ArticyIdPropertyWidget; 
    bool bShouldCustomize = true; 
 
    UClass* GetClassRestriction() const;
    bool IsExactClass() const;
    bool IsReadOnly() const;
    bool HasClassRestrictionMetaData() const;
    bool HasExactClassMetaData() const;
 
    FArticyId GetArticyId() const;
    void OnArticyIdChanged(const FArticyId& NewArticyId) const;
};