From 7430ee8d8cefbb8c58b88ee34b2f543f6a6bbb9c Mon Sep 17 00:00:00 2001
From: Shreerang Kale <shreerangkale@gmail.com>
Date: Tue, 10 Jun 2025 18:20:20 +0530
Subject: [PATCH] Update run script and compose file to initialize tmkms

---
 .../compose/docker-compose-tmkms.yml          |  9 ++++-
 stack-orchestrator/config/tmkms/run.sh        | 35 +++++++++++++++++++
 2 files changed, 43 insertions(+), 1 deletion(-)

diff --git a/stack-orchestrator/compose/docker-compose-tmkms.yml b/stack-orchestrator/compose/docker-compose-tmkms.yml
index ed15609..de08d95 100644
--- a/stack-orchestrator/compose/docker-compose-tmkms.yml
+++ b/stack-orchestrator/compose/docker-compose-tmkms.yml
@@ -3,8 +3,15 @@ services:
     restart: unless-stopped
     image: cerc/tmkms:local
     command: ["bash", "-c", "/opt/run.sh"]
+    environment:
+      CERC_CHAIN_ID: ${CERC_CHAIN_ID:-laconic-mainnet}
+      NODE_IP: ${NODE_IP:-localhost}
+      CERC_KEY_PREFIX: ${CERC_KEY_PREFIX:-laconic}
     volumes:
-      - ./tmkms:/root
+      - tmkms-data:/root
       - ../config/tmkms/run.sh:/opt/run.sh
     ports:
       - "26659"
+
+volumes:
+  tmkms-data:
diff --git a/stack-orchestrator/config/tmkms/run.sh b/stack-orchestrator/config/tmkms/run.sh
index 45a016a..3eb4908 100755
--- a/stack-orchestrator/config/tmkms/run.sh
+++ b/stack-orchestrator/config/tmkms/run.sh
@@ -7,6 +7,41 @@ fi
 set -e
 
 TMKMS_HOME=/root/.tmkms
+TMKMS_SECRETS_DIR=$TMKMS_HOME/secrets
+TMKMS_STATE_DIR=$TMKMS_HOME/state
+
+echo "Initializing tmkms configuration..."
+
+# Initialize tmkms config
+tmkms init config --home $TMKMS_HOME
+
+# Generate a new softsign key
+echo "Generating new softsign key..."
+tmkms softsign keygen $TMKMS_SECRETS_DIR/kms-identity.key --home $TMKMS_HOME
+
+# Update tmkms.toml
+echo "Updating tmkms.toml with chain_id, node IP, and key prefixes..."
+
+# Add chain configuration
+cat <<EOF >> $TMKMS_HOME/tmkms.toml
+
+[[chain]]
+id = "$CERC_CHAIN_ID"
+key_format = { type = "cosmos-json", account_key_prefix = "${CERC_KEY_PREFIX}pub", consensus_key_prefix = "${CERC_KEY_PREFIX}valconspub" }
+state_file = "$TMKMS_STATE_DIR/priv_validator_state.json"
+
+[[validator]]
+chain_id = "$CERC_CHAIN_ID"
+addr = "tcp://$NODE_IP:26659"
+secret_key = "$TMKMS_SECRETS_DIR/kms-identity.key"
+protocol_version = "v0.34"
+reconnect = true
+
+[[providers.softsign]]
+key_type = "consensus"
+path = "$TMKMS_SECRETS_DIR/priv_validator_key"
+chain_ids = ["$CERC_CHAIN_ID"]
+EOF
 
 echo "Starting tmkms..."
 tmkms start --config $TMKMS_HOME/tmkms.toml