feat: .sh scripts to interact with strategies

Author: uri-99

contracts/scripts/get_strategy_weight.sh

@@ -0,0 +1,39 @@
+#!/bin/bash
+
+if [ -z "$OUTPUT_PATH" ]; then
+    echo "OUTPUT_PATH env var is not set"
+    exit 1
+fi
+
+if  [ -z "$RPC_URL" ]; then
+    echo "RPC_URL env var is not set"
+    exit 1
+fi
+
+STAKE_REGISTRY=$(jq -r '.addresses.stakeRegistry' "$OUTPUT_PATH")
+
+## Using in this cast call:
+
+#     struct StrategyParams {
+#         IStrategy strategy; (iface -> address)
+#         uint96 multiplier;
+#     }
+
+#     /// @notice Returns the strategy and weight multiplier for the `index`'th strategy in the quorum `quorumNumber`
+#     function strategyParamsByIndex(
+#         uint8 quorumNumber, 
+#         uint256 index
+#     ) public view returns (StrategyParams memory)
+#
+
+QUORUM_NUMER=0x0 #Aligned has only 1 quorum for now
+INDEX=$1
+
+echo $STAKE_REGISTRY
+
+cast call $STAKE_REGISTRY "strategyParamsByIndex(uint8,uint256)((address,uint96))" $QUORUM_NUMER $INDEX #--rpc-url $RPC_URL
+
+# Expected output:
+# (strategy_address, multiplier)
+# example:
+# (0xc5a5C42992dECbae36851359345FE25997F5C42d, 1000000000000000000 [1e18])

contracts/scripts/remove_strategy.sh

@@ -0,0 +1,41 @@
+#!/bin/bash
+
+if [ -z "$OUTPUT_PATH" ]; then
+    echo "OUTPUT_PATH env var is not set"
+    exit 1
+fi
+
+if  [ -z "$RPC_URL" ]; then
+    echo "RPC_URL env var is not set"
+    exit 1
+fi
+
+if [ -z "$PRIVATE_KEY" ]; then
+    echo "PRIVATE_KEY env var is not set"
+    exit 1
+fi
+
+if [ -z "$INDICES_TO_REMOVE" ]; then
+    echo "INDICES_TO_REMOVE env var is not set"
+    exit 1
+fi
+if [[ ! "$INDICES_TO_REMOVE" =~ ^\[[0-9]+(,[0-9]+)*\]$ ]]; then
+  echo "The INDICES_TO_REMOVE doesn't match the required format: [0,1,...,n]"
+  exit 1
+fi
+
+STAKE_REGISTRY=$(jq -r '.addresses.stakeRegistry' "$OUTPUT_PATH")
+
+## Using in this cast call:
+
+    # /**
+    #  * @notice This function is used for removing strategies and their associated weights from the
+    #  * mapping strategyParams for a specific @param quorumNumber.
+    #  * @dev higher indices should be *first* in the list of @param indicesToRemove, since otherwise
+    #  * the removal of lower index entries will cause a shift in the indices of the other strategiesToRemove
+    #  */
+    # function removeStrategies(uint8 quorumNumber, uint256[] calldata indicesToRemove) external;
+
+QUORUM_NUMBER=0 #Aligned has only 1 quorum for now
+
+cast send $STAKE_REGISTRY "removeStrategies(uint8, uint256[])()" $QUORUM_NUMBER $INDICES_TO_REMOVE --private-key $PRIVATE_KEY --rpc-url $RPC_URL

contracts/scripts/update_strategy_weight.sh

@@ -0,0 +1,75 @@
+#!/bin/bash
+
+if [ -z "$OUTPUT_PATH" ]; then
+    echo "OUTPUT_PATH env var is not set"
+    exit 1
+fi
+
+if  [ -z "$RPC_URL" ]; then
+    echo "RPC_URL env var is not set"
+    exit 1
+fi
+
+if [ -z "$PRIVATE_KEY" ]; then
+    echo "PRIVATE_KEY env var is not set"
+    exit 1
+fi
+
+if [ -z "$STRATEGY_INDICES" ]; then
+    echo "STRATEGY_INDICES env var is not set"
+    exit 1
+fi
+if [[ ! "$STRATEGY_INDICES" =~ ^\[[0-9]+(,[0-9]+)*\]$ ]]; then
+  echo "The STRATEGY_INDICES doesn't match the required format: [0,1,...,n]"
+  exit 1
+fi
+
+if [ -z "$NEW_MULTIPLIERS" ]; then
+    echo "NEW_MULTIPLIERS env var is not set"
+    exit 1
+fi
+if [[ ! "$NEW_MULTIPLIERS" =~ ^\[[0-9]+(,[0-9]+)*\]$ ]]; then
+  echo "The NEW_MULTIPLIERS doesn't match the required format: [0,1,...,n]"
+  exit 1
+fi
+
+count_elements() {
+  local var="$1"
+  # Remove brackets and count elements by splitting on commas
+  echo "$var" | sed 's/[\[\]]//g' | awk -F',' '{print NF}'
+}
+count1=$(count_elements "$STRATEGY_INDICES")
+count2=$(count_elements "$NEW_MULTIPLIERS")
+
+
+if [[ $count1 -ne $count2 ]]; then
+    echo "STRATEGY_INDICES and NEW_MULTIPLIERS have different numbers of elements:"
+    echo "STRATEGY_INDICES: $STRATEGY_INDICES"
+    echo "NEW_MULTIPLIERS: $NEW_MULTIPLIERS"
+    exit 1
+fi
+
+STAKE_REGISTRY=$(jq -r '.addresses.stakeRegistry' "$OUTPUT_PATH")
+
+## Using in this cast call:
+
+# /**
+#  * @notice This function is used for modifying the weights of strategies that are already in the
+#  * mapping strategyParams for a specific
+#  * @param quorumNumber is the quorum number to change the strategy for
+#  * @param strategyIndices are the indices of the strategies to change
+#  * @param newMultipliers are the new multipliers for the strategies
+#  */
+# function modifyStrategyParams(
+#     uint8 quorumNumber,
+#     uint256[] calldata strategyIndices,
+#     uint96[] calldata newMultipliers
+# ) external;
+
+QUORUM_NUMBER=0 #Aligned has only 1 quorum for now
+
+echo $QUORUM_NUMBER
+
+echo $STAKE_REGISTRY
+
+cast send $STAKE_REGISTRY "modifyStrategyParams(uint8, uint256[], uint96[])()" $QUORUM_NUMBER $STRATEGY_INDICES $NEW_MULTIPLIERS --private-key $PRIVATE_KEY --rpc-url $RPC_URL