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

using System;
using System.Collections.Generic;
using System.Linq;
using MinecraftClient.Inventory;
using Tomlet.Attributes;
using static MinecraftClient.ChatBots.AutoDrop.Configs;

namespace MinecraftClient.ChatBots
{
    public class AutoDrop : ChatBot
    {
        public static Configs Config = new();

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

            public bool Enabled = false;

            [TomlInlineComment("$config.ChatBot.AutoDrop.Mode$")]
            public DropMode Mode = DropMode.include;

            public List<ItemType> Items = new() { ItemType.Cobblestone, ItemType.Dirt };

            public void OnSettingUpdate() { }

            public enum DropMode
            {
                include,    // Items in list will be dropped
                exclude,    // Items in list will be kept
                everything  // Everything will be dropped
            }
        }

        private int updateDebounce = 0;
        private readonly int updateDebounceValue = 2;
        private int inventoryUpdated = -1;

        public string CommandHandler(string cmd, string[] args)
        {
            if (args.Length > 0)
            {
                switch (args[0].ToLower())
                {
                    case "on":
                        Config.Enabled = true;
                        inventoryUpdated = 0;
                        OnUpdateFinish();
                        return Translations.Get("bot.autoDrop.on");
                    case "off":
                        Config.Enabled = false;
                        return Translations.Get("bot.autoDrop.off");
                    case "add":
                        if (args.Length >= 2)
                        {
                            if (Enum.TryParse(args[1], true, out ItemType item))
                            {
                                Config.Items.Add(item);
                                return Translations.Get("bot.autoDrop.added", item.ToString());
                            }
                            else
                            {
                                return Translations.Get("bot.autoDrop.incorrect_name", args[1]);
                            }
                        }
                        else
                        {
                            return Translations.Get("cmd.inventory.help.usage") + ": add <item name>";
                        }
                    case "remove":
                        if (args.Length >= 2)
                        {
                            if (Enum.TryParse(args[1], true, out ItemType item))
                            {
                                if (Config.Items.Contains(item))
                                {
                                    Config.Items.Remove(item);
                                    return Translations.Get("bot.autoDrop.removed", item.ToString());
                                }
                                else
                                {
                                    return Translations.Get("bot.autoDrop.not_in_list");
                                }
                            }
                            else
                            {
                                return Translations.Get("bot.autoDrop.incorrect_name", args[1]);
                            }
                        }
                        else
                        {
                            return Translations.Get("cmd.inventory.help.usage") + ": remove <item name>";
                        }
                    case "list":
                        if (Config.Items.Count > 0)
                        {
                            return Translations.Get("bot.autoDrop.list", Config.Items.Count, string.Join("\n", Config.Items));
                        }
                        else
                        {
                            return Translations.Get("bot.autoDrop.no_item");
                        }
                    case "mode":
                        if (args.Length >= 2)
                        {
                            switch (args[1].ToLower())
                            {
                                case "include":
                                    Config.Mode = DropMode.include;
                                    break;
                                case "exclude":
                                    Config.Mode = DropMode.exclude;
                                    break;
                                case "everything":
                                    Config.Mode = DropMode.everything;
                                    break;
                                default:
                                    return Translations.Get("bot.autoDrop.unknown_mode"); // Unknwon mode. Available modes: Include, Exclude, Everything
                            }
                            inventoryUpdated = 0;
                            OnUpdateFinish();
                            return Translations.Get("bot.autoDrop.switched", Config.Mode.ToString()); // Switched to {0} mode.
                        }
                        else
                        {
                            return Translations.Get("bot.autoDrop.unknown_mode");
                        }
                    default:
                        return GetHelp();
                }
            }
            else
            {
                return GetHelp();
            }
        }

        private static string GetHelp()
        {
            return Translations.Get("general.available_cmd", "on, off, add, remove, list, mode");
        }

        public override void Initialize()
        {
            if (!GetInventoryEnabled())
            {
                LogToConsoleTranslated("extra.inventory_required");
                LogToConsoleTranslated("general.bot_unload");
                UnloadBot();
                return;
            }
            RegisterChatBotCommand("autodrop", Translations.Get("bot.autoDrop.cmd"), GetHelp(), CommandHandler);
            RegisterChatBotCommand("ad", Translations.Get("bot.autoDrop.alias"), GetHelp(), CommandHandler);
        }

        public override void Update()
        {
            if (updateDebounce > 0)
            {
                updateDebounce--;
                if (updateDebounce <= 0)
                {
                    OnUpdateFinish();
                }
            }
        }

        public override void OnInventoryUpdate(int inventoryId)
        {
            if (Config.Enabled)
            {
                updateDebounce = updateDebounceValue;
                // Always interact container if available (larger ID) because they included player inventory (ID 0)
                if (inventoryId >= inventoryUpdated)
                    inventoryUpdated = inventoryId;
            }
        }

        private void OnUpdateFinish()
        {
            if (inventoryUpdated != -1)
            {
                if (!GetInventories().ContainsKey(inventoryUpdated))
                {
                    // Inventory updated but no inventory ?
                    LogDebugToConsoleTranslated("bot.autoDrop.no_inventory", inventoryUpdated);
                    return;
                }
                var inventory = GetInventories()[inventoryUpdated];
                var items = inventory.Items.ToDictionary(entry => entry.Key, entry => entry.Value);
                if (Config.Mode == DropMode.include)
                {
                    foreach (var item in items)
                    {
                        // Ingore crafting result slot
                        if (item.Key == 0)
                            continue;
                        if (Config.Items.Contains(item.Value.Type))
                        {
                            // Drop it !!
                            WindowAction(inventoryUpdated, item.Key, WindowActionType.DropItemStack);
                        }
                    }
                }
                else if (Config.Mode == DropMode.exclude)
                {
                    foreach (var item in items)
                    {
                        // Ingore crafting result slot
                        if (item.Key == 0)
                            continue;
                        if (!Config.Items.Contains(item.Value.Type))
                        {
                            // Drop it !!
                            WindowAction(inventoryUpdated, item.Key, WindowActionType.DropItemStack);
                        }
                    }
                }
                else
                {
                    foreach (var item in items)
                    {
                        // Ingore crafting result slot
                        if (item.Key == 0)
                            continue;
                        // Drop it !!
                        WindowAction(inventoryUpdated, item.Key, WindowActionType.DropItemStack);
                    }
                }
                inventoryUpdated = -1;
            }
        }
    }
}