NoCheatPlus/src/fr/neatmonster/nocheatplus/NoCheatPlus.java

package fr.neatmonster.nocheatplus;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

import net.minecraft.server.DedicatedServer;
import net.minecraft.server.EntityPlayer;
import net.minecraft.server.NetServerHandler;
import net.minecraft.server.NetworkManager;

import org.bukkit.Bukkit;
import org.bukkit.craftbukkit.CraftServer;
import org.bukkit.craftbukkit.entity.CraftPlayer;
import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.Listener;
import org.bukkit.event.player.PlayerJoinEvent;
import org.bukkit.event.player.PlayerLoginEvent;
import org.bukkit.plugin.PluginDescriptionFile;
import org.bukkit.plugin.java.JavaPlugin;

import fr.neatmonster.nocheatplus.checks.ExecuteActionsEvent;
import fr.neatmonster.nocheatplus.checks.Workarounds;
import fr.neatmonster.nocheatplus.checks.blockbreak.BlockBreakListener;
import fr.neatmonster.nocheatplus.checks.blockinteract.BlockInteractListener;
import fr.neatmonster.nocheatplus.checks.blockplace.BlockPlaceListener;
import fr.neatmonster.nocheatplus.checks.chat.ChatListener;
import fr.neatmonster.nocheatplus.checks.fight.FightListener;
import fr.neatmonster.nocheatplus.checks.inventory.InventoryListener;
import fr.neatmonster.nocheatplus.checks.moving.MovingListener;
import fr.neatmonster.nocheatplus.config.ConfPaths;
import fr.neatmonster.nocheatplus.config.ConfigManager;
import fr.neatmonster.nocheatplus.players.Permissions;
import fr.neatmonster.nocheatplus.utilities.LagMeasureTask;

/*
 * M"""""""`YM          MM'""""'YMM dP                           dP   MM"""""""`YM dP                   
 * M  mmmm.  M          M' .mmm. `M 88                           88   MM  mmmmm  M 88                   
 * M  MMMMM  M .d8888b. M  MMMMMooM 88d888b. .d8888b. .d8888b. d8888P M'        .M 88 dP    dP .d8888b. 
 * M  MMMMM  M 88'  `88 M  MMMMMMMM 88'  `88 88ooood8 88'  `88   88   MM  MMMMMMMM 88 88    88 Y8ooooo. 
 * M  MMMMM  M 88.  .88 M. `MMM' .M 88    88 88.  ... 88.  .88   88   MM  MMMMMMMM 88 88.  .88       88 
 * M  MMMMM  M `88888P' MM.     .dM dP    dP `88888P' `88888P8   dP   MM  MMMMMMMM dP `88888P' `88888P' 
 * MMMMMMMMMMM          MMMMMMMMMMM                                   MMMMMMMMMMMM                      
 */
/**
 * This is the main class of NoCheatPlus. The commands, events listeners and tasks are registered here.
 */
public class NoCheatPlus extends JavaPlugin implements Listener {
    public static long     time = System.currentTimeMillis();

    /** The listeners. */
    private List<Listener> listeners;

    /* (non-Javadoc)
     * @see org.bukkit.plugin.java.JavaPlugin#onDisable()
     */
    @Override
    public void onDisable() {
        final PluginDescriptionFile pdfFile = getDescription();

        // Stop the lag measuring task.
        LagMeasureTask.cancel();

        // Cleanup the configuration manager.
        ConfigManager.cleanup();

        // Just to be sure nothing gets left out.
        getServer().getScheduler().cancelTasks(this);

        // Tell the server administrator the we finished unloading NoCheatPlus.
        System.out.println("[NoCheatPlus] Version " + pdfFile.getVersion() + " is disabled.");
    }

    /* (non-Javadoc)
     * @see org.bukkit.plugin.java.JavaPlugin#onEnable()
     */
    @Override
    public void onEnable() {
        // Read the configuration files.
        ConfigManager.init(this);

        // List the events listeners.
        listeners = new ArrayList<Listener>();
        listeners.add(new BlockBreakListener());
        listeners.add(new BlockInteractListener());
        listeners.add(new BlockPlaceListener());
        listeners.add(new ChatListener());
        listeners.add(new FightListener());
        listeners.add(new InventoryListener());
        listeners.add(new MovingListener());
        listeners.add(new Workarounds());

        // Set up a task to monitor server lag.
        LagMeasureTask.start(this);

        // Register all listeners.
        for (final Listener listener : listeners)
            Bukkit.getPluginManager().registerEvents(listener, this);
        Bukkit.getPluginManager().registerEvents(this, this);

        // Register the commands handler.
        getCommand("nocheatplus").setExecutor(new CommandHandler(this));

        // Set the NetServerHandler of every player.
        for (final Player player : Bukkit.getOnlinePlayers())
            setCustomNetServerHandler(player);

        // Tell the server administrator that we finished loading NoCheatPlus now.
        System.out.println("[NoCheatPlus] Version " + getDescription().getVersion() + " is enabled.");
    }

    /**
     * This event handler is used to execute the actions when a violation is detected.
     * 
     * @param event
     *            the event handled
     */
    @EventHandler(
            priority = EventPriority.LOWEST)
    final void onExecuteActions(final ExecuteActionsEvent event) {
        event.executeActions();
    }
    
