70 lines
1.4 KiB
Nix
70 lines
1.4 KiB
Nix
{
|
|
stdenv
|
|
, bash
|
|
, writeText
|
|
}:
|
|
|
|
{
|
|
stages
|
|
, conf
|
|
}:
|
|
|
|
with stdenv.lib;
|
|
|
|
let
|
|
|
|
dStages = foldr (stageFn: {conf, prevStage, stages}: {
|
|
conf = conf;
|
|
prevStage = stageFn {nextStage=prevStage; conf=conf;};
|
|
stages = [ (stageFn {nextStage=prevStage; conf=conf;}) ] ++ stages;
|
|
})
|
|
{conf=conf; stages=[]; prevStage=null;} stages;
|
|
|
|
stageProgram = stage:
|
|
if stage ? programPath
|
|
then "${stage}${stage.programPath}" else "${stage}";
|
|
|
|
linkStages = imap1 (i: s: {
|
|
num = "${toString i}";
|
|
name = "${toString i}-${baseNameOf s.name}";
|
|
stage = s;
|
|
programPath = stageProgram s;
|
|
}) dStages.stages;
|
|
|
|
desc = builtins.concatStringsSep "\n"
|
|
(map (x: "# ${x.stage}") linkStages);
|
|
|
|
|
|
firstStage = (x: x.programPath) (elemAt linkStages 0);
|
|
|
|
jsonConf = writeText "garlic_config.json" (builtins.toJSON conf);
|
|
in
|
|
stdenv.mkDerivation {
|
|
name = "unit";
|
|
preferLocalBuild = true;
|
|
phases = [ "installPhase" ];
|
|
inherit desc;
|
|
installPhase = ''
|
|
cat > $out << EOF
|
|
#!/bin/sh -e
|
|
|
|
${desc}
|
|
|
|
# Set the experiment unit in the environment
|
|
export GARLIC_UNIT=$(basename $out)
|
|
|
|
# And change the working directory
|
|
mkdir \$GARLIC_UNIT
|
|
cd \$GARLIC_UNIT
|
|
|
|
# Copy the configuration for the unit to the output path
|
|
cp ${jsonConf} garlic_config.json
|
|
|
|
# Finally, execute the first stage:
|
|
exec ${firstStage}
|
|
EOF
|
|
|
|
chmod +x $out
|
|
'';
|
|
}
|