79 lines
3.0 KiB
Bash
79 lines
3.0 KiB
Bash
#!/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
|
|
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 using environment variable (better approach)
|
|
CLEAN_HEAP_SIZE=$(echo "$HEAP_SIZE" | tr -d ' ' | grep -oE '[0-9]+[mg]')
|
|
if [ -z "$CLEAN_HEAP_SIZE" ]; then
|
|
CLEAN_HEAP_SIZE="2g"
|
|
fi
|
|
export OPENSEARCH_JAVA_OPTS="-Xms${CLEAN_HEAP_SIZE} -Xmx${CLEAN_HEAP_SIZE}"
|
|
|
|
# Remove any existing heap configuration to avoid duplicates
|
|
rm -f /usr/share/opensearch/config/jvm.options.d/heap.options
|
|
rm -f /usr/share/opensearch/config/jvm.options
|
|
|
|
# Export variables for envsubst
|
|
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
|
|
PLUGINS_JSON=$(read_config 'plugins')
|
|
if [ -n "$PLUGINS_JSON" ] && [ "$PLUGINS_JSON" != "[]" ] && command -v bashio >/dev/null 2>&1; then
|
|
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 |