BotSharp/src/Plugins/BotSharp.Plugin.ChatHub/Hooks/ChatHubConversationHook.cs

using BotSharp.Abstraction.Conversations.Dtos;
using BotSharp.Abstraction.Conversations.Enums;
using BotSharp.Abstraction.Routing.Enums;
using BotSharp.Abstraction.Routing.Models;
using BotSharp.Abstraction.SideCar;
using BotSharp.Abstraction.Users.Dtos;
using Microsoft.AspNetCore.SignalR;
using System;
using System.Runtime.CompilerServices;

namespace BotSharp.Plugin.ChatHub.Hooks;

public class ChatHubConversationHook : ConversationHookBase
{
    private readonly IServiceProvider _services;
    private readonly IHubContext<SignalRHub> _chatHub;
    private readonly ILogger<ChatHubConversationHook> _logger;
    private readonly IUserIdentity _user;
    private readonly BotSharpOptions _options;
    private readonly ChatHubSettings _settings;

    public ChatHubConversationHook(
        IServiceProvider services,
        IHubContext<SignalRHub> chatHub,
        ILogger<ChatHubConversationHook> logger,
        BotSharpOptions options,
        ChatHubSettings settings,
        IUserIdentity user)
    {
        _services = services;
        _chatHub = chatHub;
        _logger = logger;
        _user = user;
        _options = options;
        _settings = settings;
        Priority = -1; // Make sure this hook is the top one.
    }

    public override async Task OnConversationInitialized(Conversation conversation)
    {
        if (!AllowSendingMessage()) return;

        var userService = _services.GetRequiredService<IUserService>();
        var conv = ConversationDto.FromSession(conversation);

        var user = await userService.GetUser(conv.User.Id);
        conv.User = UserDto.FromUser(user);

        await SendEvent(ChatEvent.OnConversationInitFromClient, conv.Id, conv);
        await base.OnConversationInitialized(conversation);
    }

    public override async Task OnMessageReceived(RoleDialogModel message)
    {
        if (!AllowSendingMessage()) return;

        var conv = _services.GetRequiredService<IConversationService>();
        var userService = _services.GetRequiredService<IUserService>();
        var sender = await userService.GetMyProfile();

        // Update console conversation UI for CSR
        var model = new ChatResponseDto()
        {
            ConversationId = conv.ConversationId,
            MessageId = message.MessageId,
            Payload = message.Payload,
            Text = !string.IsNullOrEmpty(message.SecondaryContent) ? message.SecondaryContent : message.Content,
            Sender = UserDto.FromUser(sender)
        };
        await SendEvent(ChatEvent.OnMessageReceivedFromClient, conv.ConversationId, model);

        // Send typing-on to client
        var action = new ConversationSenderActionModel
        {
            ConversationId = conv.ConversationId,
            SenderAction = SenderActionEnum.TypingOn
        };
        await SendEvent(ChatEvent.OnSenderActionGenerated, conv.ConversationId, action);
        await base.OnMessageReceived(message);
    }

    public override async Task OnFunctionExecuting(RoleDialogModel message, InvokeFunctionOptions? options = null)
    {
        await base.OnFunctionExecuting(message, options);
    }

    public override async Task OnPostbackMessageReceived(RoleDialogModel message, PostbackMessageModel replyMsg)
    {
        await this.OnMessageReceived(message);
    }

