Minecraft-Console-Client/MinecraftClient/ChatBots/DiscordBridge.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using DSharpPlus;
using DSharpPlus.Entities;
using DSharpPlus.Exceptions;
using Microsoft.Extensions.Logging;
using Tomlet.Attributes;

namespace MinecraftClient.ChatBots
{
    internal enum BridgeDirection
    {
        Both = 0,
        Minecraft,
        Discord
    }

    public class DiscordBridge : ChatBot
    {
        private static DiscordBridge? instance = null;
        public bool IsConnected { get; private set; }

        private DiscordClient? _client;
        private DiscordChannel? _channel;
        private BridgeDirection bridgeDirection = BridgeDirection.Both;

        public static Configs Config = new();

        [TomlDoNotInlineObject]
        public class Configs
        {
            [NonSerialized]
            private const string BotName = "DiscordBridge";

            public bool Enabled = false;

            [TomlInlineComment("$config.ChatBot.DiscordBridge.Token$")]
            public string Token = "your bot token here";

            [TomlInlineComment("$config.ChatBot.DiscordBridge.GuildId$")]
            public ulong GuildId = 1018553894831403028L;

            [TomlInlineComment("$config.ChatBot.DiscordBridge.ChannelId$")]
            public ulong ChannelId = 1018565295654326364L;

            [TomlInlineComment("$config.ChatBot.DiscordBridge.OwnersIds$")]
            public ulong[] OwnersIds = new[] { 978757810781323276UL };

            [TomlInlineComment("$config.ChatBot.DiscordBridge.MessageSendTimeout$")]
            public int Message_Send_Timeout = 3;

            [TomlPrecedingComment("$config.ChatBot.DiscordBridge.Formats$")]
            public string PrivateMessageFormat = "**[Private Message]** {username}: {message}";
            public string PublicMessageFormat = "{username}: {message}";
            public string TeleportRequestMessageFormat = "A new Teleport Request from **{username}**!";

            public void OnSettingUpdate()
            {
                Message_Send_Timeout = Message_Send_Timeout <= 0 ? 3 : Message_Send_Timeout;
            }
        }

        public DiscordBridge()
        {
            instance = this;
        }

        public override void Initialize()
        {
            RegisterChatBotCommand("dscbridge", "bot.DiscordBridge.desc", "dscbridge direction <both|mc|discord>", OnDscCommand);

            Task.Run(async () => await MainAsync());
        }

        ~DiscordBridge()
        {
            Disconnect();
        }

        public override void OnUnload()
        {
            Disconnect();
        }

        private void Disconnect()
        {
            if (_client != null)
            {
                try
                {
                    if (_channel != null)
                        _client.SendMessageAsync(_channel, new DiscordEmbedBuilder
                        {
                            Description = Translations.TryGet("bot.DiscordBridge.disconnected"),
                            Color = new DiscordColor(0xFF0000)
                        }).Wait(Config.Message_Send_Timeout * 1000);
                }
                catch (Exception e)
                {
                    LogToConsole("§w§l§f" + Translations.TryGet("bot.DiscordBridge.canceled_sending"));
                    LogDebugToConsole(e);
                }

                _client.DisconnectAsync().Wait();
                IsConnected = false;
            }
        }

        public static DiscordBridge? GetInstance()
        {
            return instance;
        }

        private string OnDscCommand(string cmd, string[] args)
        {
            if (args.Length == 2)
            {
                if (args[0].ToLower().Equals("direction"))
                {
                    string direction = args[1].ToLower().Trim();

                    string? bridgeName = "";

                    switch (direction)
                    {
                        case "b":
                        case "both":
                            bridgeName = "bot.DiscordBridge.direction.both";
                            bridgeDirection = BridgeDirection.Both;
                            break;

                        case "mc":
                        case "minecraft":
                            bridgeName = "bot.DiscordBridge.direction.minecraft";
                            bridgeDirection = BridgeDirection.Minecraft;
                            break;

                        case "d":
                        case "dcs":
                        case "discord":
                            bridgeName = "bot.DiscordBridge.direction.discord";
                            bridgeDirection = BridgeDirection.Discord;
                            break;

                        default:
                            return Translations.TryGet("bot.DiscordBridge.invalid_direction");
                    }

                    return Translations.TryGet("bot.DiscordBridge.direction", Translations.TryGet(bridgeName));
                };
            }

            return "dscbridge direction <both|mc|discord>";
        }

