"""v2 second_brain: pipeline_settings table

Revision ID: 7b3e8a5c1f29
Revises: 4a1e2f6c9d10
Create Date: 2026-05-25 00:00:00.000000

Adds a single-row `pipeline_settings` table that the web dashboard edits
and the workers (current dev-side scheduler + future tower-side transcribe
worker) read at the start of each run. The transcription_* fields persist
now but are only consumed once the transcribe worker exists — the
extraction_* fields are wired into the existing scheduler in this commit.

Single-row enforcement: PK is fixed at `id=1` with a CHECK constraint, so
INSERT-as-upsert-by-PK keeps the table free of stale rows.

Schema/permission notes mirror the v1 migration: lovebug lacks CREATE on
the petalbrain database, so env.py has already bootstrapped the schema —
this migration just SETs search_path and emits DDL inside it.
"""

from __future__ import annotations

from collections.abc import Sequence

from alembic import op

revision: str = "7b3e8a5c1f29"
down_revision: str | Sequence[str] | None = "4a1e2f6c9d10"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.execute("SET search_path TO second_brain")

    op.execute(
        """
        CREATE TABLE pipeline_settings (
            id                                    INTEGER PRIMARY KEY DEFAULT 1,
            transcription_enabled                 BOOLEAN NOT NULL DEFAULT TRUE,
            transcription_active_hours_start      TIME,
            transcription_active_hours_end        TIME,
            transcription_max_concurrent_gpu_jobs INTEGER NOT NULL DEFAULT 1,
            transcription_max_video_length_seconds INTEGER,
            transcription_max_items_per_run       INTEGER NOT NULL DEFAULT 10,
            extraction_enabled                    BOOLEAN NOT NULL DEFAULT TRUE,
            extraction_active_hours_start         TIME,
            extraction_active_hours_end           TIME,
            extraction_max_items_per_run          INTEGER NOT NULL DEFAULT 20,
            updated_at                            TIMESTAMP NOT NULL
                                                  DEFAULT (now() AT TIME ZONE 'UTC'),
            CONSTRAINT pipeline_settings_singleton CHECK (id = 1)
        )
        """
    )

    # Seed the singleton row with the table's column defaults. Idempotent
    # via ON CONFLICT so re-running this migration on a partly-applied DB
    # (or a manual cherry-pick) is safe.
    op.execute(
        """
        INSERT INTO pipeline_settings (id) VALUES (1)
        ON CONFLICT (id) DO NOTHING
        """
    )


def downgrade() -> None:
    op.execute("SET search_path TO second_brain")
    op.execute("DROP TABLE IF EXISTS pipeline_settings")