From a393865ee4a672f527525c630543b09b7588f539 Mon Sep 17 00:00:00 2001
From: Dmitry K <okdas@pm.me>
Date: Tue, 3 Dec 2024 18:06:04 -0800
Subject: [PATCH] alpha testnet mitigation v0.0.11

---
 .gitignore                 |  1 +
 docker-compose.yml         |  1 +
 scripts/init-poktrolld.sh  | 18 ++++++++++++++++++
 scripts/start-poktrolld.sh | 11 ++++++++++-
 4 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 93390b6..617b0f3 100755
--- a/.gitignore
+++ b/.gitignore
@@ -23,3 +23,4 @@ gateway/config/config.toml
 prometheus/data
 grafana/data
 .DS_Store
+tmp
diff --git a/docker-compose.yml b/docker-compose.yml
index 4d56282..5e36326 100755
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -22,6 +22,7 @@ services:
     entrypoint: ["sh", "/home/pocket/start-poktrolld.sh"]
     environment:
       - NODE_HOSTNAME=${NODE_HOSTNAME}
+      - NETWORK_NAME=${NETWORK_NAME}
       - POKTROLLD_LOG_LEVEL=${POKTROLLD_LOG_LEVEL}
       - DAEMON_NAME=poktrolld
       - DAEMON_HOME=/home/pocket/.poktroll
diff --git a/scripts/init-poktrolld.sh b/scripts/init-poktrolld.sh
index 7be7b79..7a391f8 100755
--- a/scripts/init-poktrolld.sh
+++ b/scripts/init-poktrolld.sh
@@ -68,6 +68,24 @@ if [ -n "$NETWORK_NAME" ]; then
             exit 1
         fi
     fi
+
+    # Add testnet-alpha specific mitigation after the binary setup
+    if [ "$NETWORK_NAME" = "testnet-alpha" ]; then
+        # Check for upgrade info file
+        UPGRADE_INFO_FILE="/home/pocket/.poktroll/data/upgrade-info.json"
+        if [ -f "$UPGRADE_INFO_FILE" ]; then
+            UPGRADE_HEIGHT=$(jq -r '.height' "$UPGRADE_INFO_FILE")
+            if [ "$UPGRADE_HEIGHT" = "83725" ]; then
+                echo "Applying testnet-alpha mitigation..."
+                # Remove upgrade info file
+                rm "$UPGRADE_INFO_FILE"
+                # Change symlink back to v0.0.10
+                rm -f /home/pocket/.poktroll/cosmovisor/current
+                ln -s /home/pocket/.poktroll/cosmovisor/upgrades/v0.0.10 /home/pocket/.poktroll/cosmovisor/current
+                echo "Successfully applied testnet-alpha mitigation"
+            fi
+        fi
+    fi
 else
     echo "NETWORK_NAME variable not set. Please set it to 'testnet-alpha', 'testnet-beta', or 'mainnet'."
     exit 1
diff --git a/scripts/start-poktrolld.sh b/scripts/start-poktrolld.sh
index 0dd540c..db8e1ac 100755
--- a/scripts/start-poktrolld.sh
+++ b/scripts/start-poktrolld.sh
@@ -12,8 +12,17 @@ set -e
 # Read the seeds from the genesis.seeds file
 export SEEDS=$(cat "$DAEMON_HOME/config/genesis.seeds")
 
+# Add skip-upgrades flag for testnet-alpha
+EXTRA_FLAGS=""
+if [ "$NETWORK_NAME" = "testnet-alpha" ]; then
+    # TODO(@okdas): move this into the separate file in `genesis` repo like we do this for seeds. That way, we can
+    # automate this in the future for all networks.
+    EXTRA_FLAGS="--unsafe-skip-upgrades=83725"
+fi
+
 # Start the binary via Cosmovisor
 exec cosmovisor run start \
   --p2p.external-address="${NODE_HOSTNAME}:26656" \
   --log_level="${POKTROLLD_LOG_LEVEL}" \
-  --p2p.seeds="${SEEDS}"
+  --p2p.seeds="${SEEDS}" \
+  ${EXTRA_FLAGS}