Sit/src/main/java/one/oth3r/sit/Events.java

package one.oth3r.sit;

import net.fabricmc.fabric.api.command.v2.CommandRegistrationCallback;
import net.fabricmc.fabric.api.event.lifecycle.v1.ServerLifecycleEvents;
import net.fabricmc.fabric.api.event.lifecycle.v1.ServerTickEvents;
import net.fabricmc.fabric.api.event.player.UseBlockCallback;
import net.fabricmc.fabric.api.networking.v1.ServerPlayConnectionEvents;
import net.minecraft.block.*;
import net.minecraft.block.enums.BlockHalf;
import net.minecraft.block.enums.SlabType;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityType;
import net.minecraft.entity.decoration.DisplayEntity;
import net.minecraft.item.BlockItem;
import net.minecraft.item.Item;
import net.minecraft.registry.Registries;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.text.Text;
import net.minecraft.util.ActionResult;
import net.minecraft.util.hit.HitResult;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Box;
import net.minecraft.util.math.Vec3d;
import net.minecraft.world.World;

import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

public class Events {
    private static int tick;
    public static HashMap<ServerPlayerEntity, Entity> entities = new HashMap<>();
    public static HashMap<ServerPlayerEntity, Integer> checkPlayers = new HashMap<>();
    public static boolean checkLogic(ServerPlayerEntity player) {
        Item mainItem = player.getMainHandStack().getItem();
        Item offItem = player.getOffHandStack().getItem();
        if (player.isSneaking()) return false;
        if (config.mainReq.equals(config.MainReq.empty) && !player.getMainHandStack().isEmpty()) return false;
        if (config.mainReq.equals(config.MainReq.restrictive)) {
            if (checkList(config.mainBlacklist,mainItem)) return false;
            if (!checkList(config.mainWhitelist,mainItem)) {
                if (config.mainBlock && (mainItem instanceof BlockItem)) return false;
                if (config.mainFood && mainItem.isFood()) return false;
                if (config.mainUsable && player.getMainHandStack().isUsedOnRelease()) return false;
            }
        }
        if (config.offReq.equals(config.OffReq.empty) && !player.getOffHandStack().isEmpty()) return false;
        if (config.offReq.equals(config.OffReq.restrictive)) {
            if (checkList(config.offBlacklist,offItem)) return false;
            if (!checkList(config.offWhitelist,offItem)) {
                if (config.offBlock && (offItem instanceof BlockItem)) return false;
                if (config.offFood && offItem.isFood()) return false;
                if (config.offUsable && player.getOffHandStack().isUsedOnRelease()) return false;
            }
        }
        return true;
    }
    public static boolean checkList(List<String> list, Item item) {
        String itemID = Registries.ITEM.getId(item).toString();
        System.out.println(itemID);
        return list.contains(itemID);
    }
    public static HashMap<String,HashMap<String,Object>> getCustomBlocks() {
        HashMap<String,HashMap<String,Object>> map = new HashMap<>();
        int i = 1;
        for (String s:config.customBlocks) {
            String[] split = s.split("\\|");
            HashMap<String,Object> data = new HashMap<>();
            data.put("block",split[0]);
            data.put("height",split[1]);
            data.put("hitbox",split[2]);
            if (split.length==4) data.put("state",split[3]);
            map.put(i+"",data);
            i++;
        }
        return map;
    }
    @SuppressWarnings("deprecation")
    public static boolean checkBlocks(BlockPos pos, World world) {
        BlockState blockState = world.getBlockState(pos);
        Block block = blockState.getBlock();
        BlockState blockStateAbove = world.getBlockState(pos.add(0,1,0));
        Block blockAbove = blockStateAbove.getBlock();
        System.out.println(blockState);
        //not the biggest fan but ah well
        if (blockStateAbove.isFullCube(world,pos.add(0,1,0)) && blockStateAbove.blocksMovement()) return false;
        else if (blockAbove instanceof StairsBlock || blockAbove instanceof SlabBlock || blockAbove instanceof CarpetBlock) return false;
        for (Entity entity:entities.values()) if (entity.getBlockPos().equals(pos) || entity.getBlockPos().add(0,1,0).equals(pos)) return false;
        if (block instanceof StairsBlock && config.stairsOn) return blockState.get(StairsBlock.HALF) == BlockHalf.BOTTOM;
        if (block instanceof SlabBlock && config.slabsOn) return blockState.get(SlabBlock.TYPE) == SlabType.BOTTOM;
        if (block instanceof CarpetBlock && config.carpetsOn) return true;
        if (blockState.isFullCube(world,pos.add(0,1,0)) && config.fullBlocksOn) return true;
        if (config.customOn && config.customBlocks.size() != 0) {
            System.out.println("checking custom");
            for (HashMap<String,Object> map:getCustomBlocks().values()) {
                String blockID = Registries.BLOCK.getId(block).toString();
                if (map.get("block").equals(blockID)) {
                    if (!map.containsKey("state")) return true;
                    String[] states = ((String) map.get("state")).split(",\\s*");
                    boolean matching = true;
                    for (String state:states) {
                        if (state.charAt(0) == '!')
                            if (blockState.toString().contains(state.substring(1))) matching = false;
                        else if (!blockState.toString().contains(state)) matching = false;
                    }
                    return matching;
                }
            }
        }
        return false;
    }
    public static void setEntity(BlockPos pos, World world, Entity entity) {
        Block block = world.getBlockState(pos).getBlock();
        entity.setCustomName(Text.of(Sit.ENTITY_NAME));
        entity.updatePositionAndAngles(pos.getX() + 0.5, pos.getY()+.47, pos.getZ() + 0.5, 0, 0);
        entity.setInvulnerable(true);
        if (block instanceof StairsBlock) {
            entity.updatePositionAndAngles(pos.getX() + 0.5, pos.getY()+.27, pos.getZ() + 0.5, 0, 0);
            entity.setBoundingBox(Box.of(Vec3d.of(pos),1.5,2,1.5));
        }
        if (block instanceof SlabBlock) {
            entity.updatePositionAndAngles(pos.getX() + 0.5, pos.getY()+.27, pos.getZ() + 0.5, 0, 0);
            entity.setBoundingBox(Box.of(Vec3d.of(pos),1.5,1,1.5));
        }
        if (block instanceof CarpetBlock) {
            entity.updatePositionAndAngles(pos.getX() + 0.5, pos.getY()-.17, pos.getZ() + 0.5, 0, 0);
            entity.setBoundingBox(Box.of(Vec3d.of(pos),1.5,0.125,1.5));
        }
        if (world.getBlockState(pos).isFullCube(world,pos.add(0,1,0))) {
            entity.updatePositionAndAngles(pos.getX() + 0.5, pos.getY()+.78, pos.getZ() + 0.5, 0, 0);
            entity.setBoundingBox(Box.of(Vec3d.of(pos),1.5,2,1.5));
        }
        if (config.customOn && config.customBlocks.size() != 0) {
            System.out.println("checking custom sit height");
            for (HashMap<String,Object> map:getCustomBlocks().values()) {
                String blockID = Registries.BLOCK.getId(block).toString();
                if (map.get("block").equals(blockID)) {
                    double input = Math.max(Math.min(Double.parseDouble((String) map.get("height")),1),0);
                    entity.updatePositionAndAngles(pos.getX() + 0.5, pos.getY()+input-.22, pos.getZ() + 0.5, 0, 0);
                    entity.setBoundingBox(Box.of(Vec3d.of(pos),1.5,Double.parseDouble((String) map.get("hitbox")),1.5));
                }
            }
        }
        //change pitch based on if player is sitting below block height or not
        if (entity.getY() <= pos.getY()+.35) entity.setPitch(90);
        else entity.setPitch(-90);
    }
    public static void register() {
        UseBlockCallback.EVENT.register((pl, world, hand, hitResult) -> {
            ServerPlayerEntity player = Sit.server.getPlayerManager().getPlayer(pl.getUuid());
            if (player == null) return ActionResult.PASS;
            if (hand == net.minecraft.util.Hand.MAIN_HAND && hitResult.getType() == HitResult.Type.BLOCK) {
                BlockPos pos = hitResult.getBlockPos();
                //todo idk find bugs and polish
                if (!checkLogic(player)) return ActionResult.PASS;
                if (checkBlocks(pos,world)) {
                    if (entities.containsKey(player)) {
                        if (!config.sitWhileSeated) return ActionResult.PASS;
                        entities.get(player).setRemoved(Entity.RemovalReason.DISCARDED);
                        entities.remove(player);
                    }
                    DisplayEntity.TextDisplayEntity entity = new DisplayEntity.TextDisplayEntity(EntityType.TEXT_DISPLAY,player.getServerWorld());
                    setEntity(pos,world,entity);
                    player.getServerWorld().spawnEntity(entity);
                    player.startRiding(entity);
                    entities.put(player,entity);
                }
            }
            return ActionResult.PASS;
        });
        ServerTickEvents.END_SERVER_TICK.register(minecraftServer -> minecraftServer.execute(Events::cleanUp));
        ServerPlayConnectionEvents.JOIN.register((handler, sender, server) -> {
            ServerPlayerEntity player = handler.player;
            checkPlayers.put(player,2);
        });
        ServerPlayConnectionEvents.DISCONNECT.register((handler, server) -> {
            ServerPlayerEntity player = handler.player;
            if (entities.containsKey(player)) {
                if (!config.keepActive) {
                    player.dismountVehicle();
                    entities.get(player).setRemoved(Entity.RemovalReason.DISCARDED);
                }
                entities.remove(player);
            }
            checkPlayers.remove(player);
        });
        ServerLifecycleEvents.SERVER_STARTED.register(s -> {
            Sit.server = s;
            Sit.commandManager = s.getCommandManager();
        });
        CommandRegistrationCallback.EVENT.register((dispatcher, registryAccess, environment) -> SitCommand.register(dispatcher));
    }
    public static void cleanUp() {
        tick++;
        if (tick >= 5) {
            tick = 0;
            Iterator<Map.Entry<ServerPlayerEntity, Entity>> entityLoop = entities.entrySet().iterator();
            while (entityLoop.hasNext()) {
                Map.Entry<ServerPlayerEntity, Entity> entry = entityLoop.next();
                ServerPlayerEntity player = entry.getKey();
                Entity entity = entry.getValue();
                if (player.getVehicle() == null || !player.getVehicle().equals(entity)) {
                    entity.setRemoved(Entity.RemovalReason.DISCARDED);
                    System.out.println("removed..");
                    entityLoop.remove();
                } else {
                    BlockPos pos = new BlockPos(entity.getBlockX(),(int) Math.floor(player.getY()),entity.getBlockZ());
                    if (entity.getPitch() == 90) pos = new BlockPos(entity.getBlockX(),(int) Math.ceil(player.getY()),entity.getBlockZ());
                    BlockState blockState = player.getWorld().getBlockState(pos);
                    if (blockState.isAir()) {
                        System.out.println(pos);
                        System.out.println(entity.getY());
                        System.out.println("dismount");
                        player.teleport(player.getX(),player.getBlockY()+1,player.getZ());
                        entity.setRemoved(Entity.RemovalReason.DISCARDED);
                        entityLoop.remove();
                    }
                }
            }
            Iterator<Map.Entry<ServerPlayerEntity, Integer>> playerCheckLoop = checkPlayers.entrySet().iterator();
            while (playerCheckLoop.hasNext()) {
                Map.Entry<ServerPlayerEntity, Integer> entry = playerCheckLoop.next();
                ServerPlayerEntity player = entry.getKey();
                int i = entry.getValue();
                checkPlayers.put(player,i-1);
                if (i<0) {
                    System.out.println("fail");
                    playerCheckLoop.remove();
                    continue;
                }
                System.out.println("trying to add back");
                if (player.getVehicle() != null) {
                    Entity entity = player.getVehicle();
                    if (entity.getName().getString().equals(Sit.ENTITY_NAME)) {
                        System.out.println("adding back");
                        setEntity(player.getBlockPos().add(0,1,0),player.getServerWorld(),entity);
                        entities.put(player,entity);
                        playerCheckLoop.remove();
                    }
                }
            }

            for (ServerPlayerEntity player:checkPlayers.keySet()) {
                int i = checkPlayers.get(player);
                checkPlayers.put(player,i-1);
                if (i<0) {
                    System.out.println("fail");
                    checkPlayers.remove(player);
                    continue;
                }
                System.out.println("trying to add back");
                if (player.getVehicle() != null) {
                    Entity entity = player.getVehicle();
                    if (entity.getName().getString().equals(Sit.ENTITY_NAME)) {
                        System.out.println("adding back");
                        setEntity(player.getBlockPos().add(0,1,0),player.getServerWorld(),entity);
                        entities.put(player,entity);
                        checkPlayers.remove(player);
                    }
                }
            }
        }
    }
}