feat: unspent nullifier tool

crates/toolkit/Cargo.toml

@@ -11,7 +11,11 @@ world-chain-pbh.workspace = true
 alloy-consensus.workspace = true
 alloy-network.workspace = true
 alloy-primitives.workspace = true
+alloy-provider = { workspace = true, features = ["reqwest", "reqwest-default-tls"] }
+alloy-rpc-client.workspace = true
+alloy-rpc-types-eth.workspace = true
 alloy-rlp.workspace = true
+alloy-sol-types.workspace = true
 
 # 3rd party
 bytes.workspace = true
@@ -25,3 +29,5 @@ tokio.workspace = true
 dotenvy.workspace = true
 chrono.workspace = true
 rand.workspace = true
+tracing.workspace = true
+tracing-subscriber.workspace = true

crates/toolkit/src/cli.rs

@@ -1,3 +1,4 @@
+use alloy_primitives::Address;
 use bytes::Bytes;
 use chrono::NaiveDate;
 use clap::Parser;
@@ -29,6 +30,8 @@ pub enum Cmd {
     /// For the inclusion proof you can fetch it dynamically from the (staging) sequencer API via `--inclusion-proof-url https://signup-orb-ethereum.stage-crypto.worldcoin.dev/inclusionProof`
     /// or `export INCLUSION_PROOF_URL=https://signup-orb-ethereum.stage-crypto.worldcoin.dev/inclusionProof` env var
     Prove(ProveArgs),
+    /// Queries the PBH entrypoint contract directly to find currently available PBH nonces.
+    CheckNonces(CheckNoncesArgs),
 }
 
 #[derive(Debug, Clone, Parser)]
@@ -68,3 +71,24 @@ pub struct ProveArgs {
 
 #[derive(Debug, Clone, Parser)]
 pub struct SendArgs {}
+
+#[derive(Debug, Clone, Parser)]
+pub struct CheckNoncesArgs {
+    /// PBH-enabled node RPC URL.
+    #[clap(long, env = "RPC_URL")]
+    pub rpc_url: String,
+
+    /// PBH entrypoint contract address.
+    #[clap(long, env = "PBH_ENTRYPOINT")]
+    pub pbh_entrypoint: Address,
+
+    /// Overrides the current date for PBH external nullifier generation.
+    /// Format: "YYYY-MM-DD"
+    ///
+    /// Dates are always assumed to be in UTC.
+    #[clap(short = 'D', long)]
+    pub custom_date: Option<NaiveDate>,
+
+    #[command(flatten)]
+    pub identity_source: IdentitySource,
+}

crates/toolkit/src/main.rs

@@ -11,6 +11,9 @@ use world_chain_pbh::{
 };
 
 mod cli;
+mod pbh_contract_client;
+
+use pbh_contract_client::PbhContractClient;
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct InclusionProof {
@@ -21,6 +24,7 @@ pub struct InclusionProof {
 #[tokio::main]
 async fn main() -> eyre::Result<()> {
     dotenvy::dotenv().ok();
+    init_tracing();
 
     let args = Opt::parse();
 
@@ -79,11 +83,33 @@ async fn main() -> eyre::Result<()> {
 
             println!("{encoded_hex}");
         }
+        Cmd::CheckNonces(check_args) => {
+            let date = check_args
+                .custom_date
+                .unwrap_or_else(|| chrono::Utc::now().naive_utc().date());
+            let identity = check_args.identity_source.load();
+            let client = PbhContractClient::new(check_args.rpc_url);
+            let available_nonces = client
+                .find_available_pbh_nonces(check_args.pbh_entrypoint, &identity, date)
+                .await?;
+
+            println!("{}", serde_json::to_string(&available_nonces)?);
+        }
     }
 
     Ok(())
 }
 
+fn init_tracing() {
+    let env_filter = tracing_subscriber::EnvFilter::try_from_default_env()
+        .unwrap_or_else(|_| tracing_subscriber::EnvFilter::new("toolkit=info"));
+
+    let _ = tracing_subscriber::fmt()
+        .with_env_filter(env_filter)
+        .with_target(false)
+        .try_init();
+}
+
 fn load_inclusion_proof_file(path: impl AsRef<std::path::Path>) -> eyre::Result<InclusionProof> {
     let file = std::fs::File::open(path)?;
     let proof = serde_json::from_reader(file)?;