#!/bin/bash

source ./paths.sh
source ./cli_args.sh

# --- setup build directory ---

mkdir -p build 
cd build

# --- generate input for the circuit ---

echo ""
echo "generating the input for the proof circuit..."
${NIMCLI_DIR}/cli $CLI_ARGS -v --output=input.json

# --- generate the witness ---

echo ""
echo "generating the witness..."
cd ${CIRCUIT_MAIN}_js
time node generate_witness.js ${CIRCUIT_MAIN}.wasm ../input.json ../witness.wtns
cd ${ORIG}/build

# --- create the proof ---

PROVER="snarkjs"
# PROVER="nim"

RS=`which rapidsnark`
if [[ ! -z "$RS" ]]
then
  PROVER="rapidsnark"
fi

echo ""
echo "creating the proof... using prover: \`$PROVER\`"

case $PROVER in
  snarkjs)
    time snarkjs groth16 prove ${CIRCUIT_MAIN}.zkey witness.wtns proof.json public.json
    ;;
  rapidsnark)
    time rapidsnark ${CIRCUIT_MAIN}.zkey witness.wtns proof.json public.json
    ;;
  nim)
    time nim-groth16 -tpv --zkey=${CIRCUIT_MAIN}.zkey --wtns=witness.wtns -o=proof.json -i=public.json
    ;;
  *)
    echo "unknown prover \`$PROVER\`"
    exit 99
    ;;
esac

# --- verify the proof ---

echo ""
echo "verifying the proof:"
snarkjs groth16 verify ${CIRCUIT_MAIN}_verification_key.json public.json proof.json

# --- create solidity verifier contract ---

echo ""
echo "creating solidity verifier contract:"
snarkjs zkey export solidityverifier ${CIRCUIT_MAIN}.zkey verifier.sol

# --- finish  ---

cd $ORIG