Merge pull request #13 from taniwha/docking-jointlock

Allow ModuleDockingNode if only IJointLockState
DBooots · Aug 14, 2022 · c44f87c · c44f87c
2 parents 1bc40a5 + 986c789
commit c44f87c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/ReCoupler/ReCouplerUtils.cs b/ReCoupler/ReCouplerUtils.cs
@@ -432,8 +432,18 @@ public static bool PartIsInvalidForPath(Part part, bool allowRoboJoints, bool al
             if (!allowKASJoints && isKASPart(part))
                 return true;
             // Accounts for ModuleGrappleNode and all of the Breaking Ground joints and servos.
-            if (!allowRoboJoints && !isKASPart(part) && (part.FindModuleImplementing<IJointLockState>() != null))
-                return true;
+            if (!allowRoboJoints && !isKASPart(part)) {
+                var mods =  part.FindModulesImplementing<IJointLockState>();
+                bool haveNonDockingNode = false;
+                for (int i = mods.Count; i-- > 0; ) {
+                    if (!(mods[i] is ModuleDockingNode)) {
+                        haveNonDockingNode = true;
+                        break;
+                    }
+                }
+                if (haveNonDockingNode)
+                    return true;
+            }
             return false;
         }