Skip to content

Commit 295d87d

Browse files
Add functions to allow changing the handler names in TrackingCompressionHandlerProvider
1 parent ab97e24 commit 295d87d

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

common/src/main/java/com/viaversion/viarewind/protocol/v1_8to1_7_6_10/provider/compression/TrackingCompressionHandlerProvider.java

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,30 +26,36 @@
2626
import io.netty.channel.ChannelPipeline;
2727

2828
public class TrackingCompressionHandlerProvider extends CompressionHandlerProvider {
29-
public static final String COMPRESS_HANDLER_NAME = "compress";
30-
public static final String DECOMPRESS_HANDLER_NAME = "decompress";
3129

3230
@Override
3331
public void onHandleLoginCompressionPacket(UserConnection user, int threshold) {
3432
final ChannelPipeline pipeline = user.getChannel().pipeline();
3533
if (user.isClientSide()) {
36-
pipeline.addBefore(Via.getManager().getInjector().getEncoderName(), COMPRESS_HANDLER_NAME, getEncoder(threshold));
37-
pipeline.addBefore(Via.getManager().getInjector().getDecoderName(), DECOMPRESS_HANDLER_NAME, getDecoder(threshold));
34+
pipeline.addBefore(Via.getManager().getInjector().getEncoderName(), compressHandlerName(), getEncoder(threshold));
35+
pipeline.addBefore(Via.getManager().getInjector().getDecoderName(), decompressHandlerName(), getDecoder(threshold));
3836
} else {
3937
setCompressionEnabled(user, true); // We need to remove compression for 1.7 clients
4038
}
4139
}
4240

41+
public String compressHandlerName() {
42+
return "compress";
43+
}
44+
45+
public String decompressHandlerName() {
46+
return "decompress";
47+
}
48+
4349
@Override
4450
public void onTransformPacket(UserConnection user) {
4551
if (isCompressionEnabled(user)) {
4652
final ChannelPipeline pipeline = user.getChannel().pipeline();
4753

4854
String compressor = null;
4955
String decompressor = null;
50-
if (pipeline.get(COMPRESS_HANDLER_NAME) != null) { // ViaVersion
51-
compressor = COMPRESS_HANDLER_NAME;
52-
decompressor = DECOMPRESS_HANDLER_NAME;
56+
if (pipeline.get(compressHandlerName()) != null) { // ViaVersion
57+
compressor = compressHandlerName();
58+
decompressor = decompressHandlerName();
5359
} else if (pipeline.get("compression-encoder") != null) { // Velocity
5460
compressor = "compression-encoder";
5561
decompressor = "compression-decoder";

0 commit comments

Comments
 (0)