Autism-Client/src/main/java/autismclient/api/event/AddonEvents.java

package autismclient.api.event;

import autismclient.AutismClientAddon;
import autismclient.addons.AddonManager;
import autismclient.modules.PackHideState;
import net.minecraft.client.Minecraft;
import net.minecraft.network.protocol.Packet;

import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.function.Consumer;
import java.util.function.Predicate;

public final class AddonEvents {
    private static final List<OwnedConsumer<Minecraft>> TICK = new CopyOnWriteArrayList<>();
    private static final List<OwnedPredicate<Packet<?>>> PACKET_SEND = new CopyOnWriteArrayList<>();
    private static final List<OwnedConsumer<Packet<?>>> PACKET_RECEIVE = new CopyOnWriteArrayList<>();
    private static final List<OwnedRunnable> GAME_JOIN = new CopyOnWriteArrayList<>();
    private static final List<OwnedRunnable> GAME_LEFT = new CopyOnWriteArrayList<>();

    private static final long ERROR_LOG_INTERVAL_MS = 5_000L;
    private static final Map<String, Long> LAST_ERROR_LOG_MS = new ConcurrentHashMap<>();

    private AddonEvents() {}

    public static void onTick(Consumer<Minecraft> listener) {
        String owner = ownerOrReject("tick");
        if (owner != null && listener != null) {
            TICK.add(new OwnedConsumer<>(owner, listener));
            AddonManager.recordAcceptedRegistration("event", "tick");
        }
    }

    public static void onPacketSend(Predicate<Packet<?>> listener) {
        String owner = ownerOrReject("packetSend");
        if (owner != null && listener != null) {
            PACKET_SEND.add(new OwnedPredicate<>(owner, listener));
            AddonManager.recordAcceptedRegistration("event", "packetSend");
        }
    }

    public static void onPacketReceive(Consumer<Packet<?>> listener) {
        String owner = ownerOrReject("packetReceive");
        if (owner != null && listener != null) {
            PACKET_RECEIVE.add(new OwnedConsumer<>(owner, listener));
            AddonManager.recordAcceptedRegistration("event", "packetReceive");
        }
    }

    public static void onGameJoin(Runnable listener) {
        String owner = ownerOrReject("gameJoin");
        if (owner != null && listener != null) {
            GAME_JOIN.add(new OwnedRunnable(owner, listener));
            AddonManager.recordAcceptedRegistration("event", "gameJoin");
        }
    }

    public static void onGameLeft(Runnable listener) {
        String owner = ownerOrReject("gameLeft");
        if (owner != null && listener != null) {
            GAME_LEFT.add(new OwnedRunnable(owner, listener));
            AddonManager.recordAcceptedRegistration("event", "gameLeft");
        }
    }

    private static String ownerOrReject(String event) {
        String owner = AddonManager.currentAddonId();
        if (owner == null || owner.isBlank()) {
            AutismClientAddon.LOG.warn("[Addons] Rejecting {} listener outside an addon lifecycle", event);
            AddonManager.recordRejectedRegistration(owner, "event", event, "registration outside an addon lifecycle - register from onInitialize() or onRegisterCategories()");
            return null;
        }
        return owner;
    }

    public static void unregisterAddon(String addonId) {
        if (addonId == null || addonId.isBlank()) return;
        TICK.removeIf(listener -> addonId.equals(listener.owner()));
        PACKET_SEND.removeIf(listener -> addonId.equals(listener.owner()));
        PACKET_RECEIVE.removeIf(listener -> addonId.equals(listener.owner()));
        GAME_JOIN.removeIf(listener -> addonId.equals(listener.owner()));
        GAME_LEFT.removeIf(listener -> addonId.equals(listener.owner()));
        LAST_ERROR_LOG_MS.keySet().removeIf(key -> key.startsWith(addonId + ":"));
    }

    public static void fireTick(Minecraft mc) {
        if (PackHideState.isHardLocked()) return;
        if (TICK.isEmpty()) return;
        for (OwnedConsumer<Minecraft> l : TICK) {
            try { l.listener().accept(mc); } catch (Throwable t) { logThrottled(l.owner(), "tick", t); }
        }
    }

    public static boolean hasTickListeners() {
        if (PackHideState.isHardLocked()) return false;
        return !TICK.isEmpty();
    }

    public static boolean hasPacketListeners() {
        if (PackHideState.isHardLocked()) return false;
        return !PACKET_SEND.isEmpty() || !PACKET_RECEIVE.isEmpty();
    }

    public static boolean firePacketSend(Packet<?> packet) {
        if (PackHideState.isHardLocked()) return false;
        if (PACKET_SEND.isEmpty()) return false;
        boolean cancel = false;
        for (OwnedPredicate<Packet<?>> l : PACKET_SEND) {
            try { cancel |= l.listener().test(packet); } catch (Throwable t) { logThrottled(l.owner(), "packetSend", t); }
        }
        return cancel;
    }

    public static void firePacketReceive(Packet<?> packet) {
        if (PackHideState.isHardLocked()) return;
        if (PACKET_RECEIVE.isEmpty()) return;
        for (OwnedConsumer<Packet<?>> l : PACKET_RECEIVE) {
            try { l.listener().accept(packet); } catch (Throwable t) { logThrottled(l.owner(), "packetReceive", t); }
        }
    }

    public static void fireGameJoin() {
        if (PackHideState.isHardLocked()) return;
        if (GAME_JOIN.isEmpty()) return;
        for (OwnedRunnable l : GAME_JOIN) {
            try { l.listener().run(); } catch (Throwable t) { logThrottled(l.owner(), "gameJoin", t); }
        }
    }

    public static void fireGameLeft() {
        if (PackHideState.isHardLocked()) return;
        if (GAME_LEFT.isEmpty()) return;
        for (OwnedRunnable l : GAME_LEFT) {
            try { l.listener().run(); } catch (Throwable t) { logThrottled(l.owner(), "gameLeft", t); }
        }
    }

    private static void logThrottled(String owner, String event, Throwable t) {
        long now = System.currentTimeMillis();
        String key = owner + ":" + event + ":" + t.getClass().getName();
        Long last = LAST_ERROR_LOG_MS.get(key);
        if (last != null && now - last < ERROR_LOG_INTERVAL_MS) return;
        LAST_ERROR_LOG_MS.put(key, now);
        AddonManager.recordRuntimeError(owner, event + " listener threw " + t.getClass().getSimpleName());
        AutismClientAddon.LOG.warn("[Addons] Addon '{}' {} event listener threw", owner, event, t);
    }

    private record OwnedConsumer<T>(String owner, Consumer<T> listener) {}
    private record OwnedPredicate<T>(String owner, Predicate<T> listener) {}
    private record OwnedRunnable(String owner, Runnable listener) {}
}