#!/bin/bash

set -e

# Runs all benchmarks for all pallets, for a given runtime, provided by $1
# Should be run on a reference machine to gain accurate benchmarks
# current reference machine: https://github.com/paritytech/substrate/pull/5848

runtime="$1"

echo "[+] Running all benchmarks for $runtime"

cargo +nightly build --profile production --locked --features=runtime-benchmarks

./target/production/polkadot benchmark \
    --chain "${runtime}-dev" \
    --list |\
  tail -n+2 |\
  cut -d',' -f1 |\
  uniq > "${runtime}_pallets"

# For each pallet found in the previous command, run benches on each function
while read -r line; do
  pallet="$(echo "$line" | cut -d' ' -f1)";
  echo "Runtime: $runtime. Pallet: $pallet";
  # '!' has the side effect of bypassing errexit / set -e
  ! ./target/production/polkadot benchmark \
    --chain="${runtime}-dev" \
    --steps=50 \
    --repeat=20 \
    --pallet="$pallet" \
    --extrinsic="*" \
    --execution=wasm \
    --wasm-execution=compiled \
    --heap-pages=4096 \
    --header=./file_header.txt \
    --output="./runtime/${runtime}/src/weights/${pallet/::/_}.rs"
done < "${runtime}_pallets"
rm "${runtime}_pallets"