CompleteConfig/lib/src/main/java/me/lortseam/completeconfig/gui/ConfigScreenBuilder.java

package me.lortseam.completeconfig.gui;

import me.lortseam.completeconfig.data.Config;
import me.lortseam.completeconfig.gui.cloth.ClothConfigScreenBuilder;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.fabricmc.loader.api.FabricLoader;
import net.minecraft.client.gui.screen.Screen;

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

public abstract class ConfigScreenBuilder {

    private static final ConfigScreenBuilder fallback = FabricLoader.getInstance().isModLoaded("cloth-config2") ? new ClothConfigScreenBuilder() : null;
    private static final Map<String, ConfigScreenBuilder> mainBuilders = new HashMap<>();

    /**
     * Sets the main screen builder for a mod. The main screen builder will be used to build the config screen if no
     * custom builder was specified.
     *
     * @param modID the mod's ID
     * @param screenBuilder the screen builder
     */
    public static void setMain(String modID, ConfigScreenBuilder screenBuilder) {
        mainBuilders.put(modID, screenBuilder);
    }

    public static Optional<ConfigScreenBuilder> getMain(String modID) {
        if (mainBuilders.containsKey(modID)) {
            return Optional.of(mainBuilders.get(modID));
        }
        if (fallback != null) {
            return Optional.of(fallback);
        }
        return Optional.empty();
    }

    /**
     * Tries to build a screen based on a config. Fails if no main screen builder was specified for the mod and no
     * fallback builder exists.
     *
     * @param parentScreen the parent screen
     * @param config the config to build the screen of
     * @return the built screen, if successful
     */
    public static Screen tryBuild(Screen parentScreen, Config config) {
        return getMain(config.getMod().getId()).orElseThrow(() -> {
            return new UnsupportedOperationException("No screen builder found for mod " + config.getMod().getId());
        }).build(parentScreen, config);
    }

    /**
     * Builds a screen based on a config.
     *
     * @param parentScreen the parent screen
     * @param config the config to build the screen of
     * @return the built screen
     */
    @Environment(EnvType.CLIENT)
    public abstract Screen build(Screen parentScreen, Config config);

}