        public override void GetText(string text)
        {
            if (!CanSendMessages())
                return;

            text = GetVerbatim(text).Trim();

            // Stop the crash when an empty text is recived somehow
            if (string.IsNullOrEmpty(text))
                return;

            string message = "";
            string username = "";
            bool teleportRequest = false;

            if (IsPrivateMessage(text, ref message, ref username))
                message = Config.PrivateMessageFormat.Replace("{username}", username).Replace("{message}", message).Replace("{timestamp}", GetTimestamp()).Trim();
            else if (IsChatMessage(text, ref message, ref username))
                message = Config.PublicMessageFormat.Replace("{username}", username).Replace("{message}", message).Replace("{timestamp}", GetTimestamp()).Trim();
            else if (IsTeleportRequest(text, ref username))
            {
                message = Config.TeleportRequestMessageFormat.Replace("{username}", username).Replace("{timestamp}", GetTimestamp()).Trim();
                teleportRequest = true;
            }
            else message = text;

            if (teleportRequest)
            {
                var messageBuilder = new DiscordMessageBuilder()
                    .WithEmbed(new DiscordEmbedBuilder
                    {
                        Description = message,
                        Color = new DiscordColor(0x3399FF)
                    })
                    .AddComponents(new DiscordComponent[]{
                        new DiscordButtonComponent(ButtonStyle.Success, "accept_teleport", "Accept"),
                        new DiscordButtonComponent(ButtonStyle.Danger, "deny_teleport", "Deny")
                    });

                SendMessage(messageBuilder);
                return;
            }
            else SendMessage(message);
        }

        public void SendMessage(string message)
        {
            if (!CanSendMessages() || string.IsNullOrEmpty(message))
                return;

            try
            {
                _client!.SendMessageAsync(_channel, message).Wait(Config.Message_Send_Timeout * 1000);
            }
            catch (Exception e)
            {
                LogToConsole("§w§l§f" + Translations.TryGet("bot.DiscordBridge.canceled_sending"));
                LogDebugToConsole(e);
            }
        }

        public void SendMessage(DiscordMessageBuilder builder)
        {
            if (!CanSendMessages())
                return;

            try
            {
                _client!.SendMessageAsync(_channel, builder).Wait(Config.Message_Send_Timeout * 1000);
            }
            catch (Exception e)
            {
                LogToConsole("§w§l§f" + Translations.TryGet("bot.DiscordBridge.canceled_sending"));
                LogDebugToConsole(e);
            }
        }

        public void SendMessage(DiscordEmbedBuilder embedBuilder)
        {
            if (!CanSendMessages())
                return;

            try
            {
                _client!.SendMessageAsync(_channel, embedBuilder).Wait(Config.Message_Send_Timeout * 1000);
            }
            catch (Exception e)
            {
                LogToConsole("§w§l§f" + Translations.TryGet("bot.DiscordBridge.canceled_sending"));
                LogDebugToConsole(e);
            }
        }
        public void SendImage(string filePath, string? text = null)
        {
            if (!CanSendMessages())
                return;

            try
            {
                using (var fs = new FileStream(filePath, FileMode.Open, FileAccess.Read))
                {
                    filePath = filePath[(filePath.IndexOf(Path.DirectorySeparatorChar) + 1)..];
                    var messageBuilder = new DiscordMessageBuilder();

                    if (text != null)
                        messageBuilder.WithContent(text);

                    messageBuilder.WithFiles(new Dictionary<string, Stream>() { { $"attachment://{filePath}", fs } });

                    _client!.SendMessageAsync(_channel, messageBuilder).Wait(Config.Message_Send_Timeout * 1000);
                }
            }
            catch (Exception e)
            {
                LogToConsole("§w§l§f" + Translations.TryGet("bot.DiscordBridge.canceled_sending"));
                LogDebugToConsole(e);
            }
        }

        public void SendFile(FileStream fileStream)
        {
            if (!CanSendMessages())
                return;

            SendMessage(new DiscordMessageBuilder().WithFile(fileStream));
        }

