-- Если база пуста, берём стартовую дату из GUC bookings.start_date (по умолчанию: 2017-01-01).
DO $$
DECLARE
    v_max_book_date timestamptz;
    v_start_date    timestamptz;
    v_end_date      timestamptz;
    v_bookings_cnt  bigint;
    v_jobs          integer := COALESCE(current_setting('bookings.jobs', true), '1')::integer;
    v_init_days     integer := COALESCE(current_setting('bookings.init_days', true), '1')::integer;
    v_start_cfg     text    := COALESCE(current_setting('bookings.start_date', true), '2017-01-01');
BEGIN
    -- Проверяем, что демобаза установлена
    IF to_regclass('bookings.bookings') IS NULL THEN
        RAISE EXCEPTION 'Таблица bookings.bookings не найдена. Сначала выполните make bookings-init или make bookings-generate.';
    END IF;

    IF v_jobs < 1 THEN
        RAISE EXCEPTION 'bookings.jobs должен быть >= 1. Текущее значение: %.', v_jobs;
    END IF;

    -- Ищем последнюю сгенерированную дату
    SELECT max(book_date) INTO v_max_book_date FROM bookings.bookings;

    IF v_max_book_date IS NULL THEN
        -- База пустая: берём стартовую дату из конфигурации (или дефолтную)
        v_start_date := date_trunc('day', v_start_cfg::timestamptz);
    ELSE
        -- Продолжаем с дня, следующего за максимальной датой
        v_start_date := date_trunc('day', v_max_book_date) + interval '1 day';
    END IF;

    -- Первая генерация вызывает generate(), последующие — continue()
    IF v_max_book_date IS NULL THEN
        v_end_date := v_start_date + (v_init_days || ' days')::interval;
        CALL generate(v_start_date, v_end_date, v_jobs);
    ELSE
        v_end_date := v_start_date + interval '1 day';
        CALL continue(v_end_date, v_jobs);
    END IF;

    -- continue() делает TRUNCATE gen.stat_jobs → AccessExclusiveLock.
    -- Без COMMIT воркеры не могут INSERT INTO gen.stat_jobs → deadlock.
    COMMIT;

    -- При jobs>1 воркерам нужно время, чтобы подключиться через dblink и
    -- выставить application_name='Airlines processor'. Без паузы busy()
    -- сразу вернёт false (воркеры ещё не видны в pg_stat_activity).
    IF v_jobs > 1 THEN
        PERFORM pg_sleep(3);
    END IF;

    -- Ждём завершения фоновых джобов генератора, чтобы данные успели записаться
    WHILE busy() LOOP
        PERFORM pg_sleep(1);
    END LOOP;
    PERFORM dblink_disconnect(unnest(dblink_get_connections()));

    -- Если данных нет, останавливаемся с понятной ошибкой
    SELECT COUNT(*) INTO v_bookings_cnt FROM bookings.bookings;
    IF v_bookings_cnt = 0 THEN
        RAISE EXCEPTION 'Генератор demodb завершился, но bookings.bookings пустая. Проверьте применение патчей и логи генератора.';
    END IF;

    -- Генерация закончена — возвращаем synchronous_commit = on и сбрасываем
    -- буферы на диск. Без этого docker compose down может убить PostgreSQL
    -- до записи WAL → данные пропадут (особенно на WSL2).
    ALTER DATABASE demo SET synchronous_commit = on;
    SET synchronous_commit = on;
END $$;

CHECKPOINT;