package emu.grasscutter.server.packet.send;

import emu.grasscutter.game.entity.EntityVehicle;
import emu.grasscutter.game.entity.GameEntity;
import emu.grasscutter.game.player.Player;
import emu.grasscutter.game.world.Position;
import emu.grasscutter.net.packet.BasePacket;
import emu.grasscutter.net.packet.PacketOpcodes;
import emu.grasscutter.net.proto.CreateVehicleRspOuterClass.CreateVehicleRsp;
import emu.grasscutter.net.proto.VehicleInteractTypeOuterClass;
import emu.grasscutter.net.proto.VehicleMemberOuterClass.VehicleMember;
import java.util.List;

public class PacketCreateVehicleRsp extends BasePacket {

    public PacketCreateVehicleRsp(
            Player player, int vehicleId, int pointId, Position pos, Position rot) {
        super(PacketOpcodes.CreateVehicleRsp);
        CreateVehicleRsp.Builder proto = CreateVehicleRsp.newBuilder();

        // Eject vehicle members and Kill previous vehicles if there are any
        List<GameEntity> previousVehicles =
                player.getScene().getEntities().values().stream()
                        .filter(
                                entity ->
                                        entity instanceof EntityVehicle
                                                && ((EntityVehicle) entity).getGadgetId() == vehicleId
                                                && ((EntityVehicle) entity).getOwner().equals(player))
                        .toList();

        previousVehicles.stream()
                .forEach(
                        entity -> {
                            List<VehicleMember> vehicleMembers =
                                    ((EntityVehicle) entity).getVehicleMembers().stream().toList();

                            vehicleMembers.stream()
                                    .forEach(
                                            vehicleMember -> {
                                                player
                                                        .getScene()
                                                        .broadcastPacket(
                                                                new PacketVehicleInteractRsp(
                                                                        ((EntityVehicle) entity),
                                                                        vehicleMember,
                                                                        VehicleInteractTypeOuterClass.VehicleInteractType
                                                                                .VEHICLE_INTERACT_TYPE_OUT));
                                            });

                            player.getScene().killEntity(entity, 0);
                        });

        EntityVehicle vehicle =
                new EntityVehicle(player.getScene(), player, vehicleId, pointId, pos, rot);
        player.getScene().addEntity(vehicle);

        proto.setVehicleId(vehicleId);
        proto.setEntityId(vehicle.getId());

        this.setData(proto.build());
    }
}