update-nix-dependencies.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

{ writeShellApplication, cargo, zkvms, }:
writeShellApplication {
  name = "update_nix_dependencies";

  runtimeInputs = [ cargo ];

  text = let
    namesAndPaths = builtins.concatStringsSep
      " "
      (builtins.map
        (zkvm: zkvm.pname + "," + zkvm.outPath)
        zkvms);
  in ''
    updatePath() {
        sed -i "s|/nix/store/[^-]\+-$1-[^/]\+|$2|" Cargo.toml
    }
    updateDep() {
        updatePath "$1" "$2"
        cargo generate-lockfile
    }

    cd zkvms
    for i in ${namesAndPaths}
    do
        IFS=',' read -r zkvm path <<< "''${i}"
        [ "$zkvm" == 'zkWasm' ] && continue
        [ "$zkvm" == 'Nexus-zkVM' ] && zkvm=nexus

        pushd "$zkvm"

        [ "$zkvm" == 'nexus' ] && zkvm=Nexus

        cd wrapper_macro
        updatePath "$zkvm" "$path"
        cd ../guest
        updateDep "$zkvm" "$path"
        cd ../host
        updateDep "$zkvm" "$path"

        popd
    done
  '';
}