feat(flake.nix): Add script for updating all zkVM package dependencies

All crates use the zkVM packages (from nix-blockchain-development) as
cargo path dependencies. For each zkVM, they're found in the host, guest
and wrapper crates.

Since there are a lot of places to update, we're introducing a script
which automates this.

2 files changed, 38 insertions, 0 deletions

diff --git a/flake.nix b/flake.nix
index 903acb4..21e9694 100644
--- a/flake.nix
+++ b/flake.nix
@@ -75,6 +75,7 @@
     in {
       packages.${system} = hostPackages // guestPackages // {
         rust-format-all = callPackage ./rust-format-all.nix { };
+        update-nix-dependencies = callPackage ./update-nix-dependencies.nix { };
       };
 
       formatter.${system} = pkgs.nixfmt;
diff --git a/update-nix-dependencies.nix b/update-nix-dependencies.nix
new file mode 100644
index 0000000..e3e4275
--- /dev/null
+++ b/update-nix-dependencies.nix
@@ -0,0 +1,37 @@
+{ writeShellApplication, cargo, }:
+writeShellApplication {
+  name = "update_nix_dependencies";
+
+  runtimeInputs = [ cargo ];
+
+  text = ''
+    updateCrate() {
+        sed -i "s|/nix/store/[^-]\+-$1-[^/]\+|$2|" Cargo.toml
+    }
+
+    cd zkvms
+    for zkvm in *
+    do
+        [ ! -d "$zkvm" ] || [ "$zkvm" == 'result' ] && continue
+        [ "$zkvm" == 'zkwasm' ] && continue
+        pushd "$zkvm"
+        newPath="$(nix build github:metacraft-labs/nix-blockchain-development#"$zkvm" --print-out-paths)"
+
+        [ "$zkvm" == 'nexus' ] && zkvm=Nexus
+
+        cd guest
+        updateCrate "$zkvm" "$newPath"
+        cd ../host
+        updateCrate "$zkvm" "$newPath"
+        cd ../wrapper_macro
+        updateCrate "$zkvm" "$newPath"
+
+        cd ../guest
+        cargo generate-lockfile
+        cd ../host
+        cargo generate-lockfile
+
+        popd
+    done
+  '';
+}