zk-update.sh

#!/bin/bash

# example usage:
# curl https://gh-proxy.com/https://raw.githubusercontent.com/EarthLedger/aleo/refs/heads/main/zk-update.sh | bash -s 0.2.3 aleo14jy6z0h384m6tyqdlx6djflk6d0wp7fug4xqz05wp95ppqaqaygsgagv4t

# if not root user, exit
if [ "$EUID" -ne 0 ]
  then echo "Please run as root"
  exit
fi

VER=0.2.5
ADDR=$1
WORKER=$(hostname -I | awk '{print $1}')

echo "This script will update ZKWORK prover in your ubuntu system, and auto configure it to run on boot"

# if no VER or ADDR quit
if [ -z "$VER" ] || [ -z "$ADDR" ] ; then
  echo "Usage: $0 <receive-address>"
  echo "Example: $0 aleo1spkkxewxj2dl2lgdps9xr28093p5nxsvjv55g2unmqfu0hmwyuysmf4qp3"
  exit 1
fi

wget -O aleo_prover-v${VER}_full.tar.gz https://ghp.ci/https://ghp.ci/https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/cuda-v0.2.5-hotfix2/aleo_prover-v0.2.5_cuda_full_hotfix2.tar.gz

tar -xvf aleo_prover-v${VER}_full.tar.gz -C /opt

# geneate run/stop scripts
echo "#!/bin/bash
cd /opt/aleo_prover
./aleo_prover --address $ADDR --pool aleo.asia1.zk.work:10003 --pool aleo.hk.zk.work:10003 --pool aleo.jp.zk.work:10003 --custom_name $WORKER >> prover.log 2>&1
echo \$! > aleo_prover.pid
" > /opt/aleo_prover/start.sh
chmod +x /opt/aleo_prover/start.sh

echo "#!/bin/bash
kill -9 \$(cat /opt/aleo_prover/aleo_prover.pid)
" > /opt/aleo_prover/stop.sh
chmod +x /opt/aleo_prover/stop.sh

# add systemd service to run PWD/aleo_prover/run_prover.sh
echo "[Unit]
Description=Aleo Prover
After=network.target

[Service]
Type=simple
User=root
WorkingDirectory=/opt/aleo_prover
ExecStart=/opt/aleo_prover/start.sh
ExecStop=/opt/aleo_prover/stop.sh
Restart=always

[Install]
WantedBy=multi-user.target
" > /etc/systemd/system/aleo.service

systemctl daemon-reload
systemctl enable aleo.service
systemctl restart aleo.service