pg_exporter 0.11.1

---
name: Coverage

on:
  workflow_call:
    secrets:
      CODECOV_TOKEN:
        required: true

jobs:
  coverage:
    name: Coverage
    runs-on: ubuntu-latest

    services:
      postgres:
        image: postgres:18
        env:
          POSTGRES_USER: postgres
          POSTGRES_PASSWORD: postgres
          POSTGRES_DB: postgres
          POSTGRES_HOST_AUTH_METHOD: trust
        options: >-
          --health-cmd "pg_isready -U postgres"
          --health-interval 5s
          --health-timeout 5s
          --health-retries 10
          --health-start-period 10s
        ports:
          - 5432:5432

    steps:
      - uses: actions/checkout@v6

      - name: Cache Rust toolchain and dependencies
        uses: actions/cache@v5
        with:
          path: |
            ~/.cargo/bin/
            ~/.cargo/registry/index/
            ~/.cargo/registry/cache/
            ~/.cargo/git/db/
            target/
          key: ${{ runner.os }}-cargo-coverage-${{ hashFiles('**/Cargo.lock') }}
          restore-keys: |
            ${{ runner.os }}-cargo-coverage-
            ${{ runner.os }}-cargo-

      - uses: dtolnay/rust-toolchain@stable
        with:
          components: llvm-tools-preview

      - name: Wait for PostgreSQL
        run: |
          timeout 30 bash -c 'until pg_isready -h localhost -p 5432 -U postgres; do sleep 1; done'
          echo "PostgreSQL is ready!"

      - name: Configure container runtime for testcontainers
        run: |
          if [[ -n "${DOCKER_HOST:-}" ]]; then
            echo "Using preconfigured DOCKER_HOST=${DOCKER_HOST}"
            echo "DOCKER_HOST=${DOCKER_HOST}" >> "${GITHUB_ENV}"
            exit 0
          fi

          if [[ -S /var/run/docker.sock ]]; then
            echo "DOCKER_HOST=unix:///var/run/docker.sock" >> "${GITHUB_ENV}"
            echo "Using Docker socket at /var/run/docker.sock"
            exit 0
          fi

          if [[ -n "${XDG_RUNTIME_DIR:-}" && -S "${XDG_RUNTIME_DIR}/podman/podman.sock" ]]; then
            echo "DOCKER_HOST=unix://${XDG_RUNTIME_DIR}/podman/podman.sock" >> "${GITHUB_ENV}"
            echo "Using Podman socket at ${XDG_RUNTIME_DIR}/podman/podman.sock"
            exit 0
          fi

          if [[ -S "/run/user/$(id -u)/podman/podman.sock" ]]; then
            echo "DOCKER_HOST=unix:///run/user/$(id -u)/podman/podman.sock" >> "${GITHUB_ENV}"
            echo "Using Podman socket at /run/user/$(id -u)/podman/podman.sock"
            exit 0
          fi

          echo "No Docker/Podman socket available for testcontainers" >&2
          exit 1

      - name: Install and configure pg_stat_statements extension
        run: |
          # Configure shared_preload_libraries (requires restart)
          psql -h localhost -U postgres -d postgres -c "ALTER SYSTEM SET shared_preload_libraries = 'pg_stat_statements';"

          # Restart PostgreSQL container to load pg_stat_statements
          docker ps -a
          CONTAINER_ID=$(docker ps --filter "ancestor=postgres:18" --format "{{.ID}}")
          echo "Restarting container $CONTAINER_ID"
          docker restart $CONTAINER_ID

          # Wait for PostgreSQL to be ready again
          sleep 5
          timeout 30 bash -c 'until pg_isready -h localhost -p 5432 -U postgres; do sleep 1; done'

          # Now set pg_stat_statements config (extension is preloaded)
          psql -h localhost -U postgres -d postgres -c "ALTER SYSTEM SET pg_stat_statements.track = 'all';"

          # Reload configuration
          psql -h localhost -U postgres -d postgres -c "SELECT pg_reload_conf();"

          # Enable pg_stat_statements extension
          psql -h localhost -U postgres -d postgres -c "CREATE EXTENSION IF NOT EXISTS pg_stat_statements;"

          # Verify extension is loaded
          psql -h localhost -U postgres -d postgres -c "SELECT extname, extversion FROM pg_extension WHERE extname = 'pg_stat_statements';"

          # Generate some test queries to populate pg_stat_statements
          psql -h localhost -U postgres -d postgres -c "SELECT 1;"
          psql -h localhost -U postgres -d postgres -c "SELECT current_timestamp;"
          psql -h localhost -U postgres -d postgres -c "SELECT COUNT(*) FROM pg_stat_statements;"

      - name: Run tests
        run: cargo test --verbose -- --nocapture
        env:
          PG_EXPORTER_DSN: postgresql://postgres:postgres@localhost:5432/postgres
          PG_EXPORTER_REQUIRE_TESTCONTAINERS: '1'
          RUST_BACKTRACE: full
          CARGO_INCREMENTAL: 0
          LLVM_PROFILE_FILE: coverage-%p-%m.profraw
          RUSTFLAGS: -Cinstrument-coverage -Ccodegen-units=1 -Clink-dead-code -Coverflow-checks=off
          RUSTDOCFLAGS: -Cinstrument-coverage -Ccodegen-units=1 -Clink-dead-code -Coverflow-checks=off

      - name: Install grcov
        run: |
          if [[ ! -e ~/.cargo/bin/grcov ]]; then
            cargo install grcov
          else
            echo "grcov already installed"
          fi

      - name: Run grcov
        run: grcov . --binary-path target/debug/ -s . -t lcov --branch --ignore-not-existing
          --ignore '../**' --ignore '/*' -o coverage.lcov

      - name: Upload to codecov.io
        uses: codecov/codecov-action@v6
        env:
          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
        with:
          files: coverage.lcov
          flags: rust

      - name: Coveralls GitHub Action
        uses: coverallsapp/github-action@v2
        with:
          github-token: ${{ secrets.GITHUB_TOKEN }}
          file: coverage.lcov
          fail-on-error: false