#!/usr/bin/env bash

set -e

# Changing user
sudo -S su user

# Disable security in elasticsearch configuration
echo "xpack security"
sudo sed -i "s/xpack.security.enabled: true/xpack.security.enabled: false/g" /etc/elasticsearch/elasticsearch.yml
echo "cluster security"
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
echo "mkdir"
sudo mkdir -p /var/run/elasticsearch
echo "chown"
sudo chown -R elasticsearch:elasticsearch /var/run/elasticsearch

# Starting elasticsearch
echo "reload"
sudo systemctl daemon-reload
echo "enable"
sudo systemctl enable elasticsearch.service
echo "start"
sudo systemctl start elasticsearch.service

# Update API_KEY and PASSWORD from users.yml
echo "API_KEY"
sudo sed -i 's/API_KEY/$API_KEY/g' /packages/users.yml
echo "ADMIN_PASSWORD $ADMIN_PASSWORD"
sudo sed -i 's,ADMIN_PASSWORD,'"$ADMIN_PASSWORD"',g' /packages/users.yml
echo "ARGILLA_PASSWORD $ARGILLA_PASSWORD"
sudo sed -i 's,ARGILLA_PASSWORD,'"$ARGILLA_PASSWORD"',g' /packages/users.yml

# Load data
echo "Loading data"
pip3 install datasets
sudo python3.9 /load_data.py &

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