#!/usr/bin/env bash

set -e

# Changing user
echo "ubuntu" | sudo -S su user

# Generate hashed passwords
admin_password=$(htpasswd -nbB "" "$ADMIN_PASSWORD" | cut -d ":" -f 2 | tr -d "\n")
argilla_password=$(htpasswd -nbB "" "$ARGILLA_PASSWORD" | cut -d ":" -f 2 | tr -d "\n")

sed -i "s/ADMIN_PASSWORD/$ADMIN_PASSWORD/g" /packages/users.yml
sed -i "s/ARGILLA_PASSWORD/$ARGILLA_PASSWORD/g" /packages/users.yml
sed -i "s/ADMIN_API_KEY/$ADMIN_API_KEY/g" /packages/users.yml
sed -i "s/ARGILLA_API_KEY/$ARGILLA_API_KEY/g" /packages/users.yml

# Disable security in elasticsearch configuration
sudo sed -i "s/xpack.security.enabled: true/xpack.security.enabled: false/g" /etc/elasticsearch/elasticsearch.yml
sudo sed -i "s/cluster.initial_master_nodes/#cluster.initial_master_nodes/g" /etc/elasticsearch/elasticsearch.yml
echo "cluster.routing.allocation.disk.threshold_enabled: false" | sudo tee -a /etc/elasticsearch/elasticsearch.yml

# Create elasticsearch directory and change ownership
sudo mkdir -p /var/run/elasticsearch
sudo chown -R elasticsearch:elasticsearch /var/run/elasticsearch

# Starting elasticsearch
sudo systemctl daemon-reload
sudo systemctl enable elasticsearch
sudo systemctl start elasticsearch

# Load data
python3.9 /app/load_data.py &

# Start argilla
uvicorn argilla:app --host "0.0.0.0"