#!/bin/bash
set -e

# Restore the original password
ORIGINAL_PASSWORD=$(echo "${SERVERLESS_CLUSTER_PASSWORD}" | sed 's/\\\$/\$/g' | sed 's/\\\!/\!/g' | sed 's/\\\^/\^/g')

# Creating the database if it doesn't exist
echo "Creating the database if it doesn't exist..."
cat > ~/.my.cnf <<EOF
[client]
host=${SERVERLESS_CLUSTER_HOST}
port=${SERVERLESS_CLUSTER_PORT}
user='${SERVERLESS_CLUSTER_USERNAME}'
password='${ORIGINAL_PASSWORD}'
EOF

chmod 600 ~/.my.cnf

mysql -e "CREATE DATABASE IF NOT EXISTS \`${SERVERLESS_CLUSTER_DATABASE_NAME}\`;"

# Clone the repository
echo "Cloning repository..."
git clone https://github.com/Icemap/tidb-lab-unify-storage-openai.git
cd tidb-lab-unify-storage-openai

# Write .env
echo "Writing .env file..."
cat > .env <<EOF
SERVERLESS_CLUSTER_HOST=${SERVERLESS_CLUSTER_HOST}
SERVERLESS_CLUSTER_PORT=${SERVERLESS_CLUSTER_PORT}
SERVERLESS_CLUSTER_USERNAME='${SERVERLESS_CLUSTER_USERNAME}'
SERVERLESS_CLUSTER_PASSWORD='${ORIGINAL_PASSWORD}'
SERVERLESS_CLUSTER_DATABASE_NAME='${SERVERLESS_CLUSTER_DATABASE_NAME}'
OPENAI_API_KEY=<Enter your OpenAI API Key>
EOF

# Run Docker compose
echo "Starting Docker compose..."
sudo docker compose up -d