Update start.sh

Author: NekoSuneVR

start.sh

@@ -33,54 +33,90 @@ case "$THETA_MODE" in
     echo "PrivateNet setup..."
 
     mkdir -p /theta/privatenet
-    cp -r /theta/theta/integration/privatenet /theta/ || true
+
+    if [ ! -d "/theta/privatenet/node" ]; then
+      echo "Copying privatenet config..."
+      cp -r /theta/theta/integration/privatenet /theta/ || true
+    else
+      echo "Privatenet config already exists. Skipping copy."
+    fi
 
     mkdir -p ~/.thetacli
-    cp -r /theta/theta/integration/privatenet/thetacli/* ~/.thetacli/ || true
-    chmod 700 ~/.thetacli/keys/encrypted || true
+
+    if [ ! -d "/root/.thetacli/keys" ]; then
+      echo "Copying CLI wallet data..."
+      cp -r /theta/theta/integration/privatenet/thetacli/* ~/.thetacli/ || true
+      chmod 700 ~/.thetacli/keys/encrypted || true
+    else
+      echo "Wallet already exists. Skipping copy."
+    fi
 
     init_wallet
     start_cli
 
     exec theta start \
       --config=/theta/privatenet/node \
-      --password-file=$PASSWORD_FILE
+      --password="$THETA_PASSWORD"
     ;;
 
   testnet)
     echo "TestNet setup..."
 
     mkdir -p /theta/testnet
-    cp -r /theta/theta/integration/testnet/walletnode /theta/testnet || true
 
-    wget -O /theta/testnet/walletnode/snapshot \
-      $(curl -k https://theta-testnet-backup.s3.amazonaws.com/snapshot/snapshot)
+    if [ ! -d "/theta/testnet/walletnode" ]; then
+      echo "Copying testnet config..."
+      cp -r /theta/theta/integration/testnet/walletnode /theta/testnet || true
+    else
+      echo "Testnet config already exists. Skipping copy."
+    fi
+
+    # Snapshot check
+    if [ ! -f "/theta/testnet/walletnode/snapshot" ]; then
+      echo "Downloading snapshot..."
+      wget -O /theta/testnet/walletnode/snapshot \
+        $(curl -k https://theta-testnet-backup.s3.amazonaws.com/snapshot/snapshot)
+    else
+      echo "Snapshot already exists. Skipping download."
+    fi
 
     init_wallet
     start_cli
 
     exec theta start \
       --config=/theta/testnet/walletnode \
-      --password-file=$PASSWORD_FILE
+      --password="$THETA_PASSWORD"
     ;;
 
   mainnet)
     echo "MainNet setup..."
 
     mkdir -p /theta/mainnet/walletnode
 
-    curl -k --output /theta/mainnet/walletnode/config.yaml \
-      $(curl -k 'https://mainnet-data.thetatoken.org/config?is_guardian=true')
-
-    wget -O /theta/mainnet/walletnode/snapshot \
-      $(curl -k https://mainnet-data.thetatoken.org/snapshot)
+    # Config check
+    if [ ! -f "/theta/mainnet/walletnode/config.yaml" ]; then
+      echo "Downloading config.yaml..."
+      curl -k --output /theta/mainnet/walletnode/config.yaml \
+        $(curl -k 'https://mainnet-data.thetatoken.org/config?is_guardian=true')
+    else
+      echo "config.yaml already exists. Skipping download."
+    fi
+
+    # Snapshot check
+    if [ ! -f "/theta/mainnet/walletnode/snapshot" ]; then
+      echo "Downloading snapshot..."
+      wget -O /theta/mainnet/walletnode/snapshot \
+        $(curl -k https://mainnet-data.thetatoken.org/snapshot)
+    else
+      echo "Snapshot already exists. Skipping download."
+    fi
 
     init_wallet
     start_cli
 
     exec theta start \
       --config=/theta/mainnet/walletnode \
-      --password-file=$PASSWORD_FILE
+      --password="$THETA_PASSWORD"
     ;;
 
   *)