diff --git a/lib.nix b/lib.nix
index 0b9f6d9..537d6d2 100644
--- a/lib.nix
+++ b/lib.nix
@@ -168,6 +168,17 @@ let
         type = flake-parts-lib.mkPerSystemType module;
       };
 
+    findInputByOutPath = outPath:
+      lib.attrsets.concatMapAttrs
+      (inputName: input:
+        (
+          if lib.strings.hasPrefix input.outPath "${outPath}" then
+            input
+          else 
+            findCurrentInput outPath input.inputs or { }
+        )
+      );
+
     # Helper function for defining a per-system option that
     # gets transposed by the usual flake system logic to a
     # top-level flake attribute.