#!/bin/bash

# Function to read configuration using bashio if available
read_config() {
    if command -v bashio >/dev/null 2>&1; then
        bashio::config "$1"
    else
        # Fallback for testing outside Home Assistant - return empty to avoid issues
        case "$1" in
            "node_role") echo "master" ;;
            "cluster_name") echo "hass-opensearch-cluster" ;;
            "node_name") echo "os-node-1" ;;
            "heap_size") echo "1g" ;;
            "http_port") echo "9200" ;;
            "transport_port") echo "9300" ;;
            "discovery_seed_hosts") echo '["os-master:9300"]' ;;
            "initial_master_nodes") echo '["os-master"]' ;;
            "security_enabled") echo "true" ;;
            "admin_password") echo "SecurePass123!" ;;
            "plugins") echo "[]" ;;
            *) echo "" ;;
        esac
    fi
}

# Read configuration values
NODE_ROLE=$(read_config 'node_role' | tr -d '[]"' | tr ',' ' ')
CLUSTER_NAME=$(read_config 'cluster_name')
NODE_NAME=$(read_config 'node_name')
HEAP_SIZE=$(read_config 'heap_size')
HTTP_PORT=$(read_config 'http_port')
TRANSPORT_PORT=$(read_config 'transport_port')
DISCOVERY_SEEDS=$(read_config 'discovery_seed_hosts' | tr -d '[]"' | tr ',' ' ')
INITIAL_MASTERS=$(read_config 'initial_master_nodes' | tr -d '[]"' | tr ',' ' ')
SECURITY_ENABLED=$(read_config 'security_enabled')
ADMIN_PASSWORD=$(read_config 'admin_password')

# Export environment variables
export OPENSEARCH_INITIAL_ADMIN_PASSWORD="${ADMIN_PASSWORD}"
export DISCOVERY_TYPE="multi-node"

# Set heap size with proper format (remove any extra spaces or invalid characters)
CLEAN_HEAP_SIZE=$(echo "$HEAP_SIZE" | tr -d ' ' | grep -oE '[0-9]+[mg]')
if [ -z "$CLEAN_HEAP_SIZE" ]; then
    CLEAN_HEAP_SIZE="1g"  # Default if invalid
fi

# Set heap size in jvm.options.d (proper format)
mkdir -p /usr/share/opensearch/config/jvm.options.d
echo "-Xms${CLEAN_HEAP_SIZE}" > /usr/share/opensearch/config/jvm.options.d/heap.options
echo "-Xmx${CLEAN_HEAP_SIZE}" >> /usr/share/opensearch/config/jvm.options.d/heap.options

# Export variables for envsubst (ensure they have proper values)
export CLUSTER_NAME="${CLUSTER_NAME:-hass-opensearch-cluster}"
export NODE_NAME="${NODE_NAME:-os-node-1}"
export NODE_ROLE="${NODE_ROLE:-master}"
export HTTP_PORT="${HTTP_PORT:-9200}"
export TRANSPORT_PORT="${TRANSPORT_PORT:-9300}"
export DISCOVERY_SEEDS="${DISCOVERY_SEEDS:-os-master:9300}"
export INITIAL_MASTERS="${INITIAL_MASTERS:-os-master}"
export SECURITY_ENABLED="${SECURITY_ENABLED:-false}"

# Generate opensearch.yml from template
envsubst < /etc/opensearch/opensearch.yml.template > /usr/share/opensearch/config/opensearch.yml

# Install plugins if specified (only if not in test mode)
PLUGINS_JSON=$(read_config 'plugins')
if [ -n "$PLUGINS_JSON" ] && [ "$PLUGINS_JSON" != "[]" ] && command -v bashio >/dev/null 2>&1; then
    # Extract plugins from JSON array properly
    PLUGINS=$(echo "$PLUGINS_JSON" | jq -r '.[]' 2>/dev/null || echo "")
    if [ -n "$PLUGINS" ]; then
        for plugin in $PLUGINS; do
            echo "Installing plugin: $plugin"
            /usr/share/opensearch/bin/opensearch-plugin install --batch "$plugin" || true
        done
    fi
fi

# Start OpenSearch
exec /usr/share/opensearch/bin/opensearch