From d746c0a3bf6781c0bb5b24093f2c21664cb2badf Mon Sep 17 00:00:00 2001
From: Shreerang Kale <shreerangkale@gmail.com>
Date: Wed, 11 Jun 2025 10:02:46 +0530
Subject: [PATCH] Update run script to use validator priv key file

---
 stack-orchestrator/compose/docker-compose-tmkms.yml |  5 +++--
 stack-orchestrator/config/tmkms/run.sh              | 13 ++++++++++++-
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/stack-orchestrator/compose/docker-compose-tmkms.yml b/stack-orchestrator/compose/docker-compose-tmkms.yml
index 1790eae..5285688 100644
--- a/stack-orchestrator/compose/docker-compose-tmkms.yml
+++ b/stack-orchestrator/compose/docker-compose-tmkms.yml
@@ -6,12 +6,13 @@ services:
     environment:
       CERC_CHAIN_ID: ${CERC_CHAIN_ID:-laconic-mainnet}
       NODE_IP: ${NODE_IP:-localhost}
+      NODE_PORT: ${NODE_PORT:-26659}
       CERC_KEY_PREFIX: ${CERC_KEY_PREFIX:-laconic}
     volumes:
       - tmkms-data:/home/tmkmsuser/tmkms
       - ../config/tmkms/run.sh:/opt/run.sh
-    ports:
-      - "26659"
+    extra_hosts:
+    - "host.docker.internal:host-gateway"
 
 volumes:
   tmkms-data:
diff --git a/stack-orchestrator/config/tmkms/run.sh b/stack-orchestrator/config/tmkms/run.sh
index 09af613..342fd99 100755
--- a/stack-orchestrator/config/tmkms/run.sh
+++ b/stack-orchestrator/config/tmkms/run.sh
@@ -7,6 +7,7 @@ fi
 set -e
 
 TMKMS_HOME=/home/tmkmsuser/tmkms
+INPUT_PRIV_KEY_FILE=$TMKMS_HOME/tmp/priv_validator_key.json
 TMKMS_SECRETS_DIR=$TMKMS_HOME/secrets
 TMKMS_STATE_DIR=$TMKMS_HOME/state
 
@@ -32,7 +33,7 @@ state_file = "$TMKMS_STATE_DIR/priv_validator_state.json"
 
 [[validator]]
 chain_id = "$CERC_CHAIN_ID"
-addr = "tcp://$NODE_IP:26659"
+addr = "tcp://$NODE_IP:$NODE_PORT"
 secret_key = "$TMKMS_SECRETS_DIR/kms-identity.key"
 protocol_version = "v0.34"
 reconnect = true
@@ -43,5 +44,15 @@ path = "$TMKMS_SECRETS_DIR/priv_validator_key"
 chain_ids = ["$CERC_CHAIN_ID"]
 EOF
 
+# Place validator key in secrets directory
+cp $INPUT_PRIV_KEY_FILE $TMKMS_SECRETS_DIR/priv_validator_key.json
+
+# Import the private validator key into tmkms
+echo "Importing private validator key into tmkms..."
+tmkms softsign import $TMKMS_SECRETS_DIR/priv_validator_key.json $TMKMS_SECRETS_DIR/priv_validator_key
+
+# Remove the JSON key file
+rm $TMKMS_SECRETS_DIR/priv_validator_key.json
+
 echo "Starting tmkms..."
 tmkms start --config $TMKMS_HOME/tmkms.toml