feat: Add zkvm cargo features

So you can make code modifications compile-time, based on the zkVM

5 files changed, 23 insertions, 4 deletions

diff --git a/guests/README.md b/guests/README.md
index 1bbb176..9afcc23 100644
--- a/guests/README.md
+++ b/guests/README.md
@@ -26,13 +26,19 @@ These are normal Rust programs, with certain specific patterns implemented for z
    guests_macro = { version = "0.1.0", path = "../../guests_macro" }
    ```
 
-   and add a `no_std` [feature](https://doc.rust-lang.org/cargo/reference/features.html) like this:
+   and add a `no_std` and zkVM [features](https://doc.rust-lang.org/cargo/reference/features.html) like this:
 
    ```toml
    no_std = []
+   jolt = []
+   nexus = []
+   risc0 = []
+   sp1 = []
+   zkm = []
+   zkwasm = []
    ```
 
-   It's ok if you don't conditionally include anything with the feature, however it **must** exist!
+   It's ok if you don't use those features in any way, however they **must** exist!
 
 4. **Update your program.**
 
diff --git a/guests/fibonacci/Cargo.toml b/guests/fibonacci/Cargo.toml
index 4fe043c..6ada0c6 100644
--- a/guests/fibonacci/Cargo.toml
+++ b/guests/fibonacci/Cargo.toml
@@ -8,3 +8,9 @@ guests_macro = { version = "0.1.0", path = "../../guests_macro" }
 
 [features]
 no_std = []
+jolt = []
+nexus = []
+risc0 = []
+sp1 = []
+zkm = []
+zkwasm = []
diff --git a/guests/graph_coloring/Cargo.toml b/guests/graph_coloring/Cargo.toml
index b097d33..14af0f8 100644
--- a/guests/graph_coloring/Cargo.toml
+++ b/guests/graph_coloring/Cargo.toml
@@ -8,3 +8,9 @@ guests_macro = { version = "0.1.0", path = "../../guests_macro" }
 
 [features]
 no_std = []
+jolt = []
+nexus = []
+risc0 = []
+sp1 = []
+zkm = []
+zkwasm = []
diff --git a/zkvmLib.nix b/zkvmLib.nix
index 7584c5a..74eb855 100644
--- a/zkvmLib.nix
+++ b/zkvmLib.nix
@@ -133,6 +133,7 @@ in {
         [features]
         guest = [] # Only used in jolt
         no_std = ["zkp/no_std"]
+        ${args.pname} = ["zkp/${args.pname}"]
       '';
     in ''
       pushd zkvms/${args.pname}/guest
@@ -154,7 +155,7 @@ in {
       pushd zkvms/${args.pname}/guest
       runHook preBuildGuest
 
-      ${args.buildGuestCommand or "cargo build --release"} \
+      ${args.buildGuestCommand or "cargo build --release --features ${args.pname}"} \
           ${if args ? guestTarget then "--target " + args.guestTarget else ""} \
           ${args.guestExtraArgs or ""}
 
diff --git a/zkvms/zkwasm/default.nix b/zkvms/zkwasm/default.nix
index 9d880f5..b353bf7 100644
--- a/zkvms/zkwasm/default.nix
+++ b/zkvms/zkwasm/default.nix
@@ -47,7 +47,7 @@ in
         export WASM_PACK_CACHE=.wasm-pack-cache
       '';
 
-      buildGuestCommand = "wasm-pack build --release --frozen";
+      buildGuestCommand = "wasm-pack build --release --frozen --features zkwasm";
 
       preBuild = ''
         export GUEST_PATH="$out/pkg/guest_bg.wasm"