From 201aa677c746c9af07cf5a4741b04aa96225f980 Mon Sep 17 00:00:00 2001
From: Prathamesh Musale <prathamesh.musale0@gmail.com>
Date: Thu, 8 Aug 2024 11:57:25 +0530
Subject: [PATCH] Separate scripts for genesis generation using allocations or
 eth holdings

---
 ...nerate-stage1-genesis-using-allocations.sh | 70 +++++++++++++++++++
 .../scripts/generate-stage1-genesis.sh        | 22 ++++--
 2 files changed, 85 insertions(+), 7 deletions(-)
 create mode 100755 stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis-using-allocations.sh

diff --git a/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis-using-allocations.sh b/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis-using-allocations.sh
new file mode 100755
index 0000000..d0551b9
--- /dev/null
+++ b/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis-using-allocations.sh
@@ -0,0 +1,70 @@
+#!/bin/bash
+
+# Exit on error
+set -e
+set -u
+
+# Check args
+if [ "$#" -ne 3 ]; then
+  echo "Usage: $0 <stage0-deployment-dir-absolute> <participant-allocation> <validator-allocation>"
+  echo "Example: $0 /home/user/stage0-deployment 1000000 1000000000000"
+  exit 1
+fi
+
+STAGE0_DEPLOYMENT_DIR="$1"
+PARTICIPANT_ALLOCATION="$2"
+VALIDATOR_ALLOCATION="$3"
+STAGE1_GENESIS_DIR=stage1-genesis
+
+# Create a temporary target directory
+mkdir -p $STAGE1_GENESIS_DIR
+
+# --------
+
+# Export onboarding module state from stage0 laconicd chain
+
+onboarding_state_file="$STAGE1_GENESIS_DIR/stage0-onboarding-state.json"
+docker run -it \
+  -v ${STAGE0_DEPLOYMENT_DIR}/data/laconicd-data:/root/stage0-deployment/.laconicd \
+  cerc/laconicd:local bash -c "laconicd export --home /root/stage0-deployment/.laconicd" \
+  | jq .app_state.onboarding > "$onboarding_state_file"
+
+stage0_participants=$(cat "$onboarding_state_file" | jq .participants)
+
+echo "Exported onboarding module state from stage 0 chain"
+
+# --------
+
+# For each participant, using ETH account holdings, figure out balance allocation for stage1
+stage1_allocations_file="$STAGE1_GENESIS_DIR/stage1-allocations.json"
+jq -n --argjson participants "$stage0_participants" --arg participant_allocation "$PARTICIPANT_ALLOCATION" --arg validator_allocation "$VALIDATOR_ALLOCATION" '
+[
+  $participants[] |
+  {
+    cosmos_address: .cosmos_address,
+    balance: (if .role == "validator" then $validator_allocation else $participant_allocation end)
+  }
+]' > $stage1_allocations_file
+
+echo "Calculated allocations for stage 1 chain"
+
+# --------
+
+# Run a script with cerc/laconicd:local to generate the genesis file
+# with onboarding module state and given allocations
+docker run -it \
+  -v ./stage1-genesis:/root/.laconicd \
+  -v ./scripts:/scripts \
+  cerc/laconicd:local bash -c "/scripts/genesis.sh"
+
+# Copy over the genesis file to output folder
+OUTPUT_DIR=output
+mkdir -p $OUTPUT_DIR
+cp ./stage1-genesis/config/genesis.json $OUTPUT_DIR/genesis.json
+
+echo "Genesis file for stage1 written to $OUTPUT_DIR/genesis.json"
+
+# --------
+
+# Clean up
+echo "Please remove the temporary data directory: sudo rm -rf stage1-genesis"
diff --git a/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis.sh b/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis.sh
index d0551b9..53fac02 100755
--- a/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis.sh
+++ b/stack-orchestrator/stacks/fixturenet-laconicd/scripts/generate-stage1-genesis.sh
@@ -5,15 +5,13 @@ set -e
 set -u
 
 # Check args
-if [ "$#" -ne 3 ]; then
-  echo "Usage: $0 <stage0-deployment-dir-absolute> <participant-allocation> <validator-allocation>"
-  echo "Example: $0 /home/user/stage0-deployment 1000000 1000000000000"
+if [ "$#" -ne 2 ]; then
+  echo "Usage: $0 <stage0-deployment-dir-absolute> <eth-account-holdings-json-file>"
   exit 1
 fi
 
 STAGE0_DEPLOYMENT_DIR="$1"
-PARTICIPANT_ALLOCATION="$2"
-VALIDATOR_ALLOCATION="$3"
+ETH_ACCOUNT_HOLDINGS_FILE="$2"
 STAGE1_GENESIS_DIR=stage1-genesis
 
 # Create a temporary target directory
@@ -21,6 +19,13 @@ mkdir -p $STAGE1_GENESIS_DIR
 
 # --------
 
+# ETH account holdings fetched from the bridge node
+eth_account_holdings=$(cat $ETH_ACCOUNT_HOLDINGS_FILE)
+
+echo "Fetched Ethereum account holdings"
+
+# --------
+
 # Export onboarding module state from stage0 laconicd chain
 
 onboarding_state_file="$STAGE1_GENESIS_DIR/stage0-onboarding-state.json"
@@ -37,12 +42,15 @@ echo "Exported onboarding module state from stage 0 chain"
 
 # For each participant, using ETH account holdings, figure out balance allocation for stage1
 stage1_allocations_file="$STAGE1_GENESIS_DIR/stage1-allocations.json"
-jq -n --argjson participants "$stage0_participants" --arg participant_allocation "$PARTICIPANT_ALLOCATION" --arg validator_allocation "$VALIDATOR_ALLOCATION" '
+jq -n --argjson holdings "$eth_account_holdings" --argjson participants "$stage0_participants" '
 [
+  $holdings[] |
+  . as $holding |
   $participants[] |
+  select((.nitro_address | ascii_downcase) == ($holding.nitro_address | ascii_downcase)) |
   {
     cosmos_address: .cosmos_address,
-    balance: (if .role == "validator" then $validator_allocation else $participant_allocation end)
+    balance: $holding.balance
   }
 ]' > $stage1_allocations_file