1 files changed, 12 insertions, 1 deletions
diff --git a/lib/by-name-overlay.nix b/lib/by-name-overlay.nix
index cc8bcb67..4d700a8f 100644
--- a/lib/by-name-overlay.nix
+++ b/lib/by-name-overlay.nix
@@ -5,6 +5,8 @@
   baseDirectory,
   fileName,
   finalizeFunction,
+  coImportsNameFunction ? null,
+  coImportsWarnMessageObject ? null,
 }: let
   # Takes a list of attrs as input and returns one merged attr set.
   flattenAttrs = list:
@@ -44,9 +46,18 @@
     else let
       mkPath = name: _type: let
         path = baseDirectory + "/${shard}/${name}" + "/${fileName}";
+        coImportPath =
+          if coImportsNameFunction != null
+          then
+            coImportsNameFunction
+            {inherit shard name;}
+          else path;
       in
         if builtins.pathExists path
-        then path
+        then
+          if builtins.pathExists coImportPath
+          then path
+          else warn "'${builtins.toString coImportPath}' does not exist. Should include ${coImportsWarnMessageObject} for '${shard}/${name}'" path
         else warn "'${builtins.toString path}' does not exist. Skipped" null;
     in
       filterAttrs (name: value: value != null)