NoDirectConnectBungee 1.0

package me.z609.nodirectconnectbungee;

import net.md_5.bungee.api.chat.TextComponent;
import net.md_5.bungee.api.connection.ProxiedPlayer;
import net.md_5.bungee.api.event.ProxyPingEvent;
import net.md_5.bungee.api.event.ServerConnectEvent;
import net.md_5.bungee.api.plugin.Listener;
import net.md_5.bungee.api.plugin.Plugin;
import net.md_5.bungee.event.EventHandler;

import java.util.ArrayList;
import java.util.List;

/**
 * This code is by Z609, and is copyright (C) 2015-2016 Z609.
 * Don't share this code with the public, please! Thanks!
 */
public class NoDirectConnectBungee extends Plugin implements Listener {

    private List<String> pinged = new ArrayList<String>();

    @Override
    public void onEnable(){
        getProxy().getPluginManager().registerListener(this, this);
    }

    @EventHandler
    public void onPing(ProxyPingEvent event){
        String ip = event.getConnection().getAddress().getAddress().getHostAddress();
        if(!pinged.contains(ip)){
            pinged.add(ip);
        }
    }

    @EventHandler
    public void onConnect(ServerConnectEvent event){
        ProxiedPlayer player = event.getPlayer();
        String ip = player.getAddress().getAddress().getHostAddress();
        if(!pinged.contains(ip)){
            event.setCancelled(true);
            player.disconnect(new TextComponent("Please add our server to your server list to connect."));
        }
    }

}