From 60e6130ece1ad9dbc377cce9efe0252da8524288 Mon Sep 17 00:00:00 2001
From: Sebastian Ullrich <sebasti@nullri.ch>
Date: Sat, 24 Jul 2021 16:36:05 +0200
Subject: [PATCH] chore: Nix: expose benchmarks as flake attributes for
 convenience

---
 nix/bootstrap.nix | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/nix/bootstrap.nix b/nix/bootstrap.nix
index b45d87dc62..c406609d82 100644
--- a/nix/bootstrap.nix
+++ b/nix/bootstrap.nix
@@ -1,6 +1,7 @@
 { debug ? false, stage0debug ? false, extraCMakeFlags ? [],
   stdenv, lib, cmake, gmp, gnumake, bash, buildLeanPackage, writeShellScriptBin, runCommand, symlinkJoin, lndir,
   ... } @ args:
+with builtins;
 rec {
   inherit stdenv;
   buildCMake = args: stdenv.mkDerivation ({
@@ -130,6 +131,14 @@ rec {
         ${update-stage0}/bin/update-stage0
         git commit -m "chore: update stage0"
       '';
+      benchmarks =
+        let
+          entries = attrNames (readDir ../tests/bench);
+          leanFiles = map (n: elemAt n 0) (filter (n: n != null) (map (match "(.*)\.lean") entries));
+        in lib.genAttrs leanFiles (n: (buildLeanPackage {
+          name = n;
+          src = filterSource (e: _: baseNameOf e == "${n}.lean") ../tests/bench;
+        }).executable);
     };
   stage1 = stage { stage = 1; prevStage = stage0; self = stage1; };
   stage2 = stage { stage = 2; prevStage = stage1; self = stage2; };