Menu description pt. 1

1 files changed, 3 insertions, 2 deletions

diff --git a/engine/UI/Button.go b/engine/UI/Button.go
index a8b9168..f865c5f 100644
--- a/engine/UI/Button.go
+++ b/engine/UI/Button.go
@@ -16,6 +16,7 @@ type Button struct {
 	DisplayElement UIElement
 	EventType DetectionType
 	ButtonType rl.MouseButton
+	OnClick func()
 
 	clicked func(rl.MouseButton) bool
 }
@@ -43,10 +44,10 @@ func (base *Button) GetOccupationWeight() float32 {
 
 func (base *Button) Draw(position *rl.Rectangle) {
 	base.DisplayElement.Draw(position)
-	if base.clicked(base.ButtonType) {
+	if base.OnClick != nil && base.clicked(base.ButtonType) {
 		var mousePos = rl.GetMousePosition()
 		if rl.CheckCollisionPointRec(mousePos, *position) {
-			rl.TraceLog(rl.LogInfo, "xdx")
+			base.OnClick()
 		}
 	}
 }