From 90824374ab69f8ec2386903b07bb7a3678d9a762 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Mon, 13 Jan 2025 11:43:11 +0200
Subject: feat: Add script for running a zkVM

---
 flake.nix               | 9 +++++++++
 zkvms/risc0/default.nix | 7 ++-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/flake.nix b/flake.nix
index 9f315ac..a28d708 100644
--- a/flake.nix
+++ b/flake.nix
@@ -34,11 +34,20 @@
 
         installPhase = ''
           runHook preInstall
+
           mkdir -p "$out"/bin
           for bin in $(find . -type f -regex ".*release/[^/]*" -executable -print)
           do
             mv "$bin" "$out"/bin/
           done
+
+          cat <<EOF > "$out"/bin/${pname}
+          #!/usr/bin/env sh
+          ${if currentPackage ? preRun then preRun else ""}
+          "$out"/bin/${hostBin} \$@
+          EOF
+          chmod +x "$out"/bin/${pname}
+
           runHook postInstall
         '';
 
diff --git a/zkvms/risc0/default.nix b/zkvms/risc0/default.nix
index c491217..f9df4ae 100644
--- a/zkvms/risc0/default.nix
+++ b/zkvms/risc0/default.nix
@@ -34,10 +34,7 @@ in
       inherit cargoArtifacts;
 
       nativeBuildInputs = [
-        just
         metacraft-labs.risc0
-        stdenv.cc
-        pkg-config
       ];
 
       postPatch = ''
@@ -52,8 +49,8 @@ in
 
       hostBin = "host-risc0";
 
-      postInstall = ''
-        ln -s "${metacraft-labs.risc0}"/bin/r0vm "$out"/bin/r0vm
+      preRun = ''
+        export PATH="\$PATH:${metacraft-labs.risc0}/bin"
       '';
 
       doCheck = false;
-- 
cgit v1.2.3