SPT-Server-Build/Libraries/SPTarkov.Server.Core/Helpers/NotificationSendHelper.cs

using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.Models.Common;
using SPTarkov.Server.Core.Models.Eft.Profile;
using SPTarkov.Server.Core.Models.Eft.Ws;
using SPTarkov.Server.Core.Models.Enums;
using SPTarkov.Server.Core.Models.Utils;
using SPTarkov.Server.Core.Servers;
using SPTarkov.Server.Core.Servers.Ws;
using SPTarkov.Server.Core.Services;
using SPTarkov.Server.Core.Utils;
using LogLevel = SPTarkov.Server.Core.Models.Spt.Logging.LogLevel;

namespace SPTarkov.Server.Core.Helpers;

[Injectable]
public class NotificationSendHelper(
    ISptLogger<NotificationSendHelper> logger,
    SptWebSocketConnectionHandler sptWebSocketConnectionHandler,
    SaveServer saveServer,
    NotificationService notificationService,
    TimeUtil timeUtil,
    JsonUtil jsonUtil
)
{
    /// <summary>
    ///     Send notification message to the appropriate channel
    /// </summary>
    /// <param name="sessionID">Session/player id</param>
    /// <param name="notificationMessage"></param>
    public void SendMessage(MongoId sessionID, WsNotificationEvent notificationMessage)
    {
        if (logger.IsLogEnabled(LogLevel.Debug))
        {
            logger.Debug(
                $"Send message for {sessionID} started, message: {jsonUtil.Serialize(notificationMessage)}"
            );
        }
        if (sptWebSocketConnectionHandler.IsWebSocketConnected(sessionID))
        {
            if (logger.IsLogEnabled(LogLevel.Debug))
            {
                logger.Debug(
                    $"Send message for {sessionID} websocket available, message being sent"
                );
            }
            sptWebSocketConnectionHandler.SendMessage(sessionID, notificationMessage);
        }
        else
        {
            if (logger.IsLogEnabled(LogLevel.Debug))
            {
                logger.Debug(
                    $"Send message for {sessionID} websocket not available, queueing into profile"
                );
            }
            notificationService.Add(sessionID, notificationMessage);
        }
    }

    /// <summary>
    ///     Send a message directly to the player
    /// </summary>
    /// <param name="sessionId">Session id</param>
    /// <param name="senderDetails">Who is sending the message to player</param>
    /// <param name="messageText">Text to send player</param>
    /// <param name="messageType">Underlying type of message being sent</param>
    public void SendMessageToPlayer(
        MongoId sessionId,
        UserDialogInfo senderDetails,
        string messageText,
        MessageType messageType
    )
    {
        var dialog = GetDialog(sessionId, messageType, senderDetails);

        dialog.New += 1;
        var message = new Message
        {
            Id = new MongoId(),
            UserId = dialog.Id,
            MessageType = messageType,
            DateTime = timeUtil.GetTimeStamp(),
            Text = messageText,
            HasRewards = null,
            RewardCollected = null,
            Items = null,
        };
        dialog.Messages.Add(message);

        var notification = new WsChatMessageReceived
        {
            EventType = NotificationEventType.new_message,
            EventIdentifier = message.Id,
            DialogId = message.UserId,
            Message = message,
        };
        SendMessage(sessionId, notification);
    }

    /// <summary>
    ///     Helper function for SendMessageToPlayer(), get new dialog for storage in profile or find existing by sender id
    /// </summary>
    /// <param name="sessionId">Session id</param>
    /// <param name="messageType">Type of message to generate</param>
    /// <param name="senderDetails">Who is sending the message</param>
    /// <returns>Dialogue</returns>
    protected Models.Eft.Profile.Dialogue GetDialog(
        MongoId sessionId,
        MessageType messageType,
        UserDialogInfo senderDetails
    )
    {
        // Use trader id if sender is trader, otherwise use nickname
        var dialogKey = senderDetails.Id;

        // Get all dialogs with pmcs/traders player has
        var dialogueData = saveServer.GetProfile(sessionId).DialogueRecords;

        // Ensure empty dialog exists based on sender details passed in
        dialogueData.TryAdd(
            dialogKey,
            GetEmptyDialogTemplate(dialogKey, messageType, senderDetails)
        );

        return dialogueData[dialogKey];
    }

    protected Models.Eft.Profile.Dialogue GetEmptyDialogTemplate(
        string dialogKey,
        MessageType messageType,
        UserDialogInfo senderDetails
    )
    {
        return new Models.Eft.Profile.Dialogue
        {
            Id = dialogKey,
            Type = messageType,
            Messages = [],
            Pinned = false,
            New = 0,
            AttachmentsNew = 0,
            Users =
                senderDetails.Info.MemberCategory == MemberCategory.Trader ? null : [senderDetails],
        };
    }
}