update install script for autorun ansible.

install-ten.sh

@@ -1,14 +1,22 @@
 #!/bin/bash
 
-# Function to prompt for input with validation
+CURRENT_DIR=$(pwd)
+
+# Function to prompt for input with validation and default value
 prompt_for_input() {
   local prompt_message=$1
   local input_variable_name=$2
   local validation_regex=$3
+  local default_value=$4
 
   while true; do
-    echo "$prompt_message"
+    echo -n "$prompt_message"
+    if [ -n "$default_value" ]; then
+      echo -n " [default: $default_value]"
+    fi
+    echo ":"
     read input_value
+    input_value=${input_value:-$default_value}
     if [[ $input_value =~ $validation_regex ]]; then
       eval $input_variable_name="'$input_value'"
       break
@@ -19,20 +27,20 @@ prompt_for_input() {
 }
 
 # Prompt for HOST_ID
-prompt_for_input "Please enter your HOST_ID (this is the public key of your validator account, e.g., 0x1234...):" HOST_ID '^0x[0-9a-fA-F]+$'
+prompt_for_input "Please enter your HOST_ID (this is the public key of your validator account, e.g., 0x1234...)" HOST_ID '^0x[0-9a-fA-F]+$'
 
 # Prompt for PRIVATE_KEY
-prompt_for_input "Please enter your PRIVATE_KEY associated with the HOST_ID (without the 0x prefix):" PRIVATE_KEY '^[0-9a-fA-F]+$'
+prompt_for_input "Please enter your PRIVATE_KEY associated with the HOST_ID (without the 0x prefix)" PRIVATE_KEY '^[0-9a-fA-F]+$'
 
 # Prompt for HOST_PUBLIC_P2P_ADDR_ROOT
-prompt_for_input "Please enter the DNS or external IP of your validator node (e.g., validator.example.com or 8.2.27.123):" HOST_PUBLIC_P2P_ADDR_ROOT '^[a-zA-Z0-9.-]+$'
+prompt_for_input "Please enter the DNS or external IP of your validator node (e.g., validator.example.com or 8.2.27.123)" HOST_PUBLIC_P2P_ADDR_ROOT '^[a-zA-Z0-9.-]+$'
 
 HOST_PUBLIC_P2P_ADDR="${HOST_PUBLIC_P2P_ADDR_ROOT}:10000"
 HOST_P2P_PORT="10000"
 LOG_LEVEL="3"
 
 # Prompt for L1_WS_URL
-prompt_for_input "Please enter the L1_WS_URL (a Sepolia client websocket URL, e.g., ws:// or wss://...):" L1_WS_URL '^wss?://.+$'
+prompt_for_input "Please enter the L1_WS_URL (a Sepolia client websocket URL, e.g., ws:// or wss://...)" L1_WS_URL '^wss?://.+$'
 
 # Prompt for DEPLOY_POSTGRES
 echo "Would you like to use an existing external Postgres instance? (Default is No - one will be deployed) [N/y]:"
@@ -41,11 +49,29 @@ read DEPLOY_POSTGRES
 if [[ "$DEPLOY_POSTGRES" =~ ^([yY][eE][sS]|[yY])$ ]]; then
   INCLUDE_POSTGRES_DB_HOST="true"
   # Prompt for POSTGRES_DB_HOST
-  prompt_for_input "Please enter the POSTGRES_DB_HOST (Postgres login URI):" POSTGRES_DB_HOST '^.+$'
+  prompt_for_input "Please enter the POSTGRES_DB_HOST (Postgres login URI)" POSTGRES_DB_HOST '^.+$'
 else
   INCLUDE_POSTGRES_DB_HOST="false"
 fi
 
+# Prompt for username with default value
+prompt_for_input "Please enter the username for SSH access to your validator node" SSH_USERNAME '^[a-zA-Z0-9._-]+$' "tenuser"
+
+# Prompt for SSH key path or password
+echo "Do you want to use an SSH key or password for authentication? (Enter 'key' or 'password', Default is key)"
+read AUTH_METHOD
+AUTH_METHOD=${AUTH_METHOD:-key}
+if [[ "$AUTH_METHOD" == "key" ]]; then
+  prompt_for_input "Please enter the path to your SSH key file (Just press Enter if you ran terraform)" SSH_KEY_PATH '^.+$' "${CURRENT_DIR}/terraform/ssh-key.pem"
+  SSH_CREDENTIAL="ansible_ssh_private_key_file=${SSH_KEY_PATH}"
+elif [[ "$AUTH_METHOD" == "password" ]]; then
+  prompt_for_input "Please enter the password for SSH authentication" SSH_PASSWORD '^.+$'
+  SSH_CREDENTIAL="ansible_ssh_pass=${SSH_PASSWORD}"
+else
+  echo "Invalid choice. Exiting."
+  exit 1
+fi
+
 # Generate a random password for Ansible Vault
 VAULT_PASSWORD=$(openssl rand -base64 32)
 
@@ -68,6 +94,19 @@ elif [ "$INCLUDE_POSTGRES_DB_HOST" = "false" ]; then
   echo "POSTGRES_DB_HOST: \"\"" >> ./ansible/files/node_secrets.yml
 fi
 
+# Create or update the hosts.ini file
+cat > ./ansible/hosts.ini <<EOL
+[local]
+localhost ansible_connection=local
+
+[ten-validator]
+${HOST_PUBLIC_P2P_ADDR_ROOT} # host or ip of your VM
+
+[ten-validator:vars]
+ansible_user=${SSH_USERNAME}
+${SSH_CREDENTIAL}
+EOL
+
 # Encrypt the file with ansible-vault using the .vaultpass file
 ansible-vault encrypt ./ansible/files/node_secrets.yml --vault-password-file .vaultpass
 
@@ -76,4 +115,9 @@ echo "The Ansible playbook will now be run. Please confirm by pressing Enter...
 read
 
 # Run the Ansible playbook
-ansible-playbook ./ansible/setup-validator-playbook.yaml
+cd ./ansible
+ansible-playbook -i hosts.ini setup-validator-playbook.yaml --vault-password-file ${CURRENT_DIR}/.vaultpass
+cd $CURRENT_DIR
+
+echo "Installation complete!"
+rm .vaultpass