        private bool CanSendMessages()
        {
            return _client == null || _channel == null || bridgeDirection == BridgeDirection.Minecraft ? false : true;
        }

        async Task MainAsync()
        {
            try
            {
                if (string.IsNullOrEmpty(Config.Token.Trim()))
                {
                    LogToConsole(Translations.TryGet("bot.DiscordBridge.missing_token"));
                    UnloadBot();
                    return;
                }

                _client = new DiscordClient(new DiscordConfiguration()
                {
                    Token = Config.Token.Trim(),
                    TokenType = TokenType.Bot,
                    AutoReconnect = true,
                    Intents = DiscordIntents.All,
                    MinimumLogLevel = Settings.Config.Logging.DebugMessages ?
                        (LogLevel.Trace | LogLevel.Information | LogLevel.Debug | LogLevel.Critical | LogLevel.Error | LogLevel.Warning) : LogLevel.None
                });

                try
                {
                    await _client.GetGuildAsync(Config.GuildId);
                }
                catch (Exception e)
                {
                    if (e is NotFoundException)
                    {
                        LogToConsole(Translations.TryGet("bot.DiscordBridge.guild_not_found", Config.GuildId));
                        UnloadBot();
                        return;
                    }

                    LogDebugToConsole("Exception when trying to find the guild:");
                    LogDebugToConsole(e);
                }

                try
                {
                    _channel = await _client.GetChannelAsync(Config.ChannelId);
                }
                catch (Exception e)
                {
                    if (e is NotFoundException)
                    {
                        LogToConsole(Translations.TryGet("bot.DiscordBridge.channel_not_found", Config.ChannelId));
                        UnloadBot();
                        return;
                    }

                    LogDebugToConsole("Exception when trying to find the channel:");
                    LogDebugToConsole(e);
                }

                _client.MessageCreated += async (source, e) =>
                {
                    if (e.Guild.Id != Config.GuildId)
                        return;

                    if (e.Channel.Id != Config.ChannelId)
                        return;

                    if (!Config.OwnersIds.Contains(e.Author.Id))
                        return;

                    string message = e.Message.Content.Trim();

                    if (string.IsNullOrEmpty(message) || string.IsNullOrWhiteSpace(message))
                        return;

                    if (bridgeDirection == BridgeDirection.Discord)
                    {
                        if (!message.StartsWith(".dscbridge"))
                            return;
                    }

                    if (message.StartsWith("."))
                    {
                        message = message[1..];
                        await e.Message.CreateReactionAsync(DiscordEmoji.FromName(_client, ":gear:"));

                        string? result = "";
                        PerformInternalCommand(message, ref result);
                        result = string.IsNullOrEmpty(result) ? "-" : result;

                        await e.Message.DeleteOwnReactionAsync(DiscordEmoji.FromName(_client, ":gear:"));
                        await e.Message.CreateReactionAsync(DiscordEmoji.FromName(_client, ":white_check_mark:"));
                        await e.Message.RespondAsync($"{Translations.TryGet("bot.DiscordBridge.command_executed")}:\n```{result}```");
                    }
                    else SendText(message);
                };

                _client.ComponentInteractionCreated += async (s, e) =>
                {
                    if (!(e.Id.Equals("accept_teleport") || e.Id.Equals("deny_teleport")))
                        return;

                    string result = e.Id.Equals("accept_teleport") ? "Accepted :white_check_mark:" : "Denied :x:";
                    SendText(e.Id.Equals("accept_teleport") ? "/tpaccept" : "/tpdeny");
                    await e.Interaction.CreateResponseAsync(InteractionResponseType.UpdateMessage, new DiscordInteractionResponseBuilder().WithContent(result));
                };

                await _client.ConnectAsync();

                await _client.SendMessageAsync(_channel, new DiscordEmbedBuilder
                {
                    Description = Translations.TryGet("bot.DiscordBridge.connected"),
                    Color = new DiscordColor(0x00FF00)
                });

                IsConnected = true;
                LogToConsole("§y§l§f" + Translations.TryGet("bot.DiscordBridge.connected"));
                await Task.Delay(-1);
            }
            catch (Exception e)
            {
                LogToConsole("§w§l§f" + Translations.TryGet("bot.DiscordBridge.unknown_error"));
                LogToConsole(e);
                return;
            }
        }
    }
}