From 617c02b2d656f59beed742ccdccac53562eca9c9 Mon Sep 17 00:00:00 2001
From: Ravis96 <ravis96@spoko.pl>
Date: Mon, 1 Jul 2024 16:07:40 +0200
Subject: [PATCH] Add fill-item builder methods

---
 .../cc/dreamcode/menu/DreamMenuBuilder.java   | 66 +++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/core/src/main/java/cc/dreamcode/menu/DreamMenuBuilder.java b/core/src/main/java/cc/dreamcode/menu/DreamMenuBuilder.java
index cb5ae01..4a073ae 100644
--- a/core/src/main/java/cc/dreamcode/menu/DreamMenuBuilder.java
+++ b/core/src/main/java/cc/dreamcode/menu/DreamMenuBuilder.java
@@ -8,6 +8,7 @@
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.stream.Stream;
 
 @Getter
 @RequiredArgsConstructor
@@ -39,6 +40,71 @@ public DreamMenuBuilder<B, T, I> fillBackground(@NonNull I i) {
         return this;
     }
 
+    public DreamMenuBuilder<B, T, I> fillTop(@NonNull I i) {
+        Stream.of(
+                MenuUtil.countSlot(1, 1),
+                MenuUtil.countSlot(1, 2),
+                MenuUtil.countSlot(1, 3),
+                MenuUtil.countSlot(1, 4),
+                MenuUtil.countSlot(1, 5),
+                MenuUtil.countSlot(1, 6),
+                MenuUtil.countSlot(1, 7),
+                MenuUtil.countSlot(1, 8),
+                MenuUtil.countSlot(1, 9)
+        ).forEach(slot -> {
+            if (!this.items.containsKey(slot)) {
+                this.items.put(slot, i);
+            }
+        });
+
+        return this;
+    }
+
+    public DreamMenuBuilder<B, T, I> fillBottom(@NonNull I i) {
+        Stream.of(
+                MenuUtil.countSlot(this.rows, 1),
+                MenuUtil.countSlot(this.rows, 2),
+                MenuUtil.countSlot(this.rows, 3),
+                MenuUtil.countSlot(this.rows, 4),
+                MenuUtil.countSlot(this.rows, 5),
+                MenuUtil.countSlot(this.rows, 6),
+                MenuUtil.countSlot(this.rows, 7),
+                MenuUtil.countSlot(this.rows, 8),
+                MenuUtil.countSlot(this.rows, 9)
+        ).forEach(slot -> {
+            if (!this.items.containsKey(slot)) {
+                this.items.put(slot, i);
+            }
+        });
+
+        return this;
+    }
+
+    public DreamMenuBuilder<B, T, I> fillTopAndBottom(@NonNull I i) {
+        this.fillTop(i);
+        this.fillBottom(i);
+
+        return this;
+    }
+
+    public DreamMenuBuilder<B, T, I> fillMargin(@NonNull I i) {
+        this.fillTop(i);
+        this.fillBottom(i);
+
+        for (int row = 2; row < this.rows; row++) {
+            Stream.of(
+                    MenuUtil.countSlot(row, 1),
+                    MenuUtil.countSlot(row, 9)
+            ).forEach(slot -> {
+                if (!this.items.containsKey(slot)) {
+                    this.items.put(slot, i);
+                }
+            });
+        }
+
+        return this;
+    }
+
     public abstract B buildEmpty();
     public abstract B buildEmpty(@NonNull Map<String, Object> replaceMap);
     public abstract B buildWithItems();