from pathlib import Path

import psycopg

from .config import settings


def main() -> None:
    migration_dir = Path(__file__).resolve().parent.parent / "migrations"
    with psycopg.connect(settings.database_url) as connection:
        connection.execute("CREATE TABLE IF NOT EXISTS schema_migrations (version TEXT PRIMARY KEY, applied_at TIMESTAMPTZ NOT NULL DEFAULT now())")
        connection.commit()
        for path in sorted(migration_dir.glob("*.sql")):
            version = path.name
            applied = connection.execute("SELECT 1 FROM schema_migrations WHERE version = %s", (version,)).fetchone()
            if applied:
                continue
            connection.execute(path.read_text(encoding="utf-8"))
            connection.execute("INSERT INTO schema_migrations(version) VALUES (%s)", (version,))
            connection.commit()
            print(f"applied {version}")


if __name__ == "__main__":
    main()