feat(nix/readTree): allow arguments to depend on tree location r/1586

This accepts a function for args, and passes it an attrset containing
the locatedAt parameter.

Change-Id: I3c0f8ca00605e02a787fda88f32b06f5ef9998f2
Reviewed-on: https://cl.tvl.fyi/c/depot/+/1639
Tested-by: BuildkiteCI
Reviewed-by: tazjin <mail@tazj.in>

1 files changed, 3 insertions, 3 deletions

diff --git a/nix/readTree/default.nix b/nix/readTree/default.nix
index 4faf20d9e5c0..9793310eeda0 100644
--- a/nix/readTree/default.nix
+++ b/nix/readTree/default.nix
@@ -17,9 +17,9 @@ let
     readDir
     substring;
 
-  argsWithPath = parts: args // {
-    locatedAt = parts;
-  };
+  argsWithPath = parts:
+    let meta.locatedAt = parts;
+    in meta // (if isAttrs args then args else args meta);
 
   readDirVisible = path:
     let