aboutsummaryrefslogtreecommitdiff
path: root/flake.nix
blob: ad73c17815851e69f4b571a4f835b32b0fdad50e (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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
{
  description = "zkVMs benchmarks";

  nixConfig = {
    extra-substituters = "https://nix-blockchain-development.cachix.org";
    extra-trusted-public-keys = "nix-blockchain-development.cachix.org-1:Ekei3RuW3Se+P/UIo6Q/oAgor/fVhFuuuX5jR8K/cdg=";
  };

  inputs = {
    mcl-blockchain.url = "github:metacraft-labs/nix-blockchain-development?ref=zkvm-packages";
    nixpkgs.follows = "mcl-blockchain/nixpkgs";
    crane.follows = "mcl-blockchain/crane";
    rust-overlay.follows = "mcl-blockchain/rust-overlay";
    # flake-utils.follows = "mcl-blockchain/flake-utils";
  };

  outputs = { self, nixpkgs, mcl-blockchain, crane, rust-overlay, ... }:
  let
    system = "x86_64-linux";
    pkgs = import nixpkgs { system = system; overlays = [
        mcl-blockchain.overlays.default
        rust-overlay.overlays.default
      ];
    };
    callPackage = pkgs.lib.callPackageWith pkgs;

    fixDeps = commonArgs: commonArgs // {
        postUnpack = ''
          ${commonArgs.postUnpack or ""}
          ln -s ../../../guests ./source/zkvms/${commonArgs.pname}/guest/
          ln -s ../../../guests_macro ./source/zkvms/${commonArgs.pname}/guest/
          ln -s ../../Cargo.lock ./source/zkvms/${commonArgs.pname}/
        '';

        preBuild = ''
          ${commonArgs.preBuild or ""}
          cd zkvms/${commonArgs.pname}
        '';
      };


    # Overrides build and install phases for use with zkVMs
    # Requirements:
    # - zkVM is inside zkvms/pname/
    # - guest crate is located at zkvms/pname/guest and is named "guest"
    withCustomPhases = currentPackage: with currentPackage; {
        buildGuestPhase = ''
          pushd guest
          runHook preBuildGuest

          cargo build --release --target ${currentPackage.guestTarget} ${currentPackage.extraGuestArgs or ""}
          ln -s ../../guest/target/${currentPackage.guestTarget}/release/guest ../host/src/guest

          runHook postBuildGuest
          popd
        '';

        buildPhase = ''
          export INPUTS="$PWD/Vertices-010.in"

          pushd zkvms/${currentPackage.pname}
          runPhase buildGuestPhase
          runHook preBuild

          cargo build --bin ${hostBin} --release

          runHook postBuild
          popd
        '';

        installPhase = ''
          runHook preInstall

          mkdir -p "$out"/bin
          for bin in $(find . -type f -regex "./zkvms/.*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
        '';

        doNotPostBuildInstallCargoBinaries = true;
      } // currentPackage;


    args-zkVM = {
      craneLib-default = crane.mkLib pkgs;
      zkVM-helpers = {
        inherit fixDeps;
        inherit withCustomPhases;
      };
    };
  in {
    packages.${system} = {
      risc0 = callPackage ./zkvms/risc0/default.nix args-zkVM;
      sp1 = callPackage ./zkvms/sp1/default.nix args-zkVM;
      zkwasm = callPackage ./zkvms/zkwasm/default.nix args-zkVM;
      zkm = callPackage ./zkvms/zkm/default.nix args-zkVM;
      jolt = callPackage ./zkvms/jolt/default.nix args-zkVM;
    };
  };
}