Native prover workflow

Cargo.toml

@@ -15,6 +15,7 @@ members = [
   "mvpoly",
   "o1vm",
   "plonk-neon",
+  "plonk-napi",
   "plonk-wasm",
   "poly-commitment",
   "poseidon",
@@ -61,6 +62,9 @@ libflate = "2"
 log = "0.4.20"
 num-bigint = { version = "0.4.4", features = ["rand", "serde"] }
 num-integer = "0.1.45"
+napi = { git = "https://github.com/o1-labs/napi-rs.git", rev = "023d1d4f31bd75e8ab55c95b5077a319b5208cdf", version = "3.3.0", default-features = false, features = ["napi7"] }
+napi-derive = { git="https://github.com/o1-labs/napi-rs.git", rev = "023d1d4f31bd75e8ab55c95b5077a319b5208cdf", version = "3.3.0", features = ["type-def"] }
+napi-build = "2.1.0"
 ocaml = { version = "0.22.2" }
 ocaml-gen = { version = "1.0.0" }
 once_cell = "=1.21.3"

plonk-napi/Cargo.toml

@@ -0,0 +1,46 @@
+[package]
+name = "plonk-napi"
+version = "0.1.0"
+authors = ["[email protected]"]
+description = "Node-API bindings for plonk proof systems"
+repository = "https://github.com/MinaProtocol/mina"
+license = "MIT/Apache-2.0"
+edition = "2021"
+
+[lib]
+name = "plonk_napi"
+crate-type = ["cdylib"] # to generate a dynamic library that is loadable by Node
+
+[dependencies]
+napi = { workspace = true, features = ["napi7"] }
+napi-derive.workspace = true
+
+# arkworks
+ark-ec.workspace = true
+ark-ff.workspace = true
+ark-poly.workspace = true
+ark-serialize.workspace = true
+arkworks.workspace = true
+
+# proof-systems
+kimchi.workspace = true
+mina-curves = { path = "../curves" }
+mina-poseidon = { path = "../poseidon" }
+o1-utils = { path = "../utils" }
+poly-commitment = { path = "../poly-commitment" }
+
+getrandom.workspace = true
+libc.workspace = true
+num-bigint.workspace = true
+once_cell.workspace = true
+paste.workspace = true
+rand.workspace = true
+rayon.workspace = true
+rmp-serde.workspace = true
+serde.workspace = true
+serde_json.workspace = true
+serde_with.workspace = true
+wasm-types.workspace = true
+
+[build-dependencies]
+napi-build.workspace = true

plonk-napi/build.rs

@@ -0,0 +1,3 @@
+fn main() {
+    napi_build::setup();
+}

plonk-napi/src/build_info.rs

@@ -0,0 +1,41 @@
+use std::sync::atomic::{AtomicU64, Ordering};
+
+use napi_derive::napi;
+
+#[cfg(target_os = "windows")]
+#[napi]
+pub const OS_NAME: &str = "Windows";
+
+#[cfg(target_os = "linux")]
+#[napi]
+pub const OS_NAME: &str = "Linux";
+
+#[cfg(target_os = "macos")]
+#[napi]
+pub const OS_NAME: &str = "macOS";
+
+#[cfg(target_arch = "x86_64")]
+#[napi]
+pub const ARCH_NAME: &str = "x86_64";
+
+#[cfg(target_arch = "arm")]
+#[napi]
+pub const ARCH_NAME: &str = "ARM";
+
+#[cfg(target_arch = "aarch64")]
+#[napi]
+pub const ARCH_NAME: &str = "AArch64";
+
+#[napi]
+pub const BACKING: &str = "native";
+
+static NATIVE_CALLS: AtomicU64 = AtomicU64::new(0);
+
+#[napi]
+pub fn get_native_calls() -> u64 {
+    NATIVE_CALLS.load(Ordering::Relaxed)
+}
+
+pub(crate) fn report_native_call() {
+    NATIVE_CALLS.fetch_add(1, Ordering::Relaxed);
+}

plonk-napi/src/circuit.rs

@@ -0,0 +1,38 @@
+use ark_ff::PrimeField;
+use kimchi::circuits::{constraints::ConstraintSystem, gate::CircuitGate};
+use mina_curves::pasta::Fp;
+use napi::bindgen_prelude::*;
+use napi_derive::napi;
+use serde::Serialize;
+
+use crate::{build_info::report_native_call, types::WasmPastaFpPlonkIndex};
+
+#[derive(Serialize)]
+struct Circuit<F>
+where
+    F: PrimeField,
+{
+    public_input_size: usize,
+    #[serde(bound = "CircuitGate<F>: Serialize")]
+    gates: Vec<CircuitGate<F>>,
+}
+
+impl<F> From<&ConstraintSystem<F>> for Circuit<F>
+where
+    F: PrimeField,
+{
+    fn from(cs: &ConstraintSystem<F>) -> Self {
+        Self {
+            public_input_size: cs.public,
+            gates: cs.gates.to_vec(),
+        }
+    }
+}
+
+#[napi(js_name = "prover_to_json")]
+pub fn prover_to_json(prover_index: &External<WasmPastaFpPlonkIndex>) -> String {
+    report_native_call();
+
+    let circuit: Circuit<Fp> = prover_index.0.cs.as_ref().into();
+    serde_json::to_string(&circuit).expect("couldn't serialize constraints")
+}

plonk-napi/src/lib.rs

@@ -0,0 +1,12 @@
+mod build_info;
+mod circuit;
+pub mod plonk_verifier_index;
+mod poseidon;
+mod prover_index;
+mod types;
+
+pub use poseidon::{caml_pasta_fp_poseidon_block_cipher, caml_pasta_fq_poseidon_block_cipher};
+
+pub use circuit::prover_to_json;
+pub use prover_index::{prover_index_from_bytes, prover_index_to_bytes};
+pub use types::WasmPastaFpPlonkIndex;