GeyserModelEngine-plugin/src/main/java/re/imc/geysermodelengine/GeyserModelEngine.java

package re.imc.geysermodelengine;

import com.github.retrooper.packetevents.PacketEvents;
import com.github.retrooper.packetevents.event.PacketListenerPriority;
import com.ticxo.modelengine.api.ModelEngineAPI;
import com.ticxo.modelengine.api.model.ActiveModel;
import com.ticxo.modelengine.api.model.ModeledEntity;
import dev.jorel.commandapi.CommandAPI;
import dev.jorel.commandapi.CommandAPIBukkitConfig;
import io.github.retrooper.packetevents.factory.spigot.SpigotPacketEventsBuilder;
import org.bukkit.Bukkit;
import org.bukkit.World;
import org.bukkit.entity.Entity;
import org.bukkit.plugin.java.JavaPlugin;
import re.imc.geysermodelengine.listener.ModelListener;
import re.imc.geysermodelengine.listener.MountPacketListener;
import re.imc.geysermodelengine.managers.ConfigManager;
import re.imc.geysermodelengine.managers.bedrock.BedrockMountControlManager;
import re.imc.geysermodelengine.managers.commands.CommandManager;
import re.imc.geysermodelengine.managers.model.EntityTaskManager;
import re.imc.geysermodelengine.managers.model.ModelManager;
import re.imc.geysermodelengine.managers.player.PlayerManager;
import re.imc.geysermodelengine.managers.server.ServerManager;
import re.imc.geysermodelengine.managers.model.data.ModelEntityData;
import re.imc.geysermodelengine.runnables.BedrockMountControlRunnable;
import re.imc.geysermodelengine.runnables.UpdateTaskRunnable;

import java.util.*;
import java.util.concurrent.*;

public class GeyserModelEngine extends JavaPlugin {

    private ConfigManager configManager;
    private ServerManager serverManager;

    private CommandManager commandManager;

    private ModelManager modelManager;
    private EntityTaskManager entityTaskManager;
    private BedrockMountControlManager bedrockMountControlManager;

    private PlayerManager playerManager;

    @Override
    public void onLoad() {
        PacketEvents.setAPI(SpigotPacketEventsBuilder.build(this));
        PacketEvents.getAPI().load();

        CommandAPI.onLoad(new CommandAPIBukkitConfig(this));
    }

    @Override
    public void onEnable() {
        loadHooks();
        loadManagers();
        loadRunnables();

        PacketEvents.getAPI().getEventManager().registerListener(new MountPacketListener(this), PacketListenerPriority.NORMAL);

        Bukkit.getPluginManager().registerEvents(new ModelListener(this), this);

        Bukkit.getScheduler().runTaskLater(this, () -> {
                    for (World world : Bukkit.getWorlds()) {

                        for (Entity entity : world.getEntities()) {

                            if (!modelManager.getEntitiesCache().containsKey(entity.getEntityId())) {

                                ModeledEntity modeledEntity = ModelEngineAPI.getModeledEntity(entity);

                                if (modeledEntity != null) {
                                    Optional<ActiveModel> model = modeledEntity.getModels().values().stream().findFirst();
                                    model.ifPresent(m -> modelManager.create(modeledEntity, m));
                                }
                            }
                        }
                    }
                }, 100);
    }

    @Override
    public void onDisable() {
        PacketEvents.getAPI().terminate();

        for (Map<ActiveModel, ModelEntityData> entities : modelManager.getEntitiesCache().values()) {
            entities.forEach((model, modelEntity) -> {
                modelEntity.getEntity().remove();
            });
        }

        CommandAPI.onDisable();
    }

    private void loadHooks() {
        PacketEvents.getAPI().init();
        CommandAPI.onEnable();
    }

    private void loadManagers() {
        this.configManager = new ConfigManager(this);
        this.serverManager = new ServerManager();

        this.commandManager = new CommandManager(this);

        this.modelManager = new ModelManager(this);
        this.entityTaskManager = new EntityTaskManager(this);
        this.bedrockMountControlManager = new BedrockMountControlManager();

        this.playerManager = new PlayerManager();
    }

    private void loadRunnables() {
        Bukkit.getAsyncScheduler().runAtFixedRate(this, new UpdateTaskRunnable(this), 10, configManager.getConfig().getLong("entity-position-update-period"), TimeUnit.MILLISECONDS);
        Bukkit.getAsyncScheduler().runAtFixedRate(this, new BedrockMountControlRunnable(this), 1, 1, TimeUnit.MILLISECONDS);
    }

    public ConfigManager getConfigManager() {
        return configManager;
    }

    public ServerManager getServerManager() {
        return serverManager;
    }

    public CommandManager getCommandManager() {
        return commandManager;
    }

    public ModelManager getModelManager() {
        return modelManager;
    }

    public EntityTaskManager getEntityTaskManager() {
        return entityTaskManager;
    }

    public BedrockMountControlManager getBedrockMountControlManager() {
        return bedrockMountControlManager;
    }

    public PlayerManager getPlayerManager() {
        return playerManager;
    }
}