Enable pressing shift to take 10 items

OCB7D2D · Sep 20, 2024 · 75c6335 · 75c6335
1 parent d75f72e
commit 75c6335
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/Harmony/XUiC_WheelItemStack.cs b/Harmony/XUiC_WheelItemStack.cs
@@ -18,10 +18,13 @@ public override void Init()
 
     private void SetDnD(ItemStack stack) => xui.dragAndDrop.CurrentStack = stack;
 
+    static bool IsShiftPressed => Input.GetKey(KeyCode.LeftShift) || Input.GetKey(KeyCode.RightShift);
+
     protected virtual int TransferItems(int amount,
         ItemStack src, Action<ItemStack> setSrc,
         ItemStack dst, Action<ItemStack> setDst)
     {
+        if (IsShiftPressed) amount *= 10;
         if (WheelItemStack.TransferItems(amount, src, setSrc, dst, setDst) > 0)
         {
             // Copied from vanilla

diff --git a/Harmony/XUiC_WheelRequiredItemStack.cs b/Harmony/XUiC_WheelRequiredItemStack.cs
@@ -13,10 +13,13 @@ public class XUiC_WheelRequiredItemStack : XUiC_RequiredItemStack
 
     private void SetDnD(ItemStack stack) => xui.dragAndDrop.CurrentStack = stack;
 
+    static bool IsShiftPressed => Input.GetKey(KeyCode.LeftShift) || Input.GetKey(KeyCode.RightShift);
+
     protected virtual int TransferItems(int amount,
         ItemStack src, Action<ItemStack> setSrc,
         ItemStack dst, Action<ItemStack> setDst)
     {
+        if (IsShiftPressed) amount *= 10;
         if (WheelItemStack.TransferItems(amount, src, setSrc, dst, setDst) > 0)
         {
             // Copied from vanilla

diff --git a/SmxCompat/XUiC_WheelQuartzItemStack.cs b/SmxCompat/XUiC_WheelQuartzItemStack.cs
@@ -38,10 +38,13 @@ public override void Init()
 
     private void SetDnD(ItemStack stack) => xui.dragAndDrop.CurrentStack = stack;
 
+    static bool IsShiftPressed => Input.GetKey(KeyCode.LeftShift) || Input.GetKey(KeyCode.RightShift);
+
     protected virtual int TransferItems(int amount,
         ItemStack src, Action<ItemStack> setSrc,
         ItemStack dst, Action<ItemStack> setDst)
     {
+        if (IsShiftPressed) amount *= 10;
         if (WheelItemStack.TransferItems(amount, src, setSrc, dst, setDst) > 0)
         {
             // Copied from vanilla