#!/usr/bin/env sh
set -eu

if [ "${1:-}" = "" ]; then
  echo "Usage: $0 path/to/osm.pbf" >&2
  exit 2
fi

PBF_PATH=$1
if [ ! -f "$PBF_PATH" ]; then
  echo "OSM PBF file not found: $PBF_PATH" >&2
  exit 2
fi

if [ -f .env ]; then
  set -a
  # shellcheck disable=SC1091
  . ./.env
  set +a
fi

DB_NAME=${POSTGRES_DB:-radio}
DB_USER=${POSTGRES_USER:-radio}
DB_PASSWORD=${DB_PASSWORD:-radio}
POSTGIS_SERVICE=${POSTGIS_SERVICE:-postgis}
OSM2PGSQL_IMAGE=${OSM2PGSQL_IMAGE:-radio-osm2pgsql:latest}

SCRIPT_DIR=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
REPO_DIR=$(CDPATH= cd -- "$SCRIPT_DIR/.." && pwd)
PBF_DIR=$(CDPATH= cd -- "$(dirname -- "$PBF_PATH")" && pwd)
PBF_FILE=$(basename -- "$PBF_PATH")
POSTGIS_CONTAINER=$(docker compose ps -q "$POSTGIS_SERVICE")

if [ -z "$POSTGIS_CONTAINER" ]; then
  echo "PostGIS container is not running. Start it with: docker compose up -d postgis" >&2
  exit 1
fi

NETWORK=$(docker inspect "$POSTGIS_CONTAINER" \
  --format '{{range $name, $_ := .NetworkSettings.Networks}}{{println $name}}{{end}}' \
  | head -n 1)

if ! docker image inspect "$OSM2PGSQL_IMAGE" >/dev/null 2>&1; then
  echo "Build local osm2pgsql image: $OSM2PGSQL_IMAGE"
  docker build \
    -t "$OSM2PGSQL_IMAGE" \
    -f "$SCRIPT_DIR/Dockerfile.osm2pgsql" \
    "$SCRIPT_DIR"
fi

echo "Ensure PostGIS extension exists"
docker compose exec -T "$POSTGIS_SERVICE" psql -U "$DB_USER" -d "$DB_NAME" \
  < "$REPO_DIR/db/init/01_postgis.sql"

echo "Drop previous buildings table"
docker compose exec -T "$POSTGIS_SERVICE" psql -U "$DB_USER" -d "$DB_NAME" \
  -c "DROP TABLE IF EXISTS buildings CASCADE;
      DROP TABLE IF EXISTS planet_osm_point CASCADE;
      DROP TABLE IF EXISTS planet_osm_line CASCADE;
      DROP TABLE IF EXISTS planet_osm_polygon CASCADE;
      DROP TABLE IF EXISTS planet_osm_roads CASCADE;
      DROP TABLE IF EXISTS planet_osm_nodes CASCADE;
      DROP TABLE IF EXISTS planet_osm_ways CASCADE;
      DROP TABLE IF EXISTS planet_osm_rels CASCADE;"

echo "Import buildings from $PBF_PATH"
docker run --rm \
  --network "$NETWORK" \
  -e PGPASSWORD="$DB_PASSWORD" \
  -v "$PBF_DIR:/osm:ro" \
  "$OSM2PGSQL_IMAGE" \
  -H "$POSTGIS_SERVICE" \
  -P 5432 \
  -U "$DB_USER" \
  -d "$DB_NAME" \
  --create \
  --slim \
  --drop \
  --latlong \
  --hstore \
  "/osm/$PBF_FILE"

echo "Normalize buildings table"
docker compose exec -T "$POSTGIS_SERVICE" psql -U "$DB_USER" -d "$DB_NAME" \
  < "$REPO_DIR/db/sql/normalize_buildings.sql"

echo "Buildings import complete"