FROM postgres:16.10-alpine AS builder

# Install curl and xz for downloading and decompressing SQL dumps
RUN apk add --no-cache curl xz

# PostgreSQL configuration
ENV POSTGRES_USER=postgres
ENV POSTGRES_PASSWORD=zoopassword
ENV POSTGRES_DB=zoodb
ENV POSTGRES_HOST_AUTH_METHOD=md5

# Copy seed data and init script
COPY seed/ /seed/
COPY init-databases.sh /usr/local/bin/
COPY fast-import.conf /tmp/fast-import.conf

# Download and decompress postmill database dump from R2
RUN curl -L https://pub-c02faf5a0de84bc3b61262b62a029c8a.r2.dev/reddit/postmill_dump_slim.sql.xz -o /seed/postmill.sql.xz && \
    xz -d /seed/postmill.sql.xz

# Accept build arg for no seeding
ARG ZOO_NO_SEED=false
ENV ZOO_NO_SEED=$ZOO_NO_SEED

# Initialize PostgreSQL with base configuration
RUN mkdir -p /var/lib/postgresql/data && \
    chown -R postgres:postgres /var/lib/postgresql && \
    su-exec postgres initdb -D /var/lib/postgresql/data && \
# Configure pg_hba.conf to allow md5 connections from Docker network
    echo "host all all 0.0.0.0/0 md5" >> /var/lib/postgresql/data/pg_hba.conf && \
# Backup clean postgresql.conf before modifying for fast import
    cp /var/lib/postgresql/data/postgresql.conf /tmp/postgresql.conf.clean && \
# Append fast import settings to postgresql.conf (includes Unix socket-only config)
    cat /tmp/fast-import.conf >> /var/lib/postgresql/data/postgresql.conf && \
# Kill any stale postgres processes (CI build safety)
    pkill -9 postgres || true && \
    sleep 1 && \
# Start PostgreSQL server with optimized settings (Unix sockets only)
    su-exec postgres pg_ctl -D /var/lib/postgresql/data -l /tmp/postgres.log start && \
# Wait for PostgreSQL to be ready using pg_isready
    until su-exec postgres pg_isready -q; do \
        echo "Waiting for PostgreSQL to be ready..." && \
        sleep 1; \
    done && \
# Now set password and create database using psql
    su-exec postgres psql -U postgres -c "ALTER USER postgres PASSWORD '$POSTGRES_PASSWORD';" && \
    su-exec postgres psql -U postgres -c "CREATE DATABASE $POSTGRES_DB;" && \
# Run the database initialization script
    /usr/local/bin/init-databases.sh || (echo "Failed to initialize databases. Logs:" && cat /tmp/postgres.log && exit 1) && \
# Run CHECKPOINT to flush all data and reduce WAL files
    su-exec postgres psql -U postgres -c "CHECKPOINT;" && \
# Stop PostgreSQL cleanly
    su-exec postgres pg_ctl -D /var/lib/postgresql/data stop -m smart && \
# Wait for PostgreSQL to fully stop
    until ! su-exec postgres pg_isready -q 2>/dev/null; do \
        echo "Waiting for PostgreSQL to stop..." && \
        sleep 1; \
    done && \
# Report directory sizes before cleanup
    echo "Directory sizes before cleanup:" && \
    du -sh /var/lib/postgresql/data/pg_wal && \
    du -sh /var/lib/postgresql/data/pg_stat_tmp 2>/dev/null || echo "pg_stat_tmp: 0" && \
    du -sh /var/lib/postgresql/data && \
# Restore clean postgresql.conf (remove fast-import settings)
    cp /tmp/postgresql.conf.clean /var/lib/postgresql/data/postgresql.conf && \
# Reset WAL to minimal state (safe after clean CHECKPOINT + shutdown)
    su-exec postgres pg_resetwal -f /var/lib/postgresql/data && \
# Remove temporary files
    rm -rf /var/lib/postgresql/data/pg_stat_tmp/* && \
# Report final size
    echo "Final data directory size:" && \
    du -sh /var/lib/postgresql/data && \
    du -sh /var/lib/postgresql/data/pg_wal && \
# Create uncompressed tar backup (same pattern as MySQL)
    cd /var/lib/postgresql && \
    tar -cf /var/lib/postgresql/data-golden.tar data

FROM postgres:16.10-alpine

# Copy the uncompressed tar backup
COPY --from=builder /var/lib/postgresql/data-golden.tar /var/lib/postgresql/data-golden.tar

# PostgreSQL configuration (must match builder stage)
ENV POSTGRES_USER=postgres
ENV POSTGRES_PASSWORD=zoopassword
ENV POSTGRES_DB=zoodb
ENV POSTGRES_HOST_AUTH_METHOD=md5

# Copy custom entrypoint script
COPY custom-entrypoint.sh /usr/local/bin/

ENTRYPOINT ["custom-entrypoint.sh"]
CMD ["postgres"]