MessageStore.h

// Copyright 2026 Stream.IO, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Message.h"
 
#include "MessageStore.generated.h"
 
using FMessageRef = TSharedRef<FMessage>;
using FMessages = TArray<FMessageRef>;
 
USTRUCT(BlueprintType)
struct STREAMCHAT_API FMessageStore
{
    GENERATED_BODY()
 
    
    void Append(const FMessageStore&);
    void Append(const TArray<FMessageDto>, UUserManager*);
 
    bool AddMessage(const FMessage&);
    bool AddMessage(const FMessageDto&, UUserManager*);
 
    void AppendReplies(const FString& ParentId, const TArray<FMessageDto>&, UUserManager*);
 
    bool IsEmpty() const;
    // Must check if store is empty first or this will crash
    const FMessage& First();
    // Must check if store is empty first or this will crash
    const FMessage& Last();
 
    TSharedPtr<FMessage> Next(const FMessage&) const;
 
    FMessages FilterRecent(const FTimespan& Since) const;
 
    const FMessages& GetMessages() const;
 
    const FMessages& GetReplies(const FString& ParentId) const;
 
private:
    void Append(const FMessages&);
    bool AddReply(const FMessage&);
    void ShowReplyInChannel(const FMessageRef&);
    void IncrementReplyCount(const FString& ParentId);
 
    FMessages Messages;
    TMap<FString, FMessages> Threads;
};