implement handling mentions, attachments, and stickers

1 files changed, 133 insertions, 133 deletions

diff --git a/DiscordToXIV/Plugin.cs b/DiscordToXIV/Plugin.cs
index c73ea58..2e40bba 100644
--- a/DiscordToXIV/Plugin.cs
+++ b/DiscordToXIV/Plugin.cs
@@ -7,50 +7,16 @@ using Dalamud.Plugin.Services;
 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;
-
+using DiscordTOXIV.Windows;
 
 
 namespace DiscordToXIV;
 
 
-public class Message
-{
-    [JsonPropertyName("id")]
-    public string? Id { get; init; }
-    [JsonPropertyName("author")]
-    public string? Author { get; init; }
-
-    [JsonPropertyName("author_name")]
-    public string? AuthorName { get; init; }
-
-    [JsonPropertyName("nickname")]
-    public string? Nickname { get; init; }
-
-    [JsonPropertyName("content")]
-    public string? Content { get; init; }
-
-    [JsonPropertyName("time")]
-    public string? Time { get; init; }
-
-    [JsonPropertyName("channel")]
-    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 =
@@ -64,16 +30,14 @@ public sealed class Plugin : IDalamudPlugin
 
     private const string CommandName = "/pdiscordtoxiv";
     private const int DefaultPort = 8765;
+    private bool IsWebSocketServerRunning = false;
 
     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; }
+    private MainWindow MainWindow { get; init; }
     public readonly WindowSystem WindowSystem = new("DiscordToXIV");
     
 
@@ -81,7 +45,9 @@ public sealed class Plugin : IDalamudPlugin
     {
         Configuration = PluginInterface.GetPluginConfig() as Configuration ?? new Configuration();
         ConfigWindow = new ConfigWindow(this);
+        MainWindow = new MainWindow(this);
         WindowSystem.AddWindow(ConfigWindow);
+        WindowSystem.AddWindow(MainWindow);
         cancellationTokenSource = new CancellationTokenSource();
         connectedClients = new List<IWebSocketConnection>();
 
@@ -91,7 +57,9 @@ public sealed class Plugin : IDalamudPlugin
         });
         PluginInterface.UiBuilder.Draw += DrawUI;
         PluginInterface.UiBuilder.OpenConfigUi += ToggleConfigUI;
-        ChatGui.Print("Plugin starting...");
+        PluginInterface.UiBuilder.OpenMainUi += ToggleMainUI;
+        if(!Configuration.HideWelcomeMessage)
+            ChatGui.Print("[DiscordToXIV] Websocket Server Ready! Use /pdiscordtoxiv <port> to start the server (Need help? /pdiscordtoxiv help)");
     }
 
     public void Dispose()
@@ -108,123 +76,158 @@ public sealed class Plugin : IDalamudPlugin
             if(args == "stop")
             {
                 StopWebSocketServer();
-                ChatGui.Print("WebSocket server stop requested");
-                return;
+                ChatGui.Print("[DiscordToXIV] WebSocket server stop requested");
+            }
+            else if (args == "start")
+            {
+                ChatGui.Print("[DiscordToXIV] Starting WebSocket server with default port " + DefaultPort);
+                StartWebSocketServer(DefaultPort);
             }
-            if (!int.TryParse(args, out port) || port <= 0 || port > 65535)
+            else if (args == "help")
+                ToggleMainUI();
+            else if(args == "config")
+                ToggleConfigUI();
+            else if (!int.TryParse(args, out port) || port <= 0 || port > 65535)
             {
-                ChatGui.PrintError($"Invalid port number: {args}. Using default port {DefaultPort}.");
-                port = DefaultPort;
+                ChatGui.PrintError($"[DiscordToXIV] Invalid port number: {args}. Using default port {DefaultPort}.");
+                StartWebSocketServer(DefaultPort);
             }
+            else
+            {
+                ToggleMainUI(); 
+            }
+            return;
+        }
+        
+        // No args provided
+        if (!IsWebSocketServerRunning)
+        {
+            ChatGui.Print("[DiscordToXIV] Starting WebSocket server with default port " + DefaultPort);
+            StartWebSocketServer(DefaultPort);
+        }
+        else
+        {
+            ChatGui.Print("[DiscordToXIV] Stopping WebSocket server...");
+            StopWebSocketServer();
         }
 
-        StartWebSocketServer(port);
-        ChatGui.Print($"WebSocket server started on port {port}");
     }
 
