#!/bin/bash
set -e

# Function to extract domain from Git SSH URL
extract_domain() {
    local url=$1

    # Handle different SSH URL formats
    if [[ $url == ssh://* ]]; then
        # For URLs starting with ssh://, extract the domain
        echo "$url" | sed -E 's|ssh://[^@]+@([^:/]+).*|\1|'
    else
        # For URLs without ssh://, extract the domain after '@' and before ':'
        echo "$url" | sed -E 's|[^@]+@([^:/]+).*|\1|'
    fi
}

# Check if the required environment variables are set
if [ -z "$REPO_URL" ]; then
    echo "Error: REPO_URL environment variable must be set."
    exit 1
fi

if [ ! -f /root/.ssh/id_rsa ]; then
    echo "Error: SSH private key not found at /root/.ssh/id_rsa"
    exit 1
fi

if [ -z "$GIT_USER" ]; then
    echo "Error: GIT_USER environment variable must be set."
    exit 1
fi

if [ -z "$GIT_EMAIL" ]; then
    echo "Error: GIT_EMAIL environment variable must be set."
    exit 1
fi

# If no teleport edition is given, fallback to oss
if [ -z "$TELEPORT_EDITION" ]; then
    TELEPORT_EDITION="oss"
fi

# Install Teleport if the environment variables are set
if [ -n "$TELEPORT_VERSION" ] && [ -n "$TELEPORT_URL" ]; then
    echo "Installing Teleport version $TELEPORT_VERSION, edition $TELEPORT_EDITION..."
    echo "Executing: curl https://goteleport.com/static/install.sh | bash -s $TELEPORT_VERSION $TELEPORT_EDITION"
    curl https://goteleport.com/static/install.sh | bash -s "${TELEPORT_VERSION}" "${TELEPORT_EDITION}"

    # If teleport token is set and there is no teleport.yaml file, create one
    if [ -n "$TELEPORT_TOKEN" ] && [ ! -f /etc/teleport.yaml ]; then
        teleport configure --roles=node --token="$TELEPORT_TOKEN" --proxy="$TELEPORT_URL" --no-acme -o file
        echo "Created teleport configuration file"
    fi
else
    echo "Error: TELEPORT_VERSION and TELEPORT_URL environment variables must be set."
    exit 1
fi

# Ensure the SSH key has the correct permissions
chmod 600 /root/.ssh/id_rsa

# Extract the domain from the REPO_URL
DOMAIN=$(extract_domain "$REPO_URL")

# Check if the repository directory already exists
if [ ! -d "/workspace/.git" ]; then
    # Add the SSH configuration for the domain
    echo "Host $DOMAIN
      HostName $DOMAIN
      IdentityFile /root/.ssh/id_rsa
      StrictHostKeyChecking no
    " >>/root/.ssh/config

    # Clone the repository if it doesn't exist
    git clone "$REPO_URL" /workspace
fi

# Add the public SSH key to authorized_keys
if [ -f /root/.ssh/hostkey.pub ]; then
    cat /root/.ssh/hostkey.pub >>/root/.ssh/authorized_keys
fi

# Change to the repository directory
cd /workspace

# Install dependencies
npm install

# Copy the environment file to the repository's environments directory
if [ -f /environment/.env ]; then
    mkdir -p /workspace/environments
    cp /environment/.env /workspace/.env
else
    echo "Error: Environment file not found at /environment/.env"
fi

# Set the git user and email
git config --global user.name "$GIT_USER"
git config --global user.email "$GIT_EMAIL"

# Get the hostname of the machine
HOSTNAME=$(hostname)

# File path
CONFIG_FILE="/etc/teleport.yaml"

# Edit the teleport.yaml file to update the nodename
sed -i "s/^  nodename:.*/  nodename: $HOSTNAME/" "$CONFIG_FILE"

# Start Teleport in the background
nohup teleport start &

# Start the application with pm2
exec pm2-runtime "npm run dev"