#!/usr/bin/env bash

set -e

# Changing user
sudo -S su user

# Update API_KEY and PASSWORD from users.yml
echo "sed"
sudo sed -i 's,API_KEY,'"$API_KEY"',g' /packages/users.yml
sudo sed -i 's,ADMIN_PASSWORD,'"$ADMIN_PASSWORD"',g' /packages/users.yml
sudo sed -i 's,ARGILLA_PASSWORD,'"$ARGILLA_PASSWORD"',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 ownerships
sudo mkdir -p /var/run/elasticsearch
sudo chown -R elasticsearch:elasticsearch /var/run/elasticsearch
sudo chown -R user:user /load_data.py

# Start elasticsearch
echo "reload"
sudo systemctl daemon-reload
echo "enable"
sudo systemctl enable elasticsearch.service
echo "starting"
sudo systemctl start elasticsearch.service
echo "started"

# Load data
echo "load data"
pip3 install datasets
python3.9 /load_data.py "$API_KEY" &

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