src/main/java/moe/ymc/safelog/mixin/ServerLoginNetworkHandlerMixin.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

package moe.ymc.safelog.mixin;

import com.mojang.authlib.GameProfile;
import moe.ymc.safelog.Safelog;
import net.minecraft.network.ClientConnection;
import net.minecraft.server.network.ServerLoginNetworkHandler;
import org.spongepowered.asm.mixin.Final;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;

import java.security.MessageDigest;

@Mixin(value = ServerLoginNetworkHandler.class)
public class ServerLoginNetworkHandlerMixin {
    @Shadow
    GameProfile profile;
    @Shadow
    @Final
    public ClientConnection connection;

    @Inject(
            method = "getConnectionInfo",
            at = @At("HEAD"),
            cancellable = true
    )
    public void getConnectionInfo(CallbackInfoReturnable<String> cir) {
        if(profile != null) {
            cir.setReturnValue(profile + " (" + Safelog.sanitize(this.connection.getAddress()) + ")");
        } else {
            cir.setReturnValue(Safelog.sanitize(this.connection.getAddress()));
        }
    }
}