#!/bin/bash

# Load the .env file

source .env

# Check if LOCKBOX_MASTER_KEY is present or empty

if [[ -z "$LOCKBOX_MASTER_KEY" ]]; then

# Generate LOCKBOX_MASTER_KEY

LOCKBOX_MASTER_KEY=$(openssl rand -hex 32)

# Update .env file

awk -v key="$LOCKBOX_MASTER_KEY" '

BEGIN { FS=OFS="=" }

/^LOCKBOX_MASTER_KEY=/ { $2=key; found=1 }

1

END { if (!found) print "LOCKBOX_MASTER_KEY="key }

' .env > temp.env && mv temp.env .env

echo "Generated a secure master key for the lockbox"

else

echo "The lockbox master key already exists."

fi

# Check if SECRET_KEY_BASE is present or empty

if [[ -z "$SECRET_KEY_BASE" ]]; then

# Generate SECRET_KEY_BASE

SECRET_KEY_BASE=$(openssl rand -hex 64)

# Update .env file

awk -v key="$SECRET_KEY_BASE" '

BEGIN { FS=OFS="=" }

/^SECRET_KEY_BASE=/ { $2=key; found=1 }

1

END { if (!found) print "SECRET_KEY_BASE="key }

' .env > temp.env && mv temp.env .env

echo "Created a secret key for secure operations."

else

echo "The secret key base is already in place."

fi

# Check if PGRST_JWT_SECRET is present or empty

if [[ -z "$PGRST_JWT_SECRET" ]]; then

# Generate PGRST_JWT_SECRET

PGRST_JWT_SECRET=$(openssl rand -hex 32)

# Update .env file

awk -v key="$PGRST_JWT_SECRET" '

BEGIN { FS=OFS="=" }

/^PGRST_JWT_SECRET=/ { $2=key; found=1 }

1

END { if (!found) print "PGRST_JWT_SECRET="key }

' .env > temp.env && mv temp.env .env

echo "Generated a unique secret for PGRST authentication."

else

echo "The PGRST JWT secret is already generated and in place."

fi

# Function to generate a random password

generate_password() {

openssl rand -base64 12 | tr -d '/+' | cut -c1-16

}

# Check if PG_USER, PG_HOST, PG_PASS, PG_DB are present or empty

if [[ -z "$PG_USER" ]] || [[ -z "$PG_HOST" ]] || [[ -z "$PG_PASS" ]] || [[ -z "$PG_DB" ]]; then

# Prompt user for values

read -p "Enter PostgreSQL database username: " PG_USER

read -p "Enter PostgreSQL database hostname: " PG_HOST

read -p "Enter PostgreSQL database password: " PG_PASS

read -p "Enter PostgreSQL database name: " PG_DB

# Update .env file

awk -v pg_user="$PG_USER" -v pg_host="$PG_HOST" -v pg_pass="$PG_PASS" -v pg_db="$PG_DB" '

BEGIN { FS=OFS="=" }

/^PG_USER=/ { $2=pg_user; found=1 }

/^PG_HOST=/ { $2=pg_host; found=1 }

/^PG_PASS=/ { $2=pg_pass; found=1 }

/^PG_DB=/ { $2=pg_db; found=1 }

1

END {

if (!found) {

print "PG_USER="pg_user

print "PG_HOST="pg_host

print "PG_PASS="pg_pass

print "PG_DB="pg_db

}

}

' .env > temp.env && mv temp.env .env

echo "Successfully updated postgresql database values .env file"

fi

# Copy values from PG to TOOLJET_DB

TOOLJET_DB_USER=$PG_USER

TOOLJET_DB_HOST=$PG_HOST

TOOLJET_DB_PASS=$PG_PASS

# Update .env file for TOOLJET_DB

awk -v tj_user="$TOOLJET_DB_USER" -v tj_host="$TOOLJET_DB_HOST" -v tj_pass="$TOOLJET_DB_PASS" '

BEGIN { FS=OFS="=" }

/^TOOLJET_DB_USER=/ { $2=tj_user; found=1 }

/^TOOLJET_DB_HOST=/ { $2=tj_host; found=1 }

/^TOOLJET_DB_PASS=/ { $2=tj_pass; found=1 }

1

END { if (!found) print "TOOLJET_DB_USER="tj_user ORS "TOOLJET_DB_HOST="tj_host ORS "TOOLJET_DB_PASS="tj_pass }

' .env > temp.env && mv temp.env .env

echo "Successfully updated tooljet database values in the .env file"

# Construct PGRST_DB_URI with user-provided values

PGRST_DB_URI="postgres://$PG_USER:$PG_PASS@$PG_HOST/tooljet_db"

# Update .env file for PGRST_DB_URI

awk -v uri="$PGRST_DB_URI" '

BEGIN { FS=OFS="=" }

/^PGRST_DB_URI=/ { $2=uri; found=1 }

1

END { if (!found) print "PGRST_DB_URI="uri }

' .env > temp.env && mv temp.env .env

echo "Successfully updated PGRST database URI"

if [[ -z $PG_USER || -z $PG_PASS || -z $PG_HOST ]]

then

echo "Please set the required PG_USER, PG_PASS, and PG_HOST values within the .env file"

exit 1

fi

export $(grep -v '^#' .env | xargs)

if psql -d postgresql://$PG_USER:$PG_PASS@$PG_HOST/postgres -c 'select now()' > /dev/null 2>&1

then

echo "Successfully pinged the database!";

else

echo "Can't connect to the database. Kindly check the credenials provided in the .env file!"

exit 1

fi

if sudo systemctl start redis-server && sudo systemctl enable redis-server

then

echo "Successfully started Redis!"

else

echo "Failed to start and enable Redis"

fi

if sudo -E systemctl start openresty

then

echo "Successfully started reverse proxy!"

else

echo "Failed to start reverse proxy"

exit 1

fi

if sudo -E systemctl start postgrest

then

echo "Successfully started PostgREST server!"

else

echo "Failed to start PostgREST server"

exit 1

fi

if [[ "$WORKFLOW_WORKER" == "true" ]]; then

echo "WORKER is true. Running the worker..."

npm run worker:prod &

else

echo "WORKER is not true. Skipping the worker execution."

fi

if sudo systemctl start neo4j && sudo systemctl enable neo4j

then

echo "Successfully started Neo4j!"

else

echo "Failed to start and enable Neo4j"

exit 1

fi

TOOLJET_EDTION=ee npm --prefix server run db:setup:prod

if sudo -E systemctl start nest

then

echo "The app will be served at ${TOOLJET_HOST}"

else

echo "Failed to start the server!"

exit 1

fi

sudo systemctl restart nest

sudo -E systemctl restart postgrest