Merge 6936d33310 into 336333acb1

api/src/main/java/net/md_5/bungee/api/event/ProtocolChangedEvent.java

@@ -0,0 +1,47 @@
+package net.md_5.bungee.api.event;
+
+import lombok.AllArgsConstructor;
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.ToString;
+import net.md_5.bungee.api.connection.Connection;
+import net.md_5.bungee.api.plugin.Event;
+import net.md_5.bungee.protocol.Protocol;
+
+/**
+ * Called when the encode or decode protocol of a {@link Connection} is changed.
+ */
+@Getter
+@ToString(callSuper = false)
+@EqualsAndHashCode(callSuper = false)
+@AllArgsConstructor
+public class ProtocolChangedEvent extends Event
+{
+
+    /**
+     * The old protocol.
+     */
+    private Protocol oldProtocol;
+
+    /**
+     * The new protocol.
+     */
+    private Protocol newProtocol;
+
+    /**
+     * The connection whose protocol is being changed.
+     */
+    private Connection connection;
+
+    /**
+     * The direction of the changed protocol.
+     * encode or decode
+     */
+    private Direction direction;
+
+    public enum Direction
+    {
+        ENCODE,
+        DECODE
+    }
+}

proxy/src/main/java/net/md_5/bungee/connection/DownstreamBridge.java

@@ -36,6 +36,7 @@ import net.md_5.bungee.api.config.ServerInfo;
 import net.md_5.bungee.api.connection.ProxiedPlayer;
 import net.md_5.bungee.api.connection.Server;
 import net.md_5.bungee.api.event.PluginMessageEvent;
+import net.md_5.bungee.api.event.ProtocolChangedEvent;
 import net.md_5.bungee.api.event.ServerConnectEvent;
 import net.md_5.bungee.api.event.ServerDisconnectEvent;
 import net.md_5.bungee.api.event.ServerKickEvent;
@@ -146,6 +147,12 @@ public class DownstreamBridge extends PacketHandler
         con.sendPacket( packet );
     }
 
+    @Override
+    public void protocolChanged(ChannelWrapper channel, Protocol oldProtocol, Protocol newProtocol, ProtocolChangedEvent.Direction direction)
+    {
+        bungee.getPluginManager().callEvent( new ProtocolChangedEvent( oldProtocol, newProtocol, server, direction ) );
+    }
+
     @Override
     public void handle(KeepAlive alive) throws Exception
     {

proxy/src/main/java/net/md_5/bungee/connection/InitialHandler.java

@@ -43,6 +43,7 @@ import net.md_5.bungee.api.event.LoginEvent;
 import net.md_5.bungee.api.event.PlayerHandshakeEvent;
 import net.md_5.bungee.api.event.PostLoginEvent;
 import net.md_5.bungee.api.event.PreLoginEvent;
+import net.md_5.bungee.api.event.ProtocolChangedEvent;
 import net.md_5.bungee.api.event.ProxyPingEvent;
 import net.md_5.bungee.api.event.ServerConnectEvent;
 import net.md_5.bungee.http.HttpClient;
@@ -181,6 +182,12 @@ public class InitialHandler extends PacketHandler implements PendingConnection
         }
     }
 
+    @Override
+    public void protocolChanged(ChannelWrapper channel, Protocol oldProtocol, Protocol newProtocol, ProtocolChangedEvent.Direction direction)
+    {
+        bungee.getPluginManager().callEvent( new ProtocolChangedEvent( oldProtocol, newProtocol, this, direction ) );
+    }
+
     @Override
     public void handle(PluginMessage pluginMessage) throws Exception
     {

proxy/src/main/java/net/md_5/bungee/connection/UpstreamBridge.java

@@ -19,6 +19,7 @@ import net.md_5.bungee.api.connection.ProxiedPlayer;
 import net.md_5.bungee.api.event.ChatEvent;
 import net.md_5.bungee.api.event.PlayerDisconnectEvent;
 import net.md_5.bungee.api.event.PluginMessageEvent;
+import net.md_5.bungee.api.event.ProtocolChangedEvent;
 import net.md_5.bungee.api.event.SettingsChangedEvent;
 import net.md_5.bungee.api.event.TabCompleteEvent;
 import net.md_5.bungee.entitymap.EntityMap;
@@ -155,6 +156,12 @@ public class UpstreamBridge extends PacketHandler
         }
     }
 
