Improve Invalid Setup Detection

src/main/java/com/nomiceu/nomilabs/mixin/betterp2p/WidgetP2PColumnMixin.java

@@ -6,6 +6,7 @@
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 import com.nomiceu.nomilabs.integration.betterp2p.LabsBetterMemoryCardModes;
 import com.nomiceu.nomilabs.network.LabsNetworkHandler;
@@ -20,7 +21,8 @@
 import kotlin.jvm.functions.Function0;
 
 /**
- * Trims text before renaming, handles add as input/output, properly refreshes renames in gui.
+ * Trims text before renaming, handles add as input/output, properly refreshes renames in gui,
+ * properly searches for inputs/outputs.
  */
 @Mixin(value = WidgetP2PColumn.class, remap = false)
 public class WidgetP2PColumnMixin {
@@ -37,6 +39,28 @@ public class WidgetP2PColumnMixin {
     @Final
     private IGuiTextField renameBar;
 
+    @Inject(method = "findInput", at = @At("HEAD"), cancellable = true)
+    private void findInputInAll(Short frequency, CallbackInfoReturnable<InfoWrapper> cir) {
+        for (InfoWrapper info : infos.getSorted()) {
+            if (info.getFrequency() == frequency && !info.getOutput()) {
+                cir.setReturnValue(info);
+                return;
+            }
+        }
+        cir.setReturnValue(null);
+    }
+
+    @Inject(method = "findOutput", at = @At("HEAD"), cancellable = true)
+    private void findOutputInAll(Short frequency, CallbackInfoReturnable<InfoWrapper> cir) {
+        for (InfoWrapper info : infos.getSorted()) {
+            if (info.getFrequency() == frequency && info.getOutput()) {
+                cir.setReturnValue(info);
+                return;
+            }
+        }
+        cir.setReturnValue(null);
+    }
+
     @Inject(method = "finishRename", at = @At("HEAD"))
     private void trimText(CallbackInfo ci) {
         renameBar.setText(renameBar.getText().trim());