add ai commands

Services/BotService.cs

@@ -10,6 +10,7 @@ public class BotService
 {
     private readonly DiscordSocketClient _client;
     private readonly InteractionService _interactionService;
+    private readonly LlmService _llmService;
     private readonly ILogger<BotService> _logger;
     private readonly IConfiguration _configuration;
     private readonly IServiceProvider _serviceProvider;
@@ -17,12 +18,14 @@ public class BotService
     public BotService(
         DiscordSocketClient client,
         InteractionService interactionService,
+        LlmService llmService,
         ILogger<BotService> logger,
         IConfiguration configuration,
         IServiceProvider serviceProvider)
     {
         _client = client;
         _interactionService = interactionService;
+        _llmService = llmService;
         _logger = logger;
         _configuration = configuration;
         _serviceProvider = serviceProvider;
@@ -35,6 +38,7 @@ public class BotService
 
         _client.Ready += ClientReadyAsync;
         _client.InteractionCreated += InteractionCreatedAsync;
+        _client.MessageReceived += MessageReceivedAsync;
 
         var token = Environment.GetEnvironmentVariable("DISCORD_BOT_TOKEN");
         if (string.IsNullOrWhiteSpace(token))
@@ -87,6 +91,105 @@ public class BotService
         }
     }
 
+    private async Task MessageReceivedAsync(SocketMessage socketMessage)
+    {
+        if (socketMessage is not SocketUserMessage message)
+            return;
+
+        if (message.Source != MessageSource.User || message.Author.IsBot || message.Author.IsWebhook)
+            return;
+
+        if (_client.CurrentUser is null)
+            return;
+
+        var botId = _client.CurrentUser.Id;
+        var hasMention = message.MentionedUsers.Any(u => u.Id == botId);
+        var isReplyToBot = false;
+
+        if (message.Reference?.MessageId.IsSpecified == true)
+        {
+            try
+            {
+                var referenced = message.ReferencedMessage ??
+                    await message.Channel.GetMessageAsync(message.Reference.MessageId.Value);
+
+                isReplyToBot = referenced?.Author.Id == botId;
+            }
+            catch (Exception ex)
+            {
+                _logger.LogDebug(ex, "Unable to resolve referenced message for reply detection.");
+            }
+        }
+
+        if (!hasMention && !isReplyToBot)
+            return;
+
+        var prompt = hasMention ? RemoveBotMentions(message.Content, botId) : message.Content.Trim();
+
+        if (string.IsNullOrWhiteSpace(prompt))
+        {
+            await message.ReplyAsync("Please include a prompt after mentioning me.");
+            return;
+        }
+
+        try
+        {
+            var accumulated = string.Empty;
+
+            await foreach (var text in _llmService.SendMessageAsync(message.Channel.Id, message.Author.Username, prompt))
+            {
+                accumulated = text;
+            }
+
+            if (string.IsNullOrWhiteSpace(accumulated))
+            {
+                await message.ReplyAsync("The model returned an empty response.");
+                return;
+            }
+
+            await SendChunkedReplyAsync(message, accumulated);
+        }
+        catch (Exception ex)
+        {
+            _logger.LogError(ex, "Error handling mention/reply LLM chat in channel {ChannelId}", message.Channel.Id);
+            await message.ReplyAsync("I hit an error while generating a response.");
+        }
+    }
+
+    private static async Task SendChunkedReplyAsync(SocketUserMessage message, string response)
+    {
+        const int maxChunkLength = 2000;
+
+        for (var start = 0; start < response.Length; start += maxChunkLength)
+        {
+            var length = Math.Min(maxChunkLength, response.Length - start);
+            var chunk = response.Substring(start, length);
+
+            if (start == 0)
+            {
+                await message.ReplyAsync(chunk);
+            }
+            else
+            {
+                await message.Channel.SendMessageAsync(chunk);
+            }
+        }
+    }
+
+    private static string RemoveBotMentions(string content, ulong botId)
+    {
+        if (string.IsNullOrWhiteSpace(content))
+            return string.Empty;
+
+        var rawMention = $"<@{botId}>";
+        var nickMention = $"<@!{botId}>";
+
+        return content
+            .Replace(rawMention, string.Empty, StringComparison.Ordinal)
+            .Replace(nickMention, string.Empty, StringComparison.Ordinal)
+            .Trim();
+    }
+
     private static Task LogAsync(LogMessage message)
     {
         Console.WriteLine($"[{message.Severity}] {message.Source}: {message.Message}");