+    @Override
+    public void protocolChanged(ChannelWrapper channel, Protocol oldProtocol, Protocol newProtocol, ProtocolChangedEvent.Direction direction)
+    {
+        BungeeCord.getInstance().getPluginManager().callEvent( new ProtocolChangedEvent( oldProtocol, newProtocol, con, direction ) );
+    }
+
     @Override
     public void handle(KeepAlive alive) throws Exception
     {

proxy/src/main/java/net/md_5/bungee/netty/ChannelWrapper.java

@@ -9,6 +9,7 @@ import java.net.SocketAddress;
 import java.util.concurrent.TimeUnit;
 import lombok.Getter;
 import lombok.Setter;
+import net.md_5.bungee.api.event.ProtocolChangedEvent;
 import net.md_5.bungee.compress.PacketCompressor;
 import net.md_5.bungee.compress.PacketDecompressor;
 import net.md_5.bungee.protocol.DefinedPacket;
@@ -43,7 +44,13 @@ public class ChannelWrapper
 
     public void setDecodeProtocol(Protocol protocol)
     {
+        Protocol oldProtocol = getDecodeProtocol();
         ch.pipeline().get( MinecraftDecoder.class ).setProtocol( protocol );
+        HandlerBoss handlerBoss = ch.pipeline().get( HandlerBoss.class );
+        if ( oldProtocol != protocol && handlerBoss != null )
+        {
+            handlerBoss.getHandler().protocolChanged( this, oldProtocol, protocol, ProtocolChangedEvent.Direction.DECODE );
+        }
     }
 
     public Protocol getEncodeProtocol()
@@ -54,7 +61,13 @@ public class ChannelWrapper
 
     public void setEncodeProtocol(Protocol protocol)
     {
+        Protocol oldProtocol = getEncodeProtocol();
         ch.pipeline().get( MinecraftEncoder.class ).setProtocol( protocol );
+        HandlerBoss handlerBoss = ch.pipeline().get( HandlerBoss.class );
+        if ( oldProtocol != protocol && handlerBoss != null )
+        {
+            handlerBoss.getHandler().protocolChanged( this, oldProtocol, protocol, ProtocolChangedEvent.Direction.ENCODE );
+        }
     }
 
     public void setProtocol(Protocol protocol)

proxy/src/main/java/net/md_5/bungee/netty/HandlerBoss.java

@@ -10,6 +10,8 @@ import io.netty.handler.timeout.ReadTimeoutException;
 import java.io.IOException;
 import java.net.InetSocketAddress;
 import java.util.logging.Level;
+import lombok.AccessLevel;
+import lombok.Getter;
 import net.md_5.bungee.api.ProxyServer;
 import net.md_5.bungee.connection.CancelSendSignal;
 import net.md_5.bungee.connection.InitialHandler;
@@ -29,6 +31,7 @@ public class HandlerBoss extends ChannelInboundHandlerAdapter
 {
 
     private ChannelWrapper channel;
+    @Getter(AccessLevel.PACKAGE)
     private PacketHandler handler;
 
     public void setHandler(PacketHandler handler)

proxy/src/main/java/net/md_5/bungee/netty/PacketHandler.java

@@ -1,6 +1,8 @@
 package net.md_5.bungee.netty;
 
+import net.md_5.bungee.api.event.ProtocolChangedEvent;
 import net.md_5.bungee.protocol.PacketWrapper;
+import net.md_5.bungee.protocol.Protocol;
 
 public abstract class PacketHandler extends net.md_5.bungee.protocol.AbstractPacketHandler
 {
@@ -32,4 +34,9 @@ public abstract class PacketHandler extends net.md_5.bungee.protocol.AbstractPac
     public void writabilityChanged(ChannelWrapper channel) throws Exception
     {
     }
+
+    public void protocolChanged(ChannelWrapper channel, Protocol oldProtocol, Protocol newProtocol, ProtocolChangedEvent.Direction direction)
+    {
+
+    }
 }