From cb0d24d0fdcdd86587f06fd8683a73995daefed6 Mon Sep 17 00:00:00 2001
From: isXander <xander@isxander.dev>
Date: Mon, 29 May 2023 07:41:20 +0100
Subject: Fix ListOption still using old controller system

---
 .../src/main/java/dev/isxander/yacl/impl/ListOptionImpl.java  | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'common/src/main/java/dev/isxander/yacl/impl')

diff --git a/common/src/main/java/dev/isxander/yacl/impl/ListOptionImpl.java b/common/src/main/java/dev/isxander/yacl/impl/ListOptionImpl.java
index 015a7c3..e81d702 100644
--- a/common/src/main/java/dev/isxander/yacl/impl/ListOptionImpl.java
+++ b/common/src/main/java/dev/isxander/yacl/impl/ListOptionImpl.java
@@ -3,6 +3,7 @@ package dev.isxander.yacl.impl;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableSet;
 import dev.isxander.yacl.api.*;
+import dev.isxander.yacl.api.controller.ControllerBuilder;
 import net.minecraft.network.chat.Component;
 import org.apache.commons.lang3.Validate;
 import org.jetbrains.annotations.ApiStatus;
@@ -239,7 +240,15 @@ public final class ListOptionImpl<T> implements ListOption<T> {
         }
 
         @Override
-        public Builder<T> controller(@NotNull Function<ListOptionEntry<T>, Controller<T>> control) {
+        public Builder<T> controller(@NotNull Function<Option<T>, ControllerBuilder<T>> controller) {
+            Validate.notNull(controller, "`controller` cannot be null");
+
+            this.controllerFunction = opt -> controller.apply(opt).build();
+            return this;
+        }
+
+        @Override
+        public Builder<T> customController(@NotNull Function<ListOptionEntry<T>, Controller<T>> control) {
             Validate.notNull(control, "`control` cannot be null");
 
             this.controllerFunction = control;
-- 
cgit