1 files changed, 30 insertions, 1 deletions
diff --git a/flake.nix b/flake.nix
index 7299b4b..9f315ac 100644
--- a/flake.nix
+++ b/flake.nix
@@ -20,7 +20,36 @@
     pkgs = import nixpkgs { system = system; overlays = [ mcl-blockchain.overlays.default ]; };
     callPackage = pkgs.lib.callPackageWith pkgs;
 
+    withZKVMPhases = currentPackage: with currentPackage; {
+        buildPhase = ''
+          root_dir="$PWD"
+          runHook preBuild
+          cd "$root_dir"
+
+          export INPUTS="$PWD/Vertices-010.in"
+          cargo build --bin ${hostBin} --release
+
+          runHook postBuild
+        '';
+
+        installPhase = ''
+          runHook preInstall
+          mkdir -p "$out"/bin
+          for bin in $(find . -type f -regex ".*release/[^/]*" -executable -print)
+          do
+            mv "$bin" "$out"/bin/
+          done
+          runHook postInstall
+        '';
+
+        doNotPostBuildInstallCargoBinaries = true;
+      } // currentPackage;
+
+    args-zkVM = {
+      craneLib-default = crane.mkLib pkgs;
+      inherit withZKVMPhases;
+    };
   in {
-    packages.${system}.risc0 = callPackage ./zkvms/risc0/default.nix {craneLib-default = crane.mkLib pkgs;};
+    packages.${system}.risc0 = callPackage ./zkvms/risc0/default.nix args-zkVM;
   };
 }