TerrainAPI Lock

UselessSolutions · Nov 13, 2023 · bd102c3 · bd102c3
1 parent 5702766
commit bd102c3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/main/java/useless/terrainapi/TerrainMain.java b/src/main/java/useless/terrainapi/TerrainMain.java
@@ -17,12 +17,17 @@ public class TerrainMain implements ModInitializer {
 	public static final Gson GSON = (new GsonBuilder()).setFieldNamingPolicy(FieldNamingPolicy.LOWER_CASE_WITH_UNDERSCORES).setPrettyPrinting().excludeFieldsWithoutExposeAnnotation().create();
     public static final String MOD_ID = "terrain-api";
     public static final Logger LOGGER = LoggerFactory.getLogger(MOD_ID);
+	/**
+	 * Prevents accidental use of API Features outside the proper initialization environment
+	 */
+	public static boolean LOCK_API = true;
     @Override
     public void onInitialize() {
         LOGGER.info("TerrainMain initialized.");
     }
 	@ApiStatus.Internal
 	public static void loadModules(){
+		LOCK_API = false;
 		new TerrainInitialization().onInitialize();
 		FabricLoader.getInstance().getEntrypoints("terrain-api", TerrainAPI.class).forEach(api -> {
 			// Make sure the method is implemented
@@ -33,5 +38,6 @@ public static void loadModules(){
 			}
 		});
 		ConfigManager.saveAll();
+		LOCK_API = true;
 	}
 }
diff --git a/src/main/java/useless/terrainapi/generation/GeneratorFeatures.java b/src/main/java/useless/terrainapi/generation/GeneratorFeatures.java
@@ -2,6 +2,7 @@
 
 import net.minecraft.core.world.generate.feature.WorldFeature;
 import org.jetbrains.annotations.ApiStatus;
+import useless.terrainapi.TerrainMain;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -24,6 +25,9 @@ public class GeneratorFeatures {
 	 * @param densityParameters Object[] of additional parameters that will be included with the Parameters object passed into the density function
 	 */
 	public void addFeature(Function<Parameters, WorldFeature> featureFunction, Object[] featureParameters, Function<Parameters, Integer> densityFunction, Object[] densityParameters){
+		if (TerrainMain.LOCK_API){
+			throw new RuntimeException("Illegal use of Terrain Initialization outside of the main terrain-api entrypoint loop!!!");
+		}
 		featureFunctionsList.add(featureFunction);
 		featureParametersList.add(featureParameters);
 		densityFunctionsList.add(densityFunction);