Migrate to FluidSlotWidget from FluidDisplaySlotWidget (#1963)

1 files changed, 17 insertions, 0 deletions

diff --git a/src/main/java/gregtech/common/gui/modularui/widget/FluidLockWidget.java b/src/main/java/gregtech/common/gui/modularui/widget/FluidLockWidget.java
new file mode 100644
index 0000000000..5627a4fb8b
--- /dev/null
+++ b/src/main/java/gregtech/common/gui/modularui/widget/FluidLockWidget.java
@@ -0,0 +1,17 @@
+package gregtech.common.gui.modularui.widget;
+
+import com.gtnewhorizons.modularui.common.widget.FluidNameHolderWidget;
+
+import gregtech.api.interfaces.metatileentity.IFluidLockable;
+
+public class FluidLockWidget extends FluidNameHolderWidget {
+
+    public FluidLockWidget(IFluidLockable fluidLockable) {
+        super(fluidLockable::getLockedFluidName, name -> {
+            if (fluidLockable.acceptsFluidLock(name)) {
+                fluidLockable.setLockedFluidName(name);
+                fluidLockable.lockFluid(name != null);
+            }
+        });
+    }
+}