path: root/src/main/java/ganarchy/friendcode/client/FriendCodeScreen.java

package ganarchy.friendcode.client;

import ganarchy.friendcode.FriendCode;
import ganarchy.friendcode.mixin.FriendCodeIntegratedServerExt;
import ganarchy.friendcode.util.KeyUtil;import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.MinecraftClient;
import net.minecraft.client.gui.screen.OpenToLanScreen;
import net.minecraft.client.gui.screen.Screen;
import net.minecraft.client.gui.widget.ButtonWidget;
import net.minecraft.client.gui.widget.CyclingButtonWidget;
import net.minecraft.client.util.NetworkUtils;
import net.minecraft.client.util.math.MatrixStack;
import net.minecraft.screen.ScreenTexts;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.text.MutableText;
import net.minecraft.text.Text;
import net.minecraft.util.WorldSavePath;import net.minecraft.world.GameMode;

import java.io.IOException;

@Environment(EnvType.CLIENT)
public class FriendCodeScreen extends Screen {
    private static final Text
        ALLOW_COMMANDS_TEXT = Text.translatable("selectWorld.allowCommands"),
        GAME_MODE_TEXT = Text.translatable("selectWorld.gameMode"),
        CODE_TYPE_TEXT = Text.translatable("friendcode.code_type"),
        OTHER_PLAYERS_TEXT = Text.translatable("lanServer.otherPlayers"),
        START_SHARING = Text.translatable("friendcode.start");
    private final Screen parent;
    private CodeType codeType = CodeType.SESSION;
    private GameMode gameMode = GameMode.SURVIVAL;
    private boolean allowCommands;

    public FriendCodeScreen(Screen parent) {
        super(Text.translatable("friendcode.title"));
        this.parent = parent;
    }

    @Override
    protected void init() {
        // title button allows going back to LAN screen
        this.addDrawableChild(new ButtonWidget(
            this.width / 2 - 155,
            50,
            310,
            20,
            this.title,
            button -> this.client.setScreen(new OpenToLanScreen(this.parent))
        ));

        // game setting buttons
        this.addDrawableChild(
            CyclingButtonWidget
            .builder(GameMode::getSimpleTranslatableName)
            .values(
                GameMode.SURVIVAL,
                GameMode.SPECTATOR,
                GameMode.CREATIVE,
                GameMode.ADVENTURE
            )
            .initially(this.gameMode)
            .build(
                this.width / 2 - 155,
                100,
                150,
                20,
                GAME_MODE_TEXT,
                (button, gameMode) -> {
                    this.gameMode = gameMode;
                }
            )
        );
        this.addDrawableChild(
            CyclingButtonWidget
            .onOffBuilder(this.allowCommands)
            .build(
                this.width / 2 + 5,
                100,
                150,
                20,
                ALLOW_COMMANDS_TEXT,
                (button, allowCommands) -> {
                    this.allowCommands = allowCommands;
                }
            )
        );

        // friend code type button
        this.addDrawableChild(
            CyclingButtonWidget
            .builder(CodeType::getSimpleTranslatableName)
            .values(CodeType.SESSION, CodeType.WORLD)
            .initially(this.codeType)
            .build(
                this.width / 2 - 155,
                125,
                310,
                20,
                CODE_TYPE_TEXT,
                (button, codeType) -> {
                    this.codeType = codeType;
                }
            )
        ).active = false;

        // start sharing
        this.addDrawableChild(new ButtonWidget(
            this.width / 2 - 155,
            this.height - 28,
            150,
            20,
            START_SHARING,
            button -> {
                // FIXME
                int i = NetworkUtils.findLocalPort();
                var samPinger = openToFriends(this.client, this.codeType, this.gameMode, this.allowCommands, i);
                MutableText text = samPinger != null ? Text.translatable("commands.publish.started", i) : Text.translatable("commands.publish.failed");
                this.client.inGameHud.getChatHud().addMessage(text);
                this.client.updateWindowTitle();
                this.client.setScreen(new WaitingForFriendCodeScreen(samPinger));
            }
        ));

        // go back to options
        this.addDrawableChild(new ButtonWidget(
            this.width / 2 + 5,
            this.height - 28,
            150,
            20,
            ScreenTexts.CANCEL,
            button -> this.client.setScreen(this.parent)
        ));
    }

    private static I2PSamPinger openToFriends(MinecraftClient client, CodeType codeType, GameMode gameMode, boolean allowCommands, int port) {
        try {
            String privateKey = null;
            if (codeType == CodeType.WORLD) {
                // FIXME this currently does nothing.
                var worldDir = client.getServer().submit(() -> client.getServer().getSavePath(WorldSavePath.ROOT)).join();
                var keyFile = worldDir.resolve("friendcode.key");
                privateKey = KeyUtil.readKeyFile(keyFile);
            }
            client.loadBlockList();
            client.getServer().getNetworkIo().bind(null, port);
            FriendCode.LOGGER.info("Started serving on {}", port);
            ((FriendCodeIntegratedServerExt) client.getServer()).lanPort(port);
            // reuse LAN pinger machinery instead of rolling our own
            var lanPinger = new I2PSamPinger(client.getServer().getServerMotd(), "" + port, privateKey);
            ((FriendCodeIntegratedServerExt) client.getServer()).lanPinger(lanPinger);
            lanPinger.start();
            ((FriendCodeIntegratedServerExt) client.getServer()).forcedGameMode(gameMode);
            client.getServer().getPlayerManager().setCheatsAllowed(allowCommands);
            int i = client.getServer().getPermissionLevel(client.player.getGameProfile());
            client.player.setClientPermissionLevel(i);
            for (ServerPlayerEntity serverPlayerEntity : client.getServer().getPlayerManager().getPlayerList()) {
                client.getServer().getCommandManager().sendCommandTree(serverPlayerEntity);
            }
            return lanPinger;
        }
        catch (IOException iOException) {
            return null;
        }
    }

    @Override
    public void render(MatrixStack matrices, int mouseX, int mouseY, float delta) {
        this.renderBackground(matrices);
        FriendCodeScreen.drawCenteredText(matrices, this.textRenderer, OTHER_PLAYERS_TEXT, this.width / 2, 82, 0xFFFFFF);
        super.render(matrices, mouseX, mouseY, delta);
    }
}