implement name color highlighting, handle stickers, and channel name replacement

1 files changed, 136 insertions, 43 deletions

diff --git a/DiscordToXIV/Plugin.cs b/DiscordToXIV/Plugin.cs
index bd8c977..c73ea58 100644
--- a/DiscordToXIV/Plugin.cs
+++ b/DiscordToXIV/Plugin.cs
@@ -8,37 +8,55 @@ using System.Collections.Generic;
 using System.Threading;
 using System.Text.Json;
 using System.Text.Json.Serialization;
+using Dalamud.Game.Text.SeStringHandling;
+using Dalamud.Game.Text.SeStringHandling.Payloads;
 using Dalamud.Interface.Windowing;
 using DiscordToXIV.Windows;
 
+
+
 namespace DiscordToXIV;
 
 
 public class Message
 {
+    [JsonPropertyName("id")]
+    public string? Id { get; init; }
     [JsonPropertyName("author")]
-    public string Author { get; set; }
+    public string? Author { get; init; }
 
     [JsonPropertyName("author_name")]
-    public string AuthorName { get; set; }
+    public string? AuthorName { get; init; }
 
     [JsonPropertyName("nickname")]
-    public string Nickname { get; set; }
+    public string? Nickname { get; init; }
 
     [JsonPropertyName("content")]
-    public string Content { get; set; }
+    public string? Content { get; init; }
 
     [JsonPropertyName("time")]
-    public string Time { get; set; }
+    public string? Time { get; init; }
 
     [JsonPropertyName("channel")]
-    public string Channel { get; set; }
-}
+    public string? Channel { get; init; }
+    
+    [JsonPropertyName("sticker_id")]
+    public string? StickerId { get; init; }
+    
+    [JsonPropertyName("sticker_name")]
+    public string? StickerName { get; init; }
 
+    [JsonIgnore]
+    public string? ChannelName { get; set; }
+}
 
 
 public sealed class Plugin : IDalamudPlugin
 {
+    private int[] nameColor =
+    {
+        45, 517, 704, 708, 52, 61
+    };
     [PluginService] internal static IDalamudPluginInterface PluginInterface { get; private set; } = null!;
     [PluginService] internal static ICommandManager CommandManager { get; private set; } = null!;
     [PluginService] internal static IChatGui ChatGui { get; private set; } = null!;
@@ -47,25 +65,31 @@ public sealed class Plugin : IDalamudPlugin
     private const string CommandName = "/pdiscordtoxiv";
     private const int DefaultPort = 8765;
 
-    private WebSocketServer _webSocketServer;
-    private CancellationTokenSource _cancellationTokenSource;
-    private List<IWebSocketConnection> _connectedClients;
+    private WebSocketServer webSocketServer;
+    private readonly CancellationTokenSource cancellationTokenSource;
+    private readonly List<IWebSocketConnection> connectedClients;
+    private readonly HashSet<string> recentMessages = new HashSet<string>();
+    private readonly Queue<string> messageQueue = new Queue<string>();
+    private const int MaxRecentMessages = 100;
+    private readonly object messageLock = new object();
     public Configuration Configuration { get; init; }
     private ConfigWindow ConfigWindow { get; init; }
-    public readonly WindowSystem WindowSystem = new("SamplePlugin");
+    public readonly WindowSystem WindowSystem = new("DiscordToXIV");
+    
 
     public Plugin()
     {
         Configuration = PluginInterface.GetPluginConfig() as Configuration ?? new Configuration();
         ConfigWindow = new ConfigWindow(this);
         WindowSystem.AddWindow(ConfigWindow);
-        _cancellationTokenSource = new CancellationTokenSource();
-        _connectedClients = new List<IWebSocketConnection>();
+        cancellationTokenSource = new CancellationTokenSource();
+        connectedClients = new List<IWebSocketConnection>();
 
         CommandManager.AddHandler(CommandName, new CommandInfo(OnCommand)
         {
             HelpMessage = "Start WebSocket server. Usage: /pdiscordtoxiv [port]"
         });
+        PluginInterface.UiBuilder.Draw += DrawUI;
         PluginInterface.UiBuilder.OpenConfigUi += ToggleConfigUI;
         ChatGui.Print("Plugin starting...");
     }
@@ -98,50 +122,112 @@ public sealed class Plugin : IDalamudPlugin
         ChatGui.Print($"WebSocket server started on port {port}");
     }
 
