Minestom/src/main/java/net/minestom/server/event/EventInterface.java

package net.minestom.server.event;

import org.jetbrains.annotations.NotNull;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.function.BiConsumer;
import java.util.function.Predicate;

public interface EventInterface<E extends Event> {

    static <E extends Event, T> @NotNull FilteredBuilder<E, T> filtered(@NotNull EventFilter<E, T> filter, @NotNull Predicate<T> predicate) {
        return new FilteredBuilder<>(filter, predicate);
    }

    @NotNull Collection<Class<? extends Event>> eventTypes();

    void call(@NotNull E event);

    class FilteredBuilder<E extends Event, T> {
        private final EventFilter<E, T> filter;
        private final Predicate<T> predicate;
        private final Map<Class<? extends Event>, BiConsumer<Object, E>> mapped = new HashMap<>();

        FilteredBuilder(EventFilter<E, T> filter, Predicate<T> predicate) {
            this.filter = filter;
            this.predicate = predicate;
        }

        public <M extends E> FilteredBuilder<E, T> map(@NotNull Class<M> eventType,
                                                       @NotNull BiConsumer<@NotNull T, @NotNull M> consumer) {
            this.mapped.put(eventType, (BiConsumer<Object, E>) consumer);
            return this;
        }

        public @NotNull EventInterface<E> build() {
            final var copy = Map.copyOf(mapped);
            final var eventTypes = copy.keySet();
            return new EventInterface<>() {
                @Override
                public @NotNull Collection<Class<? extends Event>> eventTypes() {
                    return eventTypes;
                }

                @Override
                public void call(@NotNull E event) {
                    final T handler = filter.getHandler(event);
                    if (!predicate.test(handler)) return;
                    final var consumer = copy.get(event.getClass());
                    consumer.accept(handler, event);
                }
            };
        }
    }
}