Minecraft-Console-Client/MinecraftClient/Commands/Entitycmd.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using MinecraftClient.Inventory;
using MinecraftClient.Mapping;

namespace MinecraftClient.Commands
{
    class Entitycmd : Command
    {
        public override string CMDName { get { return "entity"; } }
        public override string CMDDesc { get { return "entity <id|entitytype> <attack|use>"; } }

        public override string Run(McClient handler, string command, Dictionary<string, object> localVars)
        {
            if (handler.GetEntityHandlingEnabled())
            {
                string[] args = getArgs(command);
                if (args.Length >= 1)
                {
                    try
                    {
                        int entityID = 0;
                        int.TryParse(args[0], out entityID);
                        if (entityID != 0)
                        {
                            if (handler.GetEntities().ContainsKey(entityID))
                            {
                                string action = args.Length > 1
                                    ? args[1].ToLower()
                                    : "list";
                                switch (action)
                                {
                                    case "attack":
                                        handler.InteractEntity(entityID, 1);
                                        return "Entity attacked";
                                    case "use":
                                        handler.InteractEntity(entityID, 0);
                                        return "Entity used";
                                    default:
                                        Entity entity = handler.GetEntities()[entityID];
                                        int id = entity.ID;
                                        float health = entity.Health;
                                        int latency = entity.Latency;
                                        Item item = entity.Item;
                                        string nickname = entity.Name;
                                        string customname = entity.CustomName;
                                        EntityPose pose = entity.Pose;
                                        EntityType type = entity.Type;
                                        double distance = Math.Round(entity.Location.Distance(handler.GetCurrentLocation()), 2);

                                        string color = "§a"; // Green
                                        if (health < 10)
                                            color = "§c";  // Red
                                        else if (health < 15)
                                            color = "§e";  // Yellow

                                        string location = String.Format("X:{0}, Y:{1}, Z:{2}", Math.Round(entity.Location.X, 2), Math.Round(entity.Location.Y, 2), Math.Round(entity.Location.Y, 2));
                                        string done = String.Format("Entity: {0}\n [MCC] Type: {1}", id, type);
                                        if (!String.IsNullOrEmpty(nickname))
                                            done += String.Format("\n [MCC] Nickname: {0}", nickname);
                                        else if (!String.IsNullOrEmpty(customname))
                                            done += String.Format("\n [MCC] CustomName: {0}§8", customname.Replace("&", "§"));
                                        if (type == EntityType.Player)
                                            done += String.Format("\n [MCC] Latency: {0}", latency);
                                        else if (type == EntityType.Item || type == EntityType.ItemFrame || type == Mapping.EntityType.EyeOfEnder || type == Mapping.EntityType.Egg || type == Mapping.EntityType.EnderPearl || type == Mapping.EntityType.Potion || type == Mapping.EntityType.Fireball || type == Mapping.EntityType.FireworkRocket)
                                        {
                                            string displayName = item.DisplayName;
                                            if (String.IsNullOrEmpty(displayName))
                                                done += String.Format("\n [MCC] Item: {0} x{1}", item.Type, item.Count);
                                            else
                                                done += String.Format("\n [MCC] Item: {0} x{1} - {2}§8", item.Type, item.Count, displayName);
                                        }

                                        if (entity.Equipment.Count >= 1 && entity.Equipment != null)
                                        {
                                            done += String.Format("\n [MCC] Equipment:");
                                            if (entity.Equipment.ContainsKey(0) && entity.Equipment[0] != null)
                                                done += String.Format("\n   [MCC] MainHand: {0} x{1}", entity.Equipment[0].Type, entity.Equipment[0].Count);
                                            if (entity.Equipment.ContainsKey(1) && entity.Equipment[1] != null)
                                                done += String.Format("\n   [MCC] OffHand: {0} x{1}", entity.Equipment[1].Type, entity.Equipment[1].Count);
                                            if (entity.Equipment.ContainsKey(5) && entity.Equipment[5] != null)
                                                done += String.Format("\n   [MCC] Helmet: {0} x{1}", entity.Equipment[5].Type, entity.Equipment[5].Count);
                                            if (entity.Equipment.ContainsKey(4) && entity.Equipment[4] != null)
                                                done += String.Format("\n   [MCC] Chestplate: {0} x{1}", entity.Equipment[4].Type, entity.Equipment[4].Count);
                                            if (entity.Equipment.ContainsKey(3) && entity.Equipment[3] != null)
                                                done += String.Format("\n   [MCC] Leggings: {0} x{1}", entity.Equipment[3].Type, entity.Equipment[3].Count);
                                            if (entity.Equipment.ContainsKey(2) && entity.Equipment[2] != null)
                                                done += String.Format("\n   [MCC] Boots: {0} x{1}", entity.Equipment[2].Type, entity.Equipment[2].Count);
                                        }
                                        done += String.Format("\n [MCC] Pose: {0}", pose);
                                        done += String.Format("\n [MCC] Health: {0}", color + health + "§8");
                                        done += String.Format("\n [MCC] Distance: {0}", distance);
                                        done += String.Format("\n [MCC] Location: {0}", location);
                                        return done;
                                }
                            }
                            else return "Entity not found";
                        }
                        else
                        {
                            EntityType interacttype = EntityType.Player;
                            Enum.TryParse(args[0], out interacttype);
                            string actionst = "Entity attacked";
                            int actioncount = 0;
                            foreach (var entity2 in handler.GetEntities())
                            {
                                if (entity2.Value.Type == interacttype)
                                {
                                    string action = args.Length > 1
                                    ? args[1].ToLower()
                                    : "list";
                                    if (action == "attack")
                                    {
                                        handler.InteractEntity(entity2.Key, 1);
                                        actionst = "Entity attacked";
                                        actioncount++;
                                    }
                                    else if (action == "use")
                                    {
                                        handler.InteractEntity(entity2.Key, 0);
                                        actionst = "Entity used";
                                        actioncount++;
                                    }
                                    else return CMDDesc;
                                }
                            }
                            return actioncount + " " + actionst;
                        }
                    }
                    catch (FormatException) { return CMDDesc; }
                }
                else
                {
                    Dictionary<int, Mapping.Entity> entities = handler.GetEntities();
                    List<string> response = new List<string>();
                    response.Add("Entities:");
                    foreach (var entity2 in entities)
                    {
                        int id = entity2.Key;
                        float health = entity2.Value.Health;
                        int latency = entity2.Value.Latency;
                        string nickname = entity2.Value.Name;
                        string customname = entity2.Value.CustomName;
                        EntityPose pose = entity2.Value.Pose;
                        EntityType type = entity2.Value.Type;
                        Item item = entity2.Value.Item;
                        string location = String.Format("X:{0}, Y:{1}, Z:{2}", Math.Round(entity2.Value.Location.X, 2), Math.Round(entity2.Value.Location.Y, 2), Math.Round(entity2.Value.Location.Y, 2));

                        if (type == EntityType.Item || type == EntityType.ItemFrame || type == Mapping.EntityType.EyeOfEnder || type == Mapping.EntityType.Egg || type == Mapping.EntityType.EnderPearl || type == Mapping.EntityType.Potion || type == Mapping.EntityType.Fireball || type == Mapping.EntityType.FireworkRocket)
                            response.Add(String.Format(" #{0}: Type: {1}, Item: {2}, Location: {3}", id, type, item.Type, location));
                        else if (type == Mapping.EntityType.Player && !String.IsNullOrEmpty(nickname))
                            response.Add(String.Format(" #{0}: Type: {1}, Nickname: §8{2}§8, Latency: {3}, Health: {4}, Pose: {5}, Location: {6}", id, type, nickname, latency, health, pose, location));
                        else if (type == Mapping.EntityType.Player && !String.IsNullOrEmpty(customname))
                            response.Add(String.Format(" #{0}: Type: {1}, CustomName: §8{2}§8, Latency: {3}, Health: {4}, Pose: {5}, Location: {6}", id, type, customname.Replace("&", "§"), latency, health, pose, location));
                        else
                            response.Add(String.Format(" #{0}: Type: {1}, Health: {2}, Location: {3}", id, type, health, location));
                    }
                    response.Add(CMDDesc);
                    return String.Join("\n", response);
                }
            }
            else return "Please enable entityhandling in config to use this command.";
        }
    }
}