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

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace MinecraftClient.ChatBots
{
    /// <summary>
    /// This bot automatically re-join the server if kick message contains predefined string (Server is restarting ...)
    /// </summary>
    public class AutoRelog : ChatBot
    {
        private string[] dictionary = new string[0];
        private int attempts;
        private int delay;

        /// <summary>
        /// This bot automatically re-join the server if kick message contains predefined string
        /// </summary>
        /// <param name="DelayBeforeRelog">Delay before re-joining the server (in seconds)</param>
        /// <param name="retries">Number of retries if connection fails (-1 = infinite)</param>
        public AutoRelog(int DelayBeforeRelog, int retries)
        {
            attempts = retries;
            if (attempts == -1) { attempts = int.MaxValue; }
            McTcpClient.ReconnectionAttemptsLeft = attempts;
            delay = DelayBeforeRelog;
            if (delay < 1) { delay = 1; }
            if (Settings.DebugMessages)
                LogToConsole("Launching with " + attempts + " reconnection attempts");
        }

        public override void Initialize()
        {
            McTcpClient.ReconnectionAttemptsLeft = attempts;
            if (Settings.AutoRelog_IgnoreKickMessage)
            {
                if (Settings.DebugMessages)
                    LogToConsole("Initializing without a kick message file");
            }
            else
            {
                if (System.IO.File.Exists(Settings.AutoRelog_KickMessagesFile))
                {
                    if (Settings.DebugMessages)
                        LogToConsole("Loading messages from file: " + System.IO.Path.GetFullPath(Settings.AutoRelog_KickMessagesFile));

                    dictionary = System.IO.File.ReadAllLines(Settings.AutoRelog_KickMessagesFile);

                    for (int i = 0; i < dictionary.Length; i++)
                    {
                        if (Settings.DebugMessages)
                            LogToConsole("  Loaded message: " + dictionary[i]);
                        dictionary[i] = dictionary[i].ToLower();
                    }
                }
                else
                {
                    LogToConsole("File not found: " + System.IO.Path.GetFullPath(Settings.AutoRelog_KickMessagesFile));

                    if (Settings.DebugMessages)
                        LogToConsole("  Current directory was: " + System.IO.Directory.GetCurrentDirectory());
                }
            }
        }

        public override bool OnDisconnect(DisconnectReason reason, string message)
        {
            if (reason == DisconnectReason.UserLogout)
            {
                if (Settings.DebugMessages)
                    LogToConsole("Disconnection initiated by User or MCC bot. Ignoring.");
            }
            else
            {
                message = GetVerbatim(message);
                string comp = message.ToLower();

                if (Settings.DebugMessages)
                    LogToConsole("Got disconnected with message: " + message);

                if (Settings.AutoRelog_IgnoreKickMessage)
                {
                    if (Settings.DebugMessages)
                        LogToConsole("Ignoring kick message, reconnecting anyway.");
                    ReconnectToTheServer();
                    return true;
                }

                foreach (string msg in dictionary)
                {
                    if (comp.Contains(msg))
                    {
                        if (Settings.DebugMessages)
                            LogToConsole("Message contains '" + msg + "'. Reconnecting.");

                        LogToConsole("Waiting " + delay + " seconds before reconnecting...");
                        System.Threading.Thread.Sleep(delay * 1000);
                        McTcpClient.ReconnectionAttemptsLeft = attempts;
                        ReconnectToTheServer();
                        return true;
                    }
                }

                if (Settings.DebugMessages)
                    LogToConsole("Message not containing any defined keywords. Ignoring.");
            }

            return false;
        }

        public static bool OnDisconnectStatic(DisconnectReason reason, string message)
        {
            if (Settings.AutoRelog_Enabled)
            {
                AutoRelog bot = new AutoRelog(Settings.AutoRelog_Delay, Settings.AutoRelog_Retries);
                bot.Initialize();
                return bot.OnDisconnect(reason, message);
            }
            return false;
        }
    }
}