-private void StartWebSocketServer(int port)
-{
-    StopWebSocketServer();
-
-    webSocketServer = new WebSocketServer($"ws://0.0.0.0:{port}");
-
-    webSocketServer.Start(socket =>
+    private void StartWebSocketServer(int port)
     {
-        socket.OnOpen = () =>
-        {
-            PluginLog.Information("WebSocket connection opened.");
-            connectedClients.Add(socket);
-        };
-
-        socket.OnClose = () =>
+        StopWebSocketServer();
+        IsWebSocketServerRunning = true;
+        webSocketServer = new WebSocketServer($"ws://0.0.0.0:{port}");
+        webSocketServer.Start(socket =>
         {
-            PluginLog.Information("WebSocket connection closed.");
-            connectedClients.Remove(socket);
-        };
+            socket.OnOpen = () =>
+            {
+                PluginLog.Information("[DiscordToXIV] WebSocket connection opened.");
+                ChatGui.Print("[DiscordToXIV] Connected to BetterDiscord Relayer!");
+                connectedClients.Add(socket);
+            };
 
-        socket.OnMessage = message =>
-        {
-            try
+            socket.OnClose = () =>
             {
-                var receivedMessage = JsonSerializer.Deserialize<Message>(message);
+                PluginLog.Information("[DiscordToXIV] WebSocket connection closed.");
+                ChatGui.Print("[DiscordToXIV] Disconnected from BetterDiscord Relayer!");
+                connectedClients.Remove(socket);
+            };
 
-                if (receivedMessage == null || string.IsNullOrEmpty(receivedMessage.Id))
+            socket.OnMessage = message =>
+            {
+                try
                 {
-                    PluginLog.Error("Received message without an ID.");
-                    return;
-                }
+                    var receivedMessage = JsonSerializer.Deserialize<Message>(message);
 
-                lock (messageLock)
-                {
-                    if (recentMessages.Contains(receivedMessage.Id))
+                    if (receivedMessage == null || string.IsNullOrEmpty(receivedMessage.Id))
                     {
-                        PluginLog.Information("Duplicate message detected by ID, skipping.");
+                        PluginLog.Error("Received message without an ID.");
                         return;
                     }
+                    
 
-                    recentMessages.Add(receivedMessage.Id);
-                    messageQueue.Enqueue(receivedMessage.Id);
-
-                    if (messageQueue.Count > MaxRecentMessages)
+                    var seString = new SeString(new List<Payload>());
+                    var name = receivedMessage.Author;
+                    ushort nameColor = (ushort)ChatUtils.GetNameColor(name, this.nameColor);
+                    if (receivedMessage.Nickname != null)
                     {
-                        string oldMessageId = messageQueue.Dequeue();
-                        recentMessages.Remove(oldMessageId);
+                        if (!Configuration.HideUsernameWhenNicknameExists)
+                            name = $"{receivedMessage.Nickname} ({receivedMessage.AuthorName})";
+                        else
+                            name = receivedMessage.Nickname;
                     }
-                }
-                var seString = new SeString(new List<Payload>());
-                var name = receivedMessage.Author;
-                ushort nameColor = (ushort)GetNameColor(name);
-                if (receivedMessage.Nickname != null)
-                {
-                    if (!Configuration.HideUsernameWhenNicknameExists)
-                        name = $"{receivedMessage.Nickname} ({receivedMessage.AuthorName})";
+                    else if (receivedMessage.AuthorName != null)
+                    {
+                        name = receivedMessage.AuthorName;
+                    }   
+                    
+                    if(Configuration.ChannelMappings.TryGetValue(receivedMessage.Channel, out var channelName))
+                        receivedMessage.ChannelName = channelName;
                     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;
+                    {
+                        if (Configuration.ShowOnlyKnownChannels) return;
+                        receivedMessage.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(56));
+                    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));
+                    var messageContent = receivedMessage.Content;
+                    if (Configuration.AdjustEmoteText)
+                        messageContent = ChatUtils.FixEmoteText(messageContent);
+                    if (Configuration.AdjustMentions)
+                        messageContent = ChatUtils.ReplaceMentionsWithNames(messageContent, receivedMessage.Mentions);
+                    seString.Append($": {messageContent}");
+                    var stickerData = "";
+                    if (receivedMessage.StickerId != null)
+                        if (!Configuration.HideStickerUrls)
+                        {
+                            stickerData = $" [{receivedMessage.StickerName}](https://media.discordapp.net/stickers/{receivedMessage.StickerId}.webp?size=160&quality=lossless)";
+                        }
+                        else
+                        {
+                            stickerData = $" [{receivedMessage.StickerName}]";
+                        }
+                    var attachments = receivedMessage.Attachments;
+                    if (attachments != null)
+                    {
+                        foreach (var attachment in attachments)
+                        {
+                            seString.Append(new UIForegroundPayload(25));
+                            if (!Configuration.HideAttachmentUrls)
+                            {
+                                seString.Append($"\nAttachment:\n[{attachment.Filename}]({attachment.Url})");
+                            }
+                            else
+                            {
+                                seString.Append($"\nAttachment:\n{attachment.Filename}");
+                            }
+                            seString.Append(UIForegroundPayload.UIForegroundOff);
+                        }
+                    }
                     seString.Append(new UIForegroundPayload(25));
                     seString.Append(stickerData);
+                    if(stickerData == "" && receivedMessage.Content == null) return;
+                    seString.Append(UIForegroundPayload.UIForegroundOff);
+                    ChatGui.Print(seString);
                 }
-                seString.Append(UIForegroundPayload.UIForegroundOff);
-
-                ChatGui.Print(seString);
-            }
-            catch (Exception ex)
-            {
-                PluginLog.Error($"Failed to process message: {ex.Message}");
-            }
-        };
-    });
-}
-
-
+                catch (Exception ex)
+                {
+                    PluginLog.Error($"Failed to process message: {ex.Message}");
+                }
+            };
+        });
+    }
     private void StopWebSocketServer()
     {
         if (webSocketServer != null)
         {
+            IsWebSocketServerRunning = false;
             PluginLog.Information("Stopping WebSocket server...");
 
             foreach (var socket in connectedClients)
@@ -241,11 +244,8 @@ private void StartWebSocketServer(int port)
         }
     }
     
-    private int GetNameColor(string name)
-    {
-        var index = Math.Abs(name.GetHashCode()) % nameColor.Length;
-        return nameColor[index];
-    }
+
     public void ToggleConfigUI() => ConfigWindow.Toggle();
+    public void ToggleMainUI() => MainWindow.Toggle();
     private void DrawUI() => WindowSystem.Draw();
 }