Grasscutter/src/main/java/emu/grasscutter/game/home/HomeWorldMPSystem.java

package emu.grasscutter.game.home;

import emu.grasscutter.game.player.Player;
import emu.grasscutter.game.props.EnterReason;
import emu.grasscutter.game.world.Position;
import emu.grasscutter.game.world.World;
import emu.grasscutter.game.world.data.TeleportProperties;
import emu.grasscutter.net.proto.EnterTypeOuterClass;
import emu.grasscutter.net.proto.OtherPlayerEnterHomeNotifyOuterClass;
import emu.grasscutter.net.proto.PlayerApplyEnterHomeResultNotifyOuterClass;
import emu.grasscutter.net.proto.RetcodeOuterClass;
import emu.grasscutter.server.event.player.PlayerEnterHomeEvent;
import emu.grasscutter.server.event.player.PlayerLeaveHomeEvent;
import emu.grasscutter.server.event.player.PlayerTeleportEvent;
import emu.grasscutter.server.game.BaseGameSystem;
import emu.grasscutter.server.game.GameServer;
import emu.grasscutter.server.packet.send.*;

public class HomeWorldMPSystem extends BaseGameSystem {
    public HomeWorldMPSystem(GameServer server) {
        super(server);
    }

    public void sendEnterHomeRequest(Player requester, int ownerUid) {
        var owner = getServer().getPlayerByUid(ownerUid);
        if (owner == null) {
            requester.sendPacket(
                    new PacketPlayerApplyEnterHomeResultNotify(
                            ownerUid,
                            "",
                            false,
                            PlayerApplyEnterHomeResultNotifyOuterClass.PlayerApplyEnterHomeResultNotify.Reason
                                    .OPEN_STATE_NOT_OPEN));
            requester.sendPacket(new PacketTryEnterHomeRsp());
            return;
        }

        if (owner.getRealmList() == null) {
            requester.sendPacket(
                    new PacketPlayerApplyEnterHomeResultNotify(
                            ownerUid,
                            "",
                            false,
                            PlayerApplyEnterHomeResultNotifyOuterClass.PlayerApplyEnterHomeResultNotify.Reason
                                    .OPEN_STATE_NOT_OPEN));
            requester.sendPacket(new PacketTryEnterHomeRsp());
            return;
        }

        var request = owner.getEnterHomeRequests().get(requester.getUid());

        if (request != null && !request.isExpired()) {
            return;
        }

        if (owner.isInEditMode()) {
            requester.sendPacket(
                    new PacketPlayerApplyEnterHomeResultNotify(
                            ownerUid,
                            owner.getNickname(),
                            false,
                            PlayerApplyEnterHomeResultNotifyOuterClass.PlayerApplyEnterHomeResultNotify.Reason
                                    .HOST_IN_EDIT_MODE));
            requester.sendPacket(new PacketTryEnterHomeRsp());
            return;
        }

        request = new EnterHomeRequest(requester);
        owner.getEnterHomeRequests().put(requester.getUid(), request);

        owner.sendPacket(new PacketPlayerApplyEnterHomeNotify(requester));
    }

    public void acceptEnterHomeRequest(Player owner, int requesterUid, boolean isAgreed) {
        var request = owner.getEnterHomeRequests().get(requesterUid);
        if (request == null || request.isExpired()) {
            return;
        }

        var requester = request.getRequester();
        owner.getEnterHomeRequests().remove(requesterUid);

        if (requester.getWorld().isMultiplayer()) {
            requester.sendPacket(
                    new PacketPlayerApplyEnterHomeResultNotify(
                            owner.getUid(),
                            owner.getNickname(),
                            false,
                            PlayerApplyEnterHomeResultNotifyOuterClass.PlayerApplyEnterHomeResultNotify.Reason
                                    .SYSTEM_JUDGE));
            requester.sendPacket(new PacketTryEnterHomeRsp());
            return;
        }

        requester.sendPacket(
                new PacketPlayerApplyEnterHomeResultNotify(
                        owner.getUid(),
                        owner.getNickname(),
                        isAgreed,
                        PlayerApplyEnterHomeResultNotifyOuterClass.PlayerApplyEnterHomeResultNotify.Reason
                                .PLAYER_JUDGE));

        if (!isAgreed) {
            requester.sendPacket(new PacketTryEnterHomeRsp());
            return;
        }

        this.enterHome(requester, owner, 0, false);
    }

