From 1a0cb36739969e4d04dd0e49f8c123e4ed120ba4 Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Fri, 3 Dec 2021 20:54:01 +0300
Subject: [PATCH] More rules

---
 .../bclib/api/biomes/SurfaceRuleBuilder.java  | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/main/java/ru/bclib/api/biomes/SurfaceRuleBuilder.java b/src/main/java/ru/bclib/api/biomes/SurfaceRuleBuilder.java
index af29fc50..5a6e2001 100644
--- a/src/main/java/ru/bclib/api/biomes/SurfaceRuleBuilder.java
+++ b/src/main/java/ru/bclib/api/biomes/SurfaceRuleBuilder.java
@@ -89,6 +89,40 @@ public class SurfaceRuleBuilder {
 		return this;
 	}
 	
+	/**
+	 * Set biome ceiling with specified {@link BlockState}. Example - block of sandstone in the Overworld desert in air pockets.
+	 * @param state {@link BlockState} for the ground cover.
+	 * @return same {@link SurfaceRuleBuilder} instance.
+	 */
+	public SurfaceRuleBuilder ceil(BlockState state) {
+		entryInstance = getFromCache("ceil_" + state.toString(), () -> {
+			RuleSource rule = SurfaceRules.state(state);
+			return new SurfaceRuleEntry(1, SurfaceRules.ifTrue(SurfaceRules.ON_CEILING, rule));
+		});
+		rules.add(entryInstance);
+		return this;
+	}
+	
+	/**
+	 * Allows to add custom rule.
+	 * @param priority rule priority, lower values = higher priority (rule will be applied before others).
+	 * @param rule custom {@link SurfaceRules.RuleSource}.
+	 * @return same {@link SurfaceRuleBuilder} instance.
+	 */
+	public SurfaceRuleBuilder rule(int priority, SurfaceRules.RuleSource rule) {
+		rules.add(new SurfaceRuleEntry(priority, rule));
+		return this;
+	}
+	
+	/**
+	 * Allows to add custom rule.
+	 * @param rule
+	 * @return
+	 */
+	public SurfaceRuleBuilder rule(SurfaceRules.RuleSource rule) {
+		return rule(7, rule);
+	}
+	
 	/**
 	 * Finalise rule building process.
 	 * @return {@link SurfaceRules.RuleSource}.