MessageWidget.h

// Copyright 2026 Stream.IO, Inc. All Rights Reserved.
 
#pragma once
 
#include "AttachmentWidget.h"
#include "Channel/Message.h"
#include "Components/Overlay.h"
#include "CoreMinimal.h"
#include "MessageHoverMenuWidget.h"
#include "Reaction/MessageReactionsWidget.h"
#include "StreamWidget.h"
#include "TextBubbleWidget.h"
#include "TimestampWidget.h"
 
#include "MessageWidget.generated.h"
 
UCLASS()
class STREAMCHATUI_API UMessageWidget final : public UStreamWidget
{
    GENERATED_BODY()
 
public:
    explicit UMessageWidget();
    UFUNCTION(BlueprintCallable, Category = "Stream Chat")
    void Setup(const FMessage& InMessage, EMessageSide InSide, EMessagePosition InPosition);
 
protected:
    UPROPERTY(meta = (BindWidgetOptional))
    UTextBubbleWidget* TextBubble;
 
    // Should contain whatever needs to be horizontally aligned
    UPROPERTY(meta = (BindWidgetOptional))
    UVerticalBox* AlignPanel;
 
    // Where the mouse hover menu will be spawned
    UPROPERTY(meta = (BindWidgetOptional))
    UContentWidget* HoverMenuTargetPanel;
 
    // Where the reactions will be spawned
    UPROPERTY(meta = (BindWidgetOptional))
    UContentWidget* ReactionsTargetPanel;
 
    // Where the timestamp will (maybe) be spawned
    UPROPERTY(meta = (BindWidgetOptional))
    UContentWidget* TimestampTargetPanel;
 
    // Where the avatar will (maybe) be spawned
    UPROPERTY(meta = (BindWidgetOptional))
    UContentWidget* AvatarTargetPanel;
 
    UPROPERTY(EditDefaultsOnly, NoClear, Category = Defaults)
    TSubclassOf<UMessageHoverMenuWidget> MouseHoverMenuWidgetClass = UMessageHoverMenuWidget::StaticClass();
 
    UPROPERTY(EditDefaultsOnly, NoClear, Category = Defaults)
    TSubclassOf<UMessageReactionsWidget> ReactionsWidgetClass = UMessageReactionsWidget::StaticClass();
 
    UPROPERTY(EditDefaultsOnly, NoClear, Category = Defaults)
    TSubclassOf<UAvatarWidget> AvatarWidgetClass = UAvatarWidget::StaticClass();
 
    UPROPERTY(EditDefaultsOnly, NoClear, Category = Defaults)
    TSubclassOf<UTimestampWidget> TimestampWidgetClass = UTimestampWidget::StaticClass();
 
    UPROPERTY(EditDefaultsOnly, NoClear, Category = Defaults)
    TSubclassOf<UAttachmentWidget> AttachmentWidgetClass = UAttachmentWidget::StaticClass();
 
    UPROPERTY(EditDefaultsOnly, Category = Defaults)
    int32 AvatarSize = 36;
 
    UPROPERTY(EditDefaultsOnly, Category = Defaults)
    FMargin AttachmentPadding = FMargin{0.f, 0.f, 0.f, 2.f};
 
    UPROPERTY(EditDefaultsOnly, Category = Thread)
    FMargin ThreadFooterPadding = FMargin{2.f, 2.f, 2.f, 0.f};
 
    UPROPERTY(EditDefaultsOnly, Category = Thread)
    int32 ThreadFooterFontSize = 12;
 
    UPROPERTY(EditDefaultsOnly, Category = "Message Actions")
    float LongPressSeconds = 0.4f;
 
    UPROPERTY(EditDefaultsOnly, Category = "Message Actions")
    float LongPressMoveTolerance = 4.f;
 
    UPROPERTY(EditDefaultsOnly, Category = Moderation)
    FMargin ModerationWarningPadding = FMargin{2.f, 2.f, 2.f, 0.f};
 
    UPROPERTY(EditDefaultsOnly, Category = Moderation)
    int32 ModerationWarningFontSize = 12;
 
    UPROPERTY(EditDefaultsOnly, Category = Moderation)
    FText BouncedMessageText = NSLOCTEXT("StreamChat", "BouncedMessage", "Not sent. Hold to edit, send anyway or delete.");
 
    UPROPERTY(EditDefaultsOnly, Category = Thread)
    FText OneReplyText = NSLOCTEXT("StreamChat", "OneReply", "1 reply");
 
    UPROPERTY(EditDefaultsOnly, Category = Thread)
    FText ManyRepliesFormat = NSLOCTEXT("StreamChat", "ManyReplies", "{0} replies");
 
    UPROPERTY(EditAnywhere, BlueprintReadOnly, Category = Setup)
    FMessage Message;
    UPROPERTY(EditAnywhere, BlueprintReadOnly, Category = Setup)
    EMessagePosition Position;
    UPROPERTY(EditAnywhere, BlueprintReadOnly, Category = Setup)
    EMessageSide Side;
 
private:
    virtual void OnSetup() override;
    virtual void NativeConstruct() override;
    virtual void NativeDestruct() override;
 
    virtual void NativeOnMouseEnter(const FGeometry& InGeometry, const FPointerEvent& InMouseEvent) override;
    virtual void NativeOnMouseLeave(const FPointerEvent& InMouseEvent) override;
    virtual FReply NativeOnMouseButtonDown(const FGeometry& InGeometry, const FPointerEvent& InMouseEvent) override;
    virtual FReply NativeOnMouseButtonUp(const FGeometry& InGeometry, const FPointerEvent& InMouseEvent) override;
 
    bool ShouldDisplayHoverMenu() const;
 
    void BeginLongPress();
    void CancelLongPress();
    void OnLongPressElapsed();
    bool ShouldDisplayActionsMenu() const;
    void CreateActionsMenuAnchor();
 
    UFUNCTION()
    UUserWidget* CreateActionsMenu();
 
    void CreateAttachmentWidgets();
 
    void CreateThreadFooter();
    bool ShouldDisplayThreadFooter() const;
    FText GetThreadFooterText() const;
 
    void CreateModerationWarning();
    bool ShouldDisplayModerationWarning() const;
 
    UFUNCTION()
    void OnThreadFooterClicked();
 
    // Only valid while hovered
    UPROPERTY(Transient)
    UMessageHoverMenuWidget* MouseHoverMenu;
 
    // Only valid if message has reactions
    UPROPERTY(Transient)
    UMessageReactionsWidget* Reactions;
 
    // Live in AlignPanel rather than a panel of their own, so tracked here to be replaced on re-setup
    UPROPERTY(Transient)
    TArray<UAttachmentWidget*> Attachments;
 
    // Also lives in AlignPanel, for the same reason: WBP_Message has no slot for it
    UPROPERTY(Transient)
    UButton* ThreadFooterButton;
 
    UPROPERTY(Transient)
    UTextBlock* ThreadFooterText;
 
    UPROPERTY(Transient)
    UTextBlock* ModerationWarningText;
 
    UPROPERTY(Transient)
    UMenuAnchor* ActionsMenuAnchor;
 
    FTimerHandle LongPressTimer;
    FVector2D PressPosition = FVector2D::ZeroVector;
};