Added MenuBuilder#onBuild

This commit is contained in:
Auxilor
2022-11-07 16:42:40 +00:00
parent 9ee579f2c4
commit 6baf636e6a
2 changed files with 21 additions and 1 deletions

View File

@@ -195,6 +195,16 @@ public interface MenuBuilder extends PageBuilder {
return this;
}
/**
* Add an action to run on build.
*
* @param action The action.
* @return The builder.
*/
default MenuBuilder onBuild(@NotNull Consumer<Menu> action) {
return this;
}
/**
* Build the menu.
*

View File

@@ -22,6 +22,7 @@ class EcoMenuBuilder(
private val onOpen = mutableListOf<OpenHandler>()
private val onRender = mutableListOf<(Player, Menu) -> Unit>()
private val menuEventHandlers = mutableListOf<MenuEventHandler<*>>()
private val onBuild = mutableListOf<(Menu) -> Unit>()
private var allowsChangingHeldItem = false
override fun getRows() = rows
@@ -80,6 +81,11 @@ class EcoMenuBuilder(
return this
}
override fun onBuild(action: Consumer<Menu>): MenuBuilder {
onBuild += { action.accept(it) }
return this
}
override fun allowChangingHeldItem(): MenuBuilder {
allowsChangingHeldItem = true
return this
@@ -122,7 +128,7 @@ class EcoMenuBuilder(
}
}
return EcoMenu(
val menu = EcoMenu(
rows,
columns,
layeredComponents,
@@ -133,5 +139,9 @@ class EcoMenuBuilder(
menuEventHandlers,
allowsChangingHeldItem
)
onBuild.forEach { it(menu) } // Run on build functions.
return menu
}
}