-    private void StartWebSocketServer(int port)
+private void StartWebSocketServer(int port)
+{
+    StopWebSocketServer();
+
+    webSocketServer = new WebSocketServer($"ws://0.0.0.0:{port}");
+
+    webSocketServer.Start(socket =>
     {
-        StopWebSocketServer();
+        socket.OnOpen = () =>
+        {
+            PluginLog.Information("WebSocket connection opened.");
+            connectedClients.Add(socket);
+        };
 
-        _webSocketServer = new WebSocketServer($"ws://0.0.0.0:{port}");
+        socket.OnClose = () =>
+        {
+            PluginLog.Information("WebSocket connection closed.");
+            connectedClients.Remove(socket);
+        };
 
-        _webSocketServer.Start(socket =>
+        socket.OnMessage = message =>
         {
-            socket.OnOpen = () =>
-            {
-                PluginLog.Information("WebSocket connection opened.");
-                _connectedClients.Add(socket);
-            };
-            
-            socket.OnClose = () =>
-            {
-                PluginLog.Information("WebSocket connection closed.");
-                _connectedClients.Remove(socket);
-            };
-            
-            socket.OnMessage = message =>
+            try
             {
-                try
+                var receivedMessage = JsonSerializer.Deserialize<Message>(message);
+
+                if (receivedMessage == null || string.IsNullOrEmpty(receivedMessage.Id))
+                {
+                    PluginLog.Error("Received message without an ID.");
+                    return;
+                }
+
+                lock (messageLock)
                 {
-                    Message receivedMessage = JsonSerializer.Deserialize<Message>(message);
-                    //PluginLog.Information($"Message from {receivedMessage.Nickname}: {receivedMessage.Content}");
-                    ChatGui.Print($"[{receivedMessage.AuthorName}] {receivedMessage.Nickname}: {receivedMessage.Content}");
+                    if (recentMessages.Contains(receivedMessage.Id))
+                    {
+                        PluginLog.Information("Duplicate message detected by ID, skipping.");
+                        return;
+                    }
+
+                    recentMessages.Add(receivedMessage.Id);
+                    messageQueue.Enqueue(receivedMessage.Id);
+
+                    if (messageQueue.Count > MaxRecentMessages)
+                    {
+                        string oldMessageId = messageQueue.Dequeue();
+                        recentMessages.Remove(oldMessageId);
+                    }
                 }
-                catch (Exception ex)
+                var seString = new SeString(new List<Payload>());
+                var name = receivedMessage.Author;
+                ushort nameColor = (ushort)GetNameColor(name);
+                if (receivedMessage.Nickname != null)
                 {
-                    PluginLog.Error($"Failed to deserialize message: {ex.Message}");
+                    if (!Configuration.HideUsernameWhenNicknameExists)
+                        name = $"{receivedMessage.Nickname} ({receivedMessage.AuthorName})";
+                    else
+                        name = receivedMessage.Nickname;
+                }
+                else if (receivedMessage.AuthorName != null)
+                {
+                    name = receivedMessage.AuthorName;
                 }
-            };
 
-        });
-    }
+                receivedMessage.ChannelName = Configuration.ChannelMappings.TryGetValue(receivedMessage.Channel, out var channelName)
+                    ? channelName
+                    : receivedMessage.Channel;
+
+                seString.Append(new UIForegroundPayload(35));
+                seString.Append($"[{receivedMessage.ChannelName}] ");
+                seString.Append(UIForegroundPayload.UIForegroundOff);
+                seString.Append(new UIForegroundPayload(nameColor));
+                seString.Append(name);
+                seString.Append(UIForegroundPayload.UIForegroundOff);
+                seString.Append(new UIForegroundPayload(1));
+                seString.Append($": {receivedMessage.Content}");
+                var stickerData = "";
+                if (receivedMessage.StickerId != null)
+                    stickerData = $" [{receivedMessage.StickerName}](https://media.discordapp.net/stickers/{receivedMessage.StickerId}.webp?size=160&quality=lossless)";
+                if(stickerData == "" && receivedMessage.Content == null)
+                    return;
+                if (stickerData != "")
+                {
+                    seString.Append(new UIForegroundPayload(25));
+                    seString.Append(stickerData);
+                }
+                seString.Append(UIForegroundPayload.UIForegroundOff);
+
+                ChatGui.Print(seString);
+            }
+            catch (Exception ex)
+            {
+                PluginLog.Error($"Failed to process message: {ex.Message}");
+            }
+        };
+    });
+}
+
 
     private void StopWebSocketServer()
     {
-        if (_webSocketServer != null)
+        if (webSocketServer != null)
         {
             PluginLog.Information("Stopping WebSocket server...");
 
-            foreach (var socket in _connectedClients)
+            foreach (var socket in connectedClients)
             {
                 if (socket.IsAvailable)
                 {
@@ -149,10 +235,17 @@ public sealed class Plugin : IDalamudPlugin
                 }
             }
 
-            _webSocketServer.Dispose();
-            _cancellationTokenSource.Cancel();
+            webSocketServer.Dispose();
+            cancellationTokenSource.Cancel();
             PluginLog.Information("WebSocket server stopped.");
         }
     }
+    
+    private int GetNameColor(string name)
+    {
+        var index = Math.Abs(name.GetHashCode()) % nameColor.Length;
+        return nameColor[index];
+    }
     public void ToggleConfigUI() => ConfigWindow.Toggle();
+    private void DrawUI() => WindowSystem.Draw();
 }