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);
    }

    public void enterHome(Player requester, Player owner) {
        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();
        targetHome.getHomeSceneItem(realmId);
        targetHome.save();
        var pos = world.getSceneById(realmId).getScriptManager().getConfig().born_pos;

        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));
        player.setCurHomeWorld(this.server.getHomeWorldOrCreate(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));
        victim.setCurHomeWorld(this.server.getHomeWorldOrCreate(victim));

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