Passed strings will be treated as a relative path below the given root, which is quite convenient when using depot.path by eliminating a lot of repetition. Change-Id: I3da6058094484f4a6ffbb84f89ad4472b502a00c Reviewed-on: https://cl.tvl.fyi/c/depot/+/3704 Tested-by: BuildkiteCI Reviewed-by: tazjin <mail@tazj.in>
		
			
				
	
	
		
			70 lines
		
	
	
	
		
			2.3 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			70 lines
		
	
	
	
		
			2.3 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
# Build a “sparse” version of a given directory, only including contained files
 | 
						|
# and directories if they are listed in a supplied list:
 | 
						|
#
 | 
						|
# # A very minimal depot
 | 
						|
# sparseTree ./depot [
 | 
						|
#   ./default.nix
 | 
						|
#   ./depot/nix/readTree/default.nix
 | 
						|
#   ./third_party/nixpkgs
 | 
						|
#   ./third_party/overlays
 | 
						|
# ]
 | 
						|
{ pkgs, lib, ... }:
 | 
						|
 | 
						|
# root path to use as a reference point
 | 
						|
root:
 | 
						|
# list of paths below `root` that should be
 | 
						|
# included in the resulting directory
 | 
						|
#
 | 
						|
# If path, need to refer to the actual file / directory to be included.
 | 
						|
# If a string, it is treated as a string relative to the root.
 | 
						|
paths:
 | 
						|
 | 
						|
let
 | 
						|
  rootLength = builtins.stringLength (toString root);
 | 
						|
 | 
						|
  # Count slashes in a path.
 | 
						|
  #
 | 
						|
  # Type: path -> int
 | 
						|
  depth = path: lib.pipe path [
 | 
						|
    toString
 | 
						|
    (builtins.split "/")
 | 
						|
    (builtins.filter builtins.isList)
 | 
						|
    builtins.length
 | 
						|
  ];
 | 
						|
 | 
						|
  # (Parent) directories will be created from deepest to shallowest
 | 
						|
  # which should mean no conflicts are caused unless both a child
 | 
						|
  # and its parent directory are in the list of paths.
 | 
						|
  # TODO(sterni): improve error messages in such cases
 | 
						|
  fromDeepest = lib.sort (a: b: depth a < depth b) paths;
 | 
						|
 | 
						|
  # Create a set which contains the source path to copy / symlink and
 | 
						|
  # it's destination, so the path below the destination root including
 | 
						|
  # a leading slash. Additionally some sanity checking is done.
 | 
						|
  makeSymlink = path:
 | 
						|
    let
 | 
						|
      withLeading = p: if builtins.substring 0 1 p == "/" then p else "/" + p;
 | 
						|
      fullPath =
 | 
						|
        /**/ if builtins.isPath path then path
 | 
						|
        else if builtins.isString path then (root + withLeading path)
 | 
						|
        else builtins.throw "Unsupported path type ${builtins.typeOf path}";
 | 
						|
      strPath = toString fullPath;
 | 
						|
      contextPath = "${fullPath}";
 | 
						|
      belowRoot = builtins.substring rootLength (-1) strPath;
 | 
						|
      prefix = builtins.substring 0 rootLength strPath;
 | 
						|
    in assert toString root == prefix; {
 | 
						|
      src = contextPath;
 | 
						|
      dst = belowRoot;
 | 
						|
    };
 | 
						|
 | 
						|
  symlinks = builtins.map makeSymlink fromDeepest;
 | 
						|
in
 | 
						|
 | 
						|
# TODO(sterni): teach readTree to also read symlinked directories,
 | 
						|
# so we ln -sT instead of cp -aT.
 | 
						|
pkgs.runCommandNoCC "sparse-${builtins.baseNameOf root}" {} (
 | 
						|
  lib.concatMapStrings ({ src, dst }: ''
 | 
						|
    mkdir -p "$(dirname "$out${dst}")"
 | 
						|
    cp -aT --reflink=auto "${src}" "$out${dst}"
 | 
						|
  '') symlinks
 | 
						|
)
 |