    /**
     * Setting the net server handler at the earliest possible point.
     * @param event
     */
    @EventHandler(
            priority = EventPriority.LOWEST)
    public void onPlayerLogin(final PlayerLoginEvent event) {
    	 // Set the NetServerHandler of the player.
        setCustomNetServerHandler(event.getPlayer());
    }

    /**
     * This event handler is used to send all the disabling messages to the client.
     * 
     * @param event
     *            the event handled
     */
    @EventHandler(
            priority = EventPriority.MONITOR)
    public void onPlayerJoin(final PlayerJoinEvent event) {
        final Player player = event.getPlayer();

        // Check if we allow all the client mods.
        final boolean allowAll = ConfigManager.getConfigFile().getBoolean(ConfPaths.MISCELLANEOUS_ALLOWCLIENTMODS);
        String message = "";

        // Allow Rei's Minimap's cave mode.
        if (allowAll || player.hasPermission(Permissions.REI_CAVE))
            message = message + "§0§0§1§e§f";

        // Allow Rei's Minimap's radar.
        if (allowAll || player.hasPermission(Permissions.REI_RADAR))
            message = message + "§0§0§2§3§4§5§6§7§e§f";

        // If all the client mods are allowed, no need to go any further.
        if (allowAll)
            return;

        // Disable Zombe's fly mod.
        if (!player.hasPermission(Permissions.ZOMBE_FLY))
            message = message + "§f §f §1 §0 §2 §4";

        // Disable Zombe's noclip.
        if (!player.hasPermission(Permissions.ZOMBE_NOCLIP))
            message = message + "§f §f §4 §0 §9 §6";

        // Disable Zombe's cheat.
        if (!player.hasPermission(Permissions.ZOMBE_CHEAT))
            message = message + "§f §f §2 §0 §4 §8";

        // Disable CJB's fly mod.
        if (!player.hasPermission(Permissions.CJB_FLY))
            message = message + "§3 §9 §2 §0 §0 §1";

        // Disable CJB's xray.
        if (!player.hasPermission(Permissions.CJB_XRAY))
            message = message + "§3 §9 §2 §0 §0 §2";

        // Disable CJB's radar.
        if (!player.hasPermission(Permissions.CJB_RADAR))
            message = message + "§3 §9 §2 §0 §0 §3";

        // Disable Minecraft AutoMap's ores.
        if (!player.hasPermission(Permissions.MINECRAFTAUTOMAP_ORES))
            message = message + "§0§0§1§f§e";

        // Disable Minecraft AutoMap's cave mode.
        if (!player.hasPermission(Permissions.MINECRAFTAUTOMAP_CAVE))
            message = message + "§0§0§2§f§e";

        // Disable Minecraft AutoMap's radar.
        if (!player.hasPermission(Permissions.MINECRAFTAUTOMAP_RADAR))
            message = message + "§0§0§3§4§5§6§7§8§f§e";

        // Disable Smart Moving's climbing.
        if (!player.hasPermission(Permissions.SMARTMOVING_CLIMBING))
            message = message + "§0§1§0§1§2§f§f";

        // Disable Smart Moving's climbing.
        if (!player.hasPermission(Permissions.SMARTMOVING_SWIMMING))
            message = message + "§0§1§3§4§f§f";

        // Disable Smart Moving's climbing.
        if (!player.hasPermission(Permissions.SMARTMOVING_CRAWLING))
            message = message + "§0§1§5§f§f";

        // Disable Smart Moving's climbing.
        if (!player.hasPermission(Permissions.SMARTMOVING_SLIDING))
            message = message + "§0§1§6§f§f";

        // Disable Smart Moving's climbing.
        if (!player.hasPermission(Permissions.SMARTMOVING_JUMPING))
            message = message + "§0§1§8§9§a§b§f§f";

        // Disable Smart Moving's climbing.
        if (!player.hasPermission(Permissions.SMARTMOVING_FLYING))
            message = message + "§0§1§7§f§f";

        player.sendMessage(message);
    }

    private boolean setCustomNetServerHandler(final Player player) {
        final EntityPlayer entityPlayer = ((CraftPlayer) player).getHandle();
        final NetServerHandler oldNSH = entityPlayer.netServerHandler;
        if (oldNSH instanceof CustomNetServerHandler)
            return false;
        final DedicatedServer server = ((CraftServer) Bukkit.getServer()).getHandle().getServer();
        final NetServerHandler newNSH = new CustomNetServerHandler(server, oldNSH.networkManager, entityPlayer);
        newNSH.a(player.getLocation().getX(), player.getLocation().getY(), player.getLocation().getZ(), player
                .getLocation().getYaw(), player.getLocation().getPitch());
        entityPlayer.netServerHandler = newNSH;
        try {
            final Field field = NetworkManager.class.getDeclaredField("packetListener");
            field.setAccessible(true);
            field.set(oldNSH.networkManager, newNSH);
        } catch (final Exception e) {
            e.printStackTrace();
        }
        oldNSH.disconnected = true;
        server.ac().a(newNSH);
        return true;
    }
}