package me.xmrvizzy.skyblocker.skyblock.experiment;

import me.xmrvizzy.skyblocker.config.SkyblockerConfig;
import me.xmrvizzy.skyblocker.utils.render.gui.ContainerSolver;
import net.fabricmc.fabric.api.client.screen.v1.ScreenEvents;
import net.minecraft.client.gui.screen.Screen;
import net.minecraft.client.gui.screen.ingame.GenericContainerScreen;
import net.minecraft.item.ItemStack;

import java.util.HashMap;
import java.util.Map;

public abstract class ExperimentSolver extends ContainerSolver {
    public enum State {
        REMEMBER, WAIT, SHOW, END
    }

    private State state = State.REMEMBER;
    private final Map<Integer, ItemStack> slots = new HashMap<>();

    protected ExperimentSolver(String containerName) {
        super(containerName);
    }

    public State getState() {
        return state;
    }

    public void setState(State state) {
        this.state = state;
    }

    public Map<Integer, ItemStack> getSlots() {
        return slots;
    }

    @Override
    protected final boolean isEnabled() {
        return isEnabled(SkyblockerConfig.get().general.experiments);
    }

    protected abstract boolean isEnabled(SkyblockerConfig.Experiments experimentsConfig);

    @Override
    protected void start(GenericContainerScreen screen) {
        super.start(screen);
        state = State.REMEMBER;
        ScreenEvents.afterTick(screen).register(this::tick);
    }

    @Override
    protected void reset() {
        super.reset();
        state = State.REMEMBER;
        slots.clear();
    }

    protected abstract void tick(Screen screen);
}