#!/bin/bash
set -euo pipefail

# Load environment
source <%= @ps_env_file %>

LOG_FILE="<%= @ps_prune_log_file %>"
echo "$(date '+%Y-%m-%d %H:%M:%S') Starting prune run" >> "$LOG_FILE"

# -----------------------
# Iterate TSDB blocks
# -----------------------
for block in "$TSDB_DIR"/*/; do
    [[ -d "$block" ]] || continue
    meta="$block/meta.json"
    [[ -f "$meta" ]] || continue

    min_time=$(jq -r '.minTime' "$meta")
    max_time=$(jq -r '.maxTime' "$meta")

    # Skip last MIN_AGE_HOURS
    block_age_hours=$(( ($(date +%s) - min_time/1000) / 3600 ))
    if (( block_age_hours < MIN_AGE_HOURS + GRACE_HOURS )); then
        log "Keeping block $block (age ${block_age_hours}h)"
        continue
    fi

    # Check existence in PostgreSQL
    check_sql="${CHECK_SQL_TEMPLATE//\{min\}/$min_time}"
    check_sql="${check_sql//\{max\}/$max_time}"

    exists=$(psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d "$PGDATABASE" -t -c "$check_sql" | xargs || echo "0")

    if [[ "$exists" != "1" ]]; then
        log "Block $block not confirmed in PostgreSQL, skipping deletion"
        continue
    fi

    # Delete block
    if [[ "$DRY_RUN" == "true" ]]; then
        log "[DRY-RUN] Would delete block $block"
    else
        log "Deleting block $block"
        rm -rf "$block"
    fi
done

echo "$(date '+%Y-%m-%d %H:%M:%S') Prune run finished" >> "$LOG_FILE"