ChatSession.h

Go to the documentation of this file.

// Copyright(c) 2025 grrimgrriefer & DZnnah, see LICENSE for details.
 
#pragma once
 
#include "CoreMinimal.h"
#include "ChatMessage.h"
#include "VoxtaServiceType.h"
#include "VoxtaServiceEntryData.h"
#include "AiCharData.h"
#include "ChatSession.generated.h"

USTRUCT(BlueprintType, Category = "Voxta")
struct VOXTADATA_API FChatSession
{
    GENERATED_BODY()
 
#pragma region public API
public:
    const TArray<FChatMessage>& GetChatMessages()
    {
        return m_chatMessages;
    }

    FGuid GetSessionId() const { return m_sessionId; }

    const TMap<VoxtaServiceType, FVoxtaServiceEntryData>& GetActiveServices() const
    {
        return m_services;
    }

    void UpdateContext(const FString& newContext)
    {
        m_chatContext = newContext;
    }

    void AddChatMessage(const FChatMessage& message)
    {
        m_chatMessages.Add(message);
    }

    void RemoveChatMessage(const FGuid& messageID)
    {
        int index = m_chatMessages.IndexOfByPredicate([messageID] (const FChatMessage& InItem)
        {
            return InItem.GetMessageId() == messageID;
        });
 
        if (index != INDEX_NONE)
        {
            m_chatMessages.RemoveAt(index);
        }
    }

    FChatMessage* GetChatMessageById(const FGuid& messageId)
    {
        return m_chatMessages.FindByPredicate([&messageId] (const FChatMessage& msg)
        {
            return msg.GetMessageId() == messageId;
        });
    }

    explicit FChatSession(const TArray<const FAiCharData*>& characters,
            FGuid chatId,
            FGuid sessionId,
            const TMap<VoxtaServiceType, FVoxtaServiceEntryData>& services,
            FStringView chatContext) :
        m_chatId(chatId),
        m_sessionId(sessionId),
        m_chatContext(chatContext),
        m_characters(characters),
        m_services(services)
    {
        m_characterIds.Reserve(characters.Num());
        for (const FAiCharData* character : characters)
        {
            m_characterIds.Emplace(character->GetId());
        }
    }

    FChatSession() = default;

    FGuid GetChatId() const { return m_chatId; }

    FStringView GetChatContext() const { return m_chatContext; }

    const TArray<FGuid>& GetCharacterIds() const { return m_characterIds; }
#pragma endregion
 
#pragma region data
private:
    UPROPERTY(BlueprintReadOnly, Category = "Voxta", meta = (AllowPrivateAccess = "true", DisplayName = "Chat ID"))
    FGuid m_chatId;
 
    UPROPERTY(BlueprintReadOnly, Category = "Voxta", meta = (AllowPrivateAccess = "true", DisplayName = "Session ID"))
    FGuid m_sessionId;
 
    UPROPERTY(BlueprintReadOnly, Category = "Voxta", meta = (AllowPrivateAccess = "true", DisplayName = "Character IDs"))
    TArray<FGuid> m_characterIds;
 
    UPROPERTY(BlueprintReadOnly, Category = "Voxta", meta = (AllowPrivateAccess = "true", DisplayName = "Context"))
    FString m_chatContext;
 
    UPROPERTY(BlueprintReadOnly, Category = "Voxta", meta = (AllowPrivateAccess = "true", DisplayName = "Messages so far"))
    TArray<FChatMessage> m_chatMessages;
 
    TArray<const FAiCharData*> m_characters;
 
    UPROPERTY(BlueprintReadOnly, Category = "Voxta", meta = (AllowPrivateAccess = "true", DisplayName = "Services"))
    TMap<VoxtaServiceType, FVoxtaServiceEntryData> m_services;
#pragma endregion
};