aboutsummaryrefslogtreecommitdiff
path: root/zkvms/zkwasm/default.nix
blob: 4f4e42f47bef9b7e291fe8e04797acc886473c43 (plain) (blame)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
{ zkVM-helpers,
  lib,
  rust-bin,
  metacraft-labs,
  wasm-pack,
  wasm-bindgen-cli,
  binaryen,
  craneLib-default,
}:
let
  commonArgs = {
    pname = "zkwasm";
    version = "infdev";

    src = with lib.fileset; toSource {
      root = ../..;
      fileset = intersection (gitTracked ../..) (unions [
          ./.
          ../../guests
          ../../guests_macro
          ../../zkvms_host_io
          ../../Vertices-010.in
      ]);
    };

    cargoLock = ./Cargo.lock;
  };

  rust-toolchain = rust-bin.nightly."2024-04-09".default.override {
    targets = ["wasm32-unknown-unknown"];
  };
  craneLib = craneLib-default.overrideToolchain rust-toolchain;
  cargoArtifacts = craneLib.buildDepsOnly (zkVM-helpers.fixDeps commonArgs);
in
  craneLib.buildPackage (zkVM-helpers.withCustomPhases (commonArgs
    // {
      inherit cargoArtifacts;

      nativeBuildInputs = [
        metacraft-labs.zkwasm
        wasm-pack
        wasm-bindgen-cli
        binaryen
      ];

      postPatch = ''
        ln -s ../../../../guests/graph_coloring ./zkvms/zkwasm/guest/src/zkp
        ln -s ../../../Cargo.lock ./zkvms/zkwasm/guest/
      '';

      hostBin = "host-zkwasm";

      buildGuestPhase = ''
        pushd guest

        # Workaround from
        # https://github.com/rustwasm/wasm-pack/issues/1335
        export WASM_PACK_CACHE=.wasm-pack-cache

        wasm-pack build --release --frozen
        popd
      '';

      postInstall = ''
        mv zkvms/zkwasm/guest/pkg "$out"/
      '';

      preBuild = ''
        export GUEST_PATH="$out/pkg/guest_bg.wasm"
      '';

      preRun = ''
        export PATH="\$PATH:${metacraft-labs.zkwasm}/bin"
      '';

      doCheck = false;
    }))