HellionChat/ChatTwo/Http/Processing.cs

using System.Globalization;
using System.Net;
using ChatTwo.Code;
using ChatTwo.Http.MessageProtocol;
using ChatTwo.Util;
using Dalamud.Game.Text.SeStringHandling.Payloads;

namespace ChatTwo.Http;

public class Processing
{
    private readonly Plugin Plugin;

    public Processing(Plugin plugin)
    {
        Plugin = plugin;
    }

    internal string ReadChannelName(Chunk[] channelName)
    {
        return string.Join("", channelName.Select(chunk => ProcessChunk(chunk, noColor: true)));
    }

    internal async Task<MessageResponse[]> ReadMessageList()
    {
        var tabMessages = await Plugin.ChatLogWindow.CurrentTab!.Messages.GetCopy();
        return tabMessages.Select(ReadMessageContent).ToArray();
    }

    internal MessageResponse ReadMessageContent(Message message)
    {
        var response = new MessageResponse
        {
            Timestamp = message.Date.ToLocalTime().ToString("t", !Plugin.Config.Use24HourClock ? null : CultureInfo.CreateSpecificCulture("es-ES"))
        };

        var content = "";
        if (message.Sender.Count > 0)
            content = message.Sender.Aggregate(content, (current, chunk) => current + ProcessChunk(chunk));

        content = message.Content.Aggregate(content, (current, chunk) => current + ProcessChunk(chunk));
        response.Message = content;

        return response;
    }

    internal async Task PrepareNewClient(SSEConnection sse)
    {
        var messages = await WebserverUtil.FrameworkWrapper(ReadMessageList);
        var channels = await Plugin.Framework.RunOnTick(Plugin.ChatLogWindow.GetAvailableChannels);
        var channelName = await Plugin.Framework.RunOnTick(() => ReadChannelName(Plugin.ChatLogWindow.PreviousChannel));

        // Using the bulk message event to clear everything on the client side that may still exist
        sse.OutboundQueue.Enqueue(new BulkMessagesEvent(new Messages(messages)));
        sse.OutboundQueue.Enqueue(new SwitchChannelEvent(new SwitchChannel(channelName)));
        sse.OutboundQueue.Enqueue(new ChannelListEvent(new ChannelList(channels.ToDictionary(pair => pair.Key, pair => (uint)pair.Value))));
    }

    private string ProcessChunk(Chunk chunk, bool noColor = false)
    {
        if (chunk is IconChunk { } icon)
        {
            return IconUtil.GfdFileView.TryGetEntry((uint) icon.Icon, out _)
                ? $"<span class=\"gfd-icon gfd-icon-hq-{(uint)icon.Icon}\"></span>"
                : "";
        }

        if (chunk is TextChunk { } text)
        {
            if (chunk.Link is EmotePayload emotePayload && Plugin.Config.ShowEmotes)
            {
                var image = EmoteCache.GetEmote(emotePayload.Code);

                // The emote name should be safe, it is checked against a list from BTTV.
                // Still sanitizing it for the extra safety.
                if (image is { Failed: false })
                    return $"<span class=\"emote-icon\"><img src=\"/emote/{WebUtility.HtmlEncode(emotePayload.Code)}\"></span>";
            }

            var colour = text.Foreground;
            if (colour == null && text.FallbackColour != null)
            {
                var type = text.FallbackColour.Value;
                colour = Plugin.Config.ChatColours.TryGetValue(type, out var col) ? col : type.DefaultColor();
            }

            var color = ColourUtil.RgbaToComponents(colour ?? 0);

            var userContent = text.Content ?? "";
            if (Plugin.ChatLogWindow.ScreenshotMode)
            {
                if (chunk.Link is PlayerPayload playerPayload)
                    userContent = Plugin.ChatLogWindow.HidePlayerInString(userContent, playerPayload.PlayerName, playerPayload.World.RowId);
                else if (Plugin.ClientState.LocalPlayer is { } player)
                    userContent = Plugin.ChatLogWindow.HidePlayerInString(userContent, player.Name.TextValue, player.HomeWorld.Id);
            }

            // HTML encode any user content to prevent xss
            userContent = WebUtility.HtmlEncode(userContent);

            if (text.Link is UriPayload uri)
                userContent = $"<a href=\"{uri.Uri}\" target=\"_blank\">{userContent}</a>";

            return noColor
                ? userContent
                : $"<span style=\"color:rgba({color.r}, {color.g}, {color.b}, {color.a})\">{userContent}</span>";
        }

        return string.Empty;
    }
}