package net.minecraft.world.entity;

import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;
import net.minecraft.Util;

public interface InsideBlockEffectApplier {
	InsideBlockEffectApplier NOOP = new InsideBlockEffectApplier() {
		@Override
		public void apply(InsideBlockEffectType type) {
		}

		@Override
		public void runBefore(InsideBlockEffectType type, Consumer<Entity> effect) {
		}

		@Override
		public void runAfter(InsideBlockEffectType type, Consumer<Entity> effect) {
		}
	};

	void apply(InsideBlockEffectType type);

	void runBefore(InsideBlockEffectType type, Consumer<Entity> effect);

	void runAfter(InsideBlockEffectType type, Consumer<Entity> effect);

	public static class StepBasedCollector implements InsideBlockEffectApplier {
		private static final InsideBlockEffectType[] APPLY_ORDER = InsideBlockEffectType.values();
		private static final int NO_STEP = -1;
		private final Set<InsideBlockEffectType> effectsInStep = EnumSet.noneOf(InsideBlockEffectType.class);
		private final Map<InsideBlockEffectType, List<Consumer<Entity>>> beforeEffectsInStep = Util.makeEnumMap(
			InsideBlockEffectType.class, insideBlockEffectType -> new ArrayList()
		);
		private final Map<InsideBlockEffectType, List<Consumer<Entity>>> afterEffectsInStep = Util.makeEnumMap(
			InsideBlockEffectType.class, insideBlockEffectType -> new ArrayList()
		);
		private final List<Consumer<Entity>> finalEffects = new ArrayList();
		private int lastStep = -1;

		public void advanceStep(int step) {
			if (this.lastStep != step) {
				this.lastStep = step;
				this.flushStep();
			}
		}

		public void applyAndClear(Entity entity) {
			this.flushStep();

			for (Consumer<Entity> consumer : this.finalEffects) {
				if (!entity.isAlive()) {
					break;
				}

				consumer.accept(entity);
			}

			this.finalEffects.clear();
			this.lastStep = -1;
		}

		private void flushStep() {
			for (InsideBlockEffectType insideBlockEffectType : APPLY_ORDER) {
				List<Consumer<Entity>> list = (List<Consumer<Entity>>)this.beforeEffectsInStep.get(insideBlockEffectType);
				this.finalEffects.addAll(list);
				list.clear();
				if (this.effectsInStep.remove(insideBlockEffectType)) {
					this.finalEffects.add(insideBlockEffectType.effect());
				}

				List<Consumer<Entity>> list2 = (List<Consumer<Entity>>)this.afterEffectsInStep.get(insideBlockEffectType);
				this.finalEffects.addAll(list2);
				list2.clear();
			}
		}

		@Override
		public void apply(InsideBlockEffectType type) {
			this.effectsInStep.add(type);
		}

		@Override
		public void runBefore(InsideBlockEffectType type, Consumer<Entity> effect) {
			((List)this.beforeEffectsInStep.get(type)).add(effect);
		}

		@Override
		public void runAfter(InsideBlockEffectType type, Consumer<Entity> effect) {
			((List)this.afterEffectsInStep.get(type)).add(effect);
		}
	}
}