    public override async Task OnResponseGenerated(RoleDialogModel message)
    {
        if (!AllowSendingMessage()) return;

        var conv = _services.GetRequiredService<IConversationService>();
        var state = _services.GetRequiredService<IConversationStateService>();

        var sender = new UserDto
        {
            FirstName = "AI",
            LastName = "Assistant",
            Role = AgentRole.Assistant
        };

        var data = new ChatResponseDto()
        {
            ConversationId = conv.ConversationId,
            MessageId = message.MessageId,
            MessageLabel = message.MessageLabel,
            Text = !string.IsNullOrEmpty(message.SecondaryContent) ? message.SecondaryContent : message.Content,
            Function = message.FunctionName,
            RichContent = message.SecondaryRichContent ?? message.RichContent,
            Data = message.Data,
            States = state.GetStates(),
            IsStreaming = message.IsStreaming,
            Sender = sender
        };

        // Send type-off to client
        var action = new ConversationSenderActionModel
        {
            ConversationId = conv.ConversationId,
            SenderAction = SenderActionEnum.TypingOff
        };
        await SendEvent(ChatEvent.OnSenderActionGenerated, conv.ConversationId, action);
        await SendEvent(ChatEvent.OnMessageReceivedFromAssistant, conv.ConversationId, data);

        var wrapper = message.AdditionalMessageWrapper;
        if (wrapper?.SendingInterval > 0 && wrapper?.Messages?.Count > 0)
        {
            action.SenderAction = SenderActionEnum.TypingOn;
            await SendEvent(ChatEvent.OnSenderActionGenerated, conv.ConversationId, action);

            foreach (var item in wrapper.Messages)
            {
                if (!string.IsNullOrWhiteSpace(item.Indication))
                {
                    data = new ChatResponseDto
                    {
                        ConversationId = conv.ConversationId,
                        MessageId = item.MessageId,
                        MessageLabel = item.MessageLabel,
                        Indication = item.Indication,
                        Sender = sender
                    };
                    await SendEvent(ChatEvent.OnIndicationReceived, conv.ConversationId, data);
                }

                await Task.Delay(wrapper.SendingInterval);

                data = new ChatResponseDto
                {
                    ConversationId = conv.ConversationId,
                    MessageId = item.MessageId,
                    MessageLabel = item.MessageLabel,
                    Text = !string.IsNullOrEmpty(item.SecondaryContent) ? item.SecondaryContent : item.Content,
                    Function = item.FunctionName,
                    RichContent = item.SecondaryRichContent ?? item.RichContent,
                    Data = item.Data,
                    States = state.GetStates(),
                    IsAppend = true,
                    Sender = sender
                };
                await SendEvent(ChatEvent.OnMessageReceivedFromAssistant, conv.ConversationId, data);
            }

            action.SenderAction = SenderActionEnum.TypingOff;
            await SendEvent(ChatEvent.OnSenderActionGenerated, conv.ConversationId, action);
        }

        await base.OnResponseGenerated(message);
    }


    public override async Task OnNotificationGenerated(RoleDialogModel message)
    {
        var conv = _services.GetRequiredService<IConversationService>();
        var data = new ChatResponseDto()
        {
            ConversationId = conv.ConversationId,
            MessageId = message.MessageId,
            Text = !string.IsNullOrEmpty(message.SecondaryContent) ? message.SecondaryContent : message.Content,
            Function = message.FunctionName,
            RichContent = message.SecondaryRichContent ?? message.RichContent,
            Data = message.Data,
            Sender = new()
            {
                FirstName = "AI",
                LastName = "Assistant",
                Role = AgentRole.Assistant
            }
        };

        await SendEvent(ChatEvent.OnNotificationGenerated, conv.ConversationId, data);
        await base.OnNotificationGenerated(message);
    }


    public override async Task OnMessageDeleted(string conversationId, string messageId)
    {
        var model = new ChatResponseDto
        {
            ConversationId = conversationId,
            MessageId = messageId
        };

        await SendEvent(ChatEvent.OnMessageDeleted, conversationId, model);
        await base.OnMessageDeleted(conversationId, messageId);
    }

    #region Private methods
    private bool AllowSendingMessage()
    {
        var sidecar = _services.GetService<IConversationSideCar>();
        return sidecar == null || !sidecar.IsEnabled;
    }

    private async Task SendEvent<T>(string @event, string conversationId, T data, [CallerMemberName] string callerName = "")
    {
        var user = _services.GetRequiredService<IUserIdentity>();
        var json = JsonSerializer.Serialize(data, _options.JsonSerializerOptions);
        await EventEmitter.SendChatEvent(_services, _logger, @event, conversationId, user?.Id, json, nameof(ChatHubConversationHook), callerName);
    }
    #endregion
}