    public void enterHome(Player requester, Player owner, int teleportPoint, boolean toSafe) {
        if (requester.getWorld().isMultiplayer()) {
            return;
        }

        if (owner.getRealmList() == null) {
            // should never happen
            requester.sendPacket(
                    new PacketTryEnterHomeRsp(
                            RetcodeOuterClass.Retcode.RET_HOME_NOT_FOUND_IN_MEM_VALUE, owner.getUid()));
            return;
        }

        var world = this.server.getHomeWorldOrCreate(owner);
        var targetHome = world.getHome();

        var event = new PlayerEnterHomeEvent(requester, owner, targetHome);
        event.call();
        if (event.isCanceled()) {
            requester.sendPacket(
                    new PacketTryEnterHomeRsp(
                            RetcodeOuterClass.Retcode.RET_HOME_OWNER_REFUSE_TO_ENTER_HOME_VALUE, owner.getUid()));
            return;
        }

        if (owner.isInEditMode()) {
            requester.sendPacket(
                    new PacketTryEnterHomeRsp(
                            RetcodeOuterClass.Retcode.RET_HOME_CANT_ENTER_BY_IN_EDIT_MODE_VALUE, owner.getUid()));
            return;
        }

        int realmId = 2000 + owner.getCurrentRealmId();
        var item = targetHome.getHomeSceneItem(realmId);
        targetHome.save();
        var pos = toSafe ? world.getSceneById(realmId).getScriptManager().getConfig().born_pos : item.getBornPos();

        if (teleportPoint != 0) {
            var target = item.getTeleportPointPos(teleportPoint);
            if (target != null) {
                pos = target;
            }
        }

        requester.getPrevPosForHome().set(requester.getPosition());
        requester.setCurHomeWorld(world);
        requester.setPrevScene(requester.getSceneId());
        world.addPlayer(requester, realmId);
        requester.setSceneId(realmId);
        requester.getPosition().set(pos);

        requester.sendPacket(
                new PacketPlayerEnterSceneNotify(
                        requester,
                        owner.getUid(),
                        TeleportProperties.builder()
                                .sceneId(realmId)
                                .enterReason(EnterReason.EnterHome)
                                .teleportTo(pos)
                                .teleportType(PlayerTeleportEvent.TeleportType.INTERNAL)
                                .build(),
                        !requester.equals(owner)));
        requester.sendPacket(new PacketTryEnterHomeRsp(owner.getUid()));

        requester.setHasSentInitPacketInHome(false);
        world.getPlayers().stream()
                .filter(player -> !player.equals(requester))
                .forEach(player -> player.sendPacket(new PacketPlayerPreEnterMpNotify(requester)));
    }

    public boolean leaveCoop(Player player, int prevScene) {
        return this.leaveCoop(player, prevScene, player.getPrevPosForHome());
    }

    public boolean leaveCoop(Player player, int prevScene, Position pos) {
        // Make sure everyone's scene is loaded
        for (var p : player.getWorld().getPlayers()) {
            if (p.getSceneLoadState() != Player.SceneLoadState.LOADED) {
                return false;
            }
        }

        // Event
        var event =
                new PlayerLeaveHomeEvent(
                        player,
                        player.getCurHomeWorld().getHost(),
                        player.getCurHomeWorld().getHome(),
                        PlayerLeaveHomeEvent.Reason.PLAYER_LEAVE);
        event.call();

        player.getPosition().set(pos);
        var world = new World(player);
        world.addPlayer(player, prevScene);
        player.getCurHomeWorld().sendPacketToHostIfOnline(new PacketOtherPlayerEnterOrLeaveHomeNotify(player, OtherPlayerEnterHomeNotifyOuterClass.OtherPlayerEnterHomeNotify.Reason.LEAVE));
        var myHome = this.server.getHomeWorldOrCreate(player);
        player.setCurHomeWorld(myHome);
        myHome.getHome().onOwnerLogin(player);

        player.sendPacket(new PacketPlayerEnterSceneNotify(player, EnterTypeOuterClass.EnterType.ENTER_TYPE_BACK, EnterReason.TeamBack, prevScene, pos));

        return true;
    }

    public boolean kickPlayerFromHome(Player owner, int targetUid) {
        // Make sure player's world is multiplayer and that player is owner
        if (!owner.getCurHomeWorld().getHost().equals(owner)) {
            return false;
        }

        // Get victim and sanity checks
        var victim = owner.getServer().getPlayerByUid(targetUid);
        if (victim == null || owner.equals(victim)) {
            return false;
        }

        // Make sure victim's scene has loaded
        if (victim.getSceneLoadState() != Player.SceneLoadState.LOADED) {
            return false;
        }

        // Event
        var event =
                new PlayerLeaveHomeEvent(
                        victim, owner, victim.getCurHomeWorld().getHome(), PlayerLeaveHomeEvent.Reason.KICKED);
        event.call();

        // Kick
        victim.getPosition().set(victim.getPrevPosForHome());
        var world = new World(victim);
        world.addPlayer(victim, 3);
        victim.getCurHomeWorld().sendPacketToHostIfOnline(new PacketOtherPlayerEnterOrLeaveHomeNotify(victim, OtherPlayerEnterHomeNotifyOuterClass.OtherPlayerEnterHomeNotify.Reason.LEAVE));
        var myHome = this.server.getHomeWorldOrCreate(victim);
        victim.setCurHomeWorld(myHome);
        myHome.getHome().onOwnerLogin(victim);

        victim.sendPacket(
                new PacketPlayerEnterSceneNotify(
                        victim,
                        EnterTypeOuterClass.EnterType.ENTER_TYPE_BACK,
                        EnterReason.TeamKick,
                        victim.getScene().getId(),
                        victim.getPrevPosForHome()));
        return true;
    }
}