RegexBot/RegexBot.cs

using Discord;
using Discord.WebSocket;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;

namespace Noikoio.RegexBot
{
    /// <summary>
    /// Main class. On start, initializes bot modules and passes the DiscordSocketClient to them
    /// </summary>
    class RegexBot
    {
        private static Configuration _config;
        private readonly DiscordSocketClient _client;
        private BotModule[] _modules;

        internal static Configuration Config => _config;
        internal IEnumerable<BotModule> Modules => _modules;
        
        internal RegexBot()
        {
            // Load configuration
            _config = new Configuration(this);
            if (!_config.LoadInitialConfig())
            {
#if DEBUG
                Console.WriteLine("Press any key to exit.");
                Console.ReadKey();
#endif
                Environment.Exit(1);
            }

            _client = new DiscordSocketClient(new DiscordSocketConfig()
            {
                LogLevel = LogSeverity.Info,
                AlwaysDownloadUsers = true,
                DefaultRetryMode = RetryMode.AlwaysRetry,
                MessageCacheSize = 0
            });

            // Hook up handlers for basic functions
            _client.Connected += _client_Connected;

            // Initialize modules
            _modules = new BotModule[]
            {
                new Module.DMLogger.DMLogger(_client),
                new Module.AutoMod.AutoMod(_client),
                new Module.ModTools.ModTools(_client),
                new Module.AutoRespond.AutoRespond(_client),
                new Module.EntityCache.EntityCache(_client) // EntityCache goes before anything else that uses its data
            };
            var dlog = Logger.GetLogger("Discord.Net");
            _client.Log += async (arg) =>
                await dlog(
                    String.Format("{0}: {1}{2}", arg.Source, ((int)arg.Severity < 3 ? arg.Severity + ": " : ""),
                    arg.Message));

            // With modules initialized, finish loading configuration
            var conf = _config.ReloadServerConfig().Result;
            if (conf == false)
            {
                Console.WriteLine("Failed to load server configuration.");
#if DEBUG
                Console.WriteLine("Press any key to exit.");
                Console.ReadKey();
#endif
                Environment.Exit(1);
            }
        }
        
        internal async Task Start()
        {
            
            await _client.LoginAsync(TokenType.Bot, Config.BotUserToken);
            await _client.StartAsync();
            
            await Task.Delay(-1);
        }

        private async Task _client_Connected() => await _client.SetGameAsync(Config.CurrentGame);

        // Defined within this class because a reference to the client is required
        public void Console_CancelKeyPress(object sender, ConsoleCancelEventArgs e)
        {
            e.Cancel = true;
            Logger.GetLogger("Runtime")("Caught cancel key. Will attempt to disconnect...").Wait();
            _client.LogoutAsync().Wait();
            _client.Dispose();
#if DEBUG
            Console.WriteLine("Press enter to exit.");
            Console.ReadLine();
#endif
            Environment.Exit(0);
        }
    }
}