upgrade_proof_aggregator.sh

#!/bin/bash

# ENV VARIABLES
#
# MULTISIG=true|false whether the contract is deployed under a multisig account
#
# EXISTING_DEPLOYMENT_INFO_PATH: Path to the proof aggregator deployment output file
#   - Holesky Stage: ./script/output/holesky/proof_aggregation_service_deployment_output.stage.json
#   - Holesky Prod: ./script/output/holesky/roof_aggregation_service_deployment_output.json
#
# PROOF_AGGREGATOR_OUTPUT_PATH: Path to the proof aggregator output file
#   - Holesky Stage: ./script/output/holesky/proof_aggregation_service_deployment_output.stage.json
#   - Holesky Prod: ./script/output/holesky/proof_aggregation_service_deployment_output.json
#
# RPC_URL: The RPC URL to connect to the Ethereum network
#
# PRIVATE_KEY: The private key to use for the deployment
#
# ETHERSCAN_API_KEY: The Etherscan API key to use for verification
#

if [ -z "$MULTISIG" ]; then
  echo "Missing MULTISIG env variable"
  exit 1
fi

# cd to the directory of this script so that this can be run from anywhere
parent_path=$( cd "$(dirname "${BASH_SOURCE[0]}")" ; pwd -P )

cd "$parent_path"

cd ../

# Save the output to a variable to later extract the address of the new deployed contract
forge_output=$(forge script script/upgrade/ProofAggregatorServiceUpgrader.s.sol \
    $EXISTING_DEPLOYMENT_INFO_PATH \
    $PROOF_AGGREGATION_SERVICE_OUTPUT_PATH \
    --rpc-url $RPC_URL \
    --private-key $PRIVATE_KEY \
    --broadcast \
    --verify \
    --etherscan-api-key $ETHERSCAN_API_KEY \
    --sig "run(string memory alignedLayerDeploymentFilePath)")

echo "$forge_output"

# Extract the proof aggregator service values from the output
proof_aggregator_service_proxy=$(echo "$forge_output" | awk '/0: address/ {print $3}')
proof_aggregator_service_implementation=$(echo "$forge_output" | awk '/1: address/ {print $3}')

# Use the extracted value to replace the  batcher payment service values in alignedlayer_deployment_output.json and save it to a temporary file
jq --arg  proof_aggregator_service_implementation "$proof_aggregator_service_implementation" '.addresses.alignedProofAggregationServiceImplementation = $proof_aggregator_service_implementation' $PROOF_AGGREGATION_SERVICE_OUTPUT_PATH > "$PROOF_AGGREGATION_SERVICE_OUTPUT_PATH.temp"

# Replace the original file with the temporary file
mv "$PROOF_AGGREGATION_SERVICE_OUTPUT_PATH.temp" $PROOF_AGGREGATION_SERVICE_OUTPUT_PATH

# Delete the temporary file
rm -f "$PROOF_AGGREGATION_SERVICE_OUTPUT_PATH.temp"

echo "The new Proof Aggregator Service Implementation is $proof_aggregator_service_implementation"

data=$(cast calldata "upgradeTo(address)" $proof_aggregator_service_implementation)

echo "The new ProofAggregator Service Implementation is $proof_aggregator_service_implementation"

if [ "$MULTISIG" = false ]; then
  echo "Executing upgrade transaction"
  cast send $proof_aggregator_service_proxy $data \
    --rpc-url $RPC_URL \
    --private-key $PRIVATE_KEY
else
  echo "You can propose the upgrade transaction with the multisig using this calldata"
  echo $data
fi