#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "${SCRIPT_DIR}/functions.sh"
resolve_paths

require_env
require_docker
resolve_volume_path

mkdir -p "${VOLUME_PATH}/synapse" "${VOLUME_PATH}/mas" "${VOLUME_PATH}/postgres" "${VOLUME_PATH}/proxy"
if [[ "${WORKER_COUNT}" -gt 0 ]]; then
mkdir -p "${VOLUME_PATH}/webapp"
fi
log_info "Initializing PostgreSQL..."
docker compose -f "$COMPOSE_FILE" --env-file "$ENV_FILE" up -d --wait matrix-db

# Wait for database readiness (silent except on failure)
for ATTEMPT in $(seq 1 60); do
    if docker compose -f "$COMPOSE_FILE" exec -T matrix-db psql -U "$PG_USER" -c "SELECT 1;" > /dev/null 2>&1; then
        break
    fi
    [[ $ATTEMPT -eq 30 ]] && { log_error "Database timeout"; exit 1; }
    sleep 2
done

# Database initialization SQL
INIT_SQL=$(cat <<EOF
DO \$\$ BEGIN
    IF NOT EXISTS (SELECT 1 FROM pg_roles WHERE rolname = 'mas') THEN
        CREATE ROLE mas WITH LOGIN PASSWORD '${PG_PASSWORD_MAS}';
    END IF;
    IF NOT EXISTS (SELECT 1 FROM pg_roles WHERE rolname = 'matrix') THEN
        CREATE ROLE matrix WITH LOGIN PASSWORD '${PG_PASSWORD_MATRIX}';
    END IF;
END \$\$;

SELECT CASE WHEN NOT EXISTS (SELECT 1 FROM pg_database WHERE datname = 'auth')
    THEN format('CREATE DATABASE %I OWNER %I', 'auth', 'mas') ELSE NULL END AS cmd \gexec

SELECT CASE WHEN NOT EXISTS (SELECT 1 FROM pg_database WHERE datname = 'synapse')
    THEN format('CREATE DATABASE %I OWNER %I', 'synapse', 'matrix') ELSE NULL END AS cmd \gexec
EOF
)

echo "$INIT_SQL" | docker compose -f "$COMPOSE_FILE" --env-file "$ENV_FILE" exec -T matrix-db psql -U "$PG_USER" -f - > /dev/null

log_info "Generating Synapse configuration..."
docker run -it --rm --network matrix_network \
    -v "${VOLUME_PATH}/synapse:/data" \
    -e SYNAPSE_SERVER_NAME="$HOMESERVER_FQDN" \
    -e SYNAPSE_REPORT_STATS=no \
    ghcr.io/element-hq/synapse:latest generate || { log_error "Synapse config generation failed"; exit 1; }

SYNAPSE_CFG="${VOLUME_PATH}/synapse/homeserver.yaml"
yq -y -i '.database.name = "psycopg2"' "$SYNAPSE_CFG"
yq -y -i ".database.args.user = \"${PG_USER_MATRIX}\"" "$SYNAPSE_CFG"
yq -y -i ".database.args.password = \"${PG_PASSWORD_MATRIX}\"" "$SYNAPSE_CFG"
yq -y -i ".database.args.database = \"${PG_DB_MATRIX}\"" "$SYNAPSE_CFG"
yq -y -i ".database.args.host = \"matrix-db\"" "$SYNAPSE_CFG"
yq -y -i ".database.args.port = 5432" "$SYNAPSE_CFG"
yq -y -i ".database.args.sslmode = \"prefer\"" "$SYNAPSE_CFG"
yq -y -i ".public_baseurl = \"https://${HOMESERVER_FQDN}\"" "$SYNAPSE_CFG"
yq -y -i ".server_name = \"${HOMESERVER_FQDN}\"" "$SYNAPSE_CFG"

fix_ownership "${VOLUME_PATH}/synapse"

log_info "Generating MAS configuration..."

docker run -it --rm --network matrix_network \
    -v "${VOLUME_PATH}/mas:/data" \
    ghcr.io/element-hq/matrix-authentication-service:latest config generate \
    | grep -v '^[0-9]\{4\}-[0-9]\{2\}-[0-9]\{2\}' > "${VOLUME_PATH}/mas/config.yaml" || { log_error "Mas config generation failed"; exit 1; }

MAS_CFG="${VOLUME_PATH}/mas/config.yaml"
# Ensure 'clients' array exists (prevents errors if we try to append later)
if ! yq '.clients' "$MAS_CFG" > /dev/null 2>&1; then
    yq -y -i '.clients = []' "$MAS_CFG"
fi
yq -y -i ".public_base_url = \"https://${MAS_FQDN}\"" "$MAS_CFG"
yq -y -i ".http.listen = \"0.0.0.0:8000\"" "$MAS_CFG"
yq -y -i ".database.uri = \"postgres://${PG_USER_MAS}:${PG_PASSWORD_MAS}@matrix-db:5432/auth?sslmode=prefer\"" "$MAS_CFG"

fix_ownership "${VOLUME_PATH}/mas"

log_info "Configuration complete"