zkverify.sh

#!/bin/bash

set -e

# Ask for API key
read -p "🔐 Enter your API key: " API_KEY

if [[ -z "$API_KEY" ]]; then
  echo "❌ Missing API key!"
  exit 1
fi

# ========== STEP 1: ENV SETUP ==========
echo "✅ Installing dependencies..."
sudo apt update && sudo apt install -y curl git jq
npm install -g snarkjs circom

# ========== STEP 2: COMPILE CIRCUIT ==========
echo "✅ Compiling circuit..."
mkdir -p zkverify/{circuits,keys,proofs,input,witness,scripts}
cd zkverify

cat > circuits/add_and_multiply.circom <<EOF
template AddAndMultiply() {
    signal input a;
    signal input b;
    signal input c;
    signal input result;

    result <== (a + b) * c;
}
component main = AddAndMultiply();
EOF

circom circuits/add_and_multiply.circom --r1cs --wasm --sym -o circuits/
mv add_and_multiply.* circuits/

# ========== STEP 3: PTAU & ZKEY ==========
echo "✅ Preparing powers of tau and zkey..."
snarkjs powersoftau new bn128 12 keys/pot12_0000.ptau -v
snarkjs powersoftau contribute keys/pot12_0000.ptau keys/pot12_final.ptau --name="zkverify" -v -e="zkverify-challenge"
snarkjs powersoftau prepare phase2 keys/pot12_final.ptau keys/pot12_final_prepared.ptau

snarkjs groth16 setup circuits/add_and_multiply.r1cs keys/pot12_final_prepared.ptau keys/add_and_multiply_0000.zkey
snarkjs zkey contribute keys/add_and_multiply_0000.zkey keys/add_and_multiply_final.zkey --name="zkverify" -v -e="zkverify-contrib"
snarkjs zkey export verificationkey keys/add_and_multiply_final.zkey keys/verification_key.json

# ========== STEP 4: CREATE generatePayload.js ==========
echo "✅ Creating generatePayload.js..."

cat > scripts/generatePayload.js <<'EOF'
const fs = require("fs");

const proof = JSON.parse(fs.readFileSync("proofs/proof.json", "utf8"));
const publicSignals = JSON.parse(fs.readFileSync("proofs/public.json", "utf8"));
const vk = JSON.parse(fs.readFileSync("keys/verification_key.json", "utf8"));

const payload = {
  proofType: "groth16",
  vkRegistered: true,
  proofOptions: {
    library: "snarkjs",
    curve: "bn128"
  },
  proofData: {
    proof: {
      pi_a: proof.pi_a.map(String),
      pi_b: proof.pi_b.map(pair => pair.map(String)),
      pi_c: proof.pi_c.map(String)
    },
    publicSignals: publicSignals.map(String),
    vk
  }
};

fs.writeFileSync("payload.json", JSON.stringify(payload, null, 2));
EOF

# ========== STEP 5: SUBMIT SINGLE PROOF ==========
echo "✅ Generating input..."
A=$(( RANDOM % 50 + 1 ))
B=$(( RANDOM % 50 + 1 ))
C=$(( RANDOM % 50 + 1 ))
RESULT=$(( (A + B) * C ))
echo "{ \"a\": $A, \"b\": $B, \"c\": $C, \"result\": $RESULT }" > input/input.json

snarkjs wtns calculate circuits/add_and_multiply.wasm input/input.json witness/witness.wtns
snarkjs groth16 prove keys/add_and_multiply_final.zkey witness/witness.wtns proofs/proof.json proofs/public.json

node scripts/generatePayload.js

echo "🚀 Submitting proof..."
RESPONSE=$(curl -s -X POST https://relayer-api.horizenlabs.io/api/v1/submit-proof/$API_KEY \
  -H "Content-Type: application/json" \
  -d @payload.json)

echo "$RESPONSE" | tee -a submit.log

echo "🎉 Proof submitted. Result saved in submit.log"